package netmap import ( "fmt" "github.com/nspcc-dev/neofs-contract/netmap" "github.com/nspcc-dev/neofs-node/pkg/morph/client" ) // UpdatePeerPrm groups parameters of UpdatePeerState operation. type UpdatePeerPrm struct { key []byte state netmap.NodeState client.InvokePrmOptional } // SetKey sets public key. func (u *UpdatePeerPrm) SetKey(key []byte) { u.key = key } // SetOnline marks node to be switched into "online" state. // // Zero UpdatePeerPrm marks node as "offline". func (u *UpdatePeerPrm) SetOnline() { u.state = netmap.NodeStateOnline } // SetMaintenance marks node to be switched into "maintenance" state. // // Zero UpdatePeerPrm marks node as "offline". func (u *UpdatePeerPrm) SetMaintenance() { u.state = netmap.NodeStateMaintenance } // UpdatePeerState changes peer status through Netmap contract call. func (c *Client) UpdatePeerState(p UpdatePeerPrm) error { method := updateStateMethod if c.client.WithNotary() && c.client.IsAlpha() { // In notary environments Alphabet must calls UpdateStateIR method instead of UpdateState. // It differs from UpdateState only by name, so we can do this in the same form. // See https://github.com/nspcc-dev/neofs-contract/issues/225. method += "IR" } if p.state == 0 { p.state = netmap.NodeStateOffline } prm := client.InvokePrm{} prm.SetMethod(method) prm.SetArgs(int64(p.state), p.key) prm.InvokePrmOptional = p.InvokePrmOptional if err := c.client.Invoke(prm); err != nil { return fmt.Errorf("could not invoke smart contract: %w", err) } return nil }