添加limit

amos-boot-system-jxiop/amos-boot-module-hygf-biz/src/main/java/com/yeejoin/amos/boot/module/hygf/biz/controller/PeasantHouseholdWxController.java

@@ -2,6 +2,7 @@ package com.yeejoin.amos.boot.module.hygf.biz.controller;
 
 import cn.hutool.core.collection.CollectionUtil;
 import com.alibaba.fastjson.JSONArray;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.yeejoin.amos.boot.biz.common.controller.BaseController;
@@ -11,10 +12,12 @@ import com.yeejoin.amos.boot.module.hygf.api.dto.PeasantHouseholdDto;
 import com.yeejoin.amos.boot.module.hygf.api.dto.PeasantHouseholdPageDto;
 import com.yeejoin.amos.boot.module.hygf.api.dto.PeasantHouseholdWxDto;
 import com.yeejoin.amos.boot.module.hygf.api.entity.PeasantHousehold;
+import com.yeejoin.amos.boot.module.hygf.api.entity.RegionalCompanies;
 import com.yeejoin.amos.boot.module.hygf.api.service.IWxService;
 import com.yeejoin.amos.boot.module.hygf.api.util.BeanDtoUtils;
 import com.yeejoin.amos.boot.module.hygf.biz.service.impl.HouseholdContractServiceImpl;
 import com.yeejoin.amos.boot.module.hygf.biz.service.impl.PeasantHouseholdServiceImpl;
+import com.yeejoin.amos.boot.module.hygf.biz.service.impl.RegionalCompaniesServiceImpl;
 import com.yeejoin.amos.boot.module.hygf.biz.service.impl.SurveyInformationServiceImpl;
 import com.yeejoin.amos.component.feign.model.FeignClientResult;
 import com.yeejoin.amos.component.robot.AmosRequestContext;
@@ -70,6 +73,8 @@ public class PeasantHouseholdWxController extends BaseController {
     private String miniprogramLoginPage;
     @Autowired
     private AmosRequestContext requestContext;
+    @Autowired
+    private RegionalCompaniesServiceImpl regionalCompaniesServiceImpl;
 
     /*@TycloudOperation (ApiLevel = UserType.AGENCY, needAuth = false)
     @RequestMapping (value = "/authenticate", method = RequestMethod.POST)
@@ -279,4 +284,18 @@ public class PeasantHouseholdWxController extends BaseController {
         }
     }
 
+    @TycloudOperation(ApiLevel = UserType.AGENCY, needAuth = false)
+    @GetMapping({"/registerLimit"})
+    @ApiOperation(httpMethod = "GET", value = ", notes = ")
+    public ResponseModel<Boolean> registerLimit(@RequestParam(value = "regionalCompaniesCode", required = true) String regionalCompaniesCode) {
+        boolean limit = false;
+        LambdaQueryWrapper<RegionalCompanies> wrapper = new LambdaQueryWrapper();
+        wrapper.eq(RegionalCompanies::getRegionalCompaniesCode, regionalCompaniesCode)
+        .eq(RegionalCompanies::getLimits, Integer.valueOf(1));
+        int count = this.regionalCompaniesServiceImpl.count(wrapper);
+        if (count > 0) {
+            limit = true;
+        }
+        return ResponseHelper.buildResponse(Boolean.valueOf(limit));
+    }
 }