package layer import ( "context" errorsStd "errors" "github.com/nspcc-dev/neofs-s3-gw/api/data" "github.com/nspcc-dev/neofs-s3-gw/api/errors" cid "github.com/nspcc-dev/neofs-sdk-go/container/id" "go.uber.org/zap" ) func (n *layer) GetObjectTagging(ctx context.Context, p *data.ObjectTaggingInfo) (map[string]string, error) { var ( err error tags map[string]string ) tags = n.systemCache.GetTagging(objectTaggingCacheKey(p)) if tags != nil { return tags, nil } version, err := n.getTaggedObjectVersion(ctx, p) if err != nil { return nil, err } tags, err = n.treeService.GetObjectTagging(ctx, p.CnrID, version) if err != nil { if errorsStd.Is(err, ErrNodeNotFound) { return nil, errors.GetAPIError(errors.ErrNoSuchKey) } return nil, err } if err = n.systemCache.PutTagging(objectTaggingCacheKey(p), tags); err != nil { n.log.Error("couldn't cache system object", zap.Error(err)) } return tags, nil } func (n *layer) PutObjectTagging(ctx context.Context, p *data.ObjectTaggingInfo, tagSet map[string]string) error { version, err := n.getTaggedObjectVersion(ctx, p) if err != nil { return err } err = n.treeService.PutObjectTagging(ctx, p.CnrID, version, tagSet) if err != nil { if errorsStd.Is(err, ErrNodeNotFound) { return errors.GetAPIError(errors.ErrNoSuchKey) } return err } if err = n.systemCache.PutTagging(objectTaggingCacheKey(p), tagSet); err != nil { n.log.Error("couldn't cache system object", zap.Error(err)) } return nil } func (n *layer) DeleteObjectTagging(ctx context.Context, p *data.ObjectTaggingInfo) error { version, err := n.getTaggedObjectVersion(ctx, p) if err != nil { return err } err = n.treeService.DeleteObjectTagging(ctx, p.CnrID, version) if err != nil { if errorsStd.Is(err, ErrNodeNotFound) { return errors.GetAPIError(errors.ErrNoSuchKey) } return err } n.systemCache.Delete(objectTaggingCacheKey(p)) return nil } func (n *layer) GetBucketTagging(ctx context.Context, cnrID *cid.ID) (map[string]string, error) { var ( err error tags map[string]string ) tags = n.systemCache.GetTagging(bucketTaggingCacheKey(cnrID)) if tags != nil { return tags, nil } if tags, err = n.treeService.GetBucketTagging(ctx, cnrID); err != nil && !errorsStd.Is(err, ErrNodeNotFound) { return nil, err } if err := n.systemCache.PutTagging(bucketTaggingCacheKey(cnrID), tags); err != nil { n.log.Error("couldn't cache system object", zap.Error(err)) } return tags, nil } func (n *layer) PutBucketTagging(ctx context.Context, cnrID *cid.ID, tagSet map[string]string) error { if err := n.treeService.PutBucketTagging(ctx, cnrID, tagSet); err != nil { return err } if err := n.systemCache.PutTagging(bucketTaggingCacheKey(cnrID), tagSet); err != nil { n.log.Error("couldn't cache system object", zap.Error(err)) } return nil } func (n *layer) DeleteBucketTagging(ctx context.Context, cnrID *cid.ID) error { n.systemCache.Delete(bucketTaggingCacheKey(cnrID)) return n.treeService.DeleteBucketTagging(ctx, cnrID) } func objectTaggingCacheKey(p *data.ObjectTaggingInfo) string { return ".tagset." + p.CnrID.EncodeToString() + "." + p.ObjName + "." + p.VersionID } func bucketTaggingCacheKey(cnrID *cid.ID) string { return ".tagset." + cnrID.EncodeToString() } func (n *layer) getTaggedObjectVersion(ctx context.Context, p *data.ObjectTaggingInfo) (*data.NodeVersion, error) { var ( err error version *data.NodeVersion ) if p.VersionID == unversionedObjectVersionID { if version, err = n.treeService.GetUnversioned(ctx, p.CnrID, p.ObjName); err != nil { if errorsStd.Is(err, ErrNodeNotFound) { return nil, errors.GetAPIError(errors.ErrNoSuchKey) } return nil, err } } else if len(p.VersionID) == 0 { if version, err = n.treeService.GetLatestVersion(ctx, p.CnrID, p.ObjName); err != nil { if errorsStd.Is(err, ErrNodeNotFound) { return nil, errors.GetAPIError(errors.ErrNoSuchKey) } return nil, err } } else { versions, err := n.treeService.GetVersions(ctx, p.CnrID, p.ObjName) if err != nil { return nil, err } for _, v := range versions { if v.OID.EncodeToString() == p.VersionID { version = v break } } } if version == nil || version.DeleteMarker != nil { return nil, errors.GetAPIError(errors.ErrNoSuchKey) } return version, nil }