hd/services/pay/handle.go

package pay_service

import (
	"errors"
	"fmt"
	"github.com/xormplus/xorm"
	"hudongzhuanjia/logger"
	"hudongzhuanjia/models"
	"hudongzhuanjia/services/im"
	"hudongzhuanjia/utils"
	"hudongzhuanjia/utils/define"
	"time"
)

func init() {
	//go loopOrder()
	go utils.HandleTicker(30*time.Minute, HandleReward)    // 打赏24小时退款
	go utils.HandleTicker(30*time.Second, HandleGoodOrder) // 订单关闭信息
}

//处理支付成功之后的回调问题
func HandleSuccess(order *models.UserOrder) error {
	if order.GoodType == 4 { // 直播商品
		customerOrder := new(models.CustomerOrder)
		_, err := customerOrder.UpdateStatusBy(order.OutTradeNo, 0, 1)
		if err != nil {
			return errors.New("更新状态发送错误")
		}
		customerOrder.GetByOutTradeNO(order.OutTradeNo)
		if customerOrder.Status != 1 {
			return errors.New("商品订单状态信息出现异常")
		}
		subs, err := models.GetCustomerOrderSubsByOrderNos(customerOrder.OrderNo)
		if err != nil {
			logger.Error("商品子订单查询异常", err)
			return err
		}
		for _, sub := range subs {
			go im_service.SendGroupCustomMessage(customerOrder.BuyerId, order.ActivityId, im_service.NoticeLiveGoodBuy,
				map[string]interface{}{
					"goods_pic_url": sub["goods_pic_url"],
					"goods_num":     sub["goods_num"],
					"good_name":     sub["good_name"],
					"good_price":    sub["good_price"],
					"goods_id":      sub["goods_id"],
					"timestamp":     time.Now().Unix(),
				})
		}

		activity := new(models.Activity)
		exist, err := models.Get(activity, order.ActivityId)
		if err != nil || !exist {
			return errors.New("互动信息异常")
		}

		gift := new(models.OrderGift)
		exist, err = gift.GetByActivityId(order.ActivityId)
		if err != nil {
			err = fmt.Errorf("获取订单送礼错误: err-> %v, activity_id-> %v", err, activity.Id)
			logger.Error(err)
			return err
		}

		if !exist {
			return nil
		}

		prize := new(models.UserPrize)
		prize.UserId = order.UserId
		prize.ActivityId = order.ActivityId
		prize.RehearsalId = activity.RehearsalId
		prize.ActivityName = activity.Name
		prize.PrizeName = gift.GiftName
		prize.PrizeImg = gift.GiftPicUrl
		prize.PrizeType = 4
		prize.IsDelete = false
		prize.Status = 1
		prize.CreatedAt = time.Now()
		prize.UpdatedAt = time.Now()
		if gift.Num > 0 {
			count, err := new(models.CustomerOrder).
				Count(activity.Id, activity.RehearsalId, customerOrder.CreatedAt)
			if err != nil {
				err = fmt.Errorf("订单人数统计失败: err->%v, out_trade_no->%v", err, order.OutTradeNo)
				logger.Error(err)
				return err
			}

			if gift.Num <= int(count) { // 人数足够
				return nil
			}
		}
		_, err = prize.Add()
		if err != nil {
			err = fmt.Errorf("奖品添加失败: err->%v, out_trade_no->%v", err, order.OutTradeNo)
			logger.Error(err)
			return err
		}

		customerOrder.UserPrizeId = prize.Id
		_, err = models.Update(customerOrder.Id, customerOrder, "user_prize_id")
		if err != nil {
			err = fmt.Errorf("奖品记录添加失败: err-> %v, out_trade_no-> %v, prize_id->%v", err, order.OutTradeNo, prize.Id)
			logger.Error(err)
			return err
		}
	} else if order.GoodType == 3 { // 直播红包
		info := new(models.LiveRedEnvelopeRule)
		_, err := info.UpdateStatusByOutTradeNo(order.OutTradeNo, 1)
		if err != nil {
			err = fmt.Errorf("状态改变出现错误: err->%v, out_trade_no->%v", err, order.OutTradeNo)
			logger.Error(err.Error())
			return err
		}
		exist, err := info.GetByOutTradeNo(order.OutTradeNo)
		if err != nil || !exist || info.Status != 1 {
			err = fmt.Errorf("直播红包信息异常: err-> %v, exist->%v, status->%v, out_trade_no->%v",
				err, exist, info.Status, order.OutTradeNo)
			logger.Error(err.Error())
			return err
		}

		go im_service.SendGroupCustomMessage(info.UserId, order.ActivityId, im_service.NoticeLiveRedPackStart,
			map[string]interface{}{
				"live_red_envelope_rule_id": info.Id,
				"prompt":                    info.Prompt,
				"timestamp":                 time.Now().Unix(),
			})
	} else if order.GoodType == 2 { // 打赏
		history := new(models.RewardHistory)
		exist, err := history.GetByOutTradeNo(order.OutTradeNo)
		if err != nil || !exist || history.Status != -1 {
			err = fmt.Errorf("打赏历史异常: err->%v, out_trade_no %v", err, order.OutTradeNo)
			logger.Error(err.Error())
			return err
		}
		_, err = history.UpdateStatus(history.Id, 0)
		if err != nil {
			err = fmt.Errorf("打赏状态发送变化: err->%v, out_trade_no %v", err, order.OutTradeNo)
			logger.Error(err.Error())
			return err
		}
	} else if order.GoodType == 1 { // 霸屏
		history := new(models.BullyScreenHistory)
		exist, err := history.GetByOutTradeNo(order.OutTradeNo)
		if err != nil || !exist || history.Status != -1 {
			err = fmt.Errorf("打赏历史异常: err->%v, out_trade_no %v", err, order.OutTradeNo)
			logger.Error(err.Error())
			return err
		}

		_, err = history.UpdateStatus(history.Id, 0)
		if err != nil {
			err = fmt.Errorf("霸屏状态发送变化: err->%v, out_trade_no %v", err, order.OutTradeNo)
			logger.Error(err.Error())
		}
	}
	return nil
}

// 处理商品订单超时的状态
func HandleGoodOrder() error {
	orders, err := models.GetExpiredAtLiveCustomerOrder()
	if err != nil {
		err = fmt.Errorf("获取过期订单信息错误原因: %v", err)
	}
	for _, v := range orders {
		if v.Type != 1 || v.Status != 0 {
			continue
		}
		res, err := OrderQuery(v.OutTradeNo)
		if err != nil {
			err = fmt.Errorf("订单查询出现错误: %v, out_trade_no: %v", err, v.OutTradeNo)
			return err
		}
		if res["trade_state"] != define.CODE_TRADE_NOTPAY {
			err = fmt.Errorf("订单非未支付订单: %v, out_trade_no: %v", res["trade_state"], v.OutTradeNo)
			return err
		}

		err = Close(v.OutTradeNo)
		if err != nil {
			return err
		}
		v.Status = 9
		v.CancelTime = time.Now()
		_, err = models.Update(v.Id, v, "status", "cancel_time")
		if err != nil {
			return err
		}

		subs, err := models.GetSubOrdersByOrderNo(v.OrderNo)
		if err != nil {
			return err
		}
		for _, sub := range subs {
			_, err = new(models.CustomerGoods).IncrStockById(sub.GoodsId, sub.GoodsNum)
			if err != nil {
				return err
			}
		}
	}
	return nil
}

// 处理打赏
func HandleReward() error {
	result, err := models.GetExpireRewardHistory()
	if err != nil {
		err = fmt.Errorf("获取过期打赏错误原因: %v", err)
		return err
	}
	for _, v := range result {
		if v.RehearsalId != 0 && v.Type == 0 { // 欧轩互动h5彩排
			continue
		}
		_, err := Refund("欧轩互动-打赏过期退款", v.OutTradeNo)
		if err != nil {
			err = fmt.Errorf("欧轩互动打赏过期退款错误: %v, out_trade_no: %v", err, v.OutTradeNo)
			return err
		}
		v.Status = 4
		_, err = models.Update(v.Id, v, "status")
		if err != nil {
			err = fmt.Errorf("欧轩互动打赏过期状态改变错误: %v, out_trade_no: %v", err, v.OutTradeNo)
			return err
		}
	}
	return nil
}

func HandleCancelOrder(outTradeNo string) error {
	return models.Commit(func(session *xorm.Session) error {
		order := new(models.CustomerOrder)
		exist, err := session.Where("is_delete=0 and status=0 and out_trade_no=?",
			outTradeNo).Get(order)
		if err != nil {
			return err
		}
		if !exist || order.Status != 0 {
			return errors.New("客户订单异常")
		}
		order.Status = 9
		order.CancelTime = time.Now()
		_, err = session.Where("is_delete=0 and out_trade_no=?",
			outTradeNo).Cols("status, cancel_time").Update(order)
		if err != nil {
			return err
		}
		subs := make([]*models.CustomerOrderSub, 0)
		err = session.Where("is_delete=0 and order_no=?", order.OrderNo).Find(&subs)
		if err != nil {
			return err
		}
		for _, sub := range subs {
			_, err = session.Where("id=?", sub.GoodsId).
				Incr("stock", sub.GoodsNum).Update(&models.CustomerGoods{})

			if err != nil {
				return err
			}
		}
		return nil
	})
}