修复:数据库密码含特殊字符导致的解析问题
This commit is contained in:
@@ -3,7 +3,9 @@ package config
|
|||||||
import (
|
import (
|
||||||
"github.com/spf13/viper"
|
"github.com/spf13/viper"
|
||||||
"log"
|
"log"
|
||||||
|
"net/url"
|
||||||
"os"
|
"os"
|
||||||
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Log struct {
|
type Log struct {
|
||||||
@@ -77,6 +79,14 @@ func ReadConfig() Config {
|
|||||||
if err := viper.Unmarshal(&conf); err != nil {
|
if err := viper.Unmarshal(&conf); err != nil {
|
||||||
log.Fatalf("Failed to unmarshal config: %s", err.Error())
|
log.Fatalf("Failed to unmarshal config: %s", err.Error())
|
||||||
}
|
}
|
||||||
|
// conf.Ztimer.Db.Password 包含 %, 做url 转码
|
||||||
|
if strings.Contains(conf.Ztimer.Db.Password, "%") {
|
||||||
|
unescape, err := url.QueryUnescape(conf.Ztimer.Db.Password)
|
||||||
|
if err == nil {
|
||||||
|
conf.Ztimer.Db.Password = unescape
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return conf
|
return conf
|
||||||
}
|
}
|
||||||
// 如果在 /etc/ 目录和当前程序所在目录下均未找到配置文件,则报错
|
// 如果在 /etc/ 目录和当前程序所在目录下均未找到配置文件,则报错
|
||||||
|
|||||||
Reference in New Issue
Block a user