[#611] Split name processing from object info creation procedure

Signed-off-by: Alex Vanin <alexey@nspcc.ru>

api/layer/object.go

@@ -267,7 +267,7 @@ func (n *layer) headLastVersionIfNotDeleted(ctx context.Context, bkt *data.Bucke
 	if err != nil {
 		return nil, err
 	}
-	objInfo := objInfoFromMeta(bkt, meta)
+	objInfo := objectInfoFromMeta(bkt, meta)
 	if err = n.objCache.PutObject(objInfo); err != nil {
 		n.log.Warn("couldn't put object info to cache",
 			zap.Stringer("object id", node.OID),
@@ -330,7 +330,7 @@ func (n *layer) headVersion(ctx context.Context, bkt *data.BucketInfo, p *HeadOb
 		return nil, err
 	}
 
-	objInfo := objInfoFromMeta(bkt, meta)
+	objInfo := objectInfoFromMeta(bkt, meta)
 	if err = n.objCache.PutObject(objInfo); err != nil {
 		n.log.Warn("couldn't put obj to object cache",
 			zap.String("bucket name", objInfo.Bucket),
@@ -670,41 +670,23 @@ func triageExtendedObjects(allObjects []*data.ExtendedObjectInfo) (prefixes []st
 	return
 }
 
-func (n *layer) objectInfoFromObjectsCacheOrNeoFS(ctx context.Context, bktInfo *data.BucketInfo, obj oid.ID, prefix, delimiter string) *data.ObjectInfo {
+func (n *layer) objectInfoFromObjectsCacheOrNeoFS(ctx context.Context, bktInfo *data.BucketInfo, obj oid.ID, prefix, delimiter string) (oi *data.ObjectInfo) {
-	if objInfo := n.objCache.GetObject(newAddress(bktInfo.CID, obj)); objInfo != nil {
+	oi = n.objCache.GetObject(newAddress(bktInfo.CID, obj))
-		// that's the simplest solution
-		// consider doing something else
-		if !strings.HasPrefix(objInfo.Name, prefix) {
-			return nil
-		}
-		if len(delimiter) == 0 {
-			return objInfo
-		}
-		copiedObjInfo := *objInfo
-		tail := strings.TrimPrefix(copiedObjInfo.Name, prefix)
-		index := strings.Index(tail, delimiter)
-		if index >= 0 {
-			copiedObjInfo.IsDir = true
-			copiedObjInfo.Size = 0
-			copiedObjInfo.Headers = nil
-			copiedObjInfo.ContentType = ""
-			copiedObjInfo.Name = prefix + tail[:index+1]
-		}
-		return &copiedObjInfo
-	}
 
+	if oi == nil {
 		meta, err := n.objectHead(ctx, bktInfo, obj)
 		if err != nil {
 			n.log.Warn("could not fetch object meta", zap.Error(err))
 			return nil
 		}
 
-	objInfo := objectInfoFromMeta(bktInfo, meta, prefix, delimiter)
+		oi = objectInfoFromMeta(bktInfo, meta)
-	if err = n.objCache.PutObject(objInfo); err != nil {
+		if err = n.objCache.PutObject(oi); err != nil {
 			n.log.Warn("couldn't cache an object", zap.Error(err))
 		}
+	}
 
-	return objInfo
+	return processObjectInfoName(oi, prefix, delimiter)
 }
 
 func (n *layer) transformNeofsError(ctx context.Context, err error) error {

api/layer/util.go

@@ -68,49 +68,23 @@ func userHeaders(attrs []object.Attribute) map[string]string {
 	return result
 }
 
-func objInfoFromMeta(bkt *data.BucketInfo, meta *object.Object) *data.ObjectInfo {
+func objectInfoFromMeta(bkt *data.BucketInfo, meta *object.Object) *data.ObjectInfo {
-	return objectInfoFromMeta(bkt, meta, "", "")
-}
-
-func objectInfoFromMeta(bkt *data.BucketInfo, meta *object.Object, prefix, delimiter string) *data.ObjectInfo {
 	var (
-		isDir    bool
-		size     int64
 		mimeType string
 		creation time.Time
-		filename = filenameFromObject(meta)
 	)
 
-	if !strings.HasPrefix(filename, prefix) {
+	headers := userHeaders(meta.Attributes())
-		return nil
+	delete(headers, object.AttributeFileName)
-	}
+	if contentType, ok := headers[object.AttributeContentType]; ok {
-
-	userHeaders := userHeaders(meta.Attributes())
-	delete(userHeaders, object.AttributeFileName)
-	if contentType, ok := userHeaders[object.AttributeContentType]; ok {
 		mimeType = contentType
-		delete(userHeaders, object.AttributeContentType)
+		delete(headers, object.AttributeContentType)
 	}
-	if val, ok := userHeaders[object.AttributeTimestamp]; !ok {
+	if val, ok := headers[object.AttributeTimestamp]; !ok {
 		// ignore empty value
 	} else if dt, err := strconv.ParseInt(val, 10, 64); err == nil {
 		creation = time.Unix(dt, 0)
-		delete(userHeaders, object.AttributeTimestamp)
+		delete(headers, object.AttributeTimestamp)
-	}
-
-	if len(delimiter) > 0 {
-		tail := strings.TrimPrefix(filename, prefix)
-		index := strings.Index(tail, delimiter)
-		if index >= 0 {
-			isDir = true
-			mimeType = ""
-			filename = prefix + tail[:index+1]
-			userHeaders = nil
-		} else {
-			size = int64(meta.PayloadSize())
-		}
-	} else {
-		size = int64(meta.PayloadSize())
 	}
 
 	objID, _ := meta.ID()
@@ -118,19 +92,43 @@ func objectInfoFromMeta(bkt *data.BucketInfo, meta *object.Object, prefix, delim
 	return &data.ObjectInfo{
 		ID:    objID,
 		CID:   bkt.CID,
-		IsDir: isDir,
+		IsDir: false,
 
 		Bucket:      bkt.Name,
-		Name:        filename,
+		Name:        filenameFromObject(meta),
 		Created:     creation,
 		ContentType: mimeType,
-		Headers:     userHeaders,
+		Headers:     headers,
 		Owner:       *meta.OwnerID(),
-		Size:        size,
+		Size:        int64(meta.PayloadSize()),
 		HashSum:     hex.EncodeToString(payloadChecksum.Value()),
 	}
 }
 
+// processObjectInfoName fixes name in objectInfo structure based on prefix and
+// delimiter from user request. If name does not contain prefix, nil value is
+// returned. If name should be modified, then function returns copy of objectInfo
+// structure.
+func processObjectInfoName(oi *data.ObjectInfo, prefix, delimiter string) *data.ObjectInfo {
+	if !strings.HasPrefix(oi.Name, prefix) {
+		return nil
+	}
+	if len(delimiter) == 0 {
+		return oi
+	}
+	copiedObjInfo := *oi
+	tail := strings.TrimPrefix(copiedObjInfo.Name, prefix)
+	index := strings.Index(tail, delimiter)
+	if index >= 0 {
+		copiedObjInfo.IsDir = true
+		copiedObjInfo.Size = 0
+		copiedObjInfo.Headers = nil
+		copiedObjInfo.ContentType = ""
+		copiedObjInfo.Name = prefix + tail[:index+1]
+	}
+	return &copiedObjInfo
+}
+
 func filenameFromObject(o *object.Object) string {
 	for _, attr := range o.Attributes() {
 		if attr.Key() == object.AttributeFileName {

api/layer/util_test.go

@@ -72,7 +72,7 @@ func newTestInfo(obj oid.ID, bkt *data.BucketInfo, name string, isDir bool) *dat
 	return info
 }
 
-func Test_objectInfoFromMeta(t *testing.T) {
+func Test_objectInfoName(t *testing.T) {
 	var uid user.ID
 	var id oid.ID
 	var containerID cid.ID
@@ -158,7 +158,7 @@ func Test_objectInfoFromMeta(t *testing.T) {
 
 	for _, tc := range cases {
 		t.Run(tc.name, func(t *testing.T) {
-			info := objectInfoFromMeta(bkt, tc.object, tc.prefix, tc.delimiter)
+			info := processObjectInfoName(objectInfoFromMeta(bkt, tc.object), tc.prefix, tc.delimiter)
 			require.Equal(t, tc.result, info)
 		})
 	}