diff --git a/cmd/neofs-node/main.go b/cmd/neofs-node/main.go index 430a58c9..320458ee 100644 --- a/cmd/neofs-node/main.go +++ b/cmd/neofs-node/main.go @@ -64,5 +64,7 @@ func shutdown(c *cfg) { c.log.Info("gRPC server stopped") + goOffline(c) + c.wg.Wait() } diff --git a/cmd/neofs-node/netmap.go b/cmd/neofs-node/netmap.go index 3b1a64fa..3fea962f 100644 --- a/cmd/neofs-node/netmap.go +++ b/cmd/neofs-node/netmap.go @@ -4,6 +4,7 @@ import ( "github.com/nspcc-dev/neofs-api-go/pkg/netmap" netmapGRPC "github.com/nspcc-dev/neofs-api-go/v2/netmap/grpc" crypto "github.com/nspcc-dev/neofs-crypto" + "github.com/nspcc-dev/neofs-node/pkg/morph/client/netmap/wrapper" "github.com/nspcc-dev/neofs-node/pkg/morph/event" netmapEvent "github.com/nspcc-dev/neofs-node/pkg/morph/event/netmap" netmapTransportGRPC "github.com/nspcc-dev/neofs-node/pkg/network/transport/netmap/grpc" @@ -114,3 +115,18 @@ func initState(c *cfg) { func addNewEpochNotificationHandler(c *cfg, h event.Handler) { addNetmapNotificationHandler(c, newEpochNotification, h) } + +func goOffline(c *cfg) { + err := c.cfgNetmap.wrapper.UpdatePeerState( + crypto.MarshalPublicKey(&c.key.PublicKey), + wrapper.StateOffline, + ) + + if err != nil { + c.log.Error("could not go offline", + zap.String("error", err.Error()), + ) + } else { + c.log.Info("request to go offline successfully sent") + } +}