部分完成任务6(先提交然后修bug)

This commit is contained in:
2025-10-30 17:44:34 +08:00
parent 8ce553a9e4
commit f624a8bf5e
5 changed files with 56 additions and 50 deletions

View File

@@ -35,12 +35,13 @@ import (
"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
"git.huangwc.com/pig/pig-farm-controller/internal/infra/repository"
"github.com/gin-gonic/gin"
"github.com/labstack/echo/v4"
"github.com/labstack/echo/v4/middleware"
)
// API 结构体定义了 HTTP 服务器及其依赖
type API struct {
engine *gin.Engine // Gin 引擎实例,用于处理 HTTP 请求
echo *echo.Echo // Echo 引擎实例,用于处理 HTTP 请求
logger *logs.Logger // 日志记录器,用于输出日志信息
userRepo repository.UserRepository // 用户数据仓库接口,用于用户数据操作
tokenService token.Service // Token 服务接口,用于 JWT token 的生成和解析
@@ -58,7 +59,7 @@ type API struct {
}
// NewAPI 创建并返回一个新的 API 实例
// 负责初始化 Gin 引擎、设置全局中间件,并注入所有必要的依赖。
// 负责初始化 Echo 引擎、设置全局中间件,并注入所有必要的依赖。
func NewAPI(cfg config.ServerConfig,
logger *logs.Logger,
userRepo repository.UserRepository,
@@ -75,21 +76,19 @@ func NewAPI(cfg config.ServerConfig,
deviceService domain_device.Service,
listenHandler webhook.ListenHandler,
analysisTaskManager *scheduler.AnalysisPlanTaskManager) *API {
// 设置 Gin 模式,例如 gin.ReleaseMode (生产模式) 或 gin.DebugMode (开发模式)
// 从配置中获取 Gin 模式
gin.SetMode(cfg.Mode)
// 使用 echo.New() 创建一个 Echo 引擎实例
e := echo.New()
// 使用 gin.New() 创建一个 Gin 引擎实例,而不是 gin.Default()
// 这样可以手动添加所需的中间件,避免 gin.Default() 默认包含的 Logger 和 Recovery 中间件
engine := gin.New()
// 根据配置设置 Echo 的调试模式
e.Debug = cfg.Mode == "debug"
// 添加 Gin Recovery 中间件,用于捕获 panic 并恢复,防止服务崩溃
// gin.Logger() 已移除,因为我们使用自定义的 logger
engine.Use(gin.Recovery())
// 添加 Echo Recovery 中间件,用于捕获 panic 并恢复,防止服务崩溃
// Echo 的 Logger 中间件默认会记录请求信息,如果需要自定义,可以替换
e.Use(middleware.Recover())
// 初始化 API 结构体
api := &API{
engine: engine,
echo: e,
logger: logger,
userRepo: userRepo,
tokenService: tokenService,
@@ -123,8 +122,8 @@ func (a *API) Start() {
// 初始化标准库的 http.Server 实例
a.httpServer = &http.Server{
Addr: addr, // 服务器监听的地址从配置中获取
Handler: a.engine, // 将 Gin 引擎作为 HTTP 请求的处理程序
Addr: addr, // 服务器监听的地址从配置中获取
Handler: a.echo, // 将 Echo 引擎作为 HTTP 请求的处理程序
}
// 在独立的 goroutine 中启动服务器