temp

app/main/api/internal/logic/pay/paymentlogic.go

@@ -4,15 +4,14 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
+	"tyc-server/app/main/api/internal/service"
 	"tyc-server/app/main/api/internal/svc"
 	"tyc-server/app/main/api/internal/types"
-	"tyc-server/app/main/model"
 	"tyc-server/common/ctxdata"
 	"tyc-server/common/xerr"
 
 	"github.com/pkg/errors"
 	"github.com/zeromicro/go-zero/core/logx"
-	"github.com/zeromicro/go-zero/core/stores/sqlx"
 )
 
 type PaymentLogic struct {
@@ -34,104 +33,45 @@ func (l *PaymentLogic) Payment(req *types.PaymentReq) (resp *types.PaymentResp,
 	if getUidErr != nil {
 		return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成订单, 获取用户信息失败, %+v", getUidErr)
 	}
-
-	brand, ok := l.ctx.Value("brand").(string)
-	if !ok {
-		return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成订单, 获取平台失败, %+v", getUidErr)
-	}
+	// 从缓存中获取订单信息
 	outTradeNo := req.Id
 	redisKey := fmt.Sprintf("%d:%s", userID, outTradeNo)
 	cache, cacheErr := l.svcCtx.Redis.GetCtx(l.ctx, redisKey)
 	if cacheErr != nil {
 		return nil, cacheErr
 	}
+
 	var data types.QueryCacheLoad
 	err = json.Unmarshal([]byte(cache), &data)
 	if err != nil {
 		return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成订单, 解析缓存内容失败, %+v", err)
 	}
-
-	product, err := l.svcCtx.ProductModel.FindOneByProductEn(l.ctx, data.Product)
-	if err != nil {
-		return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成订单, 查找产品错误: %+v", err)
-	}
-	var prepayData interface{}
-	var amount float64
-	user, err := l.svcCtx.UserModel.FindOne(l.ctx, userID)
-	if err != nil {
-		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "生成订单, 获取用户信息失败: %+v", err)
-	}
-
-	if user.Inside == 1 {
-		amount = 0.01
-	} else {
-		amount = product.SellPrice
-	}
-
-	var createOrderErr error
-	if req.PayMethod == "wechat" {
-		prepayData, createOrderErr = l.svcCtx.WechatPayService.CreateWechatOrder(l.ctx, amount, product.ProductName, outTradeNo)
-	} else if req.PayMethod == "alipay" {
-		prepayData, createOrderErr = l.svcCtx.AlipayService.CreateAlipayOrder(l.ctx, amount, product.ProductName, outTradeNo, brand)
-	} else if req.PayMethod == "appleiap" {
-		prepayData = l.svcCtx.ApplePayService.GetIappayAppID(product.ProductEn)
-	}
-	if createOrderErr != nil {
-		return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成订单, 创建支付订单失败: %+v", createOrderErr)
-	}
-	var orderID int64
-	transErr := l.svcCtx.OrderModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error {
-		order := model.Order{
-			OrderNo:         outTradeNo,
-			UserId:          userID,
-			ProductId:       product.Id,
-			PaymentPlatform: req.PayMethod,
-			PaymentScene:    "tyc",
-			Amount:          amount,
-			Status:          "pending",
-		}
-		orderInsertResult, insertOrderErr := l.svcCtx.OrderModel.Insert(ctx, session, &order)
-		if insertOrderErr != nil {
-			return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成订单, 保存订单失败: %+v", insertOrderErr)
-		}
-		insertedOrderID, lastInsertIdErr := orderInsertResult.LastInsertId()
-		if lastInsertIdErr != nil {
-			return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成订单, 获取保存订单ID失败: %+v", lastInsertIdErr)
-		}
-		orderID = insertedOrderID
-
-		promoteKey, ok := l.ctx.Value("promoteKey").(string)
-		if ok && promoteKey != "" {
-			// TODO: 在这里添加处理 promoteKey 的逻辑
-			url := fmt.Sprintf("%s/%s", l.svcCtx.Config.AdminPromotion.URLDomain, promoteKey)
-			promoteLink, err := l.svcCtx.AdminPromotionLinkModel.FindOneByUrl(l.ctx, url)
-			if err != nil && !errors.Is(err, model.ErrNotFound) {
-				return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成订单, 查找推广链接失败: %+v", err)
-			}
-			if promoteLink != nil {
-				promoteOrder := &model.AdminPromotionOrder{
-					OrderId:     orderID,
-					LinkId:      promoteLink.Id,
-					UserId:      userID,
-					AdminUserId: promoteLink.AdminUserId,
-				}
-				_, insertPromoteOrderErr := l.svcCtx.AdminPromotionOrderModel.Insert(ctx, session, promoteOrder)
-				if insertPromoteOrderErr != nil {
-					return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成订单, 保存推广订单失败: %+v", insertPromoteOrderErr)
-				}
-			}
-		}
-
-		return nil
-	})
-	if transErr != nil {
-		return nil, transErr
-	}
-	switch v := prepayData.(type) {
-	case string:
-		// 如果 prepayData 是字符串类型，直接返回
-		return &types.PaymentResp{PrepayId: v, OrderID: orderID}, nil
+	var payMethod service.PayMethod
+	switch req.PayMethod {
+	case "alipay":
+		payMethod = service.PayMethodAlipay
+	case "wechat":
+		payMethod = service.PayMethodWechat
+	// case "apple":
+	// 	payMethod = service.PayMethodApple
 	default:
-		return &types.PaymentResp{PrepayData: prepayData, OrderID: orderID}, nil
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成订单, 不支持的支付方式: %+v", req.PayMethod)
 	}
+
+	// 创建支付订单
+	orderReq := &service.CreateOrderRequest{
+		PayMethod: payMethod,
+		ProductEn: data.Product,
+	}
+
+	orderResp, err := l.svcCtx.PayService.CreateOrder(l.ctx, orderReq)
+	if err != nil {
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成订单失败: %+v", err)
+	}
+
+	return &types.PaymentResp{
+		PrepayData: orderResp.PayData,
+		PrepayId:   orderResp.OrderNo,
+		OrderID:    orderResp.OrderID,
+	}, nil
 }