数据结构优化

server/utils/oss.go

@@ -0,0 +1,138 @@
+package utils
+
+import (
+	"fmt"
+	"io"
+	"mime/multipart"
+	"os"
+	"path/filepath"
+	"strings"
+	"time"
+)
+
+// StorageType 存储类型
+type StorageType string
+
+const (
+	StorageLocal  StorageType = "local"
+	StorageQCloud StorageType = "qcloud"
+	StorageAliyun StorageType = "aliyun"
+	StorageQiniu  StorageType = "qiniu"
+)
+
+// OSSConfig OSS配置
+type OSSConfig struct {
+	StorageType string
+	AccessKey   string
+	SecretKey   string
+	Bucket      string
+	Region      string
+	Domain      string
+}
+
+// OSSUploader OSS上传接口
+type OSSUploader interface {
+	Upload(file multipart.File, filename string, size int64) (string, string, error) // 返回 filePath, fileURL, error
+	Delete(filePath string) error
+}
+
+// GetOSSUploader 根据存储类型获取上传器
+func GetOSSUploader(config *OSSConfig) (OSSUploader, error) {
+	switch StorageType(config.StorageType) {
+	case StorageLocal:
+		return &LocalUploader{
+			BasePath: "./uploads",
+			BaseURL:  "/uploads",
+		}, nil
+	case StorageQCloud:
+		// TODO: 实现腾讯云COS上传
+		return nil, fmt.Errorf("qcloud storage not implemented yet")
+	case StorageAliyun:
+		// TODO: 实现阿里云OSS上传
+		return nil, fmt.Errorf("aliyun storage not implemented yet")
+	case StorageQiniu:
+		// TODO: 实现七牛云上传
+		return nil, fmt.Errorf("qiniu storage not implemented yet")
+	default:
+		return nil, fmt.Errorf("unsupported storage type: %s", config.StorageType)
+	}
+}
+
+// LocalUploader 本地存储上传器
+type LocalUploader struct {
+	BasePath string
+	BaseURL  string
+}
+
+// Upload 上传文件到本地
+func (l *LocalUploader) Upload(file multipart.File, filename string, size int64) (string, string, error) {
+	// 生成唯一文件名
+	ext := filepath.Ext(filename)
+	timestamp := time.Now().Unix()
+	randomStr := fmt.Sprintf("%d", timestamp)
+	newFilename := fmt.Sprintf("%s_%s%s", strings.TrimSuffix(filename, ext), randomStr, ext)
+
+	// 按日期创建目录
+	dateDir := time.Now().Format("2006/01/02")
+	uploadDir := filepath.Join(l.BasePath, dateDir)
+
+	// 创建目录
+	if err := os.MkdirAll(uploadDir, 0755); err != nil {
+		return "", "", fmt.Errorf("failed to create upload directory: %v", err)
+	}
+
+	// 完整文件路径
+	filePath := filepath.Join(uploadDir, newFilename)
+
+	// 创建目标文件
+	dst, err := os.Create(filePath)
+	if err != nil {
+		return "", "", fmt.Errorf("failed to create file: %v", err)
+	}
+	defer dst.Close()
+
+	// 复制文件内容
+	if _, err := io.Copy(dst, file); err != nil {
+		return "", "", fmt.Errorf("failed to copy file: %v", err)
+	}
+
+	// 生成访问URL
+	fileURL := fmt.Sprintf("%s/%s/%s", l.BaseURL, dateDir, newFilename)
+
+	return filePath, fileURL, nil
+}
+
+// Delete 删除本地文件
+func (l *LocalUploader) Delete(filePath string) error {
+	// 确保文件路径在BasePath内（安全措施）
+	absBasePath, err := filepath.Abs(l.BasePath)
+	if err != nil {
+		return err
+	}
+
+	absFilePath, err := filepath.Abs(filePath)
+	if err != nil {
+		return err
+	}
+
+	if !strings.HasPrefix(absFilePath, absBasePath) {
+		return fmt.Errorf("invalid file path: outside base directory")
+	}
+
+	return os.Remove(filePath)
+}
+
+// GetFileType 根据MIME类型判断文件类型
+func GetFileType(mimeType string) string {
+	if strings.HasPrefix(mimeType, "image/") {
+		return "image"
+	} else if strings.HasPrefix(mimeType, "video/") {
+		return "video"
+	} else if strings.HasPrefix(mimeType, "application/pdf") ||
+		strings.HasPrefix(mimeType, "application/msword") ||
+		strings.HasPrefix(mimeType, "application/vnd.openxmlformats") ||
+		strings.HasPrefix(mimeType, "text/") {
+		return "document"
+	}
+	return "other"
+}