From dd2de32317e70741dedaecee409f04643c66b84e Mon Sep 17 00:00:00 2001
From: yanghao05 <yanghao05@qianxin.com>
Date: Thu, 24 Apr 2025 09:54:46 +0800
Subject: [PATCH] feat: update oss client

---
 .DS_Store                           | Bin 6148 -> 6148 bytes
 backend/providers/ali/oss_client.go |  96 ++++++++++++++++++++++++----
 2 files changed, 84 insertions(+), 12 deletions(-)

diff --git a/.DS_Store b/.DS_Store
index ea6c208303a2242086bf15b7075147204bcbb85b..910c38814c2809ff9f5df55104d4a6ffc669b9ed 100644
GIT binary patch
delta 84
zcmZoMXfc=|#>B)qu~2NHo+2aj#DLw41(+BanJ4ow{t)C~NMp!ks9-2zC}k*ONM$JA
m{FU(&%Vu^Ceh#3X&5TUnnJ4p$SaL8jFaQw)!{z{yEzAIY!V{GM

delta 68
zcmZoMXfc=|#>B`mu~2NHo+2aD#DLwC4MbQb^D}+jtit@5Wn+T^<7Rdaeh#3T&5F$5
WnJ4p$SaL7`0V4wg)8+t?EzAIb*ATS;

diff --git a/backend/providers/ali/oss_client.go b/backend/providers/ali/oss_client.go
index 8a761ff..2754f6b 100644
--- a/backend/providers/ali/oss_client.go
+++ b/backend/providers/ali/oss_client.go
@@ -9,6 +9,25 @@ import (
 	"github.com/aliyun/alibabacloud-oss-go-sdk-v2/oss"
 )
 
+type OSSOption func(*OSSOptions)
+
+type OSSOptions struct {
+	internal bool
+	expire   *time.Duration
+}
+
+func WithExpire(expire time.Duration) OSSOption {
+	return func(o *OSSOptions) {
+		o.expire = &expire
+	}
+}
+
+func WithInternal(internal bool) OSSOption {
+	return func(o *OSSOptions) {
+		o.internal = internal
+	}
+}
+
 type OSSClient struct {
 	client         *oss.Client
 	internalClient *oss.Client
@@ -23,36 +42,70 @@ func (c *OSSClient) GetClient() *oss.Client {
 	return c.client
 }
 
-func (c *OSSClient) PreSignUpload(ctx context.Context, path, mimeType string) (*oss.PresignResult, error) {
+func (c *OSSClient) PreSignUpload(ctx context.Context, path, mimeType string, opts ...OSSOption) (*oss.PresignResult, error) {
 	request := &oss.PutObjectRequest{
 		Bucket:      oss.Ptr(c.config.Bucket),
 		Key:         oss.Ptr(c.GetSavePath(path)),
 		ContentType: oss.Ptr(mimeType),
 	}
-	return c.client.Presign(ctx, request)
+	opt := &OSSOptions{}
+	for _, o := range opts {
+		o(opt)
+	}
+
+	client := c.client
+	if opt.internal {
+		client = c.internalClient
+	}
+
+	return client.Presign(ctx, request)
 }
 
-func (c *OSSClient) Download(ctx context.Context, path, dest string) error {
+func (c *OSSClient) Download(ctx context.Context, path, dest string, opts ...OSSOption) error {
 	request := &oss.GetObjectRequest{
 		Bucket: oss.Ptr(c.config.Bucket),
 		Key:    oss.Ptr(path),
 	}
 
-	_, err := c.internalClient.GetObjectToFile(ctx, request, dest)
-	if err != nil {
+	opt := &OSSOptions{}
+	for _, o := range opts {
+		o(opt)
+	}
+
+	client := c.client
+	if opt.internal {
+		client = c.internalClient
+	}
+
+	if _, err := client.GetObjectToFile(ctx, request, dest); err != nil {
 		return err
 	}
 	return nil
 }
 
 // GetSignedUrl
-func (c *OSSClient) GetSignedUrl(ctx context.Context, path string) (string, error) {
+func (c *OSSClient) GetSignedUrl(ctx context.Context, path string, opts ...OSSOption) (string, error) {
 	request := &oss.GetObjectRequest{
 		Bucket: oss.Ptr(c.config.Bucket),
 		Key:    oss.Ptr(path),
 	}
 
-	preSign, err := c.client.Presign(ctx, request, oss.PresignExpires(time.Minute*5))
+	opt := &OSSOptions{}
+	for _, o := range opts {
+		o(opt)
+	}
+
+	expire := time.Minute * 5
+	if opt.expire != nil {
+		expire = *opt.expire
+	}
+
+	client := c.client
+	if opt.internal {
+		client = c.internalClient
+	}
+
+	preSign, err := client.Presign(ctx, request, oss.PresignExpires(expire))
 	if err != nil {
 		return "", err
 	}
@@ -60,27 +113,46 @@ func (c *OSSClient) GetSignedUrl(ctx context.Context, path string) (string, erro
 }
 
 // Delete
-func (c *OSSClient) Delete(ctx context.Context, path string) error {
+func (c *OSSClient) Delete(ctx context.Context, path string, opts ...OSSOption) error {
 	request := &oss.DeleteObjectRequest{
 		Bucket: oss.Ptr(c.config.Bucket),
 		Key:    oss.Ptr(path),
 	}
 
-	_, err := c.internalClient.DeleteObject(ctx, request)
-	if err != nil {
+	opt := &OSSOptions{}
+	for _, o := range opts {
+		o(opt)
+	}
+
+	client := c.client
+	if opt.internal {
+		client = c.internalClient
+	}
+
+	if _, err := client.DeleteObject(ctx, request); err != nil {
 		return err
 	}
 	return nil
 }
 
 // Upload
-func (c *OSSClient) Upload(ctx context.Context, input, dst string) error {
+func (c *OSSClient) Upload(ctx context.Context, input, dst string, opts ...OSSOption) error {
 	request := &oss.PutObjectRequest{
 		Bucket: oss.Ptr(c.config.Bucket),
 		Key:    oss.Ptr(dst),
 	}
 
-	if _, err := c.internalClient.PutObjectFromFile(ctx, request, input); err != nil {
+	opt := &OSSOptions{}
+	for _, o := range opts {
+		o(opt)
+	}
+
+	client := c.client
+	if opt.internal {
+		client = c.internalClient
+	}
+
+	if _, err := client.PutObjectFromFile(ctx, request, input); err != nil {
 		return err
 	}
 	return nil