256 lines
8.7 KiB
Go
256 lines
8.7 KiB
Go
package config
|
||
|
||
import (
|
||
"fmt"
|
||
"os"
|
||
"path/filepath"
|
||
"strings"
|
||
|
||
"github.com/spf13/viper"
|
||
)
|
||
|
||
// Config 应用配置
|
||
type Config struct {
|
||
Service ServiceConfig `mapstructure:"service"`
|
||
Redis RedisConfig `mapstructure:"redis"`
|
||
MySQL MySQLConfig `mapstructure:"mysql"`
|
||
MongoDB MongoDBConfig `mapstructure:"mongodb"`
|
||
OSS OSSConfig `mapstructure:"oss"`
|
||
Wechat WechatConfig `mapstructure:"wechat"`
|
||
Scheduler SchedulerConfig `mapstructure:"scheduler"`
|
||
Admin AdminConfig `mapstructure:"admin"`
|
||
}
|
||
|
||
// ServiceConfig 服务配置
|
||
type ServiceConfig struct {
|
||
Name string `mapstructure:"name"`
|
||
Version string `mapstructure:"version"`
|
||
Host string `mapstructure:"host"`
|
||
AdminPort int `mapstructure:"admin_port"`
|
||
APIPort int `mapstructure:"api_port"`
|
||
}
|
||
|
||
// RedisConfig Redis配置
|
||
type RedisConfig struct {
|
||
Host string `mapstructure:"host"`
|
||
Port int `mapstructure:"port"`
|
||
Password string `mapstructure:"password"`
|
||
DB int `mapstructure:"db"`
|
||
PoolSize int `mapstructure:"pool_size"`
|
||
MinIdleConns int `mapstructure:"min_idle_conns"`
|
||
}
|
||
|
||
// OSSConfig 阿里云OSS配置
|
||
type OSSConfig struct {
|
||
AccessKeyID string `mapstructure:"accessKeyId"`
|
||
AccessKeySecret string `mapstructure:"accessKeySecret"`
|
||
RoleARN string `mapstructure:"roleArn"`
|
||
RoleSessionName string `mapstructure:"roleSessionName"`
|
||
Region string `mapstructure:"region"`
|
||
BucketName string `mapstructure:"bucketName"`
|
||
}
|
||
|
||
// WechatConfig 微信支付配置
|
||
type WechatConfig struct {
|
||
AppID string `mapstructure:"app_id"` // 微信小程序或公众号 APPID
|
||
MchID string `mapstructure:"mch_id"` // 商户号
|
||
NotifyURL string `mapstructure:"notify_url"` // 支付结果通知地址
|
||
APIKeyV3 string `mapstructure:"api_key_v3"` // API v3密钥
|
||
SerialNo string `mapstructure:"serial_no"` // 证书序列号
|
||
PrivateKey string `mapstructure:"private_key"` // 私钥内容（PEM格式）
|
||
PrivateKeyPath string `mapstructure:"private_key_path"` // 私钥文件路径
|
||
CertPath string `mapstructure:"cert_path"` // 证书文件路径（可选）
|
||
}
|
||
|
||
// MySQLConfig MySQL配置
|
||
type MySQLConfig struct {
|
||
Host string `mapstructure:"host"`
|
||
Port int `mapstructure:"port"`
|
||
Username string `mapstructure:"username"`
|
||
Password string `mapstructure:"password"`
|
||
Database string `mapstructure:"database"`
|
||
Charset string `mapstructure:"charset"`
|
||
MaxOpenConns int `mapstructure:"max_open_conns"`
|
||
MaxIdleConns int `mapstructure:"max_idle_conns"`
|
||
ConnMaxLifetime string `mapstructure:"conn_max_lifetime"` // 如 "3600s"
|
||
}
|
||
|
||
// MongoDBConfig MongoDB配置
|
||
type MongoDBConfig struct {
|
||
URI string `mapstructure:"uri"` // 连接字符串，如: mongodb://admin:password@localhost:27017
|
||
Database string `mapstructure:"database"` // 数据库名
|
||
Timeout string `mapstructure:"timeout"` // 连接超时，如 "10s"
|
||
}
|
||
|
||
// SchedulerConfig 调度器配置
|
||
type SchedulerConfig struct {
|
||
TickInterval string `mapstructure:"tick_interval"` // 时间轮刻度间隔，如 "100ms"
|
||
SlotNum int `mapstructure:"slot_num"` // 槽位数量
|
||
}
|
||
|
||
// AdminConfig 管理员配置
|
||
type AdminConfig struct {
|
||
JWTSecret string `mapstructure:"jwt_secret"` // JWT密钥
|
||
JWTExpiresIn string `mapstructure:"jwt_expires_in"` // JWT过期时间，如 "24h"
|
||
}
|
||
|
||
// LoadConfig 加载配置文件
|
||
// 支持从多个位置查找配置文件（优先级从高到低）：
|
||
// 1. 命令行参数指定的路径
|
||
// 2. /app/data/config/config.yaml（容器内挂载路径）
|
||
// 3. ./data/config/config.yaml（开发环境）
|
||
// 4. ./config.yaml（项目根目录，兼容旧配置）
|
||
func LoadConfig(configFile string) (*Config, error) {
|
||
v := viper.New()
|
||
v.SetConfigType("yaml")
|
||
|
||
// 如果指定了配置文件且文件存在，直接使用
|
||
if configFile != "" && configFile != "config.yaml" {
|
||
if _, err := os.Stat(configFile); err == nil {
|
||
v.SetConfigFile(configFile)
|
||
}
|
||
}
|
||
|
||
// 如果还没有设置配置文件，尝试从多个位置查找
|
||
if v.ConfigFileUsed() == "" {
|
||
configPaths := []string{
|
||
"/app/data/config/config.yaml", // 容器内挂载路径
|
||
"./data/config/config.yaml", // 开发环境
|
||
"./config.yaml", // 项目根目录（兼容旧配置）
|
||
}
|
||
|
||
for _, path := range configPaths {
|
||
if _, err := os.Stat(path); err == nil {
|
||
v.SetConfigFile(path)
|
||
break
|
||
}
|
||
}
|
||
}
|
||
|
||
// 如果仍然没有找到配置文件，使用默认路径
|
||
if v.ConfigFileUsed() == "" {
|
||
if configFile != "" {
|
||
v.SetConfigFile(configFile)
|
||
} else {
|
||
v.SetConfigFile("config.yaml")
|
||
}
|
||
}
|
||
|
||
// 设置默认值
|
||
setDefaults(v)
|
||
|
||
if err := v.ReadInConfig(); err != nil {
|
||
return nil, fmt.Errorf("读取配置文件失败: %w (尝试的路径: %s)", err, v.ConfigFileUsed())
|
||
}
|
||
|
||
var config Config
|
||
if err := v.Unmarshal(&config); err != nil {
|
||
return nil, fmt.Errorf("解析配置文件失败: %w", err)
|
||
}
|
||
|
||
// 处理证书路径：如果是绝对路径，直接验证；如果是相对路径，尝试从多个位置查找
|
||
if config.Wechat.PrivateKeyPath != "" {
|
||
if filepath.IsAbs(config.Wechat.PrivateKeyPath) {
|
||
// 绝对路径：直接验证文件是否存在
|
||
if _, err := os.Stat(config.Wechat.PrivateKeyPath); err != nil {
|
||
// 如果绝对路径不存在，尝试从其他位置查找
|
||
// 提取相对路径部分（去掉 /app/data/storage/ 或 storage/ 前缀）
|
||
relPath := config.Wechat.PrivateKeyPath
|
||
if strings.HasPrefix(relPath, "/app/data/storage/") {
|
||
relPath = strings.TrimPrefix(relPath, "/app/data/storage/")
|
||
} else if strings.HasPrefix(relPath, "storage/") {
|
||
relPath = strings.TrimPrefix(relPath, "storage/")
|
||
}
|
||
|
||
// 尝试从多个位置查找
|
||
certBasePaths := []string{
|
||
"/app/data/storage", // 容器内挂载路径
|
||
"./deploy/storage", // 部署目录
|
||
"./data/storage", // 开发环境
|
||
"./storage", // 项目根目录（兼容旧配置）
|
||
}
|
||
|
||
for _, basePath := range certBasePaths {
|
||
fullPath := filepath.Join(basePath, relPath)
|
||
if _, err := os.Stat(fullPath); err == nil {
|
||
config.Wechat.PrivateKeyPath = fullPath
|
||
if config.Wechat.CertPath != "" {
|
||
certRelPath := config.Wechat.CertPath
|
||
if strings.HasPrefix(certRelPath, "/app/data/storage/") {
|
||
certRelPath = strings.TrimPrefix(certRelPath, "/app/data/storage/")
|
||
} else if strings.HasPrefix(certRelPath, "storage/") {
|
||
certRelPath = strings.TrimPrefix(certRelPath, "storage/")
|
||
}
|
||
config.Wechat.CertPath = filepath.Join(basePath, certRelPath)
|
||
}
|
||
break
|
||
}
|
||
}
|
||
}
|
||
} else {
|
||
// 相对路径：尝试从多个位置查找
|
||
certBasePaths := []string{
|
||
"/app/data/storage", // 容器内挂载路径
|
||
"./deploy/storage", // 部署目录
|
||
"./data/storage", // 开发环境
|
||
"./storage", // 项目根目录（兼容旧配置）
|
||
}
|
||
|
||
// 提取相对路径部分（去掉 storage/ 前缀）
|
||
relPath := config.Wechat.PrivateKeyPath
|
||
if strings.HasPrefix(relPath, "storage/") {
|
||
relPath = strings.TrimPrefix(relPath, "storage/")
|
||
}
|
||
|
||
for _, basePath := range certBasePaths {
|
||
fullPath := filepath.Join(basePath, relPath)
|
||
if _, err := os.Stat(fullPath); err == nil {
|
||
config.Wechat.PrivateKeyPath = fullPath
|
||
if config.Wechat.CertPath != "" {
|
||
certRelPath := config.Wechat.CertPath
|
||
if strings.HasPrefix(certRelPath, "storage/") {
|
||
certRelPath = strings.TrimPrefix(certRelPath, "storage/")
|
||
}
|
||
config.Wechat.CertPath = filepath.Join(basePath, certRelPath)
|
||
}
|
||
break
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
return &config, nil
|
||
}
|
||
|
||
// setDefaults 设置默认配置值
|
||
func setDefaults(v *viper.Viper) {
|
||
v.SetDefault("service.name", "dd_fiber_api")
|
||
v.SetDefault("service.version", "1.0.0")
|
||
v.SetDefault("service.host", "0.0.0.0")
|
||
v.SetDefault("service.admin_port", 8080)
|
||
v.SetDefault("service.api_port", 8081)
|
||
|
||
v.SetDefault("redis.host", "localhost")
|
||
v.SetDefault("redis.port", 6379)
|
||
v.SetDefault("redis.password", "")
|
||
v.SetDefault("redis.db", 0)
|
||
v.SetDefault("redis.pool_size", 10)
|
||
v.SetDefault("redis.min_idle_conns", 5)
|
||
|
||
v.SetDefault("mysql.host", "localhost")
|
||
v.SetDefault("mysql.port", 3306)
|
||
v.SetDefault("mysql.username", "root")
|
||
v.SetDefault("mysql.password", "")
|
||
v.SetDefault("mysql.database", "")
|
||
v.SetDefault("mysql.charset", "utf8mb4")
|
||
v.SetDefault("mysql.max_open_conns", 100)
|
||
v.SetDefault("mysql.max_idle_conns", 10)
|
||
v.SetDefault("mysql.conn_max_lifetime", "3600s")
|
||
|
||
v.SetDefault("scheduler.tick_interval", "100ms")
|
||
v.SetDefault("scheduler.slot_num", 3600)
|
||
|
||
v.SetDefault("admin.jwt_secret", "your-secret-key-change-in-production")
|
||
v.SetDefault("admin.jwt_expires_in", "24h")
|
||
}
|