From 2f5935ed11672046c37f733d855214f6147b4b58 Mon Sep 17 00:00:00 2001
From: admin <weikou2014>
Date: 星期一, 28 三月 2022 11:33:19 +0800
Subject: [PATCH] TDMQ兼容

---
 fanli/src/main/java/com/yeshi/fanli/util/mq/cmq/order/SuningOrderCMQManager.java |  107 +++++++++++++++++++++++++++++------------------------
 1 files changed, 58 insertions(+), 49 deletions(-)

diff --git a/fanli/src/main/java/com/yeshi/fanli/util/mq/cmq/order/SuningOrderCMQManager.java b/fanli/src/main/java/com/yeshi/fanli/util/mq/cmq/order/SuningOrderCMQManager.java
index 12cd6cc..0f80317 100644
--- a/fanli/src/main/java/com/yeshi/fanli/util/mq/cmq/order/SuningOrderCMQManager.java
+++ b/fanli/src/main/java/com/yeshi/fanli/util/mq/cmq/order/SuningOrderCMQManager.java
@@ -4,68 +4,77 @@
 import java.util.List;
 import java.util.Map;
 
+import com.yeshi.fanli.util.mq.cmq.TDMQUtil;
 import org.yeshi.utils.CMQUtil;
 
 import com.qcloud.cmq.Message;
 import com.yeshi.fanli.util.Constant;
 import com.yeshi.fanli.util.StringUtil;
+import scala.collection.immutable.Stream;
 
 public class SuningOrderCMQManager {
 
-	private static String secretId = "AKIDTlpgJhLjOozvd6QI2XnpfGbgV4NQJk25";
-	private static String secretKey = "xhCSUHo55oHUQ6XicFcmfIgspX0EEzWo";
-	private static SuningOrderCMQManager suningOrderCMQManager;
-	private static CMQUtil cmqUtil;
+    private static String secretId = "AKIDTlpgJhLjOozvd6QI2XnpfGbgV4NQJk25";
+    private static String secretKey = "xhCSUHo55oHUQ6XicFcmfIgspX0EEzWo";
+    private static SuningOrderCMQManager suningOrderCMQManager;
+    private static CMQUtil cmqUtil;
 
-	public static String SUNING_ORDER = "order-suning";
+    public static String SUNING_ORDER = "order-suning";
 
-	static {
-		cmqUtil = CMQUtil.getInstance(secretId, secretKey);
-		SUNING_ORDER += "-" + Constant.systemCommonConfig.getProjectName();
-		cmqUtil.createQueue(SUNING_ORDER);
-	}
+    static {
+        cmqUtil = CMQUtil.getInstance(secretId, secretKey);
+        TDMQUtil.getInstance().init(secretId, secretKey, Constant.TDMQ_PUBLIC);
 
-	public static SuningOrderCMQManager getInstance() {
-		if (suningOrderCMQManager == null)
-			suningOrderCMQManager = new SuningOrderCMQManager();
-		return suningOrderCMQManager;
-	}
+        SUNING_ORDER += "-" + Constant.systemCommonConfig.getProjectName();
+        TDMQUtil.getInstance().createQueue(SUNING_ORDER);
+    }
 
-	/**
-	 * 娣诲姞鍞搧浼氳鍗曞埌闃熷垪
-	 * 
-	 * @param orderId
-	 */
-	public void addSuningOrder(String orderId) {
-		if (StringUtil.isNullOrEmpty(orderId))
-			return;
-		cmqUtil.sendMsg(SUNING_ORDER, orderId);
-	}
+    public static SuningOrderCMQManager getInstance() {
+        if (suningOrderCMQManager == null)
+            suningOrderCMQManager = new SuningOrderCMQManager();
+        return suningOrderCMQManager;
+    }
 
-	/**
-	 * 娑堣垂闃熷垪娑堟伅
-	 * 
-	 * @param count
-	 * @return
-	 */
-	public Map<String, String> consumeSuningOrder(int count) {
-		List<Message> list = cmqUtil.recieveMsg(count, SUNING_ORDER);
-		Map<String, String> map = new HashMap<>();
-		if (list != null)
-			for (Message msg : list) {
-				String result = msg.msgBody;
-				map.put(msg.receiptHandle, result);
-			}
-		return map;
-	}
+    /**
+     * 娣诲姞鍞搧浼氳鍗曞埌闃熷垪
+     *
+     * @param orderId
+     */
+    public void addSuningOrder(String orderId) {
+        if (StringUtil.isNullOrEmpty(orderId))
+            return;
+        TDMQUtil.getInstance().sendMsg(SUNING_ORDER, orderId);
+    }
 
-	/**
-	 * 鍒犻櫎鍞搧浼氶槦鍒�
-	 * 
-	 * @param receiptHandle
-	 */
-	public void deleteSuningOrder(String receiptHandle) {
-		cmqUtil.deleteMsg(SUNING_ORDER, receiptHandle);
-	}
+    /**
+     * 娑堣垂闃熷垪娑堟伅
+     *
+     * @param count
+     * @return
+     */
+    public Map<String, String> consumeSuningOrder(int count) {
+        List<Message> list = Constant.TDMQ_CONSUMER ? TDMQUtil.getInstance().recieveMsg(count, SUNING_ORDER) : cmqUtil.recieveMsg(count, SUNING_ORDER);
+        Map<String, String> map = new HashMap<>();
+        if (list != null)
+            for (Message msg : list) {
+                String result = msg.msgBody;
+                map.put(msg.receiptHandle, result);
+            }
+        return map;
+    }
+
+    /**
+     * 鍒犻櫎鍞搧浼氶槦鍒�
+     *
+     * @param receiptHandle
+     */
+    public void deleteSuningOrder(String receiptHandle) {
+
+        if (Constant.TDMQ_CONSUMER) {
+            TDMQUtil.getInstance().deleteMsg(SUNING_ORDER, receiptHandle);
+        } else {
+            cmqUtil.deleteMsg(SUNING_ORDER, receiptHandle);
+        }
+    }
 
 }

--
Gitblit v1.8.0