diff --git a/internal/core/application.go b/internal/core/application.go new file mode 100644 index 0000000..80c67d6 --- /dev/null +++ b/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 +}