登录

haoban-manage3-web/src/main/java/com/gic/haoban/manage/web/controller/LoginController.java

@@ -41,6 +41,7 @@ import java.io.UnsupportedEncodingException;
 import java.net.URLEncoder;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.stream.Collectors;
 
 
 @RestController
@@ -282,6 +283,7 @@ public class LoginController extends WebBaseController{
             logger.info("解密异常");
             return resultResponse(HaoBanErrCode.ERR_5);
         }
+        logger.info(o);
         JSONObject jsonObject = JSONObject.parseObject(o);
         String gicEnterpriseId = jsonObject.getString("enterpriseId");
         String wxEnterpriseId = jsonObject.getString("wxEnterpriseId");
@@ -290,7 +292,10 @@ public class LoginController extends WebBaseController{
             logger.info("登录失败，不存在对应企业:{}", o);
             return resultResponse(HaoBanErrCode.ERR_6);
         }
-
+        if (StringUtils.isNotBlank(wxEnterpriseId)) {
+            List<WxEnterpriseDTO> collect = enterpriseDTOList.stream().filter(dto -> dto.getWxEnterpriseId().equals(wxEnterpriseId)).collect(Collectors.toList());
+            return resultResponse(HaoBanErrCode.ERR_1, collect);
+        }
         return resultResponse(HaoBanErrCode.ERR_1, enterpriseDTOList);
 
     }