u
This commit is contained in:
1 parent
ed8bee3d77
commit
7552670e4a
23 files changed
+380
-154
No files matched your search
@@ -1,25 +1,28 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"github.com/spf13/viper"
|
||||
"time"
|
||||
)
|
||||
|
||||
type ServerConfig struct {
|
||||
// server 服务器配置结构体(包内私有)
|
||||
type server struct {
|
||||
Port int
|
||||
}
|
||||
|
||||
type JWTConfig struct {
|
||||
// jwt 配置结构体(包内私有)
|
||||
type jwt struct {
|
||||
Secret string
|
||||
Expiry time.Duration
|
||||
}
|
||||
|
||||
// 全局导出变量,指向私有结构体实例
|
||||
var (
|
||||
Server ServerConfig
|
||||
JWT JWTConfig
|
||||
Server *server
|
||||
JWT *jwt
|
||||
)
|
||||
|
||||
// InitApplicationConfig 初始化配置,传入配置文件路径
|
||||
func InitApplicationConfig(configPath string) (error, error) {
|
||||
v := viper.New()
|
||||
v.SetConfigFile(configPath)
|
||||
@@ -29,13 +32,17 @@ func InitApplicationConfig(configPath string) (error, error) {
|
||||
return err, nil
|
||||
}
|
||||
|
||||
if err := v.UnmarshalKey("server", &Server); err != nil {
|
||||
var s server
|
||||
if err := v.UnmarshalKey("server", &s); err != nil {
|
||||
return err, nil
|
||||
}
|
||||
Server = &s
|
||||
|
||||
if err := v.UnmarshalKey("jwt", &JWT); err != nil {
|
||||
var j jwt
|
||||
if err := v.UnmarshalKey("jwt", &j); err != nil {
|
||||
return err, nil
|
||||
}
|
||||
JWT = &j
|
||||
|
||||
return nil, nil
|
||||
}
|
||||
Reference in new issue
Block a user