dongfeng-pay/microservice/query/pay_for_query.go

/***************************************************
 ** @Desc : 处理代付查询功能
 ** @Time : 2019/12/3 15:07
 ** @Author : yuebin
 ** @File : pay_for_query
 ** @Last Modified by : yuebin
 ** @Last Modified time: 2019/12/3 15:07
 ** @Software: GoLand
****************************************************/
package query

import (
	"fmt"
	"github.com/beego/beego/v2/core/logs"
	"github.com/go-stomp/stomp"
	"microservice/common"
	"microservice/message_queue"
	"microservice/models"
	controller "microservice/supplier"
	"microservice/utils"
	"os"
	"time"
)

type PayForQueryTask struct {
	Delay           *time.Timer
	MerchantOrderId string
	BankOrderId     string
	FirstNotifyTime string
	QueryTimes      int
	LimitTimes      int
	Status          string
}

const (
	PayForLimitTimes    = 12 //最多查询次数
	PayForQueryInterval = 5  //时间间隔为5分钟
)

func PayForQueryTimer(task PayForQueryTask) {
	for {
		select {
		case <-task.Delay.C:
			PayForSupplier(task)
			task.Delay.Stop()
			return
			//70分钟没有执行该协程，那么退出协程
		case <-time.After(time.Minute * 70):
			return
		}
	}
}

func PayForSupplier(task PayForQueryTask) {
	logs.Info(fmt.Sprintf("执行代付查询任务：%+v", task))
	payFor := models.GetPayForByBankOrderId(task.BankOrderId)
	roadInfo := models.GetRoadInfoByRoadUid(payFor.RoadUid)
	supplier := controller.GetPaySupplierByCode(roadInfo.ProductUid)
	if supplier == nil {
		logs.Error("代付查询返回supplier为空")
		return
	}
	res, _ := supplier.PayForQuery(payFor)
	if res == common.PAYFOR_SUCCESS {
		//代付成功了
		controller.PayForSuccess(payFor)
	} else if res == common.PAYFOR_FAIL {
		//代付失败
		controller.PayForFail(payFor)
	} else if res == common.PAYFOR_BANKING {
		//银行处理中，那么就继续执行查询，直到次数超过最大次数
		if task.QueryTimes <= task.LimitTimes {
			task.QueryTimes += 1
			task.Delay = time.NewTimer(time.Duration(PayForQueryInterval) * time.Minute)
			go PayForQueryTimer(task)
		} else {
			logs.Info(fmt.Sprintf("该代付订单已经超过最大查询次数，bankOrderId = %s", task.BankOrderId))
		}
	}
}

func payForQueryConsumer(bankOrderId string) {
	exist := models.IsExistPayForByBankOrderId(bankOrderId)
	if !exist {
		logs.Error(fmt.Sprintf("代付记录不存在，bankOrderId = %s", bankOrderId))
		return
	}

	payFor := models.GetPayForByBankOrderId(bankOrderId)

	if payFor.Status != common.PAYFOR_BANKING {
		logs.Info(fmt.Sprintf("代付状态不是银行处理中，不需要去查询，bankOrderId = %s", bankOrderId))
		return
	}

	payForQueryTask := PayForQueryTask{Delay: time.NewTimer(time.Duration(PayForQueryInterval) * time.Minute), MerchantOrderId: payFor.MerchantOrderId,
		BankOrderId: payFor.BankOrderId, FirstNotifyTime: utils.GetBasicDateTime(), QueryTimes: 1, LimitTimes: PayForLimitTimes, Status: payFor.Status}

	go PayForQueryTimer(payForQueryTask)
}

/*
* 创建代付查询的消费者
 */
func CreatePayForQueryConsumer() {
	//启动定时任务
	conn := message_queue.GetActiveMQConn()
	if conn == nil {
		logs.Error("启动消息队列消费者失败....")
		os.Exit(1)
	}

	logs.Notice("代付查询消费启动成功......")

	payForQuery, err := conn.Subscribe(common.MQ_PAYFOR_QUERY, stomp.AckClient)
	if err != nil {
		logs.Error("订阅代付查询失败......")
		os.Exit(1)
	}

	for {
		select {
		case v := <-payForQuery.C:
			if v != nil {
				bankOrderId := string(v.Body)
				go payForQueryConsumer(bankOrderId)
				//应答，重要
				err := conn.Ack(v)
				if err != nil {
					logs.Error("消息应答失败！")
				}
			}
		}
	}
}