feat: object storage add GetSignURL

Signed-off-by: Gaius <[email protected]>

Author: gaius-qi

client/daemon/objectstorage/objectstorage.go

@@ -146,6 +146,13 @@ func (o *objectStorage) getObject(ctx *gin.Context) {
 	urlMeta.Digest = meta.Digest
 
 	taskID := idgen.TaskID(ctx.Request.URL.String(), urlMeta)
+	o.peeTaskManager.StartStreamTask(ctx, &peer.StreamTaskRequest{
+		URL:     url,
+		URLMeta: meta,
+		Range:   rg,
+		PeerID:  peerID,
+	},
+	)
 }
 
 // createObject uses to upload object data.

pkg/objectstorage/constants.go

@@ -28,3 +28,26 @@ const (
 	// MetaDigest is key of digest meta.
 	MetaDigest = "digest"
 )
+
+// Method is the client operation method .
+type Method string
+
+const (
+	// MethodHead is the head operation.
+	MethodHead Method = "HEAD"
+
+	// MethodGet is the get operation.
+	MethodGet Method = "GET"
+
+	// MethodPut is the put operation.
+	MethodPut Method = "PUT"
+
+	// MethodPost is the post operation.
+	MethodPost Method = "POST"
+
+	// MethodDelete is the delete operation.
+	MethodDelete Method = "Delete"
+
+	// MethodList is the list operation.
+	MethodList Method = "List"
+)

pkg/objectstorage/objectstorage.go

@@ -86,6 +86,9 @@ type ObjectStorage interface {
 
 	// ListObjectMetadatas returns metadata of objects.
 	ListObjectMetadatas(ctx context.Context, bucketName, prefix, marker string, limit int64) ([]*ObjectMetadata, error)
+
+	// GetSignURL returns sign url of object.
+	GetSignURL(ctx context.Context, bucketName, objectKey string, method Method, expire time.Duration) (string, error)
 }
 
 // New object storage interface.

pkg/objectstorage/oss.go

@@ -21,6 +21,7 @@ import (
 	"fmt"
 	"io"
 	"strconv"
+	"time"
 
 	aliyunoss "github.com/aliyun/aliyun-oss-go-sdk/oss"
 	"github.com/go-http-utils/headers"
@@ -166,3 +167,31 @@ func (o *oss) ListObjectMetadatas(ctx context.Context, bucketName, prefix, marke
 
 	return metadatas, nil
 }
+
+// GetSignURL returns sign url of object.
+func (o *oss) GetSignURL(ctx context.Context, bucketName, objectKey string, method Method, expire time.Duration) (string, error) {
+	var ossHTTPMethod aliyunoss.HTTPMethod
+	switch method {
+	case MethodGet:
+		ossHTTPMethod = aliyunoss.HTTPGet
+	case MethodPut:
+		ossHTTPMethod = aliyunoss.HTTPPut
+	case MethodHead:
+		ossHTTPMethod = aliyunoss.HTTPHead
+	case MethodPost:
+		ossHTTPMethod = aliyunoss.HTTPPost
+	case MethodDelete:
+		ossHTTPMethod = aliyunoss.HTTPDelete
+	case MethodList:
+		ossHTTPMethod = aliyunoss.HTTPGet
+	default:
+		return "", fmt.Errorf("not support method %s", method)
+	}
+
+	bucket, err := o.client.Bucket(bucketName)
+	if err != nil {
+		return "", err
+	}
+
+	return bucket.SignURL(objectKey, ossHTTPMethod, int64(expire.Seconds()))
+}

pkg/objectstorage/s3.go

@@ -20,9 +20,11 @@ import (
 	"context"
 	"fmt"
 	"io"
+	"time"
 
 	"github.com/aws/aws-sdk-go/aws"
 	"github.com/aws/aws-sdk-go/aws/credentials"
+	"github.com/aws/aws-sdk-go/aws/request"
 	"github.com/aws/aws-sdk-go/aws/session"
 	awss3 "github.com/aws/aws-sdk-go/service/s3"
 )
@@ -169,3 +171,38 @@ func (s *s3) ListObjectMetadatas(ctx context.Context, bucketName, prefix, marker
 
 	return metadatas, nil
 }
+
+// GetSignURL returns sign url of object.
+func (s *s3) GetSignURL(ctx context.Context, bucketName, objectKey string, method Method, expire time.Duration) (string, error) {
+	var req *request.Request
+	switch method {
+	case MethodGet:
+		req, _ = s.client.GetObjectRequest(&awss3.GetObjectInput{
+			Bucket: aws.String(bucketName),
+			Key:    aws.String(objectKey),
+		})
+	case MethodPut:
+		req, _ = s.client.PutObjectRequest(&awss3.PutObjectInput{
+			Bucket: aws.String(bucketName),
+			Key:    aws.String(objectKey),
+		})
+	case MethodHead:
+		req, _ = s.client.HeadObjectRequest(&awss3.HeadObjectInput{
+			Bucket: aws.String(bucketName),
+			Key:    aws.String(objectKey),
+		})
+	case MethodDelete:
+		req, _ = s.client.DeleteObjectRequest(&awss3.DeleteObjectInput{
+			Bucket: aws.String(bucketName),
+			Key:    aws.String(objectKey),
+		})
+	case MethodList:
+		req, _ = s.client.ListObjectsRequest(&awss3.ListObjectsInput{
+			Bucket: aws.String(bucketName),
+		})
+	default:
+		return "", fmt.Errorf("not support method %s", method)
+	}
+
+	return req.Presign(expire)
+}