166 lines
4.0 KiB
Go
166 lines
4.0 KiB
Go
package config
|
||
|
||
import (
|
||
"fmt"
|
||
"os"
|
||
"path/filepath"
|
||
|
||
"gopkg.in/yaml.v3"
|
||
)
|
||
|
||
// DatabaseConfig 数据库配置结构 - 对应配置文件中的 database 部分
|
||
type DatabaseConfig struct {
|
||
Host string `yaml:"host"` // 数据库地址
|
||
Port string `yaml:"port"` // 数据库端口
|
||
User string `yaml:"user"` // 用户名
|
||
Password string `yaml:"pass"` // 密码
|
||
Name string `yaml:"name"` // 数据库名称
|
||
Type string `yaml:"type"` // 数据库类型（mysql, sqlite, postgres 等）
|
||
}
|
||
|
||
// Config 完整配置文件结构
|
||
type Config struct {
|
||
Database DatabaseConfig `yaml:"database"` // 数据库配置
|
||
}
|
||
|
||
// LoadFromFile 从 YAML 文件加载配置
|
||
func LoadFromFile(filePath string) (*Config, error) {
|
||
data, err := os.ReadFile(filePath)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("读取配置文件失败：%w", err)
|
||
}
|
||
|
||
var config Config
|
||
if err := yaml.Unmarshal(data, &config); err != nil {
|
||
return nil, fmt.Errorf("解析配置文件失败：%w", err)
|
||
}
|
||
|
||
// 验证必填字段
|
||
if err := config.Validate(); err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return &config, nil
|
||
}
|
||
|
||
// Validate 验证配置
|
||
func (c *Config) Validate() error {
|
||
if c.Database.Type == "" {
|
||
return fmt.Errorf("数据库类型不能为空")
|
||
}
|
||
|
||
if c.Database.Type == "sqlite" {
|
||
// SQLite 只需要 Name（作为文件路径）
|
||
if c.Database.Name == "" {
|
||
return fmt.Errorf("SQLite 数据库名称不能为空")
|
||
}
|
||
} else {
|
||
// 其他数据库需要所有字段
|
||
if c.Database.Host == "" {
|
||
return fmt.Errorf("数据库地址不能为空")
|
||
}
|
||
if c.Database.Port == "" {
|
||
return fmt.Errorf("数据库端口不能为空")
|
||
}
|
||
if c.Database.User == "" {
|
||
return fmt.Errorf("数据库用户名不能为空")
|
||
}
|
||
if c.Database.Password == "" {
|
||
return fmt.Errorf("数据库密码不能为空")
|
||
}
|
||
if c.Database.Name == "" {
|
||
return fmt.Errorf("数据库名称不能为空")
|
||
}
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// BuildDSN 根据配置构建数据源连接字符串（DSN）
|
||
func (c *DatabaseConfig) BuildDSN() string {
|
||
switch c.Type {
|
||
case "mysql":
|
||
return c.buildMySQLDSN()
|
||
case "postgres":
|
||
return c.buildPostgresDSN()
|
||
case "sqlite":
|
||
return c.buildSQLiteDSN()
|
||
default:
|
||
// 默认返回原始配置
|
||
return ""
|
||
}
|
||
}
|
||
|
||
// buildMySQLDSN 构建 MySQL DSN
|
||
func (c *DatabaseConfig) buildMySQLDSN() string {
|
||
// 格式：user:pass@tcp(host:port)/dbname?charset=utf8mb4&parseTime=True&loc=Local
|
||
dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
|
||
c.User,
|
||
c.Password,
|
||
c.Host,
|
||
c.Port,
|
||
c.Name,
|
||
)
|
||
return dsn
|
||
}
|
||
|
||
// buildPostgresDSN 构建 PostgreSQL DSN
|
||
func (c *DatabaseConfig) buildPostgresDSN() string {
|
||
// 格式：host=localhost port=5432 user=user password=password dbname=db sslmode=disable
|
||
dsn := fmt.Sprintf("host=%s port=%s user=%s password=%s dbname=%s sslmode=disable",
|
||
c.Host,
|
||
c.Port,
|
||
c.User,
|
||
c.Password,
|
||
c.Name,
|
||
)
|
||
return dsn
|
||
}
|
||
|
||
// buildSQLiteDSN 构建 SQLite DSN
|
||
func (c *DatabaseConfig) buildSQLiteDSN() string {
|
||
// SQLite 直接使用文件名作为 DSN
|
||
return c.Name
|
||
}
|
||
|
||
// GetDriverName 获取驱动名称
|
||
func (c *DatabaseConfig) GetDriverName() string {
|
||
return c.Type
|
||
}
|
||
|
||
// FindConfigFile 在项目目录下自动查找配置文件
|
||
// 支持 yaml, yml, toml, ini, json 等格式
|
||
// 只在当前目录查找，不越级查找
|
||
func FindConfigFile(searchDir string) (string, error) {
|
||
// 配置文件名优先级列表
|
||
configNames := []string{
|
||
"config.yaml", "config.yml",
|
||
"config.toml",
|
||
"config.ini",
|
||
"config.json",
|
||
".config.yaml", ".config.yml",
|
||
".config.toml",
|
||
".config.ini",
|
||
".config.json",
|
||
}
|
||
|
||
// 如果未指定搜索目录，使用当前目录
|
||
if searchDir == "" {
|
||
var err error
|
||
searchDir, err = os.Getwd()
|
||
if err != nil {
|
||
return "", fmt.Errorf("获取当前目录失败：%w", err)
|
||
}
|
||
}
|
||
|
||
// 只在当前目录下查找，不向上查找
|
||
for _, name := range configNames {
|
||
filePath := filepath.Join(searchDir, name)
|
||
if _, err := os.Stat(filePath); err == nil {
|
||
return filePath, nil
|
||
}
|
||
}
|
||
|
||
return "", fmt.Errorf("未找到配置文件（支持 yaml, yml, toml, ini, json 格式）")
|
||
}
|