u

code/go-project/base-farmework/pkg/config/application.go

@@ -1,48 +1,56 @@
 package config
 
 import (
+	"base-framework/pkg/types"
 	"github.com/spf13/viper"
-	"time"
 )
 
-// server 服务器配置结构体（包内私有）
-type server struct {
-	Port int
-}
-
-// jwt 配置结构体（包内私有）
-type jwt struct {
-	Secret string
-	Expiry time.Duration
-}
-
-// 全局导出变量，指向私有结构体实例
+// 全局导出变量，指向 types 包中的配置实例
 var (
-	Server *server
-	JWT    *jwt
+	Server    *types.Server
+	JWT       *types.Jwt
+	Log       *types.Log
+	Snowflake *types.Snowflake
 )
 
 // InitApplicationConfig 初始化配置，传入配置文件路径
-func InitApplicationConfig(configPath string) (error, error) {
+func InitApplicationConfig(configPath string) (err error, err2 error) {
 	v := viper.New()
 	v.SetConfigFile(configPath)
 	v.SetConfigType("yaml")
 
 	if err := v.ReadInConfig(); err != nil {
-		return err, nil
+		return err, err2
 	}
 
-	var s server
-	if err := v.UnmarshalKey("server", &s); err != nil {
-		return err, nil
+	// server 配置
+	var server types.Server
+	if err := v.UnmarshalKey("server", &server); err != nil {
+		return err, err2
 	}
-	Server = &s
 
-	var j jwt
-	if err := v.UnmarshalKey("jwt", &j); err != nil {
-		return err, nil
+	// jwt 配置
+	var jwt types.Jwt
+	if err := v.UnmarshalKey("jwt", &jwt); err != nil {
+		return err, err2
 	}
-	JWT = &j
 
-	return nil, nil
+	// 初始化日志模块
+	var log types.Log
+	if err := v.UnmarshalKey("log", &log); err != nil {
+		return err, err2
+	}
+
+	// 初始化日志模块
+	var snowflake types.Snowflake
+	if err := v.UnmarshalKey("snowflake", &snowflake); err != nil {
+		return err, err2
+	}
+
+	Server = &server
+	JWT = &jwt
+	Log = &log
+	Snowflake = &snowflake
+
+	return nil, err2
 }

code/go-project/base-farmework/pkg/config/database.go

@@ -1,6 +1,7 @@
 package config
 
 import (
+	"base-framework/pkg/types"
 	"database/sql"
 	"fmt"
 	"log"
@@ -13,18 +14,6 @@ import (
 	_ "github.com/lib/pq" // PostgreSQL 驱动，根据实际数据库替换
 )
 
-// DBConfig 数据库配置结构体
-type DBConfig struct {
-	Host            string
-	Port            int
-	Username        string
-	Password        string
-	Dbname          string
-	MaxOpenConns    int           // 最大打开连接数
-	MaxIdleConns    int           // 最大空闲连接数
-	ConnMaxLifetime time.Duration // 连接最大生命周期
-}
-
 // 默认连接池参数
 const (
 	defaultMaxOpenConns    = 10
@@ -33,7 +22,7 @@ const (
 )
 
 var (
-	dbConfigs   map[string]DBConfig        // 配置副本
+	dbConfigs   map[string]types.DBConfig  // 配置副本
 	dataSources = make(map[string]*sql.DB) // 连接池
 	configsDSN  = make(map[string]string)  // key -> dsn
 	mu          sync.RWMutex
@@ -83,7 +72,7 @@ func unmarshalConfigs() error {
 	mu.Lock()
 	defer mu.Unlock()
 
-	temp := make(map[string]DBConfig)
+	temp := make(map[string]types.DBConfig)
 	if err := v.Unmarshal(&temp); err != nil {
 		return fmt.Errorf("解析数据库配置失败: %w", err)
 	}
@@ -92,11 +81,11 @@ func unmarshalConfigs() error {
 }
 
 // GetDBConfigs 并发安全地返回当前数据库配置副本
-func GetDBConfigs() map[string]DBConfig {
+func GetDBConfigs() map[string]types.DBConfig {
 	mu.RLock()
 	defer mu.RUnlock()
 
-	res := make(map[string]DBConfig, len(dbConfigs))
+	res := make(map[string]types.DBConfig, len(dbConfigs))
 	for k, v := range dbConfigs {
 		res[k] = v
 	}
@@ -104,13 +93,13 @@ func GetDBConfigs() map[string]DBConfig {
 }
 
 // buildDSN 构建 PostgreSQL DSN
-func buildDSN(cfg DBConfig) string {
+func buildDSN(cfg types.DBConfig) string {
 	return fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable",
 		cfg.Host, cfg.Port, cfg.Username, cfg.Password, cfg.Dbname)
 }
 
 // ReloadDataSources 支持增量更新数据源，并验证数据库连接
-func ReloadDataSources(newConfigs map[string]DBConfig) error {
+func ReloadDataSources(newConfigs map[string]types.DBConfig) error {
 	mu.Lock()
 	defer mu.Unlock()
 
@@ -171,7 +160,7 @@ func ReloadDataSources(newConfigs map[string]DBConfig) error {
 }
 
 // applyDBConfig 设置 sql.DB 连接池参数，并应用默认值
-func applyDBConfig(db *sql.DB, cfg DBConfig) {
+func applyDBConfig(db *sql.DB, cfg types.DBConfig) {
 	maxOpen := cfg.MaxOpenConns
 	if maxOpen <= 0 {
 		maxOpen = defaultMaxOpenConns