提交新项目

jhmicro/query/order_query.go

@@ -0,0 +1,109 @@
+/***************************************************
+ ** @Desc : This file for ...
+ ** @Time : 2019/11/22 23:02
+ ** @Author : yuebin
+ ** @File : order_query
+ ** @Last Modified by : yuebin
+ ** @Last Modified time: 2019/11/22 23:02
+ ** @Software: GoLand
+****************************************************/
+package query
+
+import (
+	"fmt"
+	"github.com/astaxie/beego/logs"
+	"github.com/go-stomp/stomp"
+	"juhe/service/common"
+	"juhe/service/controller"
+	"juhe/service/message_queue"
+	"juhe/service/models"
+	"os"
+	"time"
+)
+
+type OrderQueryTask struct {
+	BankOrderId     string
+	OrderQueryTimer *time.Timer
+	Times           int
+}
+
+const (
+	DelayTime  = 5 //延时时间为5分钟查询一次
+	LimitTimes = 5 //最多查询5次
+)
+
+/*
+** 该接口是查询上游的订单
+ */
+func solveSupplierOrderQuery(task OrderQueryTask) {
+	bankOrderId := task.BankOrderId
+	orderInfo := models.GetOrderByBankOrderId(bankOrderId)
+	if orderInfo.BankOrderId == "" || len(orderInfo.BankOrderId) == 0 {
+		logs.Error("不存在这样的订单，订单查询结束")
+		return
+	}
+	if orderInfo.Status != "" && orderInfo.Status != "wait" {
+		logs.Error(fmt.Sprintf("该订单=%s，已经处理完毕，", bankOrderId))
+		return
+	}
+	supplierCode := orderInfo.PayProductCode
+	supplier := controller.GetPaySupplierByCode(supplierCode)
+	flag := supplier.PayQuery(orderInfo)
+	if flag {
+		logs.Info("订单查询成功， bankOrderId：", bankOrderId)
+	} else {
+		if task.Times <= LimitTimes {
+			task.Times += 1
+			task.OrderQueryTimer = time.NewTimer(time.Duration(5) * time.Minute)
+			DelayOrderQueryQueue(task)
+		} else {
+			logs.Notice(fmt.Sprintf("订单id=%s, 已经查询超过次数"))
+		}
+	}
+}
+
+/*
+* 延时队列
+ */
+func DelayOrderQueryQueue(task OrderQueryTask) {
+	for {
+		select {
+		case <-task.OrderQueryTimer.C:
+			logs.Info(fmt.Sprintf("订单id=%s,执行第：%d 次查询", task.BankOrderId, task.Times))
+			solveSupplierOrderQuery(task)
+			return
+		case <-time.After(time.Duration(2*DelayTime) * time.Minute):
+			return
+		}
+	}
+}
+
+/*
+** 启动消息订单查询的消息队列消费者
+ */
+func CreateSupplierOrderQueryCuConsumer() {
+	conn := message_queue.GetActiveMQConn()
+	if conn == nil {
+		logs.Error("supplier order query consumer fail")
+		os.Exit(1)
+	}
+	logs.Notice("启动订单查询的消费者成功.....")
+	orderQuerySub, _ := conn.Subscribe(common.MqOrderQuery, stomp.AckClient)
+
+	for {
+		select {
+		case v := <-orderQuerySub.C:
+			if v != nil {
+				bankOrderId := string(v.Body)
+				logs.Info("消费者正在处理订单查询： " + bankOrderId)
+				task := OrderQueryTask{BankOrderId: bankOrderId, OrderQueryTimer: time.NewTimer(time.Second * 1), Times: 1}
+				DelayOrderQueryQueue(task)
+				//应答，重要
+				err := conn.Ack(v)
+				if err != nil {
+					logs.Error("消息应答失败！")
+				}
+			}
+		}
+	}
+}

jhmicro/query/pay_for_query.go

@@ -0,0 +1,135 @@
+/***************************************************
+ ** @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/astaxie/beego/logs"
+	"github.com/go-stomp/stomp"
+	"juhe/service/common"
+	"juhe/service/controller"
+	"juhe/service/message_queue"
+	"juhe/service/models"
+	"juhe/service/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("消息应答失败！")
+				}
+			}
+		}
+	}
+}