forked from TrueCloudLab/frostfs-node
189507dc89
Eventually more parameters will be supported (#1390) and after blobstor configuration refactoring the output will certainly change. Implement the simplest approach now. Marshaling the result directly results in too ugly names and they cannot be easily customized. Marshaling the results via `jsonpb` is better but is not that flexible in terms of what we want to output. Signed-off-by: Evgenii Stratonikov <evgeniy@nspcc.ru>
114 lines
3 KiB
Go
114 lines
3 KiB
Go
package control
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/mr-tron/base58"
|
|
rawclient "github.com/nspcc-dev/neofs-api-go/v2/rpc/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/cmd/neofs-cli/internal/key"
|
|
"github.com/nspcc-dev/neofs-node/pkg/services/control"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var listShardsCmd = &cobra.Command{
|
|
Use: "list",
|
|
Short: "List shards of the storage node",
|
|
Long: "List shards of the storage node",
|
|
Run: listShards,
|
|
}
|
|
|
|
func initControlShardsListCmd() {
|
|
commonflags.InitWithoutRPC(listShardsCmd)
|
|
|
|
flags := listShardsCmd.Flags()
|
|
|
|
flags.String(controlRPC, controlRPCDefault, controlRPCUsage)
|
|
flags.Bool(commonflags.JSON, false, "Print shard info as a JSON array")
|
|
}
|
|
|
|
func listShards(cmd *cobra.Command, _ []string) {
|
|
pk := key.Get(cmd)
|
|
|
|
req := new(control.ListShardsRequest)
|
|
req.SetBody(new(control.ListShardsRequest_Body))
|
|
|
|
signRequest(cmd, pk, req)
|
|
|
|
cli := getClient(cmd, pk)
|
|
|
|
var resp *control.ListShardsResponse
|
|
var err error
|
|
err = cli.ExecRaw(func(client *rawclient.Client) error {
|
|
resp, err = control.ListShards(client, req)
|
|
return err
|
|
})
|
|
common.ExitOnErr(cmd, "rpc error: %w", err)
|
|
|
|
verifyResponse(cmd, resp.GetSignature(), resp.GetBody())
|
|
|
|
isJSON, _ := cmd.Flags().GetBool(commonflags.JSON)
|
|
if isJSON {
|
|
prettyPrintShardsJSON(cmd, resp.GetBody().GetShards())
|
|
} else {
|
|
prettyPrintShards(cmd, resp.GetBody().GetShards())
|
|
}
|
|
}
|
|
|
|
func prettyPrintShardsJSON(cmd *cobra.Command, ii []*control.ShardInfo) {
|
|
out := make([]map[string]interface{}, 0, len(ii))
|
|
for _, i := range ii {
|
|
out = append(out, map[string]interface{}{
|
|
"shard_id": base58.Encode(i.Shard_ID),
|
|
"mode": shardModeToString(i.GetMode()),
|
|
"metabase": i.GetMetabasePath(),
|
|
"blobstor": i.GetBlobstorPath(),
|
|
"writecache": i.GetWritecachePath(),
|
|
"error_count": i.GetErrorCount(),
|
|
})
|
|
}
|
|
|
|
buf := bytes.NewBuffer(nil)
|
|
enc := json.NewEncoder(buf)
|
|
enc.SetIndent("", " ")
|
|
common.ExitOnErr(cmd, "cannot shard info to JSON: %w", enc.Encode(out))
|
|
|
|
cmd.Print(buf.String()) // pretty printer emits newline, to no need for Println
|
|
}
|
|
|
|
func prettyPrintShards(cmd *cobra.Command, ii []*control.ShardInfo) {
|
|
for _, i := range ii {
|
|
pathPrinter := func(name, path string) string {
|
|
if path == "" {
|
|
return ""
|
|
}
|
|
|
|
return fmt.Sprintf("%s: %s\n", name, path)
|
|
}
|
|
|
|
cmd.Printf("Shard %s:\nMode: %s\n"+
|
|
pathPrinter("Metabase", i.GetMetabasePath())+
|
|
pathPrinter("Blobstor", i.GetBlobstorPath())+
|
|
pathPrinter("Write-cache", i.GetWritecachePath())+
|
|
fmt.Sprintf("Error count: %d\n", i.GetErrorCount()),
|
|
base58.Encode(i.Shard_ID),
|
|
shardModeToString(i.GetMode()),
|
|
)
|
|
}
|
|
}
|
|
|
|
func shardModeToString(m control.ShardMode) string {
|
|
switch m {
|
|
case control.ShardMode_READ_WRITE:
|
|
return "read-write"
|
|
case control.ShardMode_READ_ONLY:
|
|
return "read-only"
|
|
case control.ShardMode_DEGRADED:
|
|
return "degraded"
|
|
default:
|
|
return "unknown"
|
|
}
|
|
}
|