134 lines
3.4 KiB
Go
134 lines
3.4 KiB
Go
package core
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"git.huangwc.com/pig/pig-farm-controller/internal/app/api"
|
|
"git.huangwc.com/pig/pig-farm-controller/internal/infra/config"
|
|
"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
|
|
)
|
|
|
|
// Application 是整个应用的核心，封装了所有组件和生命周期。
|
|
type Application struct {
|
|
Config *config.Config
|
|
Ctx context.Context
|
|
API *api.API
|
|
|
|
Infra *Infrastructure
|
|
Domain *DomainServices
|
|
App *AppServices
|
|
}
|
|
|
|
// 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)
|
|
}
|
|
// 初始化全局日志记录器
|
|
logs.InitDefaultLogger(cfg.Log)
|
|
|
|
// 为 Application 本身创建 Ctx
|
|
selfCtx := logs.AddCompName(context.Background(), "Application")
|
|
ctx := logs.AddFuncName(selfCtx, selfCtx, "NewApplication")
|
|
|
|
// 2. 初始化所有分层服务
|
|
infra, err := initInfrastructure(ctx, cfg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("初始化基础设施失败: %w", err)
|
|
}
|
|
domain := initDomainServices(ctx, cfg, infra)
|
|
appServices := initAppServices(ctx, infra, domain)
|
|
|
|
// 3. 初始化 API 入口点
|
|
apiServer := api.NewAPI(
|
|
cfg.Server,
|
|
infra.repos.userRepo,
|
|
appServices.pigFarmService,
|
|
appServices.pigBatchService,
|
|
appServices.monitorService,
|
|
appServices.deviceService,
|
|
appServices.planService,
|
|
appServices.userService,
|
|
infra.tokenService,
|
|
appServices.auditService,
|
|
infra.lora.listenHandler,
|
|
)
|
|
|
|
// 4. 组装 Application 对象
|
|
app := &Application{
|
|
Config: cfg,
|
|
Ctx: selfCtx,
|
|
API: apiServer,
|
|
Infra: infra,
|
|
Domain: domain,
|
|
App: appServices,
|
|
}
|
|
|
|
return app, nil
|
|
}
|
|
|
|
// Start 启动应用的所有组件并阻塞，直到接收到关闭信号。
|
|
func (app *Application) Start() error {
|
|
startCtx, logger := logs.Trace(app.Ctx, app.Ctx, "Start")
|
|
logger.Info("应用启动中...")
|
|
|
|
// 1. 启动底层监听器
|
|
if err := app.Infra.lora.loraListener.Listen(); err != nil {
|
|
return fmt.Errorf("启动 LoRa Mesh 监听器失败: %w", err)
|
|
}
|
|
|
|
// 2. 初始化应用状态 (清理、刷新任务等)
|
|
if err := app.initializeState(startCtx); err != nil {
|
|
return fmt.Errorf("初始化应用状态失败: %w", err)
|
|
}
|
|
|
|
// 3. 启动后台工作协程
|
|
app.Domain.planService.Start()
|
|
|
|
// 4. 启动 API 服务器
|
|
app.API.Start()
|
|
|
|
// 5. 等待关闭信号
|
|
quit := make(chan os.Signal, 1)
|
|
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
|
<-quit
|
|
|
|
// 接收到信号后，执行优雅关闭
|
|
return app.Stop()
|
|
}
|
|
|
|
// Stop 优雅地关闭应用的所有组件。
|
|
func (app *Application) Stop() error {
|
|
logger := logs.TraceLogger(app.Ctx, app.Ctx, "Stop")
|
|
logger.Info("应用关闭中...")
|
|
|
|
// 关闭 API 服务器
|
|
app.API.Stop()
|
|
|
|
// 关闭任务执行器
|
|
app.Domain.planService.Stop()
|
|
|
|
// 断开数据库连接
|
|
if err := app.Infra.storage.Disconnect(); err != nil {
|
|
logger.Errorw("数据库连接断开失败", "error", err)
|
|
}
|
|
|
|
// 关闭 LoRa Mesh 监听器
|
|
if err := app.Infra.lora.loraListener.Stop(); err != nil {
|
|
logger.Errorw("LoRa Mesh 监听器关闭失败", "error", err)
|
|
}
|
|
|
|
// 刷新日志缓冲区
|
|
_ = logger.Sync()
|
|
|
|
logger.Info("应用已成功关闭")
|
|
return nil
|
|
}
|