hepa-calc-api/api/controller/CallBack.go

package controller

import (
	"github.com/gin-gonic/gin"
	"hepa-calc-api/api/dao"
	"hepa-calc-api/api/responses"
	"hepa-calc-api/api/service"
	"hepa-calc-api/extend/weChat"
	"hepa-calc-api/global"
	"hepa-calc-api/utils"
	"net/http"
	"time"
)

// CallBack 回调
type CallBack struct{}

// WxPaySingle 微信支付回调-单项
func (r *CallBack) WxPaySingle(c *gin.Context) {
	notifyReq, transaction, err := weChat.ParseNotify(c)
	if err != nil {
		responses.FailWithMessage(err.Error(), c)
		return
	}

	// 记录日志
	utils.LogJsonErr("微信支付回调-单项", notifyReq)
	utils.LogJsonErr("微信支付回调-单项", transaction)

	if transaction == nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": "ERROR", "message": "缺少订单数据"})
		return
	}

	if transaction.OutTradeNo == nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": "ERROR", "message": "缺少外部订单号"})
		return
	}

	// 查询订单
	orderSingleDao := dao.OrderSingleDao{}
	maps := make(map[string]interface{})
	maps["order_no"] = transaction.OutTradeNo
	orderSingle, err := orderSingleDao.GetOrderSingle(maps)
	if err != nil || orderSingle == nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": "ERROR", "message": "无订单数据"})
		return
	}

	// 验证订单状态-订单状态（1:待支付 2:已完成 3:已取消）
	if orderSingle.OrderStatus != 1 {
		message := "订单状态:" + utils.OrderSingleStatusToString(orderSingle.OrderStatus) + " 无需处理"
		c.JSON(http.StatusOK, gin.H{"code": "SUCCESS", "message": message})
		return
	}

	// 处理支付状态
	if transaction.TradeState == nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": "ERROR", "message": "缺少支付状态"})
		return
	}

	// 处理支付状态
	wxPayResult, err := weChat.HandlePayStatus(transaction)
	if err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": "ERROR", "message": err})
		return
	}

	// 开始事务
	tx := global.Db.Begin()
	defer func() {
		if r := recover(); r != nil {
			tx.Rollback()
		}
	}()

	// 修改订单
	orderSingleData := make(map[string]interface{})
	if wxPayResult.OrderStatus != nil {
		orderSingleData["order_status"] = &wxPayResult.OrderStatus
	}
	orderSingleData["pay_status"] = &wxPayResult.PayStatus
	if wxPayResult.PayTime != nil {
		orderSingleData["pay_time"] = &wxPayResult.PayTime
	}
	orderSingleData["escrow_trade_no"] = transaction.TransactionId
	orderSingleData["updated_at"] = time.Now().Format("2006-01-02 15:04:05")
	err = orderSingleDao.EditOrderSingleById(tx, orderSingle.OrderId, orderSingleData)
	if err != nil {
		tx.Rollback()
		c.JSON(http.StatusBadRequest, gin.H{"code": "ERROR", "message": err})
		return
	}

	// 增加题目支付次数
	questionDao := dao.QuestionDao{}
	err = questionDao.Inc(tx, orderSingle.QuestionId, "pay_count", 1)
	if err != nil {
		tx.Rollback()
		responses.FailWithMessage("内部错误", c)
		return
	}

	tx.Commit()
	c.JSON(http.StatusOK, gin.H{"code": "SUCCESS", "message": "OK"})
}

// WxPayMember 微信支付回调-会员
func (r *CallBack) WxPayMember(c *gin.Context) {
	notifyReq, transaction, err := weChat.ParseNotify(c)
	if err != nil {
		responses.FailWithMessage(err.Error(), c)
		return
	}

	// 记录日志
	utils.LogJsonErr("微信支付回调-会员", notifyReq)
	utils.LogJsonErr("微信支付回调-会员", transaction)

	if transaction == nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": "ERROR", "message": "缺少订单数据"})
		return
	}

	if transaction.OutTradeNo == nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": "ERROR", "message": "缺少外部订单号"})
		return
	}

	// 查询订单
	orderMemberDao := dao.OrderMemberDao{}
	maps := make(map[string]interface{})
	maps["order_no"] = transaction.OutTradeNo
	orderMember, err := orderMemberDao.GetOrderMember(maps)
	if err != nil || orderMember == nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": "ERROR", "message": "无订单数据"})
		return
	}

	// 验证订单状态-订单状态（1:待支付 2:已完成 3:已取消）
	if orderMember.OrderStatus != 1 {
		message := "订单状态:" + utils.OrderSingleStatusToString(orderMember.OrderStatus) + " 无需处理"
		c.JSON(http.StatusOK, gin.H{"code": "SUCCESS", "message": message})
		return
	}

	// 处理支付状态
	if transaction.TradeState == nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": "ERROR", "message": "缺少支付状态"})
		return
	}

	// 处理支付状态
	wxPayResult, err := weChat.HandlePayStatus(transaction)
	if err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": "ERROR", "message": err})
		return
	}

	// 获取用户数据
	userDao := dao.UserDao{}
	user, err := userDao.GetUserById(orderMember.UserId)
	if err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": "ERROR", "message": "用户数据错误"})
		return
	}

	// 开始事务
	tx := global.Db.Begin()
	defer func() {
		if r := recover(); r != nil {
			tx.Rollback()
		}
	}()

	// 修改订单
	orderSingleData := make(map[string]interface{})
	if wxPayResult.OrderStatus != nil {
		orderSingleData["order_status"] = &wxPayResult.OrderStatus
	}
	orderSingleData["pay_status"] = &wxPayResult.PayStatus
	if wxPayResult.PayTime != nil {
		orderSingleData["pay_time"] = &wxPayResult.PayTime
	}
	orderSingleData["escrow_trade_no"] = transaction.TransactionId
	orderSingleData["updated_at"] = time.Now().Format("2006-01-02 15:04:05")
	err = orderMemberDao.EditOrderMemberById(tx, orderMember.OrderId, orderSingleData)
	if err != nil {
		tx.Rollback()
		c.JSON(http.StatusInternalServerError, gin.H{"code": "ERROR", "message": err})
		return
	}

	// 处理未支付单项订单
	if *wxPayResult.OrderStatus == 2 {
		orderSingleService := service.OrderSingleService{}
		res, err := orderSingleService.CompleteUnPayOrderSingle(tx, orderMember.UserId)
		if err != nil || res == false {
			tx.Rollback()
			c.JSON(http.StatusInternalServerError, gin.H{"code": "ERROR", "message": err})
			return
		}
	}

	// 处理用户会员过期时间
	if *wxPayResult.OrderStatus == 2 {
		// 获取订单配置
		systemMemberDao := dao.SystemMemberDao{}
		systemMember, err := systemMemberDao.GetSystemMemberById(orderMember.SystemMemberId)
		if err != nil {
			tx.Rollback()
			c.JSON(http.StatusInternalServerError, gin.H{"code": "ERROR", "message": "会员配置错误"})
			return
		}

		userService := service.UserService{}
		res := userService.AddUserMemberValidDate(tx, user, int(systemMember.MemberDays))
		if res == false {
			tx.Rollback()
			c.JSON(http.StatusInternalServerError, gin.H{"code": "ERROR", "message": "增加用户会员到期时间失败"})
			return
		}
	}

	tx.Commit()
	c.JSON(http.StatusOK, gin.H{"code": "SUCCESS", "message": "OK"})
}