p2p/simulations/examples: use atomic.Int64 (#27861)

This commit is contained in:
ucwong 2023-08-07 17:30:05 +01:00 committed by GitHub
parent eeebb07c73
commit 6d2bcb911a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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")