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

This commit is contained in:
2026-08-09 01:16:53 +08:00
parent 20e92dc82f
commit 66bc98ca8a
7 changed files with 106 additions and 92 deletions
+2 -26
View File
@@ -57,12 +57,6 @@ const docTemplate = `{
} }
] ]
} }
},
"401": {
"description": "Unauthorized",
"schema": {
"$ref": "#/definitions/utils.APIResponse"
}
} }
} }
} }
@@ -108,12 +102,6 @@ const docTemplate = `{
} }
] ]
} }
},
"401": {
"description": "Unauthorized",
"schema": {
"$ref": "#/definitions/utils.APIResponse"
}
} }
} }
} }
@@ -142,8 +130,8 @@ const docTemplate = `{
} }
], ],
"responses": { "responses": {
"201": { "200": {
"description": "Created", "description": "OK",
"schema": { "schema": {
"allOf": [ "allOf": [
{ {
@@ -159,18 +147,6 @@ const docTemplate = `{
} }
] ]
} }
},
"400": {
"description": "Bad Request",
"schema": {
"$ref": "#/definitions/utils.APIResponse"
}
},
"409": {
"description": "Conflict",
"schema": {
"$ref": "#/definitions/utils.APIResponse"
}
} }
} }
} }
+2 -26
View File
@@ -49,12 +49,6 @@
} }
] ]
} }
},
"401": {
"description": "Unauthorized",
"schema": {
"$ref": "#/definitions/utils.APIResponse"
}
} }
} }
} }
@@ -100,12 +94,6 @@
} }
] ]
} }
},
"401": {
"description": "Unauthorized",
"schema": {
"$ref": "#/definitions/utils.APIResponse"
}
} }
} }
} }
@@ -134,8 +122,8 @@
} }
], ],
"responses": { "responses": {
"201": { "200": {
"description": "Created", "description": "OK",
"schema": { "schema": {
"allOf": [ "allOf": [
{ {
@@ -151,18 +139,6 @@
} }
] ]
} }
},
"400": {
"description": "Bad Request",
"schema": {
"$ref": "#/definitions/utils.APIResponse"
}
},
"409": {
"description": "Conflict",
"schema": {
"$ref": "#/definitions/utils.APIResponse"
}
} }
} }
} }
+2 -18
View File
@@ -108,10 +108,6 @@ paths:
data: data:
$ref: '#/definitions/middlewares.TokenPair' $ref: '#/definitions/middlewares.TokenPair'
type: object type: object
"401":
description: Unauthorized
schema:
$ref: '#/definitions/utils.APIResponse'
summary: 用户登录 summary: 用户登录
tags: tags:
- auth - auth
@@ -138,10 +134,6 @@ paths:
data: data:
$ref: '#/definitions/middlewares.TokenPair' $ref: '#/definitions/middlewares.TokenPair'
type: object type: object
"401":
description: Unauthorized
schema:
$ref: '#/definitions/utils.APIResponse'
summary: 刷新并轮换 Token summary: 刷新并轮换 Token
tags: tags:
- auth - auth
@@ -159,8 +151,8 @@ paths:
produces: produces:
- application/json - application/json
responses: responses:
"201": "200":
description: Created description: OK
schema: schema:
allOf: allOf:
- $ref: '#/definitions/utils.APIResponse' - $ref: '#/definitions/utils.APIResponse'
@@ -168,14 +160,6 @@ paths:
data: data:
$ref: '#/definitions/auth.UserResponse' $ref: '#/definitions/auth.UserResponse'
type: object type: object
"400":
description: Bad Request
schema:
$ref: '#/definitions/utils.APIResponse'
"409":
description: Conflict
schema:
$ref: '#/definitions/utils.APIResponse'
summary: 注册示例用户 summary: 注册示例用户
tags: tags:
- auth - auth
+1 -1
View File
@@ -136,7 +136,7 @@ func BearerToken(header string) (string, error) {
func abortUnauthorized(c *gin.Context, message string) { func abortUnauthorized(c *gin.Context, message string) {
Logger.Warn("JWT认证失败", zap.String("message", message), zap.String("path", c.Request.URL.Path)) Logger.Warn("JWT认证失败", zap.String("message", message), zap.String("path", c.Request.URL.Path))
c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"code": 401, "message": message, "data": nil}) c.AbortWithStatusJSON(http.StatusOK, gin.H{"code": 401, "message": message, "data": nil})
} }
func GetCurrentUser(c *gin.Context) (int, string, bool) { func GetCurrentUser(c *gin.Context) (int, string, bool) {
+22 -1
View File
@@ -1,6 +1,13 @@
package middlewares package middlewares
import "testing" import (
"net/http"
"net/http/httptest"
"testing"
"github.com/gin-gonic/gin"
"go.uber.org/zap"
)
func TestGenerateTokenPair(t *testing.T) { func TestGenerateTokenPair(t *testing.T) {
InitJWT("test-secret", 15, 24, "test-issuer") InitJWT("test-secret", 15, 24, "test-issuer")
@@ -24,6 +31,20 @@ func TestGenerateTokenPair(t *testing.T) {
} }
} }
func TestJWTAuthFailureUsesHTTP200(t *testing.T) {
Logger = zap.NewNop()
gin.SetMode(gin.TestMode)
router := gin.New()
router.GET("/private", JWTAuth(), func(c *gin.Context) {
c.Status(http.StatusNoContent)
})
recorder := httptest.NewRecorder()
router.ServeHTTP(recorder, httptest.NewRequest(http.MethodGet, "/private", nil))
if recorder.Code != http.StatusOK {
t.Fatalf("HTTP status = %d, want 200", recorder.Code)
}
}
func TestBearerToken(t *testing.T) { func TestBearerToken(t *testing.T) {
if token, err := BearerToken("Bearer abc"); err != nil || token != "abc" { if token, err := BearerToken("Bearer abc"); err != nil || token != "abc" {
t.Fatalf("BearerToken() = %q, %v", token, err) t.Fatalf("BearerToken() = %q, %v", token, err)
+16 -20
View File
@@ -25,27 +25,25 @@ func NewController() *Controller { return &Controller{} }
// @Accept json // @Accept json
// @Produce json // @Produce json
// @Param body body Credentials true "注册信息" // @Param body body Credentials true "注册信息"
// @Success 201 {object} utils.APIResponse{data=UserResponse} // @Success 200 {object} utils.APIResponse{data=UserResponse}
// @Failure 400 {object} utils.APIResponse
// @Failure 409 {object} utils.APIResponse
// @Router /auth/register [post] // @Router /auth/register [post]
func (c *Controller) Register(ctx *gin.Context) { func (c *Controller) Register(ctx *gin.Context) {
var input Credentials var input Credentials
if err := ctx.ShouldBindJSON(&input); err != nil { 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 return
} }
hash, err := bcrypt.GenerateFromPassword([]byte(input.Password), bcrypt.DefaultCost) hash, err := bcrypt.GenerateFromPassword([]byte(input.Password), bcrypt.DefaultCost)
if err != nil { if err != nil {
ctx.JSON(http.StatusInternalServerError, utils.Failure(500, "密码处理失败")) ctx.JSON(http.StatusOK, utils.Failure(500, "密码处理失败"))
return return
} }
user := models.User{Username: input.Username, Password: string(hash)} user := models.User{Username: input.Username, Password: string(hash)}
if err := database.GetDB().Create(&user).Error; err != nil { if err := database.GetDB().Create(&user).Error; err != nil {
ctx.JSON(http.StatusConflict, utils.Failure(409, "用户名已存在或数据无效")) ctx.JSON(http.StatusOK, utils.Failure(409, "用户名已存在或数据无效"))
return 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 // Login godoc
@@ -55,31 +53,30 @@ func (c *Controller) Register(ctx *gin.Context) {
// @Produce json // @Produce json
// @Param body body Credentials true "登录信息" // @Param body body Credentials true "登录信息"
// @Success 200 {object} utils.APIResponse{data=middlewares.TokenPair} // @Success 200 {object} utils.APIResponse{data=middlewares.TokenPair}
// @Failure 401 {object} utils.APIResponse
// @Router /auth/login [post] // @Router /auth/login [post]
func (c *Controller) Login(ctx *gin.Context) { func (c *Controller) Login(ctx *gin.Context) {
var input Credentials var input Credentials
if err := ctx.ShouldBindJSON(&input); err != nil { 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 return
} }
var user models.User var user models.User
err := database.GetDB().Where("username = ?", input.Username).First(&user).Error err := database.GetDB().Where("username = ?", input.Username).First(&user).Error
if errors.Is(err, gorm.ErrRecordNotFound) { if errors.Is(err, gorm.ErrRecordNotFound) {
ctx.JSON(http.StatusUnauthorized, utils.Failure(401, "用户名或密码错误")) ctx.JSON(http.StatusOK, utils.Failure(401, "用户名或密码错误"))
return return
} }
if err != nil { if err != nil {
ctx.JSON(http.StatusInternalServerError, utils.Failure(500, "查询用户失败")) ctx.JSON(http.StatusOK, utils.Failure(500, "查询用户失败"))
return return
} }
if bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(input.Password)) != nil { 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 return
} }
pair, err := middlewares.GenerateTokenPair(int(user.ID), user.Username) pair, err := middlewares.GenerateTokenPair(int(user.ID), user.Username)
if err != nil { if err != nil {
ctx.JSON(http.StatusInternalServerError, utils.Failure(500, "生成 token 失败")) ctx.JSON(http.StatusOK, utils.Failure(500, "生成 token 失败"))
return return
} }
ctx.JSON(http.StatusOK, utils.Success(pair)) ctx.JSON(http.StatusOK, utils.Success(pair))
@@ -92,31 +89,30 @@ func (c *Controller) Login(ctx *gin.Context) {
// @Produce json // @Produce json
// @Param body body RefreshRequest true "Refresh Token" // @Param body body RefreshRequest true "Refresh Token"
// @Success 200 {object} utils.APIResponse{data=middlewares.TokenPair} // @Success 200 {object} utils.APIResponse{data=middlewares.TokenPair}
// @Failure 401 {object} utils.APIResponse
// @Router /auth/refresh [post] // @Router /auth/refresh [post]
func (c *Controller) Refresh(ctx *gin.Context) { func (c *Controller) Refresh(ctx *gin.Context) {
var input RefreshRequest var input RefreshRequest
if err := ctx.ShouldBindJSON(&input); err != nil { 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 return
} }
claims, err := middlewares.ParseToken(input.RefreshToken) claims, err := middlewares.ParseToken(input.RefreshToken)
if err != nil || claims.TokenType != middlewares.TokenTypeRefresh { 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 return
} }
revoked, err := database.IsTokenRevoked(ctx.Request.Context(), claims.ID) revoked, err := database.IsTokenRevoked(ctx.Request.Context(), claims.ID)
if err != nil || revoked { if err != nil || revoked {
ctx.JSON(http.StatusUnauthorized, utils.Failure(401, "refresh token 已撤销")) ctx.JSON(http.StatusOK, utils.Failure(401, "refresh token 已撤销"))
return return
} }
if err := revokeClaims(ctx, claims); err != nil { if err := revokeClaims(ctx, claims); err != nil {
ctx.JSON(http.StatusInternalServerError, utils.Failure(500, "撤销旧 token 失败")) ctx.JSON(http.StatusOK, utils.Failure(500, "撤销旧 token 失败"))
return return
} }
pair, err := middlewares.GenerateTokenPair(claims.UserID, claims.Username) pair, err := middlewares.GenerateTokenPair(claims.UserID, claims.Username)
if err != nil { if err != nil {
ctx.JSON(http.StatusInternalServerError, utils.Failure(500, "生成 token 失败")) ctx.JSON(http.StatusOK, utils.Failure(500, "生成 token 失败"))
return return
} }
ctx.JSON(http.StatusOK, utils.Success(pair)) 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) { func (c *Controller) Logout(ctx *gin.Context) {
claims, _ := middlewares.GetCurrentClaims(ctx) claims, _ := middlewares.GetCurrentClaims(ctx)
if err := revokeClaims(ctx, claims); err != nil { 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 return
} }
var input LogoutRequest var input LogoutRequest
+61
View File
@@ -0,0 +1,61 @@
//go:build cgo
package auth
import (
"bytes"
"encoding/json"
"net/http"
"net/http/httptest"
"testing"
"skeleton/config"
"skeleton/database"
"skeleton/middlewares"
"skeleton/models"
"github.com/gin-gonic/gin"
"go.uber.org/zap"
)
func TestBusinessErrorsUseHTTP200(t *testing.T) {
cfg := config.DatabaseConfig{Driver: "sqlite", DSN: ":memory:", MaxIdleConns: 1, MaxOpenConns: 1}
if err := database.Init(&cfg, zap.NewNop()); err != nil {
t.Fatal(err)
}
t.Cleanup(func() { _ = database.Close() })
if err := database.GetDB().AutoMigrate(&models.User{}); err != nil {
t.Fatal(err)
}
middlewares.InitJWT("test-secret", 15, 24, "test")
gin.SetMode(gin.TestMode)
controller := NewController()
router := gin.New()
router.POST("/register", controller.Register)
router.POST("/login", controller.Login)
assertBusinessResponse(t, router, "/register", `{"username":"x","password":"short"}`, 400)
assertBusinessResponse(t, router, "/register", `{"username":"demo","password":"change-me-123"}`, 200)
assertBusinessResponse(t, router, "/login", `{"username":"demo","password":"wrong-password"}`, 401)
}
func assertBusinessResponse(t *testing.T, handler http.Handler, path, body string, wantCode int) {
t.Helper()
request := httptest.NewRequest(http.MethodPost, path, bytes.NewBufferString(body))
request.Header.Set("Content-Type", "application/json")
recorder := httptest.NewRecorder()
handler.ServeHTTP(recorder, request)
if recorder.Code != http.StatusOK {
t.Fatalf("HTTP status = %d, want 200; body=%s", recorder.Code, recorder.Body.String())
}
var response struct {
Code int `json:"code"`
}
if err := json.Unmarshal(recorder.Body.Bytes(), &response); err != nil {
t.Fatal(err)
}
if response.Code != wantCode {
t.Fatalf("business code = %d, want %d; body=%s", response.Code, wantCode, recorder.Body.String())
}
}