60 lines
1.8 KiB
Go
60 lines
1.8 KiB
Go
// Package config 负责加载并全局提供 config.yaml 中的配置
|
||
package config
|
||
|
||
import (
|
||
"fmt"
|
||
"os"
|
||
|
||
"gopkg.in/yaml.v3"
|
||
)
|
||
|
||
// LLMConf 单个大模型提供方的连接配置
|
||
type LLMConf struct {
|
||
APIKey string `yaml:"api_key"` // API 密钥（为空则该模型不可用，走规则 AI 兜底）
|
||
BaseURL string `yaml:"base_url"` // OpenAI 兼容接口的基础地址
|
||
Model string `yaml:"model"` // 模型名称（如 lite / deepseek-chat）
|
||
}
|
||
|
||
// Config 后端全部配置的根结构
|
||
type Config struct {
|
||
Server struct {
|
||
Port int `yaml:"port"` // HTTP 服务监听端口
|
||
} `yaml:"server"`
|
||
MySQL struct {
|
||
Host string `yaml:"host"` // 数据库主机
|
||
Port int `yaml:"port"` // 数据库端口
|
||
User string `yaml:"user"` // 数据库账号
|
||
Password string `yaml:"password"` // 数据库密码
|
||
Database string `yaml:"database"` // 库名
|
||
} `yaml:"mysql"`
|
||
JWT struct {
|
||
Secret string `yaml:"secret"` // JWT 签名密钥
|
||
ExpireHours int `yaml:"expire_hours"` // Token 有效期（小时）
|
||
} `yaml:"jwt"`
|
||
AI struct {
|
||
Spark LLMConf `yaml:"spark"` // 讯飞星火 Lite 配置
|
||
DeepSeek LLMConf `yaml:"deepseek"` // DeepSeek 配置
|
||
} `yaml:"ai"`
|
||
}
|
||
|
||
// C 全局配置实例（Load 成功后可直接读取）
|
||
var C Config
|
||
|
||
// Load 从指定路径读取 YAML 配置并解析到全局变量 C
|
||
func Load(path string) error {
|
||
data, err := os.ReadFile(path)
|
||
if err != nil {
|
||
return fmt.Errorf("读取配置文件失败: %w", err)
|
||
}
|
||
if err := yaml.Unmarshal(data, &C); err != nil {
|
||
return fmt.Errorf("解析配置文件失败: %w", err)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// DSN 拼接 GORM 使用的 MySQL 连接串
|
||
func (c *Config) DSN() string {
|
||
return fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=false&loc=Local",
|
||
c.MySQL.User, c.MySQL.Password, c.MySQL.Host, c.MySQL.Port, c.MySQL.Database)
|
||
}
|