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/app/listener" "git.huangwc.com/pig/pig-farm-controller/internal/app/listener/chirp_stack" "git.huangwc.com/pig/pig-farm-controller/internal/infra/config" "git.huangwc.com/pig/pig-farm-controller/internal/infra/logs" "git.huangwc.com/pig/pig-farm-controller/internal/infra/transport/lora" ) // Application 是整个应用的核心,封装了所有组件和生命周期。 type Application struct { cfgPath string 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) 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, err := initDomainServices(ctx, cfg, infra) if err != nil { return nil, fmt.Errorf("初始化领域服务失败: %w", err) } appServices := initAppServices(ctx, infra, domain) // 3. 【核心组装逻辑】创建应用层监听器并注入到基础设施层 // 此时,所有依赖项(repos, domain services, comm)都已准备就绪 upstreamHandler := listener.NewLoRaListener( selfCtx, infra.repos.areaControllerRepo, infra.repos.pendingCollectionRepo, infra.repos.deviceRepo, infra.repos.sensorDataRepo, infra.repos.deviceCommandLogRepo, infra.repos.otaRepo, domain.deviceCommunicator, ) // 根据 LoRa 模式完成最终的绑定 if cfg.Lora.Mode == config.LoraMode_LoRaWAN { // 对于 LoRaWAN,创建真正的 Webhook 处理器并替换掉占位符 infra.lora.listenHandler = chirp_stack.NewChirpStackListener(selfCtx, upstreamHandler) } else { // 对于 LoRa Mesh,将处理器注入到已创建的 transport 实例中 if tp, ok := infra.lora.loraListener.(*lora.LoRaMeshUartPassthroughTransport); ok { tp.SetHandler(upstreamHandler) } } // 4. 初始化 API 入口点 (现在可以安全地传入 listenHandler) apiServer := api.NewAPI( cfg.Server, logs.AddCompName(context.Background(), "API"), infra.repos.userRepo, appServices.pigFarmService, appServices.pigBatchService, appServices.monitorService, appServices.deviceService, appServices.deviceTemplateService, appServices.areaControllerService, appServices.planService, appServices.userService, appServices.auditService, appServices.thresholdAlarmService, appServices.nutrientService, appServices.rawMaterialService, appServices.pigBreedService, appServices.pigAgeStageService, appServices.pigTypeService, appServices.recipeService, appServices.inventoryService, infra.tokenGenerator, infra.lora.listenHandler, // 此处传入的是最终组装好的 handler ) // 5. 组装 Application 对象 app := &Application{ cfgPath: configPath, 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(startCtx); err != nil { return fmt.Errorf("启动 LoRa Mesh 监听器失败: %w", err) } // 2. 初始化应用状态 (清理、刷新任务等) if err := app.initializeState(startCtx, app.cfgPath); err != nil { return fmt.Errorf("初始化应用状态失败: %w", err) } // 3. 启动后台工作协程 app.Domain.planService.Start(startCtx) // 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 { stopCtx, logger := logs.Trace(app.Ctx, app.Ctx, "Stop") logger.Info("应用关闭中...") // 关闭 API 服务器 app.API.Stop() // 关闭任务执行器 app.Domain.planService.Stop(stopCtx) // 断开数据库连接 if err := app.Infra.storage.Disconnect(stopCtx); err != nil { logger.Errorw("数据库连接断开失败", "error", err) } // 关闭 LoRa Mesh 监听器 if err := app.Infra.lora.loraListener.Stop(stopCtx); err != nil { logger.Errorw("LoRa Mesh 监听器关闭失败", "error", err) } // 刷新日志缓冲区 _ = logger.Sync() logger.Info("应用已成功关闭") return nil }