mirror of
https://github.com/yangjian102621/geekai.git
synced 2025-09-18 01:06:39 +08:00
76 lines
1.4 KiB
Go
76 lines
1.4 KiB
Go
package handler
|
|
|
|
import (
|
|
"chatplus/service"
|
|
"chatplus/service/payment"
|
|
"chatplus/store/model"
|
|
"chatplus/utils"
|
|
"chatplus/utils/resp"
|
|
"fmt"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type TestHandler struct {
|
|
db *gorm.DB
|
|
snowflake *service.Snowflake
|
|
js *payment.PayJS
|
|
}
|
|
|
|
func NewTestHandler(db *gorm.DB, snowflake *service.Snowflake, js *payment.PayJS) *TestHandler {
|
|
return &TestHandler{db: db, snowflake: snowflake, js: js}
|
|
}
|
|
|
|
func (h *TestHandler) Test(c *gin.Context) {
|
|
//h.initUserNickname(c)
|
|
//h.initMjTaskId(c)
|
|
|
|
orderId, _ := h.snowflake.Next(false)
|
|
params := payment.JPayReq{
|
|
TotalFee: 12345,
|
|
OutTradeNo: orderId,
|
|
Subject: "支付测试",
|
|
}
|
|
r := h.js.Pay(params)
|
|
if !r.IsOK() {
|
|
resp.ERROR(c, r.ReturnMsg)
|
|
return
|
|
}
|
|
resp.SUCCESS(c, r)
|
|
|
|
}
|
|
|
|
func (h *TestHandler) initUserNickname(c *gin.Context) {
|
|
var users []model.User
|
|
tx := h.db.Find(&users)
|
|
if tx.Error != nil {
|
|
resp.ERROR(c, tx.Error.Error())
|
|
return
|
|
}
|
|
|
|
for _, u := range users {
|
|
u.Nickname = fmt.Sprintf("极客学长@%d", utils.RandomNumber(6))
|
|
h.db.Updates(&u)
|
|
}
|
|
|
|
resp.SUCCESS(c)
|
|
}
|
|
|
|
func (h *TestHandler) initMjTaskId(c *gin.Context) {
|
|
var jobs []model.MidJourneyJob
|
|
tx := h.db.Find(&jobs)
|
|
if tx.Error != nil {
|
|
resp.ERROR(c, tx.Error.Error())
|
|
return
|
|
}
|
|
|
|
for _, job := range jobs {
|
|
id, _ := h.snowflake.Next(true)
|
|
job.TaskId = id
|
|
h.db.Updates(&job)
|
|
}
|
|
|
|
resp.SUCCESS(c)
|
|
}
|