pig/pig-farm-controller
1
0
Fork 0
Code Issues 11 Pull Requests Actions Packages Projects Releases Wiki Activity

修改程序入口

Browse Source
This commit is contained in:
huang
2025-11-05 16:00:43 +08:00
parent bee5104661
commit aaa2f1b22f
5 changed files with 129 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
internal/core/application.go
View File

@@ -1,6 +1,7 @@
package core
import (
"context"
"fmt"
"os"
"os/signal"
@@ -14,7 +15,7 @@ import (
// Application 是整个应用的核心，封装了所有组件和生命周期。
type Application struct {
Config *config.Config
Logger *logs.Logger
Ctx context.Context
API *api.API
Infra *Infrastructure
@@ -30,20 +31,24 @@ func NewApplication(configPath string) (*Application, error) {
if err := cfg.Load(configPath); err != nil {
return nil, fmt.Errorf("无法加载配置: %w", err)
}
logger := logs.NewLogger(cfg.Log)
// 初始化全局日志记录器
logs.InitDefaultLogger(cfg.Log)
// 为 Application 本身创建 Ctx
selfCtx := logs.AddCompName(context.Background(), "Application")
ctx := logs.AddFuncName(selfCtx, selfCtx, "NewApplication")
// 2. 初始化所有分层服务
infra, err := initInfrastructure(cfg, logger)
infra, err := initInfrastructure(ctx, cfg)
if err != nil {
return nil, fmt.Errorf("初始化基础设施失败: %w", err)
}
domain := initDomainServices(cfg, infra, logger)
appServices := initAppServices(infra, domain, logger)
domain := initDomainServices(ctx, cfg, infra)
appServices := initAppServices(ctx, infra, domain)
// 3. 初始化 API 入口点
apiServer := api.NewAPI(
cfg.Server,
logger,
infra.repos.userRepo,
appServices.pigFarmService,
appServices.pigBatchService,
@@ -59,7 +64,7 @@ func NewApplication(configPath string) (*Application, error) {
// 4. 组装 Application 对象
app := &Application{
Config: cfg,
Logger: logger,
Ctx: selfCtx,
API: apiServer,
Infra: infra,
Domain: domain,
@@ -71,7 +76,8 @@ func NewApplication(configPath string) (*Application, error) {
// Start 启动应用的所有组件并阻塞，直到接收到关闭信号。
func (app *Application) Start() error {
app.Logger.Info("应用启动中...")
startCtx, logger := logs.Trace(app.Ctx, app.Ctx, "Start")
logger.Info("应用启动中...")
// 1. 启动底层监听器
if err := app.Infra.lora.loraListener.Listen(); err != nil {
@@ -79,7 +85,7 @@ func (app *Application) Start() error {
}
// 2. 初始化应用状态 (清理、刷新任务等)
if err := app.initializeState(); err != nil {
if err := app.initializeState(startCtx); err != nil {
return fmt.Errorf("初始化应用状态失败: %w", err)
}
@@ -100,7 +106,8 @@ func (app *Application) Start() error {
// Stop 优雅地关闭应用的所有组件。
func (app *Application) Stop() error {
app.Logger.Info("应用关闭中...")
logger := logs.TraceLogger(app.Ctx, app.Ctx, "Stop")
logger.Info("应用关闭中...")
// 关闭 API 服务器
app.API.Stop()
@@ -110,17 +117,17 @@ func (app *Application) Stop() error {
// 断开数据库连接
if err := app.Infra.storage.Disconnect(); err != nil {
app.Logger.Errorw("数据库连接断开失败", "error", err)
logger.Errorw("数据库连接断开失败", "error", err)
}
// 关闭 LoRa Mesh 监听器
if err := app.Infra.lora.loraListener.Stop(); err != nil {
app.Logger.Errorw("LoRa Mesh 监听器关闭失败", "error", err)
logger.Errorw("LoRa Mesh 监听器关闭失败", "error", err)
}
// 刷新日志缓冲区
_ = app.Logger.Sync()
_ = logger.Sync()
app.Logger.Info("应用已成功关闭")
logger.Info("应用已成功关闭")
return nil
}