增加操作历史model

This commit is contained in:
2025-09-07 21:18:28 +08:00
parent 565cf3fa6a
commit d4ce0aa238
6 changed files with 405 additions and 12 deletions

View File

@@ -9,7 +9,9 @@ import (
"net/http"
"time"
"git.huangwc.com/pig/pig-farm-controller/internal/api/middleware"
"git.huangwc.com/pig/pig-farm-controller/internal/config"
"git.huangwc.com/pig/pig-farm-controller/internal/controller/operation"
"git.huangwc.com/pig/pig-farm-controller/internal/controller/user"
"git.huangwc.com/pig/pig-farm-controller/internal/logs"
"git.huangwc.com/pig/pig-farm-controller/internal/storage/repository"
@@ -31,13 +33,19 @@ type API struct {
// userController 用户控制器
userController *user.Controller
// operationController 操作历史控制器
operationController *operation.Controller
// authMiddleware 鉴权中间件
authMiddleware *middleware.AuthMiddleware
// logger 日志记录器
logger *logs.Logger
}
// NewAPI 创建并返回一个新的API实例
// 初始化Gin引擎和相关配置
func NewAPI(cfg *config.Config, userRepo repository.UserRepo) *API {
func NewAPI(cfg *config.Config, userRepo repository.UserRepo, operationHistoryRepo repository.OperationHistoryRepo) *API {
// 设置Gin为发布模式
gin.SetMode(gin.ReleaseMode)
@@ -64,11 +72,19 @@ func NewAPI(cfg *config.Config, userRepo repository.UserRepo) *API {
// 创建用户控制器
userController := user.NewController(userRepo)
// 创建操作历史控制器
operationController := operation.NewController(operationHistoryRepo)
// 创建鉴权中间件
authMiddleware := middleware.NewAuthMiddleware(userRepo)
return &API{
engine: engine,
config: cfg,
userController: userController,
logger: logs.NewLogger(),
engine: engine,
config: cfg,
userController: userController,
operationController: operationController,
authMiddleware: authMiddleware,
logger: logs.NewLogger(),
}
}
@@ -129,6 +145,43 @@ func (a *API) setupRoutes() {
userGroup.POST("/login", a.userController.Login)
}
// 需要鉴权的路由组
protectedGroup := a.engine.Group("/api/v1")
protectedGroup.Use(a.authMiddleware.Handle())
{
// 操作历史相关路由
operationGroup := protectedGroup.Group("/operation")
{
operationGroup.POST("/", a.operationController.Create)
operationGroup.GET("/list", a.operationController.ListByUser)
operationGroup.GET("/:id", a.operationController.Get)
}
// 示例受保护路由
protectedGroup.GET("/profile", func(c *gin.Context) {
// 从上下文中获取用户信息
userValue, exists := c.Get("user")
if !exists {
c.JSON(http.StatusUnauthorized, gin.H{"error": "无法获取用户信息"})
return
}
user, ok := userValue.(*middleware.AuthUser)
if !ok {
c.JSON(http.StatusInternalServerError, gin.H{"error": "用户信息格式错误"})
return
}
c.JSON(http.StatusOK, gin.H{
"message": "获取用户信息成功",
"user": map[string]interface{}{
"id": user.ID,
"username": user.Username,
},
})
})
}
// TODO: 添加更多路由
}