github/ping
1
0
Fork 0
mirror of https://github.com/go-ping/ping.git synced 2025-06-25 21:51:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor CheckInFlightPackets to use packet UUID

Browse Source
This commit is contained in:
thatmattlove 2022-02-17 16:30:58 -07:00
parent 46c548aa5a
commit ccc75a38ee
1 changed files with 14 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
ping.go
View File

@ -92,13 +92,12 @@ func New(addr string) *Pinger {
var firstSequence = map[uuid.UUID]map[int]InFlightPacket{}
firstSequence[firstUUID] = make(map[int]InFlightPacket)
return &Pinger{
Count: -1,
Interval: time.Second,
RecordRtts: true,
Size: timeSliceLength + trackerLength,
Timeout: time.Duration(math.MaxInt64),
PacketTimeout: time.Duration(math.MaxInt64),
Tracker: r.Uint64(),
Count: -1,
Interval: time.Second,
RecordRtts: true,
Size: timeSliceLength + trackerLength,
Timeout: time.Duration(math.MaxInt64),
Tracker: r.Uint64(),
addr: addr,
done: make(chan interface{}),
@ -129,10 +128,6 @@ type Pinger struct {
// packets have been received.
Timeout time.Duration
// PacketTimeout specifies a timeout before the OnTimeout function is called
// for a packet not being received
PacketTimeout time.Duration
// Count tells pinger to stop after sending (and receiving) Count echo
// packets. If this option is not specified, pinger will operate until
// interrupted.
@ -522,15 +517,17 @@ func (p *Pinger) runLoop(
func (p *Pinger) CheckInFlightPackets() {
// Loop through each item in map
if p.PacketTimeout == maxPacketTimeout {
if time.Duration(p.TTL) == maxPacketTimeout {
return
}
currentTime := time.Now()
for seq, pkt := range p.InFlightPackets {
if pkt.DispatchedTime.Add(p.PacketTimeout).Before(currentTime) {
delete(p.InFlightPackets, seq)
if p.OnTimeout != nil {
p.OnTimeout(seq, &pkt)
for id, inflight := range p.InFlightPackets {
for seq, pkt := range inflight {
if pkt.DispatchedTime.Add(time.Duration(p.TTL)).Before(currentTime) {
delete(p.InFlightPackets[id], seq)
if p.OnTimeout != nil {
p.OnTimeout(seq, &pkt)
}
}
}
}