feat: 添更新配置选项支持添加SQL导入功能

This commit is contained in:
TsMask
2025-03-06 18:03:05 +08:00
parent b52d92eaf9
commit 0e53b5c61d
3 changed files with 124 additions and 15 deletions

View File

@@ -31,6 +31,10 @@ func initFlag() {
// --c /etc/restconf.yaml
// -c /etc/restconf.yaml
pflag.StringP("config", "c", "./etc/restconf.yaml", "Specify Configuration File")
// --sqlPath ./sql/20250228.sql
pflag.String("sqlPath", "", "Execution SQL File Path")
// --sqlSource default
pflag.String("sqlSource", "default", "Execution SQL Database Source")
// --version
// -V
pVersion := pflag.BoolP("version", "V", false, "Output program version")
@@ -96,6 +100,21 @@ func initViper(configDir *embed.FS) {
conf.Set("runTime", time.Now())
}
// readExternalConfig 读取外部文件配置(放弃旧的配置序列化时候才用)
func readExternalConfig(configPaht string) {
f, err := os.Open(configPaht)
if err != nil {
log.Fatalf("config external file read error: %s", err)
return
}
defer f.Close()
if err = conf.MergeConfig(f); err != nil {
log.Fatalf("config external file read error: %s", err)
return
}
}
// 配置文件读取进行内部参数合并
func configInMerge(configFile string) {
// 指定配置文件读取序列化
@@ -156,21 +175,6 @@ func SetAssetsDirFS(assetsDir *embed.FS) {
conf.Set("AssetsDir", assetsDir)
}
// readExternalConfig 读取外部文件配置
func readExternalConfig(configPaht string) {
f, err := os.Open(configPaht)
if err != nil {
log.Fatalf("config external file read error: %s", err)
return
}
defer f.Close()
if err = conf.MergeConfig(f); err != nil {
log.Fatalf("config external file read error: %s", err)
return
}
}
// IsSystemUser 用户是否为系统管理员
func IsSystemUser(userId int64) bool {
if userId <= 0 {