erigon-pulse/cmd/lightclient/sentinel/handlers/heartbeats.go
Giulio rebuffo 98ceb7cc9c
separation of handlers (#5581)
Co-authored-by: giuliorebuffo <giuliorebuffo@system76-pc.localdomain>
2022-09-30 19:07:13 +02:00

48 lines
1.4 KiB
Go

package handlers
import (
"github.com/ledgerwatch/erigon/cmd/lightclient/sentinel/proto"
"github.com/ledgerwatch/erigon/cmd/lightclient/sentinel/proto/p2p"
"github.com/ledgerwatch/erigon/cmd/lightclient/utils"
"github.com/ledgerwatch/log/v3"
)
// These below are the handlers for heartbeat functions
func (c *ConsensusHandlers) goodbyeHandler(ctx *proto.StreamContext, dat *p2p.Goodbye) error {
//log.Info("[Lightclient] Received", "goodbye", dat.Reason)
_, err := ctx.Codec.WritePacket(dat)
if err != nil {
return err
}
c.peers.DisconnectPeer(ctx.Stream.Conn().RemotePeer())
return nil
}
// type safe handlers which all have access to the original stream & decompressed data
// ping handler
func pingHandler(ctx *proto.StreamContext, dat *p2p.Ping) error {
// since packets are just structs, they can be resent with no issue
_, err := ctx.Codec.WritePacket(dat)
if err != nil {
return err
}
return nil
}
// TODO: respond with proper metadata
func metadataHandler(ctx *proto.StreamContext, dat *proto.EmptyPacket) error {
return nil
}
func statusHandler(ctx *proto.StreamContext, dat *p2p.Status) error {
log.Info("[ReqResp] Status",
"epoch", dat.FinalizedEpoch,
"final root", utils.BytesToHex(dat.FinalizedRoot),
"head root", utils.BytesToHex(dat.HeadRoot),
"head slot", dat.HeadSlot,
"fork digest", utils.BytesToHex(dat.ForkDigest),
)
return nil
}