用户登录和接口鉴权

This commit is contained in:
2025-09-07 21:13:15 +08:00
parent c4522b974b
commit 565cf3fa6a
380 changed files with 18330 additions and 16854 deletions

View File

@@ -10,7 +10,9 @@ import (
"time"
"git.huangwc.com/pig/pig-farm-controller/internal/config"
"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"
"github.com/gin-gonic/gin"
)
@@ -26,13 +28,16 @@ type API struct {
// config 应用配置
config *config.Config
// userController 用户控制器
userController *user.Controller
// logger 日志记录器
logger *logs.Logger
}
// NewAPI 创建并返回一个新的API实例
// 初始化Gin引擎和相关配置
func NewAPI(cfg *config.Config) *API {
func NewAPI(cfg *config.Config, userRepo repository.UserRepo) *API {
// 设置Gin为发布模式
gin.SetMode(gin.ReleaseMode)
@@ -56,10 +61,14 @@ func NewAPI(cfg *config.Config) *API {
engine.Use(gin.Recovery())
// 创建用户控制器
userController := user.NewController(userRepo)
return &API{
engine: engine,
config: cfg,
logger: logs.NewLogger(),
engine: engine,
config: cfg,
userController: userController,
logger: logs.NewLogger(),
}
}
@@ -79,11 +88,11 @@ func (a *API) Start() error {
}
// 启动HTTP服务器
a.logger.Info(fmt.Sprintf("Starting HTTP server on %s:%d", a.config.Server.Host, a.config.Server.Port))
a.logger.Info(fmt.Sprintf("正在启动HTTP服务器 %s:%d", a.config.Server.Host, a.config.Server.Port))
go func() {
if err := a.server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
a.logger.Error(fmt.Sprintf("HTTP server startup failed: %v", err))
a.logger.Error(fmt.Sprintf("HTTP服务器启动失败: %v", err))
}
}()
@@ -92,7 +101,7 @@ func (a *API) Start() error {
// Stop 停止HTTP服务器
func (a *API) Stop() error {
a.logger.Info("Stopping HTTP server")
a.logger.Info("正在停止HTTP服务器")
// 创建一个5秒的超时上下文
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
@@ -100,11 +109,11 @@ func (a *API) Stop() error {
// 优雅地关闭服务器
if err := a.server.Shutdown(ctx); err != nil {
a.logger.Error(fmt.Sprintf("HTTP server shutdown error: %v", err))
a.logger.Error(fmt.Sprintf("HTTP服务器关闭错误: %v", err))
return err
}
a.logger.Info("HTTP server stopped")
a.logger.Info("HTTP服务器已停止")
return nil
}
@@ -113,6 +122,13 @@ func (a *API) setupRoutes() {
// 基础路由示例
a.engine.GET("/health", a.healthHandler)
// 用户相关路由
userGroup := a.engine.Group("/api/v1/user")
{
userGroup.POST("/register", a.userController.Register)
userGroup.POST("/login", a.userController.Login)
}
// TODO: 添加更多路由
}
@@ -127,6 +143,6 @@ func (a *API) setupRoutes() {
func (a *API) healthHandler(c *gin.Context) {
c.JSON(http.StatusOK, gin.H{
"status": "ok",
"message": "Pig Farm Controller API is running",
"message": "猪场控制器API正在运行",
})
}