新增企业的设备总数统计

amos-boot-system-tzs/amos-boot-module-tzs-biz/src/main/java/com/yeejoin/amos/boot/module/tzs/biz/controller/TzsAppController.java

@@ -199,5 +199,13 @@ public class TzsAppController {
         return ResponseHelper.buildResponse(appService.wxUserLogin(wx));
     }
 
+    //企业的设备总数统计
+    @TycloudOperation(ApiLevel = UserType.AGENCY)
+    @ApiOperation(httpMethod = "GET", value = "企业的认领设备总数", notes = "企业的认领设备总数")
+    @GetMapping(value = "/equipmentCount")
+    public ResponseModel<Object> equipClaimOverview(String unitCode) {
+        return ResponseHelper.buildResponse(appService.equipmentCount(unitCode));
+    }
+
 
     }

amos-boot-system-tzs/amos-boot-module-tzs-biz/src/main/java/com/yeejoin/amos/boot/module/tzs/biz/service/impl/TzsAppService.java

@@ -6,6 +6,8 @@ import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.core.MybatisConfiguration;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.yeejoin.amos.boot.biz.common.bo.ReginParams;
 import com.yeejoin.amos.boot.biz.common.utils.QRCodeUtil;
 import com.yeejoin.amos.boot.biz.common.utils.RedisKey;
 import com.yeejoin.amos.boot.biz.common.utils.RedisUtils;
@@ -13,8 +15,11 @@ import com.yeejoin.amos.boot.module.tzs.api.common.CommonException;
 import com.yeejoin.amos.boot.module.tzs.api.common.MobileLoginParam;
 import com.yeejoin.amos.boot.module.tzs.api.dto.*;
 import com.yeejoin.amos.boot.module.tzs.api.entity.*;
+import com.yeejoin.amos.boot.module.tzs.api.enums.EquipmentClassifityEnum;
 import com.yeejoin.amos.boot.module.tzs.api.mapper.CategoryOtherInfoMapper;
+import com.yeejoin.amos.boot.module.tzs.api.mapper.EquipmentCategoryMapper;
 import com.yeejoin.amos.boot.module.tzs.biz.utils.HttpUtils;
+import com.yeejoin.amos.boot.module.tzs.biz.utils.JsonUtils;
 import com.yeejoin.amos.boot.module.tzs.flc.api.entity.RegUnitInfo;
 import com.yeejoin.amos.boot.module.tzs.flc.api.mapper.RegUnitInfoMapper;
 import com.yeejoin.amos.component.feign.model.FeignClientResult;
@@ -32,6 +37,7 @@ import org.apache.ibatis.builder.MapperBuilderAssistant;
 import org.bouncycastle.jce.provider.BouncyCastleProvider;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
+import org.springframework.core.io.Resource;
 import org.springframework.mock.web.MockMultipartFile;
 import org.springframework.stereotype.Service;
 import org.springframework.util.ObjectUtils;
@@ -62,6 +68,9 @@ public class TzsAppService {
     DesignInfoService designInfoService;
 
     @Autowired
+    EquipmentCategoryMapper equipmentCategoryMapper;
+
+    @Autowired
     ProduceInfoService produceInfoService;
 
     @Autowired
@@ -79,6 +88,9 @@ public class TzsAppService {
     @Autowired
     EquipTechParamLiftingService liftingService;
 
+    @Value("classpath:/json/equipCategory.json")
+    private Resource equipCategory;
+
     @Autowired
     EquipTechParamPipelineService pipelineService;
 
@@ -460,4 +472,30 @@ public class TzsAppService {
         JSONObject response = JSONObject.parseObject(responseStr);
         return response;
     }
+
+
+
+    public Page equipmentCount(String companyCode) {
+        Page page = new Page<>();
+        Map<String, List<Map<String, Object>>> resourceJson = JsonUtils.getResourceJson(equipCategory);
+        List<Map<String, Object>> mapList = resourceJson.get(EquipmentClassifityEnum.BDLS.getCode());
+        List<Map<String, Object>> list = new ArrayList<>();
+        List<Map<String, Object>> listMap = equipmentCategoryMapper.getCategoryCount("company", null, companyCode);
+        for (Map<String, Object> map : mapList) {
+            for (Map<String, Object> map1 : listMap) {
+                if (map.get("code").equals(map1.get("category"))) {
+                    map.put("waitClaim", map1.get("waitClaim"));
+                    map.put("alreadyClaim", map1.get("alreadyClaim"));
+                    map.put("refuseClaim", map1.get("refuseClaim"));
+                    Long sum = Long.valueOf(map1.get("waitClaim").toString()) + Long.valueOf(map1.get("alreadyClaim").toString()) + Long.valueOf(map1.get("refuseClaim").toString());
+                    map.put("sum", sum);
+                }
+            }
+            list.add(map);
+        }
+        page.setCurrent(1);
+        page.setTotal(list.size());
+        page.setRecords(list);
+        return page;
+    }
 }