diff --git a/pkg/service/tree/tree.go b/pkg/service/tree/tree.go index fbab21a..f67e536 100644 --- a/pkg/service/tree/tree.go +++ b/pkg/service/tree/tree.go @@ -565,7 +565,7 @@ func (c *Tree) GetVersions(ctx context.Context, bktInfo *data.BucketInfo, filepa } func (c *Tree) GetLatestVersion(ctx context.Context, bktInfo *data.BucketInfo, objectName string) (*data.NodeVersion, error) { - meta := []string{oidKV, isUnversionedKV, isDeleteMarkerKV, etagKV, sizeKV, md5KV} + meta := []string{oidKV, isCombinedKV, isUnversionedKV, isDeleteMarkerKV, etagKV, sizeKV, md5KV} path := pathFromName(objectName) p := &GetNodesParams{ @@ -1191,7 +1191,7 @@ func (c *Tree) clearOutdatedVersionInfo(ctx context.Context, bktInfo *data.Bucke } func (c *Tree) getVersions(ctx context.Context, bktInfo *data.BucketInfo, treeID, filepath string, onlyUnversioned bool) ([]*data.NodeVersion, error) { - keysToReturn := []string{oidKV, isUnversionedKV, isDeleteMarkerKV, etagKV, sizeKV, md5KV} + keysToReturn := []string{oidKV, isCombinedKV, isUnversionedKV, isDeleteMarkerKV, etagKV, sizeKV, md5KV} path := pathFromName(filepath) p := &GetNodesParams{ BktInfo: bktInfo, diff --git a/pkg/service/tree/tree_client_in_memory.go b/pkg/service/tree/tree_client_in_memory.go index 3c3128b..bee3982 100644 --- a/pkg/service/tree/tree_client_in_memory.go +++ b/pkg/service/tree/tree_client_in_memory.go @@ -7,6 +7,7 @@ import ( "time" "git.frostfs.info/TrueCloudLab/frostfs-s3-gw/api/data" + "golang.org/x/exp/slices" ) type nodeMeta struct { @@ -183,6 +184,22 @@ func NewTreeServiceClientMemory() (*ServiceClientMemory, error) { }, nil } +type nodeResponseWrapper struct { + nodeResponse + allAttr bool + attrs []string +} + +func (n nodeResponseWrapper) GetMeta() []Meta { + res := make([]Meta, 0, len(n.meta)) + for _, value := range n.meta { + if n.allAttr || slices.Contains(n.attrs, value.key) { + res = append(res, value) + } + } + return res +} + func (c *ServiceClientMemory) GetNodes(_ context.Context, p *GetNodesParams) ([]NodeResponse, error) { cnr, ok := c.containers[p.BktInfo.CID.EncodeToString()] if !ok { @@ -205,7 +222,11 @@ func (c *ServiceClientMemory) GetNodes(_ context.Context, p *GetNodesParams) ([] res2 := make([]NodeResponse, len(res)) for i, n := range res { - res2[i] = n + res2[i] = nodeResponseWrapper{ + nodeResponse: n, + allAttr: p.AllAttrs, + attrs: p.Meta, + } } return res2, nil