Claude Opus 5
+325
@@ -0,0 +1,325 @@
|
||||
package ai_text_to_image_ser
|
||||
|
||||
import (
|
||||
"91porn-server/common"
|
||||
"91porn-server/common/db"
|
||||
"91porn-server/common/log"
|
||||
"91porn-server/common/stderr"
|
||||
"91porn-server/models/v/aiimagetovideomod"
|
||||
"91porn-server/models/v/aiplazamod"
|
||||
"91porn-server/models/v/txnmod"
|
||||
"91porn-server/models/v/walletmod"
|
||||
"91porn-server/web/webg"
|
||||
"errors"
|
||||
"fmt"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"91porn-server/models/cache/aitexttoimagedata"
|
||||
"91porn-server/models/commod"
|
||||
"91porn-server/models/v/aitexttoimagemod"
|
||||
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/bson/primitive"
|
||||
)
|
||||
|
||||
type WebListReq struct {
|
||||
ID *string `json:"id" form:"id"` // ID
|
||||
UID *uint64 `json:"uid" form:"uid"` // 用户ID
|
||||
Status *int `json:"status" form:"status"` // 状态
|
||||
commod.Page
|
||||
}
|
||||
|
||||
func (q *WebListReq) Filter() primitive.M {
|
||||
filter := bson.M{}
|
||||
if q.ID != nil {
|
||||
id, _ := primitive.ObjectIDFromHex(*q.ID)
|
||||
filter["_id"] = id
|
||||
}
|
||||
if q.Status != nil {
|
||||
filter["status"] = q.Status
|
||||
}
|
||||
if q.UID != nil {
|
||||
filter["uid"] = q.UID
|
||||
}
|
||||
return filter
|
||||
}
|
||||
|
||||
type WebListRes struct {
|
||||
Total int64 `json:"total"`
|
||||
HasNext bool `json:"hasNext"`
|
||||
List []aitexttoimagemod.AiTextToImage `json:"list"`
|
||||
}
|
||||
|
||||
// GetList 获取列表
|
||||
func (q *WebListReq) GetList() (res WebListRes, err error) {
|
||||
res.List, res.Total, res.HasNext, err = aitexttoimagemod.GetList(q.Filter(), int64(q.Skip()), int64(q.Limit()), q.GetSort())
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
type WebUpdateReq struct {
|
||||
ID string `json:"id" binding:"required"` // ID
|
||||
Status *int `json:"status"` // 状态
|
||||
Remark *string `json:"remark"` // 备注
|
||||
}
|
||||
|
||||
func (p *WebUpdateReq) UpdateCond() primitive.M {
|
||||
data := make(map[string]interface{})
|
||||
if p.Status != nil {
|
||||
data["status"] = *p.Status
|
||||
}
|
||||
if p.Remark != nil {
|
||||
data["remark"] = *p.Remark
|
||||
}
|
||||
data["updatedAt"] = time.Now()
|
||||
return data
|
||||
}
|
||||
|
||||
// Update 更新数据
|
||||
func (p *WebUpdateReq) Update() error {
|
||||
id, _ := primitive.ObjectIDFromHex(p.ID)
|
||||
data, err := aitexttoimagemod.GetInfo(id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if data.ID.IsZero() {
|
||||
return errors.New("data is null")
|
||||
}
|
||||
|
||||
if p.Status != nil {
|
||||
if data.Status == int(aitexttoimagemod.StatusRefunded) || data.Status == int(aiimagetovideomod.StatusGenerationFailed) {
|
||||
return errors.New("Please do not refund repeatedly.")
|
||||
}
|
||||
|
||||
switch aitexttoimagemod.AiTextToImageStatus(*p.Status) {
|
||||
case aitexttoimagemod.StatusRefunded:
|
||||
if err = webg.VideoDB.Trans(func(t *db.MongoTool) error {
|
||||
if _, err = aitexttoimagemod.UpdateByID(t, id, p.UpdateCond()); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 退款
|
||||
if data.DebitAmountCoin == 0 && data.DebitIncomeCoin == 0 { // 总金额为0则不需要退款
|
||||
return nil
|
||||
}
|
||||
|
||||
// 退款进入钱包余额
|
||||
wallet, err := walletmod.ReturnAmountAndIncome(t, -data.DebitAmountCoin, -data.DebitIncomeCoin, data.UID)
|
||||
if err != nil {
|
||||
log.Error(fmt.Sprintf("Handle Edit walletmod.DebitAmount error:%+v:", err), log.Any("uid", data.UID))
|
||||
return err
|
||||
}
|
||||
|
||||
var tl []txnmod.TransactionLog
|
||||
if data.DebitAmountCoin > 0 {
|
||||
tl = append(tl, txnmod.TransactionLog{
|
||||
TransNo: primitive.NewObjectID(),
|
||||
UID: data.UID,
|
||||
Amount: -data.DebitAmountCoin,
|
||||
ActualAmount: float64(-data.DebitAmountCoin),
|
||||
TranType: txnmod.AiTextToImageDebitGoldReturn.Key(),
|
||||
TranTypeInt: int64(txnmod.AiTextToImageDebitGoldReturn),
|
||||
Desc: fmt.Sprintf("退还AI绘图金币-%d", data.DebitAmountCoin),
|
||||
RealAmount: walletmod.GetRealAmount(wallet),
|
||||
})
|
||||
}
|
||||
if data.DebitIncomeCoin > 0 {
|
||||
tl = append(tl, txnmod.TransactionLog{
|
||||
TransNo: primitive.NewObjectID(),
|
||||
UID: data.UID,
|
||||
Amount: -data.DebitIncomeCoin,
|
||||
ActualAmount: float64(-data.DebitIncomeCoin),
|
||||
TranType: txnmod.AiTextToImageDebitIncomeGoldReturn.Key(),
|
||||
TranTypeInt: int64(txnmod.AiTextToImageDebitIncomeGoldReturn),
|
||||
Desc: fmt.Sprintf("退还AI绘图收益金币-%d", data.DebitIncomeCoin),
|
||||
RealAmount: walletmod.GetRealAmount(wallet),
|
||||
})
|
||||
}
|
||||
|
||||
if len(tl) > 0 {
|
||||
txnErr := txnmod.InsertManyTransactionLog(t, tl)
|
||||
if txnErr != nil {
|
||||
log.Error(fmt.Sprintf("Handle imagetovideo Edit txnmod.InsertTransactionLog error:%+v:", txnErr), log.Any("uid", data.UID))
|
||||
return stderr.ErrDbQueryError
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}); err != nil {
|
||||
if e, ok := err.(stderr.Code); ok {
|
||||
return e
|
||||
}
|
||||
return stderr.ErrDbQueryError
|
||||
}
|
||||
default:
|
||||
}
|
||||
}
|
||||
if _, err = aitexttoimagemod.UpdateByID(nil, id, p.UpdateCond()); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
type WebDeleteReq struct {
|
||||
ID primitive.ObjectID `json:"id" binding:"required"`
|
||||
}
|
||||
|
||||
// Delete 删除数据
|
||||
func (p *WebDeleteReq) Delete() error {
|
||||
data, err := aitexttoimagemod.GetInfo(p.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if err = validateDeleteStatus(data.Status); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if err := aitexttoimagedata.DeleteData(nil, p.ID); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func validateDeleteStatus(status int) error {
|
||||
if status == int(aitexttoimagemod.StatusOrderSuccess) ||
|
||||
status == int(aitexttoimagemod.StatusSubmitted) {
|
||||
return stderr.AiGenningDelForbidden
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// CallbackReq 换脸订单回调请求
|
||||
type CallbackReq struct {
|
||||
ImgUrl string `json:"imgUrl"` // AI图生视频图片地址
|
||||
AppOrderNum string `json:"appOrderNum"` // app订单号
|
||||
Status int `json:"status"` // 订单状态 1:成功，2:失败
|
||||
Msg string `json:"msg"` // 消息
|
||||
}
|
||||
|
||||
func (p *CallbackReq) Callback() error {
|
||||
id, _ := primitive.ObjectIDFromHex(p.AppOrderNum)
|
||||
// 查询该笔订单是否存在
|
||||
data, err := aitexttoimagemod.GetInfo(id)
|
||||
if err != nil {
|
||||
return stderr.ErrDbQueryError
|
||||
}
|
||||
|
||||
if data.ID.IsZero() {
|
||||
return errors.New("ai imagetovideo data is null")
|
||||
}
|
||||
|
||||
if data.Status != int(aitexttoimagemod.StatusSubmitted) {
|
||||
log.Error("AI文生图-重复回调或者状态不正确", log.Any("id", data.ID.Hex()), log.Any("status", data.Status))
|
||||
return nil
|
||||
}
|
||||
|
||||
var status = aitexttoimagemod.StatusGenerationSuccess
|
||||
|
||||
// 处理金币退款以及后续操作
|
||||
set := bson.M{}
|
||||
if p.ImgUrl != "" {
|
||||
set["newImgUrl"] = strings.TrimSpace(p.ImgUrl)
|
||||
}
|
||||
if p.Status != 1 {
|
||||
set["remark"] = p.Msg
|
||||
set["newImgUrl"] = ""
|
||||
status = aitexttoimagemod.StatusGenerationFailed
|
||||
}
|
||||
set["status"] = status
|
||||
set["updatedAt"] = time.Now()
|
||||
|
||||
switch status {
|
||||
case aitexttoimagemod.StatusGenerationSuccess:
|
||||
if _, err = aitexttoimagemod.UpdateByID(nil, id, set); err != nil {
|
||||
return stderr.ErrDbUpdateError
|
||||
}
|
||||
common.Go(func() {
|
||||
if data.ShareStatus != 1 {
|
||||
return
|
||||
}
|
||||
// 新增一个分享
|
||||
alPlaza := aiplazamod.AiPlaza{
|
||||
Type: 5,
|
||||
OrderId: data.ID,
|
||||
Template: "",
|
||||
Uid: data.UID,
|
||||
Title: data.ShareTitle,
|
||||
OriginalImage: "",
|
||||
OriginalVideo: "",
|
||||
OriginContent: data.Text,
|
||||
GenerateImage: p.ImgUrl,
|
||||
GenerateVideo: "",
|
||||
Status: aiplazamod.DefaultStatus,
|
||||
}
|
||||
_, err = aiplazamod.Insert(nil, alPlaza)
|
||||
if err != nil {
|
||||
log.Error("新增ai广场帖子失败", log.Any("alPlaza", alPlaza), log.E(err))
|
||||
return
|
||||
}
|
||||
})
|
||||
return nil
|
||||
case aitexttoimagemod.StatusGenerationFailed: // 退款
|
||||
if err = webg.VideoDB.Trans(func(t *db.MongoTool) error {
|
||||
if _, err = aitexttoimagemod.UpdateByID(t, id, set); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 退款
|
||||
if data.DebitAmountCoin == 0 && data.DebitIncomeCoin == 0 { // 总金额为0则不需要退款
|
||||
return nil
|
||||
}
|
||||
|
||||
// 退款进入钱包余额
|
||||
wallet, err := walletmod.ReturnAmountAndIncome(t, -data.DebitAmountCoin, -data.DebitIncomeCoin, data.UID)
|
||||
if err != nil {
|
||||
log.Error(fmt.Sprintf("Handle Callback Edit walletmod.DebitAmount error:%+v:", err), log.Any("uid", data.UID))
|
||||
return err
|
||||
}
|
||||
|
||||
var tl []txnmod.TransactionLog
|
||||
if data.DebitAmountCoin > 0 {
|
||||
tl = append(tl, txnmod.TransactionLog{
|
||||
TransNo: primitive.NewObjectID(),
|
||||
UID: data.UID,
|
||||
Amount: -data.DebitAmountCoin,
|
||||
ActualAmount: float64(-data.DebitAmountCoin),
|
||||
TranType: txnmod.AiTextToImageDebitGoldReturn.Key(),
|
||||
TranTypeInt: int64(txnmod.AiTextToImageDebitGoldReturn),
|
||||
Desc: fmt.Sprintf("退还AI绘图金币-%d", data.DebitAmountCoin),
|
||||
RealAmount: walletmod.GetRealAmount(wallet),
|
||||
})
|
||||
}
|
||||
if data.DebitIncomeCoin > 0 {
|
||||
tl = append(tl, txnmod.TransactionLog{
|
||||
TransNo: primitive.NewObjectID(),
|
||||
UID: data.UID,
|
||||
Amount: -data.DebitIncomeCoin,
|
||||
ActualAmount: float64(-data.DebitIncomeCoin),
|
||||
TranType: txnmod.AiTextToImageDebitIncomeGoldReturn.Key(),
|
||||
TranTypeInt: int64(txnmod.AiTextToImageDebitIncomeGoldReturn),
|
||||
Desc: fmt.Sprintf("退还AI绘图收益金币-%d", data.DebitIncomeCoin),
|
||||
RealAmount: walletmod.GetRealAmount(wallet),
|
||||
})
|
||||
}
|
||||
|
||||
if len(tl) > 0 {
|
||||
txnErr := txnmod.InsertManyTransactionLog(t, tl)
|
||||
if txnErr != nil {
|
||||
log.Error(fmt.Sprintf("Handle ai text to image Callback txnmod.InsertTransactionLog error:%+v:", txnErr), log.Any("uid", data.UID))
|
||||
return stderr.ErrDbQueryError
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}); err != nil {
|
||||
if e, ok := err.(stderr.Code); ok {
|
||||
return e
|
||||
}
|
||||
return stderr.ErrDbQueryError
|
||||
}
|
||||
return nil
|
||||
default:
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user