121 lines
3.6 KiB
Go
121 lines
3.6 KiB
Go
|
package storage
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"carrotskin/pkg/config"
|
|||
|
|
|
|||
|
|
"github.com/minio/minio-go/v7"
|
|||
|
|
"github.com/minio/minio-go/v7/pkg/credentials"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// StorageClient S3兼容对象存储客户端包装 (支持RustFS、MinIO等)
|
|||
|
|
type StorageClient struct {
|
|||
|
|
client *minio.Client
|
|||
|
|
buckets map[string]string
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewStorage 创建新的对象存储客户端 (S3兼容，支持RustFS)
|
|||
|
|
func NewStorage(cfg config.RustFSConfig) (*StorageClient, error) {
|
|||
|
|
// 创建S3兼容客户端
|
|||
|
|
// minio-go SDK支持所有S3兼容的存储，包括RustFS
|
|||
|
|
// 不指定Region，让SDK自动检测
|
|||
|
|
client, err := minio.New(cfg.Endpoint, &minio.Options{
|
|||
|
|
Creds: credentials.NewStaticV4(cfg.AccessKey, cfg.SecretKey, ""),
|
|||
|
|
Secure: cfg.UseSSL,
|
|||
|
|
})
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("创建对象存储客户端失败: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 测试连接（如果AccessKey和SecretKey为空，跳过测试）
|
|||
|
|
if cfg.AccessKey != "" && cfg.SecretKey != "" {
|
|||
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|||
|
|
defer cancel()
|
|||
|
|
|
|||
|
|
_, err = client.ListBuckets(ctx)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("对象存储连接测试失败: %w", err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
storageClient := &StorageClient{
|
|||
|
|
client: client,
|
|||
|
|
buckets: cfg.Buckets,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return storageClient, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetClient 获取底层S3客户端
|
|||
|
|
func (s *StorageClient) GetClient() *minio.Client {
|
|||
|
|
return s.client
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetBucket 获取存储桶名称
|
|||
|
|
func (s *StorageClient) GetBucket(name string) (string, error) {
|
|||
|
|
bucket, exists := s.buckets[name]
|
|||
|
|
if !exists {
|
|||
|
|
return "", fmt.Errorf("存储桶 %s 不存在", name)
|
|||
|
|
}
|
|||
|
|
return bucket, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GeneratePresignedURL 生成预签名上传URL (PUT方法)
|
|||
|
|
func (s *StorageClient) GeneratePresignedURL(ctx context.Context, bucketName, objectName string, expires time.Duration) (string, error) {
|
|||
|
|
url, err := s.client.PresignedPutObject(ctx, bucketName, objectName, expires)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", fmt.Errorf("生成预签名URL失败: %w", err)
|
|||
|
|
}
|
|||
|
|
return url.String(), nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// PresignedPostPolicyResult 预签名POST策略结果
|
|||
|
|
type PresignedPostPolicyResult struct {
|
|||
|
|
PostURL string // POST的URL
|
|||
|
|
FormData map[string]string // 表单数据
|
|||
|
|
FileURL string // 文件的最终访问URL
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GeneratePresignedPostURL 生成预签名POST URL (支持表单上传)
|
|||
|
|
// 注意：使用时必须确保file字段是表单的最后一个字段
|
|||
|
|
func (s *StorageClient) GeneratePresignedPostURL(ctx context.Context, bucketName, objectName string, minSize, maxSize int64, expires time.Duration, useSSL bool, endpoint string) (*PresignedPostPolicyResult, error) {
|
|||
|
|
// 创建上传策略
|
|||
|
|
policy := minio.NewPostPolicy()
|
|||
|
|
|
|||
|
|
// 设置策略的基本信息
|
|||
|
|
policy.SetBucket(bucketName)
|
|||
|
|
policy.SetKey(objectName)
|
|||
|
|
policy.SetExpires(time.Now().UTC().Add(expires))
|
|||
|
|
|
|||
|
|
// 设置文件大小限制
|
|||
|
|
if err := policy.SetContentLengthRange(minSize, maxSize); err != nil {
|
|||
|
|
return nil, fmt.Errorf("设置文件大小限制失败: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 使用MinIO客户端和策略生成预签名的POST URL和表单数据
|
|||
|
|
postURL, formData, err := s.client.PresignedPostPolicy(ctx, policy)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("生成预签名POST URL失败: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 移除form_data中多余的bucket字段（MinIO Go SDK可能会添加这个字段，但会导致签名错误）
|
|||
|
|
// 注意：在Go中直接delete不存在的key是安全的
|
|||
|
|
delete(formData, "bucket")
|
|||
|
|
|
|||
|
|
// 构造文件的永久访问URL
|
|||
|
|
protocol := "http"
|
|||
|
|
if useSSL {
|
|||
|
|
protocol = "https"
|
|||
|
|
}
|
|||
|
|
fileURL := fmt.Sprintf("%s://%s/%s/%s", protocol, endpoint, bucketName, objectName)
|
|||
|
|
|
|||
|
|
return &PresignedPostPolicyResult{
|
|||
|
|
PostURL: postURL.String(),
|
|||
|
|
FormData: formData,
|
|||
|
|
FileURL: fileURL,
|
|||
|
|
}, nil
|
|||
|
|
}
|