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 {
return runtime.GOMAXPROCS(0) * 4
return runtime.NumCPU() * 4
}
type packet struct {
@ -420,19 +420,26 @@ func (p *Pinger) run(once bool) {
defer conn6.Close()
}
recv := make(chan *packet, 1)
recv := make(chan *packet, len(p.addrs))
recvCtx := newContext()
wg := new(sync.WaitGroup)
p.debugln("Run(): call recvICMP()")
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)
}
}
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)
}
}
p.debugln("Run(): call sendICMP()")
queue, err := p.sendICMP(conn, conn6)
@ -476,6 +483,13 @@ mainloop:
p.debugln("Run(): wait recvICMP()")
wg.Wait()
// process all queued results
close(recv)
for r := range recv {
p.debugln("Run(): <-recv")
p.procRecv(r, queue)
}
p.mu.Lock()
p.ctx.err = err
p.mu.Unlock()