首次提交：初始化项目代码

2026-07-31 15:24:14 +08:00
commit 8d3c97bd01
73 changed files with 11720 additions and 0 deletions
@@ -0,0 +1,93 @@
+package storage
+
+import (
+	"context"
+	"time"
+
+	"cloudnest/internal/pkg/logger"
+	"github.com/minio/minio-go/v7"
+	"github.com/minio/minio-go/v7/pkg/credentials"
+)
+
+// PresignDuration is the default duration for pre-signed URLs.
+// PresignDuration 是预签名 URL 的默认有效期。
+var PresignDuration = 24 * time.Hour
+
+// MinioClient provides MinIO object storage operations.
+// MinioClient 提供 MinIO 对象存储操作。
+//
+// This struct wraps the MinIO client and provides methods for storage operations.
+//
+// 此结构体封装了 MinIO 客户端，并提供了存储操作的方法。
+type MinioClient struct {
+	Client *minio.Client
+}
+
+// NewMinioClient creates a new MinIO client.
+// NewMinioClient 创建一个新的 MinIO 客户端。
+//
+// Parameters:
+//   - endpoint: The MinIO server endpoint (host:port).
+//   - accessKey: The MinIO access key.
+//   - secretKey: The MinIO secret key.
+//   - useSSL: Whether to use SSL/TLS for connections.
+//
+// Returns:
+//   - *MinioClient: A pointer to the new MinioClient instance.
+//   - error: An error if the connection fails.
+//
+// 参数:
+//   - endpoint: MinIO 服务器端点（host:port）。
+//   - accessKey: MinIO 访问密钥。
+//   - secretKey: MinIO 秘密密钥。
+//   - useSSL: 是否使用 SSL/TLS 连接。
+//
+// 返回值:
+//   - *MinioClient: 新创建的 MinioClient 实例指针。
+//   - error: 如果连接失败则返回错误。
+func NewMinioClient(endpoint, accessKey, secretKey string, useSSL bool) (*MinioClient, error) {
+	client, err := minio.New(endpoint, &minio.Options{
+		Creds:  credentials.NewStaticV4(accessKey, secretKey, ""),
+		Secure: useSSL,
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	logger.Info("minio connected")
+	return &MinioClient{Client: client}, nil
+}
+
+// EnsureBucket creates the bucket if it doesn't exist.
+// EnsureBucket 如果桶不存在则创建桶。
+//
+// Parameters:
+//   - bucketName: The name of the bucket to ensure.
+//
+// Returns:
+//   - error: An error if the operation fails.
+//
+// Note:
+//   - This method is idempotent - it can be called safely even if the bucket exists.
+//
+// 参数:
+//   - bucketName: 要确保存在的桶名称。
+//
+// 返回值:
+//   - error: 如果操作失败则返回错误。
+//
+// 注意:
+//   - 此方法是幂等的 - 即使桶已存在也可以安全调用。
+func (c *MinioClient) EnsureBucket(bucketName string) error {
+	exists, err := c.Client.BucketExists(context.Background(), bucketName)
+	if err != nil {
+		return err
+	}
+	if !exists {
+		if err := c.Client.MakeBucket(context.Background(), bucketName, minio.MakeBucketOptions{}); err != nil {
+			return err
+		}
+		logger.Info("bucket created", "bucket", bucketName)
+	}
+	return nil
+}