package control import ( internalclient "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/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" rawclient "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/rpc/client" cid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id" oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id" "github.com/spf13/cobra" ) var locateObjectCmd = &cobra.Command{ Use: "locate-object", Short: "List shards storing the object", Long: "List shards storing the object", Run: locateObject, } func initControlLocateObjectCmd() { commonflags.Init(locateObjectCmd) flags := locateObjectCmd.Flags() flags.String(commonflags.CIDFlag, "", commonflags.CIDFlagUsage) _ = locateObjectCmd.MarkFlagRequired(commonflags.CIDFlag) flags.String(commonflags.OIDFlag, "", commonflags.OIDFlagUsage) _ = locateObjectCmd.MarkFlagRequired(commonflags.OIDFlag) flags.Bool(commonflags.JSON, false, "Print shard info as a JSON array") } func locateObject(cmd *cobra.Command, _ []string) { var cnr cid.ID var obj oid.ID _ = readObjectAddress(cmd, &cnr, &obj) pk := key.Get(cmd) body := new(control.ListShardsForObjectRequest_Body) body.SetContainerId(cnr.EncodeToString()) body.SetObjectId(obj.EncodeToString()) req := new(control.ListShardsForObjectRequest) req.SetBody(body) signRequest(cmd, pk, req) cli := internalclient.GetSDKClientByFlag(cmd, pk, commonflags.RPC) var err error var resp *control.ListShardsForObjectResponse err = cli.ExecRaw(func(client *rawclient.Client) error { resp, err = control.ListShardsForObject(client, req) return err }) commonCmd.ExitOnErr(cmd, "rpc error: %w", err) verifyResponse(cmd, resp.GetSignature(), resp.GetBody()) shards := resp.GetBody().GetShards() isJSON, _ := cmd.Flags().GetBool(commonflags.JSON) if isJSON { prettyPrintShardsJSON(cmd, shards) } else { prettyPrintShards(cmd, shards) } } func readObjectAddress(cmd *cobra.Command, cnr *cid.ID, obj *oid.ID) oid.Address { err := cnr.DecodeString(cmd.Flag(commonflags.CIDFlag).Value.String()) commonCmd.ExitOnErr(cmd, "decode container ID string: %w", err) err = obj.DecodeString(cmd.Flag(commonflags.OIDFlag).Value.String()) commonCmd.ExitOnErr(cmd, "decode object ID string: %w", err) var addr oid.Address addr.SetContainer(*cnr) addr.SetObject(*obj) return addr }