98 lines
3.0 KiB
Go
98 lines
3.0 KiB
Go
package container
|
|
|
|
import (
|
|
"strings"
|
|
|
|
v2object "git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/object"
|
|
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"
|
|
objectCli "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-cli/modules/object"
|
|
commonCmd "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/internal/common"
|
|
objectSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object"
|
|
oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// flags of list-object command.
|
|
const (
|
|
flagListObjectPrintAttr = "with-attr"
|
|
)
|
|
|
|
// flag vars of list-objects command.
|
|
var (
|
|
flagVarListObjectsPrintAttr bool
|
|
)
|
|
|
|
var listContainerObjectsCmd = &cobra.Command{
|
|
Use: "list-objects",
|
|
Short: "List existing objects in container",
|
|
Long: `List existing objects in container`,
|
|
Run: func(cmd *cobra.Command, _ []string) {
|
|
id := parseContainerID(cmd)
|
|
|
|
filters := new(objectSDK.SearchFilters)
|
|
filters.AddRootFilter() // search only user created objects
|
|
|
|
cli := internalclient.GetSDKClientByFlag(cmd, key.GetOrGenerate(cmd), commonflags.RPC)
|
|
|
|
var prmSearch internalclient.SearchObjectsPrm
|
|
var prmHead internalclient.HeadObjectPrm
|
|
|
|
prmSearch.SetClient(cli)
|
|
|
|
if flagVarListObjectsPrintAttr {
|
|
prmHead.SetClient(cli)
|
|
objectCli.Prepare(cmd, &prmSearch, &prmHead)
|
|
} else {
|
|
objectCli.Prepare(cmd, &prmSearch)
|
|
}
|
|
|
|
prmSearch.SetContainerID(id)
|
|
prmSearch.SetFilters(*filters)
|
|
|
|
res, err := internalclient.SearchObjects(cmd.Context(), prmSearch)
|
|
commonCmd.ExitOnErr(cmd, "rpc error: %w", err)
|
|
|
|
objectIDs := res.IDList()
|
|
|
|
for i := range objectIDs {
|
|
cmd.Println(objectIDs[i].String())
|
|
|
|
if flagVarListObjectsPrintAttr {
|
|
var addr oid.Address
|
|
addr.SetContainer(id)
|
|
addr.SetObject(objectIDs[i])
|
|
prmHead.SetAddress(addr)
|
|
|
|
resHead, err := internalclient.HeadObject(cmd.Context(), prmHead)
|
|
if err == nil {
|
|
attrs := resHead.Header().Attributes()
|
|
for i := range attrs {
|
|
attrKey := attrs[i].Key()
|
|
if !strings.HasPrefix(attrKey, v2object.SysAttributePrefix) && !strings.HasPrefix(attrKey, v2object.SysAttributePrefixNeoFS) {
|
|
// FIXME(@cthulhu-rider): https://git.frostfs.info/TrueCloudLab/frostfs-sdk-go/issues/97
|
|
// Use dedicated method to skip system attributes.
|
|
cmd.Printf(" %s: %s\n", attrKey, attrs[i].Value())
|
|
}
|
|
}
|
|
} else {
|
|
cmd.Printf(" failed to read attributes: %v\n", err)
|
|
}
|
|
}
|
|
}
|
|
},
|
|
}
|
|
|
|
func initContainerListObjectsCmd() {
|
|
commonflags.Init(listContainerObjectsCmd)
|
|
objectCli.InitBearer(listContainerObjectsCmd)
|
|
|
|
flags := listContainerObjectsCmd.Flags()
|
|
|
|
flags.StringVar(&containerID, commonflags.CIDFlag, "", commonflags.CIDFlagUsage)
|
|
flags.BoolVar(&flagVarListObjectsPrintAttr, flagListObjectPrintAttr, false,
|
|
"Request and print user attributes of each object",
|
|
)
|
|
}
|