package control import ( "bytes" "encoding/json" "fmt" "sort" "strings" rawclient "git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/rpc/client" "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/commonflags" "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/key" commonCmd "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/internal/common" "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/control" "github.com/mr-tron/base58" "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() { initControlFlags(listShardsCmd) flags := listShardsCmd.Flags() 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 }) commonCmd.ExitOnErr(cmd, "rpc error: %w", err) verifyResponse(cmd, resp.GetSignature(), resp.GetBody()) shards := resp.GetBody().GetShards() sortShardsByID(shards) isJSON, _ := cmd.Flags().GetBool(commonflags.JSON) if isJSON { prettyPrintShardsJSON(cmd, shards) } else { prettyPrintShards(cmd, shards) } } func prettyPrintShardsJSON(cmd *cobra.Command, ii []control.ShardInfo) { out := make([]map[string]any, 0, len(ii)) for _, i := range ii { out = append(out, map[string]any{ "shard_id": base58.Encode(i.GetShard_ID()), "mode": shardModeToString(i.GetMode()), "metabase": i.GetMetabasePath(), "blobstor": i.GetBlobstor(), "writecache": i.GetWritecachePath(), "pilorama": i.GetPiloramaPath(), "error_count": i.GetErrorCount(), "evacuation_in_progress": i.GetEvacuationInProgress(), }) } buf := bytes.NewBuffer(nil) enc := json.NewEncoder(buf) enc.SetIndent("", " ") commonCmd.ExitOnErr(cmd, "cannot shard info to JSON: %w", enc.Encode(out)) cmd.Print(buf.String()) // pretty printer emits newline, so 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) } var sb strings.Builder sb.WriteString("Blobstor:\n") for j, info := range i.GetBlobstor() { sb.WriteString(fmt.Sprintf("\tPath %d: %s\n\tType %d: %s\n", j, info.GetPath(), j, info.GetType())) } cmd.Printf("Shard %s:\nMode: %s\n"+ pathPrinter("Metabase", i.GetMetabasePath())+ sb.String()+ pathPrinter("Write-cache", i.GetWritecachePath())+ pathPrinter("Pilorama", i.GetPiloramaPath())+ fmt.Sprintf("Error count: %d\n", i.GetErrorCount())+ fmt.Sprintf("Evacuation in progress: %t\n", i.GetEvacuationInProgress()), base58.Encode(i.GetShard_ID()), shardModeToString(i.GetMode()), ) } } func shardModeToString(m control.ShardMode) string { strMode, ok := lookUpShardModeString(m) if ok { return strMode } return "unknown" } func sortShardsByID(ii []control.ShardInfo) { sort.Slice(ii, func(i, j int) bool { return bytes.Compare(ii[i].GetShard_ID(), ii[j].GetShard_ID()) < 0 }) }