feat: count response size

b4nst · Sep 3, 2024 · ef11c05 · ef11c05
1 parent 7bf815a
commit ef11c05
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/cmd/icmperf/main.go b/cmd/icmperf/main.go
@@ -26,7 +26,7 @@ func main() {
 	pinger.OnRecv(func(m *icmp.Message, t time.Time) error {
 		body := m.Body.(*icmp.Echo)
 		id := uint64(body.ID)<<32 | uint64(body.Seq)
-		record.PacketReceived(id, t)
+		record.PacketReceived(id, len(body.Data), t)
 		return nil
 	})
 

diff --git a/pkg/recorder/record.go b/pkg/recorder/record.go
@@ -45,7 +45,7 @@ func (s *Record) PacketSent(id uint64, size int, sent time.Time) {
 	}
 }
 
-func (s *Record) PacketReceived(id uint64, received time.Time) {
+func (s *Record) PacketReceived(id uint64, size int, received time.Time) {
 	// Remove the packet from the inflight map
 	s.inlock.Lock()
 	tm, ok := s.inflight[id]
@@ -63,6 +63,7 @@ func (s *Record) PacketReceived(id uint64, received time.Time) {
 	} else {
 		// Record bandwidth ping
 		tm.Received = received
+		tm.Bytes += size
 		s.tms = append(s.tms, tm)
 	}
 }