|
|
@ -91,7 +91,7 @@ func main() { |
|
|
|
type pingPongService struct { |
|
|
|
type pingPongService struct { |
|
|
|
id enode.ID |
|
|
|
id enode.ID |
|
|
|
log log.Logger |
|
|
|
log log.Logger |
|
|
|
received int64 |
|
|
|
received atomic.Int64 |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
func newPingPongService(id enode.ID) *pingPongService { |
|
|
|
func newPingPongService(id enode.ID) *pingPongService { |
|
|
@ -125,7 +125,7 @@ func (p *pingPongService) Info() interface{} { |
|
|
|
return struct { |
|
|
|
return struct { |
|
|
|
Received int64 `json:"received"` |
|
|
|
Received int64 `json:"received"` |
|
|
|
}{ |
|
|
|
}{ |
|
|
|
atomic.LoadInt64(&p.received), |
|
|
|
p.received.Load(), |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
@ -162,7 +162,7 @@ func (p *pingPongService) Run(peer *p2p.Peer, rw p2p.MsgReadWriter) error { |
|
|
|
return |
|
|
|
return |
|
|
|
} |
|
|
|
} |
|
|
|
log.Info("received message", "msg.code", msg.Code, "msg.payload", string(payload)) |
|
|
|
log.Info("received message", "msg.code", msg.Code, "msg.payload", string(payload)) |
|
|
|
atomic.AddInt64(&p.received, 1) |
|
|
|
p.received.Add(1) |
|
|
|
if msg.Code == pingMsgCode { |
|
|
|
if msg.Code == pingMsgCode { |
|
|
|
log.Info("sending pong") |
|
|
|
log.Info("sending pong") |
|
|
|
go p2p.Send(rw, pongMsgCode, "PONG") |
|
|
|
go p2p.Send(rw, pongMsgCode, "PONG") |
|
|
|