hepa-calc-api/api/service/OrderMember.go

package service

import (
	"context"
	"errors"
	"fmt"
	"github.com/wechatpay-apiv3/wechatpay-go/services/payments/app"
	"github.com/wechatpay-apiv3/wechatpay-go/services/payments/jsapi"
	"gorm.io/gorm"
	"hepa-calc-api/api/dao"
	"hepa-calc-api/api/model"
	"hepa-calc-api/config"
	"hepa-calc-api/extend/weChat"
	"hepa-calc-api/global"
	"hepa-calc-api/utils"
	"time"
)

type OrderMemberService struct {
}

// AddOrderMember 创建会员订单
func (r *OrderMemberService) AddOrderMember(tx *gorm.DB, UserId, SystemMemberId int64, UserCouponId *int64, payChannel int, orderPrice float64) (orderMember *model.OrderMember, err error) {
	// 检测并发请求
	redisKey := "AddOrderMember" + fmt.Sprintf("%d", UserId) + fmt.Sprintf("%d", SystemMemberId)
	res, _ := global.Redis.Get(context.Background(), redisKey).Result()
	if res != "" {
		return nil, errors.New("请勿重复操作")
	}

	defer func(redisKey string) {
		global.Redis.Del(context.Background(), redisKey)
	}(redisKey)

	// 添加缓存
	_, err = global.Redis.Set(context.Background(), redisKey, "1", (10)*time.Second).Result()
	if err != nil {
		return nil, errors.New("生成订单失败")
	}

	// 获取会员配置数据
	systemMemberDao := dao.SystemMemberDao{}
	systemMember, err := systemMemberDao.GetSystemMemberById(SystemMemberId)
	if err != nil {
		return nil, errors.New("会员配置异常")
	}

	// 获取用户数据
	userDao := dao.UserDao{}
	user, err := userDao.GetUserById(UserId)
	if err != nil || user == nil {
		return nil, errors.New("用户错误")
	}

	// 检测会员有效期
	if user.IsMember == 1 {
		now := time.Now()

		diff := user.MemberExpireDate.Sub(now)
		// 将差值转换为天数
		diffDays := int(diff.Hours() / 24)
		totalDays := diffDays + int(systemMember.MemberDays)
		if totalDays > 180 {
			return nil, errors.New("超出会员有效期")
		}
	}

	var amountTotal float64        // 总金额
	var couponAmountTotal float64  // 优惠卷总金额
	var paymentAmountTotal float64 // 实际付款金额
	var orderStatus int            // 订单状态（1:待支付 2:已完成 3:已取消）
	var payStatus int              // 支付状态（1:未支付 2:已支付 3:支付中 4:支付失败 5:支付超时 6:支付关闭 7:已撤销 8:转入退款）
	var payTime *time.Time         // 支付时间
	var escrowTradeNo string       // 第三方支付流水号

	// 获取会员购买价格
	systemMemberService := &SystemMemberService{}
	amountTotal = systemMemberService.GetSystemMemberBuyPrice(systemMember)
	if err != nil {
		return nil, err
	}

	// 检测用户优惠卷
	var userCoupon *model.UserCoupon
	if UserCouponId != nil {
		// 检测用户是否购买过会员
		userService := &UserService{}
		isBuy := userService.CheckUserBuyOrderMember(UserId)
		if isBuy == true {
			// 已购买过会员，无法使用优惠卷
			return nil, errors.New("优惠券异常")
		}

		// 获取优惠卷数据
		UserCouponDao := dao.UserCouponDao{}
		userCoupon, err = UserCouponDao.GetUserCouponPreloadById(*UserCouponId)
		if err != nil {
			return nil, errors.New("优惠券异常")
		}

		userCouponService := &UserCouponService{}
		isCanUse, err := userCouponService.CheckUserCoupon(userCoupon, SystemMemberId, 2, amountTotal)
		if err != nil || isCanUse == false {
			return nil, errors.New("价格异常")
		}

		// 优惠卷总金额
		couponAmountTotal = userCoupon.Coupon.CouponPrice
	}

	// 实际付款金额
	paymentAmountTotal = amountTotal - couponAmountTotal // 实际付款金额
	if paymentAmountTotal < 0 {
		return nil, errors.New("价格异常")
	}

	if orderPrice != paymentAmountTotal {
		return nil, errors.New("价格异常")
	}

	orderStatus = 1 // 订单状态（1:待支付 2:已完成 3:已取消）
	payStatus = 1   // 支付状态（1:未支付 2:已支付 3:支付中 4:支付失败 5:支付超时 6:支付关闭 7:已撤销 8:转入退款）
	payTime = nil   // 支付时间

	// 生成订单号
	orderNo := global.Snowflake.Generate().String()

	// 创建订单
	orderMember = &model.OrderMember{
		UserId:             UserId,
		SystemMemberId:     SystemMemberId,
		OrderStatus:        orderStatus,
		IsDelete:           0,
		PayChannel:         payChannel,
		PayStatus:          payStatus,
		PayTime:            payTime,
		RefundStatus:       0,
		OrderNo:            orderNo,
		EscrowTradeNo:      escrowTradeNo,
		AmountTotal:        amountTotal,
		CouponAmountTotal:  couponAmountTotal,
		PaymentAmountTotal: paymentAmountTotal,
		CancelStatus:       0,
		CancelTime:         nil,
		CancelRemarks:      "",
		OrderRemarks:       "",
	}

	orderMemberDao := dao.OrderMemberDao{}
	orderMember, err = orderMemberDao.AddOrderMember(tx, orderMember)
	if err != nil {
		return nil, errors.New("订单创建失败")
	}

	// 创建优惠卷表
	if userCoupon != nil {
		orderMemberCoupon := &model.OrderMemberCoupon{
			OrderId:        orderMember.OrderId,
			UserCouponId:   *UserCouponId,
			CouponName:     userCoupon.Coupon.CouponName,
			CouponUsePrice: userCoupon.Coupon.CouponPrice,
		}

		orderMemberCouponDao := dao.OrderMemberCouponDao{}
		orderMemberCoupon, err = orderMemberCouponDao.AddOrderMemberCoupon(tx, orderMemberCoupon)
		if err != nil {
			tx.Rollback()
			return nil, errors.New("订单创建失败")
		}

		// 修改优惠卷使用状态
		userCouponDao := dao.UserCouponDao{}

		userCouponData := make(map[string]interface{})
		userCouponData["user_coupon_status"] = 1
		userCouponData["coupon_use_date"] = time.Now().Format("2006-01-02 15:04:05")
		err := userCouponDao.EditUserCouponById(tx, userCoupon.UserCouponId, userCouponData)
		if err != nil {
			return nil, errors.New("订单创建失败")
		}
	}

	return orderMember, nil
}

// PutCancelOrderMember 取消会员订单
// cancelReason:订单取消原因（1:主动取消 2:后台取消 3:支付超时取消）
func (r *OrderMemberService) PutCancelOrderMember(tx *gorm.DB, userId, orderId int64, cancelReason int) (bool, error) {
	// 检测多次请求
	redisKey := "PutCancelOrderSingle" + fmt.Sprintf("%d", userId) + fmt.Sprintf("%d", orderId)
	res, _ := global.Redis.Get(context.Background(), redisKey).Result()
	if res != "" {
		return false, errors.New("请勿重复操作")
	}

	defer func(redisKey string) {
		global.Redis.Del(context.Background(), redisKey)
	}(redisKey)

	// 添加缓存
	_, err := global.Redis.Set(context.Background(), redisKey, "1", (10)*time.Second).Result()
	if err != nil {
		return false, errors.New("取消订单失败")
	}

	// 获取订单数据
	orderSingleDao := dao.OrderSingleDao{}
	maps := make(map[string]interface{})
	maps["user_id"] = userId
	maps["order_id"] = orderId
	orderSingle, err := orderSingleDao.GetOrderSingle(maps)
	if err != nil {
		return false, errors.New("订单异常")
	}

	// 订单状态（1:待支付 2:已完成 3:已取消）
	if orderSingle.OrderStatus == 2 {
		return false, errors.New("订单已完成,无法取消")
	}

	if orderSingle.OrderStatus == 3 {
		return false, errors.New("订单已取消,请勿重复操作")
	}

	// 取消状态（0:否 1:是）
	if orderSingle.CancelStatus == 1 {
		return false, errors.New("订单已取消")
	}

	// 支付状态（1:未支付 2:已支付 3:支付中 4:支付失败 5:支付超时 6:支付关闭 7:已撤销 8:转入退款）
	if orderSingle.PayStatus == 2 {
		return false, errors.New("订单已支付,无法取消")
	}

	// 订单退款状态（0:无退款 1:申请退款 2:退款中 3:退款成功 4:拒绝退款 5:退款关闭 6:退款异常 7:部分退款）
	if orderSingle.RefundStatus == 1 {
		return false, errors.New("订单已申请退款")
	}

	if orderSingle.RefundStatus == 2 {
		return false, errors.New("订单退款中")
	}

	if orderSingle.RefundStatus == 3 {
		return false, errors.New("订单已退款成功")
	}

	if orderSingle.RefundStatus == 6 {
		return false, errors.New("订单退款异常")
	}

	// 修改订单为取消
	orderSingleData := make(map[string]interface{})
	orderSingleData["order_status"] = 3
	if cancelReason == 3 {
		// 支付超时取消
		orderSingleData["pay_status"] = 5
	}
	orderSingleData["cancel_status"] = 1
	orderSingleData["cancel_time"] = time.Now().Format("2006-01-02 15:04:05")
	orderSingleData["cancel_remarks"] = utils.OrderCancelReasonToString(cancelReason)
	orderSingleData["updated_at"] = time.Now().Format("2006-01-02 15:04:05")
	err = orderSingleDao.EditOrderSingleById(tx, orderId, orderSingleData)
	if err != nil {
		return false, errors.New("订单取消失败")
	}

	// 退还订单优惠卷
	if orderSingle.CouponAmountTotal != 0 {
		// 获取订单优惠卷数据
		orderSingleCouponDao := dao.OrderSingleCouponDao{}
		orderSingleCoupon, err := orderSingleCouponDao.GetOrderSingleCouponByOrderId(orderId)
		if err != nil {
			tx.Rollback()
			return false, errors.New("订单取消失败")
		}

		userCouponService := &UserCouponService{}
		userCouponService.ReturnUserCoupon(tx, orderSingleCoupon.UserCouponId)
	}

	return true, nil
}

// GetJsapiPrepay 获取jsapi预支付交易会话标识
func (r *OrderMemberService) GetJsapiPrepay(m *model.OrderMember) (prepay *jsapi.PrepayWithRequestPaymentResponse, err error) {
	// 获取用户数据
	userDao := dao.UserDao{}
	user, err := userDao.GetUserById(m.UserId)
	if err != nil || user == nil {
		return nil, errors.New("用户错误")
	}

	if user.OpenId != "" {
		return nil, errors.New("发起支付失败")
	}

	jsapiRequest := weChat.JsapiRequest{
		AppId:       config.C.Wechat.AppId,
		MchId:       config.C.Wechat.Pay1281030301.MchId,
		Description: "肝病算一算",
		OutTradeNo:  m.OrderNo,
		NotifyUrl:   config.C.Wechat.RefundNotifyDomain + config.C.Wechat.RefundNotifyUrl,
		Amount: weChat.JsapiRequestAmountRequest{
			Total:    int64(m.PaymentAmountTotal * 100),
			Currency: "CNY",
		},
		Payer: weChat.JsapiRequestPayerRequest{OpenId: user.OpenId},
	}

	prepay, err = jsapiRequest.GetJsapiPrepay()
	if err != nil {
		return nil, err
	}

	return prepay, nil
}

// GetAppPrepay 获取app预支付交易会话标识
func (r *OrderMemberService) GetAppPrepay(m *model.OrderMember) (prepay *app.PrepayWithRequestPaymentResponse, err error) {
	// 获取用户数据
	userDao := dao.UserDao{}
	user, err := userDao.GetUserById(m.UserId)
	if err != nil || user == nil {
		return nil, errors.New("用户错误")
	}

	if user.OpenId != "" {
		return nil, errors.New("发起支付失败")
	}

	appRequest := weChat.AppRequest{
		AppId:       config.C.Wechat.AppId,
		MchId:       config.C.Wechat.Pay1281030301.MchId,
		Description: "肝病算一算",
		OutTradeNo:  m.OrderNo,
		NotifyUrl:   config.C.Wechat.RefundNotifyDomain + config.C.Wechat.RefundNotifyUrl,
		Amount: weChat.AppRequestAmountRequest{
			Total:    int64(m.PaymentAmountTotal * 100),
			Currency: "CNY",
		},
	}

	prepay, err = appRequest.GetAppPrepay()
	if err != nil {
		return nil, err
	}

	return prepay, nil
}