129 lines
3.2 KiB
Go
129 lines
3.2 KiB
Go
package walletctrl
|
|
|
|
import (
|
|
"91porn-server/app/appg"
|
|
"91porn-server/app/service/walletser"
|
|
"91porn-server/common"
|
|
"91porn-server/common/stderr"
|
|
"91porn-server/common/store"
|
|
"91porn-server/models/v/walletmod"
|
|
"encoding/json"
|
|
"errors"
|
|
"github.com/gin-gonic/gin"
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
func WalletBalance(ctx *gin.Context) {
|
|
req := struct {
|
|
Uid uint64 `json:"uid" form:"uid"`
|
|
}{}
|
|
err := ctx.ShouldBind(&req)
|
|
if err != nil {
|
|
common.ServeJSON(ctx, stderr.ErrParamError, err)
|
|
return
|
|
}
|
|
w, err := walletmod.GetWallet(req.Uid)
|
|
if err != nil {
|
|
common.ServeJSON(ctx, stderr.ErrServerUnavailable, err.Error())
|
|
return
|
|
}
|
|
resp := struct {
|
|
Balance int64 `json:"balance"`
|
|
}{}
|
|
resp.Balance = w.Amount + w.Income
|
|
ctx.JSON(http.StatusOK, gin.H{
|
|
"code": stderr.Success,
|
|
"msg": stderr.Success.Msg(),
|
|
"tip": stderr.Success.Tip(),
|
|
"data": resp,
|
|
"time": time.Now().UTC().Format("2006-01-02T15:04:05.000Z"),
|
|
})
|
|
}
|
|
|
|
func WalletDeduct(ctx *gin.Context) {
|
|
req := struct {
|
|
Uid uint64 `json:"uid" form:"uid"`
|
|
OrderId primitive.ObjectID `json:"orderId" form:"orderId"`
|
|
Source int `json:"source" form:"source"`
|
|
Amount int64 `json:"amount" form:"amount"` // 扣除金额
|
|
Desc string `json:"desc" form:"desc"`
|
|
}{}
|
|
err := ctx.ShouldBind(&req)
|
|
if err != nil {
|
|
common.ServeJSON(ctx, stderr.ErrParamError, err)
|
|
return
|
|
}
|
|
if req.Uid <= 0 || req.OrderId.IsZero() || req.Amount <= 0 {
|
|
common.ServeJSON(ctx, stderr.ErrParamError, err)
|
|
return
|
|
}
|
|
err = Check(ctx, req)
|
|
if err != nil {
|
|
common.ServeJSON(ctx, stderr.ErrParamError, err)
|
|
return
|
|
}
|
|
resp := struct {
|
|
Balance int64 `json:"balance"`
|
|
}{}
|
|
balance, code := walletser.StoreDeductBalance(req.Uid, req.Amount, req.OrderId, req.Source, req.Desc)
|
|
resp.Balance = balance
|
|
ctx.JSON(http.StatusOK, gin.H{
|
|
"code": code,
|
|
"msg": code.Msg(),
|
|
"tip": code.Tip(),
|
|
"data": resp,
|
|
"time": time.Now().UTC().Format("2006-01-02T15:04:05.000Z"),
|
|
})
|
|
}
|
|
|
|
func Check(ctx *gin.Context, param interface{}) (err error) {
|
|
b, _ := json.Marshal(param)
|
|
sign, err := store.Encrypt(b, appg.Conf.Base.StoreEncryptKey)
|
|
if err != nil {
|
|
return
|
|
}
|
|
headerSign := ctx.Request.Header.Get("sign")
|
|
if sign != headerSign {
|
|
return errors.New("校验失败")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func WalletRefund(ctx *gin.Context) {
|
|
req := struct {
|
|
Uid uint64 `json:"uid"`
|
|
OrderId primitive.ObjectID `json:"orderId"`
|
|
Source int `json:"source"`
|
|
Amount int64 `json:"amount"`
|
|
Desc string `json:"desc"`
|
|
}{}
|
|
err := ctx.ShouldBind(&req)
|
|
if err != nil {
|
|
common.ServeJSON(ctx, stderr.ErrParamError, err)
|
|
return
|
|
}
|
|
if req.Uid <= 0 || req.OrderId.IsZero() || req.Amount <= 0 {
|
|
common.ServeJSON(ctx, stderr.ErrParamError, err)
|
|
return
|
|
}
|
|
err = Check(ctx, req)
|
|
if err != nil {
|
|
common.ServeJSON(ctx, stderr.ErrParamError, err)
|
|
return
|
|
}
|
|
resp := struct {
|
|
Balance int64 `json:"balance"`
|
|
}{}
|
|
balance, code := walletser.StoreRefund(req.Uid, req.Amount, req.OrderId, req.Source, req.Desc)
|
|
resp.Balance = balance
|
|
ctx.JSON(http.StatusOK, gin.H{
|
|
"code": code,
|
|
"msg": code.Msg(),
|
|
"tip": code.Tip(),
|
|
"data": resp,
|
|
"time": time.Now().UTC().Format("2006-01-02T15:04:05.000Z"),
|
|
})
|
|
}
|