xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-08 10:13:44 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

redeem code function is ready

Browse Source
This commit is contained in:
RockYang
2024-08-09 18:19:51 +08:00
parent 4b717109d2
commit dfba3c30a5
18 changed files with 979 additions and 123 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/handler/dalle_handler.go
View File

@@ -241,11 +241,11 @@ func (h *DallJobHandler) Remove(c *gin.Context) {
}
var user model.User
h.DB.Where("id = ?", job.UserId).First(&user)
tx.Where("id = ?", job.UserId).First(&user)
err = tx.Create(&model.PowerLog{
UserId: user.Id,
Username: user.Username,
Type: types.PowerConsume,
Type: types.PowerRefund,
Amount: job.Power,
Balance: user.Power,
Mark: types.PowerAdd,

4
api/handler/mj_handler.go
View File

@@ -472,11 +472,11 @@ func (h *MidJourneyHandler) Remove(c *gin.Context) {
return
}
var user model.User
h.DB.Where("id = ?", job.UserId).First(&user)
tx.Where("id = ?", job.UserId).First(&user)
err = tx.Create(&model.PowerLog{
UserId: user.Id,
Username: user.Username,
Type: types.PowerConsume,
Type: types.PowerRefund,
Amount: job.Power,
Balance: user.Power,
Mark: types.PowerAdd,

4
api/handler/redeem_handler.go
View File

@@ -78,7 +78,7 @@ func (h *RedeemHandler) Verify(c *gin.Context) {
// 记录算力充值日志
var user model.User
err = h.DB.Where("id", userId).First(&user).Error
err = tx.Where("id", userId).First(&user).Error
if err != nil {
tx.Rollback()
resp.ERROR(c, err.Error())
@@ -88,7 +88,7 @@ func (h *RedeemHandler) Verify(c *gin.Context) {
h.DB.Create(&model.PowerLog{
UserId: userId,
Username: user.Username,
Type: types.PowerReward,
Type: types.PowerRedeem,
Amount: item.Power,
Balance: user.Power,
Mark: types.PowerAdd,

4
api/handler/sd_handler.go
View File

@@ -297,11 +297,11 @@ func (h *SdJobHandler) Remove(c *gin.Context) {
return
}
var user model.User
h.DB.Where("id = ?", job.UserId).First(&user)
tx.Where("id = ?", job.UserId).First(&user)
err = tx.Create(&model.PowerLog{
UserId: user.Id,
Username: user.Username,
Type: types.PowerConsume,
Type: types.PowerRefund,
Amount: job.Power,
Balance: user.Power,
Mark: types.PowerAdd,

5
api/handler/suno_handler.go
View File

@@ -94,6 +94,7 @@ func (h *SunoHandler) Create(c *gin.Context) {
RefTaskId: data.RefTaskId,
ExtendSecs: data.ExtendSecs,
Power: h.App.SysConfig.SunoPower,
SongId: utils.RandString(32),
}
if data.Lyrics != "" {
job.Prompt = data.Lyrics
@@ -226,11 +227,11 @@ func (h *SunoHandler) Remove(c *gin.Context) {
return
}
var user model.User
h.DB.Where("id = ?", job.UserId).First(&user)
tx.Where("id = ?", job.UserId).First(&user)
err = tx.Create(&model.PowerLog{
UserId: user.Id,
Username: user.Username,
Type: types.PowerConsume,
Type: types.PowerRefund,
Amount: job.Power,
Balance: user.Power,
Mark: types.PowerAdd,

37
api/handler/test_handler.go
View File

@@ -3,7 +3,9 @@ package handler
import (
"geekai/service"
"geekai/service/payment"
"github.com/gin-gonic/gin"
"gorm.io/gorm"
"net/http"
)
type TestHandler struct {
@@ -15,3 +17,38 @@ type TestHandler struct {
func NewTestHandler(db *gorm.DB, snowflake *service.Snowflake, js *payment.JPayService) *TestHandler {
return &TestHandler{db: db, snowflake: snowflake, js: js}
}
func (h *TestHandler) SseTest(c *gin.Context) {
//c.Header("Content-Type", "text/event-stream")
//c.Header("Cache-Control", "no-cache")
//c.Header("Connection", "keep-alive")
//
//
//// 模拟实时数据更新
//for i := 0; i < 10; i++ {
// // 发送 SSE 数据
// _, err := fmt.Fprintf(c.Writer, "data: %v\n\n", data)
// if err != nil {
// return
// }
// c.Writer.Flush() // 确保立即发送数据
// time.Sleep(1 * time.Second) // 每秒发送一次数据
//}
//c.Abort()
}
func (h *TestHandler) PostTest(c *gin.Context) {
var data struct {
Message string `json:"message"`
UserId uint `json:"user_id"`
}
if err := c.ShouldBindJSON(&data); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
// 将参数存储在上下文中
c.Set("data", data)
c.Next()
}