支持多数据库

bootstraps/bootstrap.go

@@ -14,6 +14,11 @@ import (
 	"git.x2erp.com/qdy/go-base/logger"
 )
 
+// 定义接口
+type ShutdownHandler interface {
+	OnShutdown()
+}
+
 // Bootstrapper 服务启动器
 type Bootstrapper struct {
 	serviceName    string
@@ -81,7 +86,7 @@ func (b *Bootstrapper) StartService() *Bootstrapper {
 }
 
 // Run 运行服务
-func (b *Bootstrapper) Run() {
+func (b *Bootstrapper) Run(handler ShutdownHandler) {
 	log.Printf("服务 %s 开始运行...", b.serviceName)
 
 	// 设置信号监听 - 在这里设置，而不是在New中
@@ -96,11 +101,11 @@ func (b *Bootstrapper) Run() {
 	}()
 
 	// 等待中断信号
-	b.waitForShutdown()
+	b.waitForShutdown(handler)
 }
 
 // RunTLS 运行HTTPS服务
-func (b *Bootstrapper) RunTLS(certFile, keyFile string) {
+func (b *Bootstrapper) RunTLS(certFile, keyFile string, handler ShutdownHandler) {
 	log.Printf("服务 %s 开始运行(HTTPS)...", b.serviceName)
 
 	// 设置信号监听
@@ -115,11 +120,11 @@ func (b *Bootstrapper) RunTLS(certFile, keyFile string) {
 	}()
 
 	// 等待中断信号
-	b.waitForShutdown()
+	b.waitForShutdown(handler)
 }
 
 // waitForShutdown 等待关闭信号
-func (b *Bootstrapper) waitForShutdown() {
+func (b *Bootstrapper) waitForShutdown(handler ShutdownHandler) {
 	log.Println("按 Ctrl+C 停止服务")
 
 	// 等待信号
@@ -137,6 +142,11 @@ func (b *Bootstrapper) waitForShutdown() {
 		log.Println("HTTP服务器已关闭")
 	}
 
+	// 执行关闭处理
+	if handler != nil {
+		handler.OnShutdown()
+	}
+
 	// 停止日志写入
 	logger.StopESWriter()
 	log.Println("服务优雅关闭完成")

config/config.go

@@ -4,11 +4,17 @@ import (
 	"sync"
 
 	"git.x2erp.com/qdy/go-base/config/subconfigs"
+	"git.x2erp.com/qdy/go-base/logger"
 )
 
-// IConfig 主配置接口
+// IConfig 主配置接口（更新支持多数据库）
 type IConfig interface {
-	GetDatabase() *subconfigs.DatabaseConfig
+	// 多数据库支持
+	GetDatabase() *subconfigs.DatabaseConfig                  // 获取默认数据库（向后兼容）
+	GetDatabases() *subconfigs.DatabasesConfig                // 获取所有数据库配置
+	GetDatabaseConfig(name string) *subconfigs.DatabaseConfig // 按名称获取数据库配置
+
+	// 其他配置（保持不变）
 	GetRedis() *subconfigs.RedisConfig
 	GetDoris() *subconfigs.DorisConfig
 	GetRabbitMQ() *subconfigs.RabbitMQConfig
@@ -18,48 +24,74 @@ type IConfig interface {
 	GetMicro() *subconfigs.MicroConfig
 	GetLog() *subconfigs.LogConfig
 	GetInitError() error
+
+	// 配置检查（更新）
 	IsDatabaseConfigured() bool
+	IsDatabaseConfiguredByName(name string) bool
 	IsDorisConfigured() bool
 	IsRedisConfigured() bool
+
 	SetServiceName(name string)
 	SetServiceVersion(version string)
 	SetEnv(name string)
 }
 
-// 根据启动参数设置环境变量
-func (c *Config) SetEnv(name string) {
-	if c.GetService().Env == "" {
-		c.GetService().Env = name
-	}
+// Config 主配置结构体 - 访问门面
+type Config struct {
+	initError error // 初始化错误
 }
 
-// 设置微服务名称。。如果配置文件设置了，就按配置文件
-func (c *Config) SetServiceName(name string) {
-	if c.GetService().ServiceName == "" {
-		c.GetService().ServiceName = name
-	}
+// ========== 数据库相关方法（更新） ==========
 
+// GetDatabases 获取所有数据库配置
+func (c *Config) GetDatabases() *subconfigs.DatabasesConfig {
+	if config := subconfigs.GetRegisteredConfig("databases"); config != nil {
+		if dbs, ok := config.(*subconfigs.DatabasesConfig); ok {
+			return dbs
+		}
+	}
+	return nil
 }
 
-// 设置微服务版本号
-func (c *Config) SetServiceVersion(version string) {
+// GetDatabase 获取默认数据库配置（向后兼容）
+func (c *Config) GetDatabase() *subconfigs.DatabaseConfig {
+	dbs := c.GetDatabases()
+	if dbs == nil {
+		logger.Errorf("获取默认数据库连接配置发生错误：配置为空！")
+		return nil
+	}
+	return dbs.GetDefaultDatabase()
+}
 
-	c.GetService().ServiceVersion = version
+// GetDatabaseConfig 按名称获取数据库配置
+func (c *Config) GetDatabaseConfig(name string) *subconfigs.DatabaseConfig {
+	dbs := c.GetDatabases()
+	if dbs == nil {
+		return nil
+	}
+	return dbs.GetDatabase(name)
 }
 
-// Config 主配置结构体 - 访问门面
-type Config struct {
-	initError error // 初始化错误
+// IsDatabaseConfigured 检查默认数据库是否配置
+func (c *Config) IsDatabaseConfigured() bool {
+	dbs := c.GetDatabases()
+	if dbs == nil {
+		return false
+	}
+	return dbs.IsConfigured()
 }
 
-// 实现IConfig接口 - 从注册表获取配置
-func (c *Config) GetDatabase() *subconfigs.DatabaseConfig {
-	if config := subconfigs.GetRegisteredConfig("database"); config != nil {
-		return config.(*subconfigs.DatabaseConfig)
+// IsDatabaseConfiguredByName 检查指定名称的数据库是否配置
+func (c *Config) IsDatabaseConfiguredByName(name string) bool {
+	dbs := c.GetDatabases()
+	if dbs == nil {
+		return false
 	}
-	return nil
+	return dbs.IsConfigured()
 }
 
+// ========== 其他配置方法（保持不变） ==========
+
 func (c *Config) GetRedis() *subconfigs.RedisConfig {
 	if config := subconfigs.GetRegisteredConfig("redis"); config != nil {
 		return config.(*subconfigs.RedisConfig)
@@ -120,17 +152,45 @@ func (c *Config) GetInitError() error {
 	return c.initError
 }
 
-// GetService 包级便捷函数 - 获取服务配置
-func GetService() *subconfigs.ServiceConfig {
-	// 直接使用全局的 cfgInstance，假设初始化已完成
+// 实现 IConfig 接口的其他方法
+func (c *Config) IsDorisConfigured() bool {
+	doris := c.GetDoris()
+	return doris != nil && doris.IsConfigured()
+}
+
+func (c *Config) IsRedisConfigured() bool {
+	redis := c.GetRedis()
+	return redis != nil && redis.IsConfigured()
+}
+
+// 设置方法
+func (c *Config) SetEnv(name string) {
+	if c.GetService().Env == "" {
+		c.GetService().Env = name
+	}
+}
+
+func (c *Config) SetServiceName(name string) {
+	if c.GetService().ServiceName == "" {
+		c.GetService().ServiceName = name
+	}
+}
+
+func (c *Config) SetServiceVersion(version string) {
+	c.GetService().ServiceVersion = version
+}
+
+// ========== 包级便捷函数（更新） ==========
+
+// GetDatabases 包级便捷函数 - 获取所有数据库配置
+func GetDatabases() *subconfigs.DatabasesConfig {
 	if cfgInstance == nil {
-		// 这种情况不应该发生，但为了安全返回默认值
-		return &subconfigs.ServiceConfig{}
+		return nil
 	}
-	return cfgInstance.GetService()
+	return cfgInstance.GetDatabases()
 }
 
-// GetDatabase 包级便捷函数 - 获取数据库配置
+// GetDatabase 包级便捷函数 - 获取默认数据库配置（向后兼容）
 func GetDatabase() *subconfigs.DatabaseConfig {
 	if cfgInstance == nil {
 		return &subconfigs.DatabaseConfig{}
@@ -138,6 +198,14 @@ func GetDatabase() *subconfigs.DatabaseConfig {
 	return cfgInstance.GetDatabase()
 }
 
+// GetDatabaseConfig 包级便捷函数 - 按名称获取数据库配置
+func GetDatabaseConfig(name string) *subconfigs.DatabaseConfig {
+	if cfgInstance == nil {
+		return nil
+	}
+	return cfgInstance.GetDatabaseConfig(name)
+}
+
 // GetRedis 包级便捷函数 - 获取Redis配置
 func GetRedis() *subconfigs.RedisConfig {
 	if cfgInstance == nil {
@@ -167,7 +235,6 @@ func GetMicro() *subconfigs.MicroConfig {
 		return &subconfigs.MicroConfig{}
 	}
 	return cfgInstance.GetMicro()
-
 }
 
 func GetAuth() *subconfigs.AuthConfig {
@@ -177,6 +244,15 @@ func GetAuth() *subconfigs.AuthConfig {
 	return cfgInstance.GetAuth()
 }
 
+func GetService() *subconfigs.ServiceConfig {
+	if cfgInstance == nil {
+		return &subconfigs.ServiceConfig{}
+	}
+	return cfgInstance.GetService()
+}
+
+// ========== 单例管理 ==========
+
 var (
 	cfgInstance *Config
 	once        sync.Once
@@ -201,21 +277,3 @@ func GetConfigWithError() (IConfig, error) {
 	cfg, err := GetConfig()
 	return cfg, err
 }
-
-// 实现 IConfig 接口的 IsDatabaseConfigured 方法
-func (c *Config) IsDatabaseConfigured() bool {
-
-	return c.GetDatabase().IsConfigured()
-}
-
-// 实现 IConfig 接口的 IsDorisConfigured 方法
-func (c *Config) IsDorisConfigured() bool {
-
-	return c.GetDoris().IsConfigured()
-}
-
-// 实现 IConfig 接口的 IsDorisConfigured 方法
-func (c *Config) IsRedisConfigured() bool {
-
-	return c.GetRedis().IsConfigured()
-}

config/subconfigs/database_config.go

@@ -1,6 +1,14 @@
 package subconfigs
 
-import "fmt"
+import (
+	"fmt"
+	"log"
+)
+
+type DatabasesConfig struct {
+	BaseConfig
+	Databases map[string]*DatabaseConfig `yaml:"databases"`
+}
 
 // DatabaseConfig 数据库配置
 type DatabaseConfig struct {
@@ -16,39 +24,238 @@ type DatabaseConfig struct {
 	ConnMaxLifetime int    `yaml:"conn_max_lifetime"`
 }
 
-// NewDatabaseConfig 创建数据库配置实例
-func NewDatabaseConfig() *DatabaseConfig {
-	return &DatabaseConfig{}
+// ========== DatabasesConfig 实现 ConfigLoader 接口 ==========
+
+// SetDefaults 设置默认值 - 实现 ConfigLoader 接口
+func (c *DatabasesConfig) SetDefaults() {
+	if c.Databases == nil {
+		c.Databases = make(map[string]*DatabaseConfig)
+		log.Println("⚠️  数据库配置为空，已初始化空的数据库配置")
+		return
+	}
+
+	// 为每个数据库配置设置默认值
+	for _, db := range c.Databases {
+		if db != nil {
+			db.SetDefaults()
+		}
+	}
+}
+
+func (c *DatabasesConfig) Load(data map[string]interface{}) error {
+	// 初始化
+	c.Databases = make(map[string]*DatabaseConfig)
+
+	// 遍历所有数据库配置
+	for dbName, dbData := range data {
+		dbDataMap, ok := dbData.(map[interface{}]interface{})
+		if !ok {
+			return fmt.Errorf("database '%s' config is not a map", dbName)
+		}
+
+		// 转换为 map[string]interface{}
+		stringMap := make(map[string]interface{})
+		for k, v := range dbDataMap {
+			if key, ok := k.(string); ok {
+				stringMap[key] = v
+			}
+		}
+
+		// 加载单个数据库配置
+		dbConfig := &DatabaseConfig{}
+		if err := dbConfig.Load(stringMap); err != nil {
+			return fmt.Errorf("failed to load database '%s' config: %v", dbName, err)
+		}
+
+		c.Databases[dbName] = dbConfig
+	}
+
+	// 设置默认值和验证
+	c.SetDefaults()
+	return c.Validate()
 }
 
-// SetDefaults 设置默认值
+// ========== DatabaseConfig 实现 ConfigLoader 接口 ==========
+
+// SetDefaults 设置默认值 - 实现 ConfigLoader 接口
 func (c *DatabaseConfig) SetDefaults() {
-	c.Type = "mysql"
-	c.Port = 3306
-	c.MaxOpenConns = 100
-	c.MaxIdleConns = 20
-	c.ConnMaxLifetime = 3600
+	if c.Type == "" {
+		c.Type = "mysql"
+	}
+	if c.Port == 0 {
+		c.Port = 3306
+	}
+	if c.MaxOpenConns == 0 {
+		c.MaxOpenConns = 100
+	}
+	if c.MaxIdleConns == 0 {
+		c.MaxIdleConns = 20
+	}
+	if c.ConnMaxLifetime == 0 {
+		c.ConnMaxLifetime = 3600
+	}
 }
 
-// Load 从yaml数据加载
+// Load 从yaml数据加载 - 实现 ConfigLoader 接口
 func (c *DatabaseConfig) Load(data map[string]interface{}) error {
+	// 虽然可能不会被直接调用，但为了接口完整性还是要实现
 	return c.LoadFromYAML(data, c)
 }
 
-// Validate 验证配置
+// ========== 业务方法 ==========
+
+// Validate 验证配置（数据库配置）
 func (c *DatabaseConfig) Validate() error {
 	if c.Type == "" {
 		return fmt.Errorf("database type is required")
 	}
+	if c.Host == "" {
+		return fmt.Errorf("database host is required")
+	}
+	if c.Port <= 0 || c.Port > 65535 {
+		return fmt.Errorf("invalid database port: %d", c.Port)
+	}
+	if c.Database == "" {
+		return fmt.Errorf("database name is required")
+	}
+	return nil
+}
+
+// Validate 验证配置（多数据库配置）
+func (c *DatabasesConfig) Validate() error {
+	if c.Databases == nil || len(c.Databases) == 0 {
+		log.Println("❌ 错误: 未找到任何数据库配置，请检查配置文件中的 'databases' 配置")
+		return fmt.Errorf("no database configurations found")
+	}
+
+	// 必须要有默认数据库
+	if _, exists := c.Databases["default"]; !exists {
+		availableDBs := c.GetAllDatabaseNames()
+		log.Printf("❌ 错误: 默认数据库未找到。可用的数据库: %v。请在配置文件中添加 'default' 数据库", availableDBs)
+		return fmt.Errorf("default database not found")
+	}
+
+	// 验证每个数据库配置
+	for name, db := range c.Databases {
+		if db == nil {
+			log.Printf("❌ 错误: 数据库 '%s' 配置为空", name)
+			return fmt.Errorf("database '%s' configuration is nil", name)
+		}
+
+		if err := db.Validate(); err != nil {
+			log.Printf("❌ 错误: 数据库 '%s' 验证失败: %v", name, err)
+			return fmt.Errorf("database '%s' validation failed: %v", name, err)
+		}
+	}
+
+	log.Printf("✅ 数据库配置验证通过，找到 %d 个数据库配置", len(c.Databases))
 	return nil
 }
 
-// IsConfigured 判断是否已配置
+// IsConfigured 判断是否已配置（数据库配置）
 func (c *DatabaseConfig) IsConfigured() bool {
-	return c.Host != "" && c.Port > 0 && c.Username != "" && c.Database != ""
+	if c.Host == "" {
+		log.Println("⚠️  警告: 数据库 Host 未配置")
+		return false
+	}
+	if c.Port <= 0 {
+		log.Println("⚠️  警告: 数据库 Port 未配置或无效")
+		return false
+	}
+	if c.Username == "" {
+		log.Println("⚠️  警告: 数据库 Username 未配置")
+		return false
+	}
+	if c.Database == "" {
+		log.Println("⚠️  警告: 数据库名称未配置")
+		return false
+	}
+	return true
+}
+
+// IsConfigured 判断是否已配置（多数据库配置）
+func (c *DatabasesConfig) IsConfigured() bool {
+	defaultDB := c.GetDefaultDatabase()
+	if defaultDB == nil {
+		log.Println("❌ 错误: 默认数据库不存在")
+		return false
+	}
+
+	if !defaultDB.IsConfigured() {
+		log.Println("❌ 错误: 默认数据库配置不完整")
+		return false
+	}
+
+	return true
+}
+
+// GetDefaultDatabase 获取默认数据库（必须存在）
+func (c *DatabasesConfig) GetDefaultDatabase() *DatabaseConfig {
+	if c.Databases == nil {
+		log.Println("❌ 错误: 尝试获取默认数据库时，数据库配置为空")
+		return nil
+	}
+
+	if db, exists := c.Databases["default"]; exists {
+		return db
+	}
+
+	// 如果没有名为default的，尝试返回第一个（但会记录警告）
+	for name, db := range c.Databases {
+		log.Printf("⚠️  警告: 未找到名为 'default' 的数据库，使用第一个数据库 '%s' 作为默认", name)
+		return db
+	}
+
+	log.Println("❌ 错误: 数据库中没有任何配置")
+	return nil
+}
+
+// GetDatabase 按名称获取数据库（为空时记录错误）
+func (c *DatabasesConfig) GetDatabase(name string) *DatabaseConfig {
+	if c.Databases == nil {
+		log.Printf("❌ 错误: 尝试获取数据库 '%s' 时，数据库配置为空", name)
+		return nil
+	}
+
+	db := c.Databases[name]
+	if db == nil {
+		availableDBs := c.GetAllDatabaseNames()
+		log.Printf("❌ 错误: 数据库 '%s' 不存在。可用的数据库: %v", name, availableDBs)
+	}
+
+	return db
+}
+
+// GetAllDatabaseNames 获取所有数据库名称
+func (c *DatabasesConfig) GetAllDatabaseNames() []string {
+	if c.Databases == nil {
+		log.Println("⚠️  警告: 尝试获取数据库名称时，数据库配置为空")
+		return []string{}
+	}
+
+	names := make([]string, 0, len(c.Databases))
+	for name := range c.Databases {
+		names = append(names, name)
+	}
+	return names
+}
+
+// GetDatabaseNamesWithoutDefault 获取除default外的所有数据库名称
+func (c *DatabasesConfig) GetDatabaseNamesWithoutDefault() []string {
+	if c.Databases == nil {
+		return []string{}
+	}
+
+	names := make([]string, 0, len(c.Databases)-1)
+	for name := range c.Databases {
+		if name != "default" {
+			names = append(names, name)
+		}
+	}
+	return names
 }
 
 // 自动注册
 func init() {
-	Register("database", &DatabaseConfig{})
+	Register("databases", &DatabasesConfig{})
 }