调用兰图数据比对接口 限流2

epmet-commons/epmet-commons-tools/src/main/java/com/epmet/commons/tools/utils/YtHsResUtils.java

@@ -6,11 +6,14 @@ import com.epmet.commons.tools.constant.StrConstant;
 import com.epmet.commons.tools.dto.result.YtDataSyncResDTO;
 import com.epmet.commons.tools.dto.result.YtHscyResDTO;
 import com.epmet.commons.tools.dto.result.YtHsjcResDTO;
+import com.google.common.util.concurrent.RateLimiter;
 import lombok.extern.slf4j.Slf4j;
 
 import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.TimeUnit;
 
 /**
  * @Description
@@ -24,6 +27,11 @@ public class YtHsResUtils {
     private static final String CARD_NO = "card_no";
     private static final String ROW_NUM = "ROWNUM";
     private static final String PAGE_SIZE = "PAGESIZE";
+    /**
+     * rate is "2 permits per second"
+     */
+    private static final Map<String,RateLimiter> rateLimiterMap = new ConcurrentHashMap<>();
+
 
     /**
      * desc:核酸采样查询
@@ -32,6 +40,8 @@ public class YtHsResUtils {
      */
     public static YtHscyResDTO hscy(String cardNo, Integer rowNum, Integer pageSize) {
         try {
+            tryAcquire("hscy");
+
             Map<String, Object> param = new HashMap<>();
             param.put(APP_KEY, "tz45j4kuWcnnjoJOVSlzGWJgI");
             param.put(CARD_NO, cardNo);
@@ -60,6 +70,12 @@ public class YtHsResUtils {
         return resultResult;
     }
 
+    private static void tryAcquire(String hscy) {
+        RateLimiter rateLimiter = rateLimiterMap.getOrDefault(hscy, RateLimiter.create(2.0));
+        rateLimiterMap.putIfAbsent(hscy,rateLimiter);
+        rateLimiter.tryAcquire(1, TimeUnit.SECONDS);
+    }
+
     /**
      * desc:核酸结果查询
      *
@@ -67,6 +83,8 @@ public class YtHsResUtils {
      */
     public static YtHsjcResDTO hsjc(String cardNo, Integer rowNum, Integer pageSize) {
         try {
+            tryAcquire("hsjc");
+
             Map<String, Object> param = new HashMap<>();
             param.put(APP_KEY, "DR4jF5Be7sCsqDmCamq2tmYCl");
             param.put(CARD_NO, cardNo);
@@ -102,6 +120,7 @@ public class YtHsResUtils {
      */
     public static YtDataSyncResDTO siWang(String cardNo, String userName) {
         try {
+            tryAcquire("siWang");
 //            1)appkey秘钥
 //            2)name姓名 必填
 //            3)idcard身份证号 必填
@@ -159,6 +178,7 @@ public class YtHsResUtils {
      * @remark：默认失败 因为一旦成功没有数据 会影响残疾人的数据 接口失败数据不做任何处理
      */
     public static YtDataSyncResDTO canji(String idCard, String userName) {
+        tryAcquire("canji");
 
         YtDataSyncResDTO failResult = new YtDataSyncResDTO();
         failResult.setCode(500);