自建通讯录回调

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

@@ -154,6 +154,20 @@ public class MessageApiServiceImpl implements MessageApiService {
         }
     }
 
+    private String getWxEnterpriseId(String corpid) {
+        String wxEnterpriseId = null ;
+        TabHaobanWxApplication wxApplication = wxApplicationService.selectByCorpId(corpid);
+        if (null == wxApplication) {
+            TabHaobanWxEnterprise wxEnterprise =  this.wxEnterpriseService.getEnterpriseBycorpId(corpid) ;
+            if(null != wxEnterprise) {
+                wxEnterpriseId = wxEnterprise.getWxEnterpriseId() ;
+            }
+        }else {
+            wxEnterpriseId = wxApplication.getWxEnterpriseId();
+        }
+        return wxEnterpriseId ;
+    }
+
     @Override
     public void dealWxMessage(String param) {
         QywxCallBackDTO dto = JSON.parseObject(param, QywxCallBackDTO.class);
@@ -163,12 +177,11 @@ public class MessageApiServiceImpl implements MessageApiService {
             log.info("通讯录只处理自建应用回调 dto={}", JSON.toJSONString(param));
             return;
         }
-        TabHaobanWxApplication wxApplication = wxApplicationService.selectByCorpId(dto.getAuthCorpId());
-        if (Objects.isNull(wxApplication)) {
+        String wxEnterpriseId = this.getWxEnterpriseId(dto.getAuthCorpId()) ;
+        if(null == wxEnterpriseId) {
             log.info("授权企业不存在，{}", dto.getAuthCorpId());
             return;
         }
-        String wxEnterpriseId = wxApplication.getWxEnterpriseId();
         WxEnterpriseQwDTO qwDTO = this.wxEnterpriseService.getQwInfo(wxEnterpriseId) ;
         if(qwDTO.isSelf() && dto.getSelfCallback()==0) {
             log.info("非自建应用回调");
@@ -437,12 +450,11 @@ public class MessageApiServiceImpl implements MessageApiService {
         String authCorpId = qywxCallBackDTO.getAuthCorpId();
         DepartmentDTO departmentDTO = new DepartmentDTO();
         departmentDTO.setIsStore(0);
-        TabHaobanWxApplication wxApplication = wxApplicationService.selectByCorpId(authCorpId);
-        if (Objects.isNull(wxApplication)) {
-            log.info("授权企业不存在，{}", qywxCallBackDTO.getAuthCorpId());
+        String wxEnterpriseId = this.getWxEnterpriseId(authCorpId) ;
+        if(null == wxEnterpriseId) {
+            log.info("授权企业不存在，{}", authCorpId);
             return;
         }
-        String wxEnterpriseId = wxApplication.getWxEnterpriseId();
         WxEnterpriseQwDTO qwDTO = this.wxEnterpriseService.getQwInfo(wxEnterpriseId) ;
         SecretSettingDTO secretSetting = secretSettingService.getSecretSetting(wxEnterpriseId,
                 SecretTypeEnum.CUSTOMIZED_APP.getVal());