61 lines
1.6 KiB
Go
61 lines
1.6 KiB
Go
package middleware
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"tyc-server/app/main/model"
|
|
"tyc-server/common/ctxdata"
|
|
"tyc-server/common/result"
|
|
"tyc-server/common/xerr"
|
|
|
|
"github.com/zeromicro/go-zero/rest/httpx"
|
|
)
|
|
|
|
// UserDisableInterceptorMiddleware 检查用户是否被封禁，封禁用户直接拦截并返回错误
|
|
type UserDisableInterceptorMiddleware struct {
|
|
UserModel model.UserModel
|
|
}
|
|
|
|
// NewUserDisableInterceptorMiddleware 创建用户封禁检查中间件
|
|
func NewUserDisableInterceptorMiddleware(userModel model.UserModel) *UserDisableInterceptorMiddleware {
|
|
return &UserDisableInterceptorMiddleware{
|
|
UserModel: userModel,
|
|
}
|
|
}
|
|
|
|
// Handle 处理请求：当用户已登录且被封禁时，直接返回错误
|
|
func (m *UserDisableInterceptorMiddleware) Handle(next http.HandlerFunc) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
claims, err := ctxdata.GetClaimsFromCtx(r.Context())
|
|
if err != nil {
|
|
// 无登录信息，放行（由后续认证中间件处理）
|
|
next(w, r)
|
|
return
|
|
}
|
|
|
|
// 管理员不检查 user 表的 disable 字段
|
|
if claims.UserType == model.UserTypeAdmin {
|
|
next(w, r)
|
|
return
|
|
}
|
|
|
|
// 查询用户是否被封禁
|
|
user, err := m.UserModel.FindOne(r.Context(), claims.UserId)
|
|
if err != nil {
|
|
// 用户不存在（如管理员账号）或查询失败，放行
|
|
next(w, r)
|
|
return
|
|
}
|
|
|
|
// 1 表示已禁用
|
|
if user.Disable == 1 {
|
|
errcode := xerr.USER_DISABLED
|
|
errmsg := xerr.MapErrMsg(errcode)
|
|
httpx.WriteJson(w, http.StatusOK, result.Error(errcode, errmsg))
|
|
return
|
|
}
|
|
|
|
next(w, r)
|
|
}
|
|
}
|