feat: add DownloadFile method to Client for file retrieval support

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

Author: TBXark

storage/qiniu/qiniu.go

@@ -29,8 +29,8 @@ type Config struct {
 }
 
 type Client struct {
-	mac    *qbox.Mac
 	config *Config
+	mac    *qbox.Mac
 }
 
 func NewClient(config *Config) *Client {
@@ -41,9 +41,10 @@ func NewClient(config *Config) *Client {
 			config.Host = u.Host
 		}
 	}
+	mac := qbox.NewMac(config.AccessKey, config.SecretKey)
 	return &Client{
-		mac:    qbox.NewMac(config.AccessKey, config.SecretKey),
 		config: config,
+		mac:    mac,
 	}
 }
 
@@ -162,3 +163,8 @@ func (n *Client) UploadLocalFile(ctx context.Context, file string, key string) (
 		Key: ret.Key,
 	}, nil
 }
+
+func (n *Client) DownloadFile(ctx context.Context, key string) (io.ReadCloser, error) {
+	manager := storage.NewBucketManager(n.mac, &storage.Config{})
+	return manager.Get(n.config.Bucket, key, nil)
+}

storage/s3/s3.go

@@ -116,3 +116,7 @@ func (s *Client) UploadLocalFile(ctx context.Context, file string, key string) (
 		Key: info.Key,
 	}, nil
 }
+
+func (s *Client) DownloadFile(ctx context.Context, key string) (io.ReadCloser, error) {
+	return s.client.GetObject(ctx, s.config.Bucket, key, minio.GetObjectOptions{})
+}

storage/storage.go

@@ -23,8 +23,13 @@ type FileUploader interface {
 	UploadLocalFile(ctx context.Context, file string, key string) (*models.FileUploadResult, error)
 }
 
+type FileDownloader interface {
+	DownloadFile(ctx context.Context, key string) (io.ReadCloser, error)
+}
+
 type Storage interface {
 	URLHandler
-	FileUploader
 	TokenGenerator
+	FileUploader
+	FileDownloader
 }