初始化

internal/config/config.go

@@ -0,0 +1,62 @@
+package config
+
+import (
+	"errors"
+	"fmt"
+	"os"
+	"strings"
+
+	"gopkg.in/yaml.v3"
+)
+
+// Config 是服务的全部运行配置，来自 YAML 文件（默认 ./config.yaml，可用 -config 指定）。
+type Config struct {
+	// Env 为 dev 时启动执行 AutoMigrate；其余一律视为 prod，绝不执行 DDL。
+	Env    string `yaml:"env"`
+	Listen string `yaml:"listen"`
+	// BaseURL 拼接文件访问 URL 的前缀（客户端可达的地址）；留空则按请求 Host 推断。
+	BaseURL string `yaml:"base_url"`
+	// APIKey 是客户端上传凭证：请求需带 Authorization: Bearer <api_key>。
+	APIKey      string `yaml:"api_key"`
+	StorageDir  string `yaml:"storage_dir"`
+	MaxUploadMB int64  `yaml:"max_upload_mb"`
+	MySQL       struct {
+		DSN string `yaml:"dsn"`
+	} `yaml:"mysql"`
+}
+
+// MaxUploadBytes 返回单文件字节上限。
+func (c *Config) MaxUploadBytes() int64 { return c.MaxUploadMB << 20 }
+
+// Load 读取并校验配置，缺省值：prod / :8788 / ./uploads / 20MB。
+func Load(path string) (*Config, error) {
+	b, err := os.ReadFile(path)
+	if err != nil {
+		return nil, fmt.Errorf("读取配置失败: %w", err)
+	}
+	c := &Config{}
+	if err := yaml.Unmarshal(b, c); err != nil {
+		return nil, fmt.Errorf("解析配置失败: %w", err)
+	}
+	if c.Env != "dev" {
+		c.Env = "prod"
+	}
+	if strings.TrimSpace(c.Listen) == "" {
+		c.Listen = ":8788"
+	}
+	if strings.TrimSpace(c.StorageDir) == "" {
+		c.StorageDir = "./uploads"
+	}
+	if c.MaxUploadMB <= 0 {
+		c.MaxUploadMB = 20
+	}
+	c.BaseURL = strings.TrimRight(strings.TrimSpace(c.BaseURL), "/")
+	c.APIKey = strings.TrimSpace(c.APIKey)
+	if c.APIKey == "" {
+		return nil, errors.New("配置缺少 api_key（客户端上传凭证，不允许留空开放上传）")
+	}
+	if strings.TrimSpace(c.MySQL.DSN) == "" {
+		return nil, errors.New("配置缺少 mysql.dsn")
+	}
+	return c, nil
+}