加好友队列

haoban-manage3-service/src/main/java/com/gic/haoban/manage/service/service/out/impl/MemberUnionidRelatedApiServiceImpl.java

@@ -17,7 +17,9 @@ import org.apache.commons.collections.CollectionUtils;
 import org.apache.commons.lang3.StringUtils;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
+import org.redisson.api.RAtomicLong;
 import org.redisson.api.RRateLimiter;
+import org.redisson.api.RSet;
 import org.redisson.api.RateIntervalUnit;
 import org.redisson.api.RateType;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -45,6 +47,7 @@ import com.gic.haoban.app.customer.enums.GicQywxSyncTypeEnum;
 import com.gic.haoban.app.customer.enums.QywxSyncTaskTypeEnum;
 import com.gic.haoban.app.customer.service.api.service.CustomerApiService;
 import com.gic.haoban.base.api.common.ServiceResponse;
+import com.gic.haoban.common.utils.DingUtils;
 import com.gic.haoban.manage.api.constants.Manage3Constants;
 import com.gic.haoban.manage.api.dto.ContentMaterialDTO;
 import com.gic.haoban.manage.api.dto.ExternalClerkRelatedDTO;
@@ -395,20 +398,59 @@ public class MemberUnionidRelatedApiServiceImpl implements MemberUnionidRelatedA
         }
         return null;
     }
-    
+    // 加好友量大的迁移到mq2
+    int maxCount = 2 ;
     private boolean mqCheck(QwFrientNoticeDTO dto) {
         boolean flag = false ;
         Object obj = RedisUtil.getCache("dealQywxExternalUserMq2") ;
+        String corpid = dto.getCorpid() ;
         if(null != obj) {
             String str = obj.toString() ;
-            String corpid = dto.getCorpid() ;
             if(StringUtils.isNotBlank(corpid) && str.contains(corpid)) {
                 flag = true ;
+                log.info("加好友计数,手动配置到userMq2,corpid={}",corpid);
+                return flag ;
+            }
+        }
+        TabHaobanWxEnterprise wxEnterprise = this.wxEnterpriseService.getEnterpriseBycorpId(corpid) ;
+        if(null != wxEnterprise) {
+            String wxEnterpriseId = wxEnterprise.getWxEnterpriseId() ;
+            flag = this.needToMq2(wxEnterpriseId) ;
+            if(!flag) {
+                long times = this.getTimes(wxEnterpriseId) ;
+                if(times>maxCount) {
+                    if(!needToMq2(wxEnterpriseId)) {
+                        DingUtils.send("加好友mq切换,企业="+ wxEnterpriseId, false);
+                    }
+                    log.info("加好友计数,自动到userMq2,{}",wxEnterpriseId);
+                    this.addToMq2Mark(wxEnterpriseId);
+                    flag = true ;
+                }
             }
         }
         return flag ;
     }
     
+    private long getTimes(String wxEnterpriseId) {
+        String key = "qwadd:count:"+wxEnterpriseId ;
+        RAtomicLong cache = RedisUtil.getRedisClient().getAtomicLong(key);
+        long val = cache.addAndGet(1);
+        if (val <= 1) {
+            cache.expire(1, TimeUnit.MINUTES);
+        }
+        return val ;
+    }
+    
+    private void addToMq2Mark(String wxEnterpriseId) {
+        String key = "qwadd:count2:"+wxEnterpriseId ;
+        RedisUtil.setCache(key, 1,600L);
+    }
+    
+    private boolean needToMq2(String wxEnterpriseId) {
+        String key = "qwadd:count2:"+wxEnterpriseId ;
+        return null != RedisUtil.getCache(key) ;
+    }
+    
     @Override
     public String addMemberUnionidRelatedNew(MemberUnionidRelatedDTO dto, String wxUserId) {
         String corpId = dto.getCorpid();