-
Notifications
You must be signed in to change notification settings - Fork 38
/
nodes.go
79 lines (62 loc) · 2.56 KB
/
nodes.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
package object
import (
"github.com/nspcc-dev/neofs-node/cmd/internal/cmdprinter"
internalclient "github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/client"
"github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/common"
"github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/commonflags"
"github.com/nspcc-dev/neofs-node/pkg/morph/client/netmap"
cid "github.com/nspcc-dev/neofs-sdk-go/container/id"
oid "github.com/nspcc-dev/neofs-sdk-go/object/id"
"github.com/spf13/cobra"
)
const (
shortFlag = "short"
shortFlagUsage = "Short node output info"
)
// object lock command.
var objectNodesCmd = &cobra.Command{
Use: "nodes",
Short: "Show nodes for an object",
Long: "Show nodes taking part in an object placement at the current epoch.",
Args: cobra.NoArgs,
Run: func(cmd *cobra.Command, _ []string) {
ctx, cancel := commonflags.GetCommandContext(cmd)
defer cancel()
var cnrID cid.ID
readCID(cmd, &cnrID)
var oID oid.ID
readOID(cmd, &oID)
cli := internalclient.GetSDKClientByFlag(ctx, cmd, commonflags.RPC)
var prmSnap internalclient.NetMapSnapshotPrm
prmSnap.SetClient(cli)
resmap, err := internalclient.NetMapSnapshot(ctx, prmSnap)
common.ExitOnErr(cmd, "could not get netmap snapshot", err)
var prmCnr internalclient.GetContainerPrm
prmCnr.SetClient(cli)
prmCnr.SetContainer(cnrID)
res, err := internalclient.GetContainer(ctx, prmCnr)
common.ExitOnErr(cmd, "could not get container: %w", err)
cnr := res.Container()
policy := cnr.PlacementPolicy()
var cnrNodes [][]netmap.NodeInfo
cnrNodes, err = resmap.NetMap().ContainerNodes(policy, cnrID)
common.ExitOnErr(cmd, "could not build container nodes for the given container: %w", err)
placementNodes, err := resmap.NetMap().PlacementVectors(cnrNodes, oID)
common.ExitOnErr(cmd, "could not build placement nodes for the given container: %w", err)
short, _ := cmd.Flags().GetBool(shortFlag)
for i := range placementNodes {
cmd.Printf("Descriptor #%d, REP %d:\n", i+1, policy.ReplicaNumberByIndex(i))
for j := range placementNodes[i] {
cmdprinter.PrettyPrintNodeInfo(cmd, placementNodes[i][j], j, "\t", short)
}
}
},
}
func initObjectNodesCmd() {
ff := objectNodesCmd.Flags()
ff.StringP(commonflags.RPC, commonflags.RPCShorthand, commonflags.RPCDefault, commonflags.RPCUsage)
ff.DurationP(commonflags.Timeout, commonflags.TimeoutShorthand, commonflags.TimeoutDefault, commonflags.TimeoutUsage)
ff.String(commonflags.CIDFlag, "", commonflags.CIDFlagUsage)
ff.String(commonflags.OIDFlag, "", commonflags.OIDFlagUsage)
ff.Bool(shortFlag, false, shortFlagUsage)
}