From 73420a13a6d75c8be6163b193142b3954324869e Mon Sep 17 00:00:00 2001
From: jianjun <jianjun4833@163.com>
Date: Wed, 26 Oct 2022 16:23:57 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E7=94=A8=E5=85=B0=E5=9B=BE=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E6=AF=94=E5=AF=B9=E6=8E=A5=E5=8F=A3=20=E9=99=90?=
 =?UTF-8?q?=E6=B5=812?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../commons/tools/utils/YtHsResUtils.java     | 20 +++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/epmet-commons/epmet-commons-tools/src/main/java/com/epmet/commons/tools/utils/YtHsResUtils.java b/epmet-commons/epmet-commons-tools/src/main/java/com/epmet/commons/tools/utils/YtHsResUtils.java
index d30e1087cb..9a4ac14de1 100644
--- a/epmet-commons/epmet-commons-tools/src/main/java/com/epmet/commons/tools/utils/YtHsResUtils.java
+++ b/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);