116 lines
3.4 KiB
Go
116 lines
3.4 KiB
Go
|
// Package config 负责加载和管理应用配置
|
||
|
|
// 使用 viper 从 YAML 配置文件中读取配置项
|
||
|
|
package config
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
|
||
|
|
"github.com/spf13/viper"
|
||
|
|
)
|
||
|
|
|
||
|
|
// AppConfig 全局配置实例
|
||
|
|
var AppConfig *Config
|
||
|
|
|
||
|
|
// Config 应用配置结构体
|
||
|
|
type Config struct {
|
||
|
|
Server ServerConfig `mapstructure:"server"`
|
||
|
|
Database DatabaseConfig `mapstructure:"database"`
|
||
|
|
JWT JWTConfig `mapstructure:"jwt"`
|
||
|
|
Storage StorageConfig `mapstructure:"storage"`
|
||
|
|
Backup BackupConfig `mapstructure:"backup"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// BackupConfig 备份工具配置
|
||
|
|
type BackupConfig struct {
|
||
|
|
MysqlDumpPath string `mapstructure:"mysqldump_path"` // mysqldump 路径
|
||
|
|
MysqlPath string `mapstructure:"mysql_path"` // mysql 路径
|
||
|
|
}
|
||
|
|
|
||
|
|
// ServerConfig 服务器配置
|
||
|
|
type ServerConfig struct {
|
||
|
|
Port int `mapstructure:"port"`
|
||
|
|
Mode string `mapstructure:"mode"` // debug / release
|
||
|
|
PublicURL string `mapstructure:"public_url"` // 公网访问地址, 如 http://files.example.com
|
||
|
|
}
|
||
|
|
|
||
|
|
// DatabaseConfig 数据库配置
|
||
|
|
type DatabaseConfig struct {
|
||
|
|
Host string `mapstructure:"host"`
|
||
|
|
Port int `mapstructure:"port"`
|
||
|
|
Username string `mapstructure:"username"`
|
||
|
|
Password string `mapstructure:"password"`
|
||
|
|
DBName string `mapstructure:"dbname"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// DSN 返回数据库连接字符串
|
||
|
|
func (d *DatabaseConfig) DSN() string {
|
||
|
|
return fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
|
||
|
|
d.Username, d.Password, d.Host, d.Port, d.DBName)
|
||
|
|
}
|
||
|
|
|
||
|
|
// JWTConfig JWT配置
|
||
|
|
type JWTConfig struct {
|
||
|
|
Secret string `mapstructure:"secret"`
|
||
|
|
Expiration int `mapstructure:"expiration"` // 过期时间(小时)
|
||
|
|
}
|
||
|
|
|
||
|
|
// StorageConfig 存储配置
|
||
|
|
type StorageConfig struct {
|
||
|
|
DefaultPolicyID uint `mapstructure:"default_policy_id"`
|
||
|
|
LocalBasePath string `mapstructure:"local_base_path"`
|
||
|
|
CDNHost string `mapstructure:"cdn_host"` // CDN地址, 如 http://localhost:3090
|
||
|
|
CDNPathPrefix string `mapstructure:"cdn_path_prefix"` // CDN路径前缀, 如 /public
|
||
|
|
}
|
||
|
|
|
||
|
|
// Load 加载配置文件
|
||
|
|
// 配置文件搜索路径: ./config.yaml, ./config/config.yaml
|
||
|
|
func Load() (*Config, error) {
|
||
|
|
viper.SetConfigName("config")
|
||
|
|
viper.SetConfigType("yaml")
|
||
|
|
viper.AddConfigPath(".")
|
||
|
|
viper.AddConfigPath("./config")
|
||
|
|
viper.AddConfigPath("./server")
|
||
|
|
|
||
|
|
// 设置默认值
|
||
|
|
setDefaults()
|
||
|
|
|
||
|
|
if err := viper.ReadInConfig(); err != nil {
|
||
|
|
if _, ok := err.(viper.ConfigFileNotFoundError); !ok {
|
||
|
|
return nil, fmt.Errorf("读取配置文件失败: %w", err)
|
||
|
|
}
|
||
|
|
log.Println("未找到配置文件, 使用默认配置")
|
||
|
|
}
|
||
|
|
|
||
|
|
AppConfig = &Config{}
|
||
|
|
if err := viper.Unmarshal(AppConfig); err != nil {
|
||
|
|
return nil, fmt.Errorf("解析配置失败: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return AppConfig, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// setDefaults 设置配置默认值
|
||
|
|
func setDefaults() {
|
||
|
|
// 服务器默认配置
|
||
|
|
viper.SetDefault("server.port", 8080)
|
||
|
|
viper.SetDefault("server.mode", "debug")
|
||
|
|
|
||
|
|
// 数据库默认配置
|
||
|
|
viper.SetDefault("database.host", "127.0.0.1")
|
||
|
|
viper.SetDefault("database.port", 3306)
|
||
|
|
viper.SetDefault("database.username", "root")
|
||
|
|
viper.SetDefault("database.password", "root")
|
||
|
|
viper.SetDefault("database.dbname", "seeyon_fs")
|
||
|
|
|
||
|
|
// JWT默认配置
|
||
|
|
viper.SetDefault("jwt.secret", "seeyon-filesystem-jwt-secret-key-2024")
|
||
|
|
viper.SetDefault("jwt.expiration", 24)
|
||
|
|
|
||
|
|
// 存储默认配置
|
||
|
|
viper.SetDefault("storage.default_policy_id", 1)
|
||
|
|
viper.SetDefault("storage.local_base_path", "./data/storage")
|
||
|
|
viper.SetDefault("storage.cdn_host", "")
|
||
|
|
viper.SetDefault("storage.cdn_path_prefix", "/public")
|
||
|
|
}
|