go-base/bootstraps/bootstrap.go

package bootstraps

import (
	"log"

	"git.x2erp.com/qdy/go-base/config"
	"git.x2erp.com/qdy/go-base/logger"
	"git.x2erp.com/qdy/go-base/myservice"
	"git.x2erp.com/qdy/go-db/factory/database"
	"go-micro.dev/v4/web"
)

// ServiceBootstrapper 服务启动器
type ServiceBootstrapper struct {
	serviceName    string
	serviceVersion string
	Cfg            config.IConfig
	DbFactory      *database.DBFactory
	WebService     web.Service
}

// NewServiceBootstrapper 创建服务启动器
func NewServiceBootstrapper(name, version string) *ServiceBootstrapper {
	return &ServiceBootstrapper{
		serviceName:    name,
		serviceVersion: version,
	}
}

// 初始化配置
func (b *ServiceBootstrapper) InitConfig() {
	// 1. 初始化启动日志
	b.initBootLogger()
	defer b.closeBootLogger()

	// 2. 加载配置
	b.loadConfig()

	// 3. 初始化数据库
	//b.initDatabase()
	//defer b.cleanupDatabase()

	// 4. 初始化运行时日志
	b.initRuntimeLogger()

	// 5. 设置优雅关闭
	b.setupGracefulShutdown()

}

// Run 启动服务
func (b *ServiceBootstrapper) Run(routeRegistrar func(web.Service, *database.DBFactory)) {

	if b.DbFactory == nil {
		log.Fatal("数据库工厂未初始化，请先调用InitDatabase()")
	}

	// 6. 启动微服务
	b.startWebService()

	// 7. 注册路由
	routeRegistrar(b.WebService, b.DbFactory)

	// 8. 记录服务启动信息
	b.logServiceInfo()

	// 9. 运行服务
	b.runWebService()
}

// initBootLogger 初始化启动日志
func (b *ServiceBootstrapper) initBootLogger() {
	if err := logger.InitBootLog(b.serviceName); err != nil {
		log.Fatal("无法初始化启动日志: ", err)
	}
}

// closeBootLogger 关闭启动日志
func (b *ServiceBootstrapper) closeBootLogger() {
	logger.CloseBootLogger()
}

// loadConfig 加载配置
func (b *ServiceBootstrapper) loadConfig() {

	log.Printf("开始加载配置...")

	cfg, err := config.GetConfig()
	if err != nil {
		log.Fatalf("加载配置文件错误: %v", err)
	}

	// 设置版本
	cfg.SetServiceName(b.serviceName)
	cfg.SetServiceVersion(b.serviceVersion)

	b.Cfg = cfg
}

// initDatabase 初始化数据库
func (b *ServiceBootstrapper) InitDatabase() {

	// 记录配置信息
	//serviceCfg := cfg.GetService()
	//microCfg := cfg.GetMicro()
	dbCfg := b.Cfg.GetDatabase()

	//log.Printf("服务名称: %s", serviceCfg.ServiceName)
	//log.Printf("服务端口: %d", serviceCfg.Port)
	//log.Printf("注册中心: %s", microCfg.RegistryAddress)
	log.Printf("数据库: %s:%d/%s", dbCfg.Host, dbCfg.Port, dbCfg.Database)

	log.Printf("初始化数据库连接...")

	dbFactory, err := database.GetDBFactory()
	if err != nil {
		log.Fatalf("数据库连接失败: %v", err)
	}
	b.DbFactory = dbFactory

	log.Printf("数据库连接测试通过!")

}

// cleanupDatabase 清理数据库资源
func (b *ServiceBootstrapper) CleanupDatabase() {
	if b.DbFactory != nil {
		if err := b.DbFactory.Close(); err != nil {
			logger.Error("数据库关闭错误: %v", err)
		}
	}
}

// initRuntimeLogger 初始化运行时日志
func (b *ServiceBootstrapper) initRuntimeLogger() {
	logger.InitRuntimeLogger(b.serviceName, b.serviceVersion, b.Cfg.GetLog())
	log.Printf("运行时日志系统初始化完成")
	log.Printf("开始启动微服务...")

	// 记录服务模式
	mode := "独立模式"
	if b.Cfg.GetMicro().RegistryAddress != "" {
		mode = "使用服务发现 (注册中心)"
	}
	log.Printf("服务模式: %s", mode)
}

// setupGracefulShutdown 设置优雅关闭
func (b *ServiceBootstrapper) setupGracefulShutdown() {
	setupGracefulShutdown(b.DbFactory, logger.StopESWriter)
}

// startWebService 启动Web服务
func (b *ServiceBootstrapper) startWebService() {
	microCfg := b.Cfg.GetMicro()
	serviceCfg := b.Cfg.GetService()

	var svc web.Service
	if microCfg.RegistryAddress != "" {
		svc = myservice.Start(b.Cfg)
	} else {
		svc = myservice.StartStandalone(b.Cfg)
	}

	b.WebService = svc
	log.Printf("Web服务实例创建完成 (端口: %d)", serviceCfg.Port)
}

// logServiceInfo 记录服务启动信息
func (b *ServiceBootstrapper) logServiceInfo() {
	serviceCfg := b.Cfg.GetService()

	log.Printf("服务启动成功")

	if serviceCfg.TrustedProxies != "" {
		log.Printf("   • 受信代理: %s", serviceCfg.TrustedProxies)
	}
}

// runWebService 运行Web服务
func (b *ServiceBootstrapper) runWebService() {
	// 测试日志
	logger.Debug("测试日志是否写入ES.")

	log.Printf("服务开始运行...")
	if err := b.WebService.Run(); err != nil {
		log.Fatal("服务运行失败:", err)
	}
}