mirror of
https://github.com/yangjian102621/geekai.git
synced 2025-09-18 01:06:39 +08:00
109 lines
2.6 KiB
Go
109 lines
2.6 KiB
Go
package handler
|
||
|
||
// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
||
// * Copyright 2023 The Geek-AI Authors. All rights reserved.
|
||
// * Use of this source code is governed by a Apache-2.0 license
|
||
// * that can be found in the LICENSE file.
|
||
// * @Author yangjian102621@163.com
|
||
// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
||
|
||
import (
|
||
"fmt"
|
||
"geekai/core"
|
||
"geekai/core/types"
|
||
"geekai/store/model"
|
||
"geekai/store/vo"
|
||
"geekai/utils"
|
||
"geekai/utils/resp"
|
||
"github.com/gin-gonic/gin"
|
||
"gorm.io/gorm"
|
||
"math"
|
||
"strings"
|
||
"sync"
|
||
"time"
|
||
)
|
||
|
||
type RewardHandler struct {
|
||
BaseHandler
|
||
lock sync.Mutex
|
||
}
|
||
|
||
func NewRewardHandler(app *core.AppServer, db *gorm.DB) *RewardHandler {
|
||
return &RewardHandler{BaseHandler: BaseHandler{App: app, DB: db}}
|
||
}
|
||
|
||
// Verify 打赏码核销
|
||
func (h *RewardHandler) Verify(c *gin.Context) {
|
||
var data struct {
|
||
TxId string `json:"tx_id"`
|
||
}
|
||
if err := c.ShouldBindJSON(&data); err != nil {
|
||
resp.ERROR(c, types.InvalidArgs)
|
||
return
|
||
}
|
||
|
||
user, err := h.GetLoginUser(c)
|
||
if err != nil {
|
||
resp.HACKER(c)
|
||
return
|
||
}
|
||
|
||
// 移除转账单号中间的空格，防止有人复制的时候多复制了空格
|
||
data.TxId = strings.ReplaceAll(data.TxId, " ", "")
|
||
|
||
h.lock.Lock()
|
||
defer h.lock.Unlock()
|
||
|
||
var item model.Reward
|
||
res := h.DB.Where("tx_id = ?", data.TxId).First(&item)
|
||
if res.Error != nil {
|
||
resp.ERROR(c, "无效的众筹交易流水号！")
|
||
return
|
||
}
|
||
|
||
if item.Status {
|
||
resp.ERROR(c, "当前众筹交易流水号已经被核销，请不要重复核销！")
|
||
return
|
||
}
|
||
|
||
tx := h.DB.Begin()
|
||
exchange := vo.RewardExchange{}
|
||
power := math.Ceil(item.Amount / h.App.SysConfig.PowerPrice)
|
||
exchange.Power = int(power)
|
||
res = tx.Model(&user).UpdateColumn("power", gorm.Expr("power + ?", exchange.Power))
|
||
if res.Error != nil {
|
||
tx.Rollback()
|
||
logger.Error("添加应用失败：", res.Error)
|
||
resp.ERROR(c, "更新数据库失败！")
|
||
return
|
||
}
|
||
|
||
// 更新核销状态
|
||
item.Status = true
|
||
item.UserId = user.Id
|
||
item.Exchange = utils.JsonEncode(exchange)
|
||
res = tx.Updates(&item)
|
||
if res.Error != nil {
|
||
tx.Rollback()
|
||
logger.Error("添加应用失败：", res.Error)
|
||
resp.ERROR(c, "更新数据库失败！")
|
||
return
|
||
}
|
||
|
||
// 记录算力充值日志
|
||
h.DB.Create(&model.PowerLog{
|
||
UserId: user.Id,
|
||
Username: user.Username,
|
||
Type: types.PowerReward,
|
||
Amount: exchange.Power,
|
||
Balance: user.Power + exchange.Power,
|
||
Mark: types.PowerAdd,
|
||
Model: "众筹支付",
|
||
Remark: fmt.Sprintf("众筹充值算力，金额：%f，价格：%f", item.Amount, h.App.SysConfig.PowerPrice),
|
||
CreatedAt: time.Now(),
|
||
})
|
||
tx.Commit()
|
||
resp.SUCCESS(c)
|
||
|
||
}
|