This commit is contained in:
oneao committed 2025-08-13 14:51:19 +08:00
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
}