138 lines
4.0 KiB
Go
138 lines
4.0 KiB
Go
package coinledger
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"hyapp-admin-server/internal/appctx"
|
|
"hyapp-admin-server/internal/integration/walletclient"
|
|
"hyapp-admin-server/internal/middleware"
|
|
"hyapp-admin-server/internal/modules/shared"
|
|
"hyapp-admin-server/internal/response"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type Handler struct {
|
|
service *Service
|
|
audit shared.OperationLogger
|
|
}
|
|
|
|
func New(userDB *sql.DB, walletDB *sql.DB, adminDB *sql.DB, walletClient walletclient.Client, audit shared.OperationLogger) *Handler {
|
|
return &Handler{service: NewService(userDB, walletDB, adminDB, walletClient), audit: audit}
|
|
}
|
|
|
|
func (h *Handler) ListCoinLedger(c *gin.Context) {
|
|
query, ok := parseListQuery(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
items, total, err := h.service.ListCoinLedger(c.Request.Context(), appctx.FromContext(c.Request.Context()), query)
|
|
if err != nil {
|
|
response.ServerError(c, "获取金币流水失败")
|
|
return
|
|
}
|
|
response.OK(c, response.Page{Items: items, Page: query.Page, PageSize: query.PageSize, Total: total})
|
|
}
|
|
|
|
func (h *Handler) ListCoinAdjustments(c *gin.Context) {
|
|
query, ok := parseListQuery(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
items, total, err := h.service.ListCoinAdjustments(c.Request.Context(), appctx.FromContext(c.Request.Context()), query)
|
|
if err != nil {
|
|
response.ServerError(c, "获取金币增减列表失败")
|
|
return
|
|
}
|
|
response.OK(c, response.Page{Items: items, Page: query.Page, PageSize: query.PageSize, Total: total})
|
|
}
|
|
|
|
func (h *Handler) LookupCoinAdjustmentTarget(c *gin.Context) {
|
|
user, err := h.service.LookupCoinAdjustmentTarget(c.Request.Context(), appctx.FromContext(c.Request.Context()), firstQuery(c, "user_id", "userId", "keyword"))
|
|
if err != nil {
|
|
if errors.Is(err, errCoinAdjustmentTargetNotFound) {
|
|
response.BadRequest(c, "用户不存在")
|
|
return
|
|
}
|
|
response.ServerError(c, "查询用户失败")
|
|
return
|
|
}
|
|
response.OK(c, user)
|
|
}
|
|
|
|
func (h *Handler) CreateCoinAdjustment(c *gin.Context) {
|
|
var req coinAdjustmentRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
response.BadRequest(c, "金币增减参数不正确")
|
|
return
|
|
}
|
|
result, err := h.service.CreateCoinAdjustment(
|
|
c.Request.Context(),
|
|
appctx.FromContext(c.Request.Context()),
|
|
int64(shared.ActorFromContext(c).UserID),
|
|
middleware.CurrentRequestID(c),
|
|
req,
|
|
)
|
|
if err != nil {
|
|
if errors.Is(err, errCoinAdjustmentTargetNotFound) {
|
|
response.BadRequest(c, "用户不存在")
|
|
return
|
|
}
|
|
response.BadRequest(c, err.Error())
|
|
return
|
|
}
|
|
shared.OperationLogWithResourceID(c, h.audit, "coin-adjustment-create", "wallet_entries", result.TransactionID, "success",
|
|
fmt.Sprintf("target_user_id=%v amount=%d transaction_id=%s reason=%q", req.TargetUserID, req.Amount, result.TransactionID, strings.TrimSpace(req.Reason)))
|
|
response.Created(c, result)
|
|
}
|
|
|
|
func parseListQuery(c *gin.Context) (listQuery, bool) {
|
|
options := shared.ListOptions(c)
|
|
startAtMS, ok := optionalInt64(c, "start_at_ms", "startAtMs")
|
|
if !ok {
|
|
response.BadRequest(c, "开始时间不正确")
|
|
return listQuery{}, false
|
|
}
|
|
endAtMS, ok := optionalInt64(c, "end_at_ms", "endAtMs")
|
|
if !ok {
|
|
response.BadRequest(c, "结束时间不正确")
|
|
return listQuery{}, false
|
|
}
|
|
if startAtMS > 0 && endAtMS > 0 && startAtMS >= endAtMS {
|
|
response.BadRequest(c, "时间区间不正确")
|
|
return listQuery{}, false
|
|
}
|
|
return normalizeListQuery(listQuery{
|
|
Page: options.Page,
|
|
PageSize: options.PageSize,
|
|
UserKeyword: strings.TrimSpace(firstQuery(c, "user_keyword", "userKeyword", "user_id", "userId", "keyword")),
|
|
StartAtMS: startAtMS,
|
|
EndAtMS: endAtMS,
|
|
}), true
|
|
}
|
|
|
|
func optionalInt64(c *gin.Context, keys ...string) (int64, bool) {
|
|
value := strings.TrimSpace(firstQuery(c, keys...))
|
|
if value == "" {
|
|
return 0, true
|
|
}
|
|
parsed, err := strconv.ParseInt(value, 10, 64)
|
|
if err != nil || parsed < 0 {
|
|
return 0, false
|
|
}
|
|
return parsed, true
|
|
}
|
|
|
|
func firstQuery(c *gin.Context, keys ...string) string {
|
|
for _, key := range keys {
|
|
if value := c.Query(key); value != "" {
|
|
return value
|
|
}
|
|
}
|
|
return ""
|
|
}
|