重写application包

internal/core/application.go

@@ -0,0 +1,107 @@
+// Package core 提供了应用的核心结构和生命周期管理。
+package core
+
+import (
+	"fmt"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"git.huangwc.com/pig/pig-farm-controller/internal/config"
+	"git.huangwc.com/pig/pig-farm-controller/internal/logs"
+	"git.huangwc.com/pig/pig-farm-controller/internal/storage/db"
+	"git.huangwc.com/pig/pig-farm-controller/internal/task"
+)
+
+// Application 是整个应用的核心，封装了所有组件和生命周期。
+type Application struct {
+	Config   *config.Config
+	Logger   *logs.Logger
+	Storage  db.Storage
+	Executor *task.Executor
+}
+
+// NewApplication 创建并初始化一个新的 Application 实例。
+// 这是应用的“组合根”，所有依赖都在这里被创建和注入。
+func NewApplication(configPath string) (*Application, error) {
+	// 1. 加载配置
+	cfg := config.NewConfig()
+	if err := cfg.Load(configPath); err != nil {
+		return nil, fmt.Errorf("无法加载配置: %w", err)
+	}
+
+	// 2. 初始化日志记录器
+	logger := logs.NewLogger(cfg.Log)
+
+	// 3. 初始化数据库存储
+	storage, err := initStorage(cfg.Database, logger)
+	if err != nil {
+		return nil, err // 错误已在 initStorage 中被包装
+	}
+
+	// 4. 初始化任务执行器
+	executor := task.NewExecutor(cfg.Heartbeat.Concurrency, logger)
+
+	// 5. 组装 Application 对象
+	app := &Application{
+		Config:   cfg,
+		Logger:   logger,
+		Storage:  storage,
+		Executor: executor,
+	}
+
+	return app, nil
+}
+
+// Start 启动应用的所有组件并阻塞，直到接收到关闭信号。
+func (app *Application) Start() error {
+	app.Logger.Info("应用启动中...")
+
+	// 启动任务执行器
+	app.Executor.Start()
+
+	// 等待关闭信号
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+	<-quit
+
+	// 接收到信号后，执行优雅关闭
+	return app.Stop()
+}
+
+// Stop 优雅地关闭应用的所有组件。
+func (app *Application) Stop() error {
+	app.Logger.Info("应用关闭中...")
+
+	// 关闭任务执行器
+	app.Executor.Stop()
+
+	// 断开数据库连接
+	if err := app.Storage.Disconnect(); err != nil {
+		app.Logger.Errorw("数据库连接断开失败", "error", err)
+	}
+
+	// 刷新日志缓冲区
+	_ = app.Logger.Sync()
+
+	app.Logger.Info("应用已成功关闭")
+	return nil
+}
+
+// initStorage 封装了数据库的初始化、连接和迁移逻辑。
+func initStorage(cfg config.DatabaseConfig, logger *logs.Logger) (db.Storage, error) {
+	// 创建存储实例
+	storage := db.NewStorage(cfg, logger)
+	if err := storage.Connect(); err != nil {
+		// 错误已在 Connect 内部被记录，这里只需包装并返回
+		return nil, fmt.Errorf("数据库连接失败: %w", err)
+	}
+
+	// 执行数据库迁移
+	var dbModels = []interface{}{ /* &User{}, &Product{} */ }
+	if err := storage.Migrate(dbModels...); err != nil {
+		return nil, fmt.Errorf("数据库迁移失败: %w", err)
+	}
+
+	return storage, nil
+}