142 lines
4.3 KiB
Go
142 lines
4.3 KiB
Go
package main
|
||
|
||
import (
|
||
"fmt"
|
||
|
||
"github.com/BurntSushi/toml"
|
||
)
|
||
|
||
// DatabaseDriver 表示配置文件中指定的数据库驱动类型，对应旧后端的 DatabaseDriver(StrEnum)。
|
||
// 用 const 配合 iota 模拟数值枚举，字符串形式由 databaseDriverFromString 转换。
|
||
type DatabaseDriver int
|
||
|
||
const (
|
||
// DatabaseDriverSqlite 表示使用 SQLite 数据库。
|
||
DatabaseDriverSqlite DatabaseDriver = iota
|
||
// DatabaseDriverMysql 表示使用 MySQL 数据库。
|
||
DatabaseDriverMysql
|
||
)
|
||
|
||
// databaseDriverFromString 将配置文件中的 driver 字符串转换为 DatabaseDriver，
|
||
// 无法识别的字符串将返回错误。
|
||
func databaseDriverFromString(s string) (DatabaseDriver, error) {
|
||
switch s {
|
||
case "sqlite":
|
||
return DatabaseDriverSqlite, nil
|
||
case "mysql":
|
||
return DatabaseDriverMysql, nil
|
||
default:
|
||
return 0, fmt.Errorf("invalid database driver: %q", s)
|
||
}
|
||
}
|
||
|
||
// SqliteDatabaseConfig 是 SQLite 数据库的连接配置，对应旧后端 SqliteDatabaseConfig。
|
||
type SqliteDatabaseConfig struct {
|
||
// Path 是数据库文件的路径。
|
||
Path string `toml:"path"`
|
||
}
|
||
|
||
// MysqlDatabaseConfig 是 MySQL 数据库的连接配置，对应旧后端 MysqlDatabaseConfig。
|
||
type MysqlDatabaseConfig struct {
|
||
// Host 是数据库主机地址。
|
||
Host string `toml:"host"`
|
||
// Port 是数据库端口。
|
||
Port int `toml:"port"`
|
||
// User 是数据库用户名。
|
||
User string `toml:"user"`
|
||
// Password 是数据库密码。
|
||
Password string `toml:"password"`
|
||
// Database 是数据库名称。
|
||
Database string `toml:"database"`
|
||
}
|
||
|
||
// DatabaseConfig 描述数据库驱动及其对应的具体配置，对应旧后端 DatabaseConfig。
|
||
// 其中 Driver 起判别作用，决定 Sqlite 或 Mysql 中的哪一个被填充。
|
||
type DatabaseConfig struct {
|
||
// Driver 是当前选择的数据库驱动。
|
||
Driver DatabaseDriver
|
||
// Sqlite 在 Driver 为 DatabaseDriverSqlite 时非空。
|
||
Sqlite *SqliteDatabaseConfig
|
||
// Mysql 在 Driver 为 DatabaseDriverMysql 时非空。
|
||
Mysql *MysqlDatabaseConfig
|
||
}
|
||
|
||
// WebConfig 是 Web 服务器的配置，对应旧后端 WebConfig。
|
||
type WebConfig struct {
|
||
// Port 是 Web 服务器监听端口。
|
||
Port int `toml:"port"`
|
||
}
|
||
|
||
// OthersConfig 是其余杂项配置，对应旧后端 OthersConfig。
|
||
type OthersConfig struct {
|
||
// Debug 指示是否启用调试模式。
|
||
Debug bool `toml:"debug"`
|
||
// AutoTokenCleanDuration 是自动清理过期 token 的时间间隔（秒）。
|
||
AutoTokenCleanDuration int `toml:"auto-token-clean-duration"`
|
||
}
|
||
|
||
// Config 是从配置文件解析得到的完整配置，对应旧后端 Config。
|
||
type Config struct {
|
||
// Database 是数据库相关配置。
|
||
Database DatabaseConfig
|
||
// Web 是 Web 服务器相关配置。
|
||
Web WebConfig
|
||
// Others 是其余杂项配置。
|
||
Others OthersConfig
|
||
}
|
||
|
||
// rawDatabaseConfig 是仅用于 TOML 解析的内部结构。
|
||
// 其 Config 字段以 toml.Primitive 暂存，以便依据 Driver 做两段式分发解码。
|
||
type rawDatabaseConfig struct {
|
||
Driver string `toml:"driver"`
|
||
Config toml.Primitive `toml:"config"`
|
||
}
|
||
|
||
// rawConfig 是仅用于 TOML 解析的内部结构，配合两段式解码使用。
|
||
type rawConfig struct {
|
||
Database rawDatabaseConfig `toml:"database"`
|
||
Web WebConfig `toml:"web"`
|
||
Others OthersConfig `toml:"others"`
|
||
}
|
||
|
||
// LoadConfig 从给定路径读取并解析 TOML 配置文件，返回解析后的配置或错误。
|
||
// 它合并了旧后端的 setup_config 与 get_config：第一阶段用 toml.DecodeFile 暂存 database.config，
|
||
// 第二阶段依据 driver 用 PrimitiveDecode 分发解码到 SqliteDatabaseConfig 或 MysqlDatabaseConfig。
|
||
func LoadConfig(path string) (*Config, error) {
|
||
var raw rawConfig
|
||
meta, err := toml.DecodeFile(path, &raw)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
driver, err := databaseDriverFromString(raw.Database.Driver)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
cfg := &Config{
|
||
Web: raw.Web,
|
||
Others: raw.Others,
|
||
Database: DatabaseConfig{
|
||
Driver: driver,
|
||
},
|
||
}
|
||
|
||
switch driver {
|
||
case DatabaseDriverSqlite:
|
||
var s SqliteDatabaseConfig
|
||
if err := meta.PrimitiveDecode(raw.Database.Config, &s); err != nil {
|
||
return nil, err
|
||
}
|
||
cfg.Database.Sqlite = &s
|
||
case DatabaseDriverMysql:
|
||
var m MysqlDatabaseConfig
|
||
if err := meta.PrimitiveDecode(raw.Database.Config, &m); err != nil {
|
||
return nil, err
|
||
}
|
||
cfg.Database.Mysql = &m
|
||
}
|
||
|
||
return cfg, nil
|
||
}
|