yml2213
b377f8350b
完成模块： - auth: 3 个 Repository 方法 + Service + Handler + Middleware - wallet: 已有 context 支持，修复依赖调用 - payment: 已有 context 支持，修复 wallet 调用 - adminaudit: 1 个方法 - notification: 2 个方法 - realname: 2 个方法 - systemconfig: 4 个方法 - adminauth: 7 个方法 - adminuser: 6 个方法 所有数据库调用已改为 r.db.WithContext(ctx)，完整传递 context 链路。 待完成模块: order, listing, chat 等 12 个模块（约 157 个方法）
146 lines
3.1 KiB
Go
146 lines
3.1 KiB
Go
package adminuser
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"hfb_sys/backend/internal/middleware"
|
|
"hfb_sys/backend/pkg/response"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type Handler struct {
|
|
service *Service
|
|
}
|
|
|
|
func NewHandler(service *Service) *Handler {
|
|
return &Handler{service: service}
|
|
}
|
|
|
|
func parsePagination(c *gin.Context) (int, int) {
|
|
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
|
|
pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "20"))
|
|
if page < 1 {
|
|
page = 1
|
|
}
|
|
if pageSize < 1 {
|
|
pageSize = 20
|
|
}
|
|
if pageSize > 100 {
|
|
pageSize = 100
|
|
}
|
|
return page, pageSize
|
|
}
|
|
|
|
func (h *Handler) List(c *gin.Context) {
|
|
page, pageSize := parsePagination(c)
|
|
result, err := h.service.List(c.Request.Context(), page, pageSize)
|
|
if err != nil {
|
|
writeAdminUserError(c, err)
|
|
return
|
|
}
|
|
response.OK(c, result)
|
|
}
|
|
|
|
func (h *Handler) Freeze(c *gin.Context) {
|
|
adminID, ok := currentAdminID(c)
|
|
if !ok {
|
|
response.Unauthorized(c, "缺少管理员上下文")
|
|
return
|
|
}
|
|
userID, ok := parseID(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
var req FreezeRequest
|
|
_ = c.ShouldBindJSON(&req)
|
|
item, err := h.service.Freeze(c.Request.Context(), adminID, userID, req, auditMeta(c))
|
|
if err != nil {
|
|
writeAdminUserError(c, err)
|
|
return
|
|
}
|
|
response.OK(c, item)
|
|
}
|
|
|
|
func (h *Handler) Unfreeze(c *gin.Context) {
|
|
adminID, ok := currentAdminID(c)
|
|
if !ok {
|
|
response.Unauthorized(c, "缺少管理员上下文")
|
|
return
|
|
}
|
|
userID, ok := parseID(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
item, err := h.service.Unfreeze(c.Request.Context(), adminID, userID, auditMeta(c))
|
|
if err != nil {
|
|
writeAdminUserError(c, err)
|
|
return
|
|
}
|
|
response.OK(c, item)
|
|
}
|
|
|
|
func (h *Handler) SetDepositFreeQuota(c *gin.Context) {
|
|
adminID, ok := currentAdminID(c)
|
|
if !ok {
|
|
response.Unauthorized(c, "缺少管理员上下文")
|
|
return
|
|
}
|
|
userID, ok := parseID(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
var req DepositFreeQuotaRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
response.BadRequest(c, "免押额度不正确")
|
|
return
|
|
}
|
|
item, err := h.service.SetDepositFreeQuota(c.Request.Context(), adminID, userID, req, auditMeta(c))
|
|
if err != nil {
|
|
writeAdminUserError(c, err)
|
|
return
|
|
}
|
|
response.OK(c, item)
|
|
}
|
|
|
|
func currentAdminID(c *gin.Context) (uint64, bool) {
|
|
value, ok := c.Get(middleware.ContextAdminID)
|
|
if !ok {
|
|
return 0, false
|
|
}
|
|
adminID, ok := value.(uint64)
|
|
return adminID, ok
|
|
}
|
|
|
|
func parseID(c *gin.Context) (uint64, bool) {
|
|
id, err := strconv.ParseUint(c.Param("id"), 10, 64)
|
|
if err != nil || id == 0 {
|
|
response.BadRequest(c, "ID 不正确")
|
|
return 0, false
|
|
}
|
|
return id, true
|
|
}
|
|
|
|
func auditMeta(c *gin.Context) AuditMeta {
|
|
return AuditMeta{
|
|
IP: c.ClientIP(),
|
|
UserAgent: c.GetHeader("User-Agent"),
|
|
RequestID: middleware.GetRequestID(c),
|
|
}
|
|
}
|
|
|
|
func writeAdminUserError(c *gin.Context, err error) {
|
|
switch {
|
|
case errors.Is(err, ErrDependencyUnavailable):
|
|
response.ServiceUnavailable(c, "数据库未连接")
|
|
case errors.Is(err, ErrInvalidUser):
|
|
response.BadRequest(c, "用户不符合规则")
|
|
case IsNotFound(err):
|
|
response.Error(c, http.StatusNotFound, "not_found", "用户不存在")
|
|
default:
|
|
response.Error(c, http.StatusInternalServerError, "internal_error", "用户管理服务暂时不可用")
|
|
}
|
|
}
|