forked from TrueCloudLab/frostfs-node
66 lines
1.8 KiB
Go
66 lines
1.8 KiB
Go
package netmap
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/client"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/netmap"
|
|
)
|
|
|
|
var errFailedToRemovePeerWithoutNotary = errors.New("peer can be forcefully removed only in notary environment")
|
|
|
|
// AddPeerPrm groups parameters of AddPeer operation.
|
|
type AddPeerPrm struct {
|
|
nodeInfo netmap.NodeInfo
|
|
|
|
client.InvokePrmOptional
|
|
}
|
|
|
|
// SetNodeInfo sets new peer NodeInfo.
|
|
func (a *AddPeerPrm) SetNodeInfo(nodeInfo netmap.NodeInfo) {
|
|
a.nodeInfo = nodeInfo
|
|
}
|
|
|
|
// AddPeer registers peer in FrostFS network through
|
|
// Netmap contract call.
|
|
func (c *Client) AddPeer(ctx context.Context, p AddPeerPrm) error {
|
|
method := addPeerMethod
|
|
|
|
if c.client.WithNotary() && c.client.IsAlpha() {
|
|
// In notary environments Alphabet must calls AddPeerIR method instead of AddPeer.
|
|
// It differs from AddPeer only by name, so we can do this in the same form.
|
|
// See https://github.com/nspcc-dev/frostfs-contract/issues/154.
|
|
method += "IR"
|
|
}
|
|
|
|
prm := client.InvokePrm{}
|
|
prm.SetMethod(method)
|
|
prm.SetArgs(p.nodeInfo.Marshal())
|
|
prm.InvokePrmOptional = p.InvokePrmOptional
|
|
|
|
if _, err := c.client.Invoke(ctx, prm); err != nil {
|
|
return fmt.Errorf("could not invoke method (%s): %w", method, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// ForceRemovePeer marks the given peer as offline via a notary control transaction.
|
|
// If vub > 0, vub will be used as valid until block value.
|
|
func (c *Client) ForceRemovePeer(ctx context.Context, nodeInfo netmap.NodeInfo, vub uint32) (uint32, error) {
|
|
if !c.client.WithNotary() {
|
|
return 0, errFailedToRemovePeerWithoutNotary
|
|
}
|
|
|
|
prm := UpdatePeerPrm{}
|
|
prm.SetKey(nodeInfo.PublicKey())
|
|
prm.SetControlTX(true)
|
|
prm.SetVUB(vub)
|
|
|
|
vub, err := c.UpdatePeerState(ctx, prm)
|
|
if err != nil {
|
|
return 0, fmt.Errorf("updating peer state: %v", err)
|
|
}
|
|
return vub, nil
|
|
}
|