taigrr/go-fastping
1
0
Fork 0
mirror of https://github.com/taigrr/go-fastping synced 2025-01-18 05:03:15 -08:00
Code Issues Projects Releases Wiki Activity

Multi-thread receive of packets

Browse Source
This commit is contained in:
Anton Skorokhod 2015-09-21 13:59:13 +02:00 committed by Tatsushi Demachi
parent 04c22ac9c6
commit 42755b085b
1 changed files with 23 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
fastping.go
View File

@ -107,7 +107,7 @@ func ipv4Payload(b []byte) []byte {
} }
func numGoRoutines() int { func numGoRoutines() int {
return runtime.GOMAXPROCS(0) * 4 return runtime.NumCPU() * 4
} }
type packet struct { type packet struct {
@ -420,19 +420,26 @@ func (p *Pinger) run(once bool) {
defer conn6.Close() defer conn6.Close()
} }
recv := make(chan *packet, 1) recv := make(chan *packet, len(p.addrs))
recvCtx := newContext() recvCtx := newContext()
wg := new(sync.WaitGroup) wg := new(sync.WaitGroup)
p.debugln("Run(): call recvICMP()") p.debugln("Run(): call recvICMP()")
if conn != nil { if conn != nil {
wg.Add(1) routines := runtime.NumCPU()
wg.Add(routines)
for i := 0; i < routines; i++ {
go p.recvICMP(conn, recv, recvCtx, wg) go p.recvICMP(conn, recv, recvCtx, wg)
} }
}
if conn6 != nil { if conn6 != nil {
wg.Add(1) routines := runtime.NumCPU()
wg.Add(routines)
for i := 0; i < routines; i++ {
go p.recvICMP(conn6, recv, recvCtx, wg) go p.recvICMP(conn6, recv, recvCtx, wg)
} }
}
p.debugln("Run(): call sendICMP()") p.debugln("Run(): call sendICMP()")
queue, err := p.sendICMP(conn, conn6) queue, err := p.sendICMP(conn, conn6)
@ -476,6 +483,13 @@ mainloop:
p.debugln("Run(): wait recvICMP()") p.debugln("Run(): wait recvICMP()")
wg.Wait() wg.Wait()
// process all queued results
close(recv)
for r := range recv {
p.debugln("Run(): <-recv")
p.procRecv(r, queue)
}
p.mu.Lock() p.mu.Lock()
p.ctx.err = err p.ctx.err = err
p.mu.Unlock() p.mu.Unlock()