feat: 添更新配置选项支持添加SQL导入功能
This commit is contained in:
@@ -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 {
|
||||
|
||||
Reference in New Issue
Block a user