From cc906040ba149c4244d968c1a264401e15765acd Mon Sep 17 00:00:00 2001
From: sxu <602087911@qq.com>
Date: Tue, 17 Dec 2024 11:11:39 +0800
Subject: [PATCH] getCompanyCount

---
 .../material/device/controller/DevInfoController.java     | 8 ++++++++
 .../com/bonus/material/device/service/DevInfoService.java | 2 ++
 .../material/device/service/impl/DevInfoServiceImpl.java  | 7 +++++++
 3 files changed, 17 insertions(+)

diff --git a/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/device/controller/DevInfoController.java b/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/device/controller/DevInfoController.java
index b8bb748..8f1918e 100644
--- a/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/device/controller/DevInfoController.java
+++ b/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/device/controller/DevInfoController.java
@@ -30,6 +30,7 @@ import javax.validation.Valid;
 import javax.validation.constraints.NotNull;
 import java.io.IOException;
 import java.util.List;
+import java.util.Map;
 
 
 /**
@@ -58,6 +59,13 @@ public class DevInfoController extends BaseController {
         return AjaxResult.success(ListPagingUtil.paging(pageIndex, pageSize, list));
     }
 
+    @ApiOperation(value = "统计公司上架数")
+    @PostMapping("/getCompanyCount")
+    public AjaxResult getCompanyCount(@RequestBody DevInfoVo devInfo) {
+        Map<String, Long> map = devInfoService.getCompanyCount(devInfo);
+        return AjaxResult.success(map);
+    }
+
     /**
      * 获取公司列表(供装备新增及首页所属公司筛选使用)
      * @param obj
diff --git a/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/device/service/DevInfoService.java b/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/device/service/DevInfoService.java
index bf351a5..d67f2e1 100644
--- a/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/device/service/DevInfoService.java
+++ b/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/device/service/DevInfoService.java
@@ -35,6 +35,8 @@ public interface DevInfoService {
      */
     List<DevInfoVo> selectDevInfoList(DevInfoVo devInfo);
 
+    Map<String, Long> getCompanyCount(DevInfoVo devInfo);
+
     List<DevInfoVo> selectUserCollectList(DevInfoVo devInfo);
 
     AjaxResult selectDevInfoHotList(DevInfoVo devInfo);
diff --git a/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/device/service/impl/DevInfoServiceImpl.java b/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/device/service/impl/DevInfoServiceImpl.java
index 878667e..69644d5 100644
--- a/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/device/service/impl/DevInfoServiceImpl.java
+++ b/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/device/service/impl/DevInfoServiceImpl.java
@@ -171,6 +171,13 @@ public class DevInfoServiceImpl implements DevInfoService {
         return voList;
     }
 
+    @Override
+    public Map<String, Long> getCompanyCount(DevInfoVo devInfo) {
+        List<DevInfoVo> voList = devInfoMapper.selectDevInfoList(devInfo);
+        Map<String, Long> map = voList.stream().collect(Collectors.groupingBy(DevInfoVo::getCompanyId, Collectors.counting()));
+        return map;
+    }
+
     /**
      * 查询设备收藏列表
      *