mirror of
https://github.com/truewhile/MeBox.git
synced 2026-09-28 11:16:37 +08:00
truewhile
49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
// SaveDatabaseConfig updates or creates config.yaml with the specified database configuration.
|
|
func SaveDatabaseConfig(dbType, dsn string) error {
|
|
configPath := "config.yaml"
|
|
data := make(map[string]any)
|
|
|
|
content, err := os.ReadFile(configPath)
|
|
if err == nil {
|
|
if err := yaml.Unmarshal(content, &data); err != nil {
|
|
data = make(map[string]any)
|
|
}
|
|
} else if !os.IsNotExist(err) {
|
|
return fmt.Errorf("read config.yaml: %w", err)
|
|
}
|
|
|
|
dbSection, ok := data["database"].(map[string]any)
|
|
if !ok {
|
|
dbSection = make(map[string]any)
|
|
}
|
|
dbSection["type"] = dbType
|
|
dbSection["dsn"] = dsn
|
|
data["database"] = dbSection
|
|
|
|
out, err := yaml.Marshal(data)
|
|
if err != nil {
|
|
return fmt.Errorf("marshal config.yaml: %w", err)
|
|
}
|
|
|
|
// 原子写：临时文件 + rename，避免进程崩溃/断电留下截断的 config.yaml
|
|
// （下次启动会硬失败）；DSN 含数据库密码，权限收窄到 0600。
|
|
tmp := configPath + ".tmp"
|
|
if err := os.WriteFile(tmp, out, 0o600); err != nil {
|
|
return fmt.Errorf("write config.yaml.tmp: %w", err)
|
|
}
|
|
if err := os.Rename(tmp, configPath); err != nil {
|
|
_ = os.Remove(tmp)
|
|
return fmt.Errorf("replace config.yaml: %w", err)
|
|
}
|
|
return nil
|
|
}
|