xchou/go-skeleton
Template
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: 统一业务响应状态码约定

Browse Source
This commit is contained in:
xchou
2026-08-09 01:16:53 +08:00
parent 20e92dc82f
commit 66bc98ca8a
7 changed files with 106 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/auth/controller.go
+16 -20
View File
@@ -25,27 +25,25 @@ func NewController() *Controller { return &Controller{} }
// @Accept json
// @Produce json
// @Param body body Credentials true "注册信息"
// @Success 201 {object} utils.APIResponse{data=UserResponse}
// @Failure 400 {object} utils.APIResponse
// @Failure 409 {object} utils.APIResponse
// @Success 200 {object} utils.APIResponse{data=UserResponse}
// @Router /auth/register [post]
func (c *Controller) Register(ctx *gin.Context) {
var input Credentials
if err := ctx.ShouldBindJSON(&input); err != nil {
ctx.JSON(http.StatusBadRequest, utils.Failure(400, err.Error()))
ctx.JSON(http.StatusOK, utils.Failure(400, err.Error()))
return
}
hash, err := bcrypt.GenerateFromPassword([]byte(input.Password), bcrypt.DefaultCost)
if err != nil {
ctx.JSON(http.StatusInternalServerError, utils.Failure(500, "密码处理失败"))
ctx.JSON(http.StatusOK, utils.Failure(500, "密码处理失败"))
return
}
user := models.User{Username: input.Username, Password: string(hash)}
if err := database.GetDB().Create(&user).Error; err != nil {
ctx.JSON(http.StatusConflict, utils.Failure(409, "用户名已存在或数据无效"))
ctx.JSON(http.StatusOK, utils.Failure(409, "用户名已存在或数据无效"))
return
}
ctx.JSON(http.StatusCreated, utils.Success(UserResponse{ID: user.ID, Username: user.Username}))
ctx.JSON(http.StatusOK, utils.Success(UserResponse{ID: user.ID, Username: user.Username}))
}
// Login godoc
@@ -55,31 +53,30 @@ func (c *Controller) Register(ctx *gin.Context) {
// @Produce json
// @Param body body Credentials true "登录信息"
// @Success 200 {object} utils.APIResponse{data=middlewares.TokenPair}
// @Failure 401 {object} utils.APIResponse
// @Router /auth/login [post]
func (c *Controller) Login(ctx *gin.Context) {
var input Credentials
if err := ctx.ShouldBindJSON(&input); err != nil {
ctx.JSON(http.StatusBadRequest, utils.Failure(400, err.Error()))
ctx.JSON(http.StatusOK, utils.Failure(400, err.Error()))
return
}
var user models.User
err := database.GetDB().Where("username = ?", input.Username).First(&user).Error
if errors.Is(err, gorm.ErrRecordNotFound) {
ctx.JSON(http.StatusUnauthorized, utils.Failure(401, "用户名或密码错误"))
ctx.JSON(http.StatusOK, utils.Failure(401, "用户名或密码错误"))
return
}
if err != nil {
ctx.JSON(http.StatusInternalServerError, utils.Failure(500, "查询用户失败"))
ctx.JSON(http.StatusOK, utils.Failure(500, "查询用户失败"))
return
}
if bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(input.Password)) != nil {
ctx.JSON(http.StatusUnauthorized, utils.Failure(401, "用户名或密码错误"))
ctx.JSON(http.StatusOK, utils.Failure(401, "用户名或密码错误"))
return
}
pair, err := middlewares.GenerateTokenPair(int(user.ID), user.Username)
if err != nil {
ctx.JSON(http.StatusInternalServerError, utils.Failure(500, "生成 token 失败"))
ctx.JSON(http.StatusOK, utils.Failure(500, "生成 token 失败"))
return
}
ctx.JSON(http.StatusOK, utils.Success(pair))
@@ -92,31 +89,30 @@ func (c *Controller) Login(ctx *gin.Context) {
// @Produce json
// @Param body body RefreshRequest true "Refresh Token"
// @Success 200 {object} utils.APIResponse{data=middlewares.TokenPair}
// @Failure 401 {object} utils.APIResponse
// @Router /auth/refresh [post]
func (c *Controller) Refresh(ctx *gin.Context) {
var input RefreshRequest
if err := ctx.ShouldBindJSON(&input); err != nil {
ctx.JSON(http.StatusBadRequest, utils.Failure(400, err.Error()))
ctx.JSON(http.StatusOK, utils.Failure(400, err.Error()))
return
}
claims, err := middlewares.ParseToken(input.RefreshToken)
if err != nil || claims.TokenType != middlewares.TokenTypeRefresh {
ctx.JSON(http.StatusUnauthorized, utils.Failure(401, "refresh token 无效或已过期"))
ctx.JSON(http.StatusOK, utils.Failure(401, "refresh token 无效或已过期"))
return
}
revoked, err := database.IsTokenRevoked(ctx.Request.Context(), claims.ID)
if err != nil || revoked {
ctx.JSON(http.StatusUnauthorized, utils.Failure(401, "refresh token 已撤销"))
ctx.JSON(http.StatusOK, utils.Failure(401, "refresh token 已撤销"))
return
}
if err := revokeClaims(ctx, claims); err != nil {
ctx.JSON(http.StatusInternalServerError, utils.Failure(500, "撤销旧 token 失败"))
ctx.JSON(http.StatusOK, utils.Failure(500, "撤销旧 token 失败"))
return
}
pair, err := middlewares.GenerateTokenPair(claims.UserID, claims.Username)
if err != nil {
ctx.JSON(http.StatusInternalServerError, utils.Failure(500, "生成 token 失败"))
ctx.JSON(http.StatusOK, utils.Failure(500, "生成 token 失败"))
return
}
ctx.JSON(http.StatusOK, utils.Success(pair))
@@ -134,7 +130,7 @@ func (c *Controller) Refresh(ctx *gin.Context) {
func (c *Controller) Logout(ctx *gin.Context) {
claims, _ := middlewares.GetCurrentClaims(ctx)
if err := revokeClaims(ctx, claims); err != nil {
ctx.JSON(http.StatusInternalServerError, utils.Failure(500, "撤销 access token 失败"))
ctx.JSON(http.StatusOK, utils.Failure(500, "撤销 access token 失败"))
return
}
var input LogoutRequest