feat: 建立 MySQL 8 数据库基础 (#78)

This commit is contained in:
chengma
2026-08-10 01:22:22 +08:00
parent 93bbca88a8
commit 7e74bf210d
12 changed files with 623 additions and 41 deletions
+60
View File
@@ -23,6 +23,66 @@ import (
// 本项目有意不做心跳,见 docs/admin/04-client-api.md §3。
const OnlineThreshold = 10 * time.Minute
const (
databaseHostEnv = "CMAUTOBUY_DB_HOST"
databasePortEnv = "CMAUTOBUY_DB_PORT"
databaseNameEnv = "CMAUTOBUY_DB_NAME"
databaseUserEnv = "CMAUTOBUY_DB_USER"
databasePasswordEnv = "CMAUTOBUY_DB_PASSWORD"
)
// DatabaseConfig 是生产 MySQL 8 的连接配置。
// 密码只从环境变量读取,不能写进 config.yaml、日志或工单。
type DatabaseConfig struct {
Host string
Port string
Name string
User string
Password string
}
// String 永远隐藏密码,防止排错时用 %v 把凭据写进日志。
func (c DatabaseConfig) String() string {
password := "(空)"
if c.Password != "" {
password = "****"
}
return fmt.Sprintf("DatabaseConfig{Host:%s Port:%s Name:%s User:%s Password:%s}",
c.Host, c.Port, c.Name, c.User, password)
}
// LoadDatabaseFromEnv 读取生产数据库配置。
// Host/Port 使用线上同机部署的安全默认值;库名、账号和密码必须显式提供。
func LoadDatabaseFromEnv() (DatabaseConfig, error) {
cfg := DatabaseConfig{
Host: strings.TrimSpace(os.Getenv(databaseHostEnv)),
Port: strings.TrimSpace(os.Getenv(databasePortEnv)),
Name: strings.TrimSpace(os.Getenv(databaseNameEnv)),
User: strings.TrimSpace(os.Getenv(databaseUserEnv)),
Password: os.Getenv(databasePasswordEnv),
}
if cfg.Host == "" {
cfg.Host = "127.0.0.1"
}
if cfg.Port == "" {
cfg.Port = "3307"
}
var missing []string
if cfg.Name == "" {
missing = append(missing, databaseNameEnv)
}
if cfg.User == "" {
missing = append(missing, databaseUserEnv)
}
if cfg.Password == "" {
missing = append(missing, databasePasswordEnv)
}
if len(missing) > 0 {
return DatabaseConfig{}, fmt.Errorf("缺少 MySQL 配置环境变量:%s", strings.Join(missing, "、"))
}
return cfg, nil
}
// DataDir 返回可写数据目录,不存在就创建。
//
// 打包成 exe 后 = exe 旁边的 data/