From 33b887e46e66a19a4d7eb1c41c0da5239ad40692 Mon Sep 17 00:00:00 2001
From: yiming <wang.yiming@hzleaper.com>
Date: Wed, 11 Jan 2023 16:33:05 +0800
Subject: [PATCH] =?UTF-8?q?=E5=BA=93=E5=8C=BA=E8=AE=BE=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../web/controller/RealTimeController.java    |  7 +++++++
 .../leaper/web/service/RealTimeService.java   | 21 +++++++++++++++++++
 2 files changed, 28 insertions(+)

diff --git a/web/src/main/java/com/leaper/web/controller/RealTimeController.java b/web/src/main/java/com/leaper/web/controller/RealTimeController.java
index b95e2f3..7b40c62 100644
--- a/web/src/main/java/com/leaper/web/controller/RealTimeController.java
+++ b/web/src/main/java/com/leaper/web/controller/RealTimeController.java
@@ -48,6 +48,13 @@ public class RealTimeController {
         return Result.success(realTimeService.allAreas());
     }
 
+    @GetMapping("/allCamerasByArea")
+    @ApiOperation(value = "球机按库区分组")
+    //@SessionHandler
+    public Result<List<Camera>> allCamerasByArea(String area) {
+        return Result.success(realTimeService.getAllCamerasByArea(area));
+    }
+
     @GetMapping("/wallStyle")
     @ApiOperation(value = "视频墙样式")
     //@SessionHandler
diff --git a/web/src/main/java/com/leaper/web/service/RealTimeService.java b/web/src/main/java/com/leaper/web/service/RealTimeService.java
index 675d945..7a4e9a8 100644
--- a/web/src/main/java/com/leaper/web/service/RealTimeService.java
+++ b/web/src/main/java/com/leaper/web/service/RealTimeService.java
@@ -1,5 +1,6 @@
 package com.leaper.web.service;
 
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.leaper.web.config.ConfigProperties;
 import com.leaper.web.entity.Camera;
 import com.leaper.web.entity.Street;
@@ -8,6 +9,7 @@ import com.leaper.web.mapper.StreetMapper;
 import com.leaper.web.pojo.realTime.RealTime;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.util.StringUtils;
 
 import javax.annotation.Resource;
 import java.util.*;
@@ -67,5 +69,24 @@ public class RealTimeService {
         return streets.stream().collect(Collectors.groupingBy(Street::getArea)).keySet();
     }
 
+    public List<Camera> getAllCamerasByArea(String area){
+        List<Camera> cameras = new ArrayList<>();
+        if(StringUtils.isEmpty(area)){
+            return cameras;
+        }
+        List<Street> streets = streetMapper.selectList(new QueryWrapper<Street>().eq("`area`",area));
+        List<Integer> cameraIds = new ArrayList<>();
+        streets.forEach(street -> {
+            if(street.getCamera1Id() != null){
+                cameraIds.add(street.getCamera1Id());
+            }
+            if(street.getCamera2Id() != null){
+                cameraIds.add(street.getCamera2Id());
+            }
+        });
+        cameras = cameraMapper.selectBatchIds(cameraIds);
+        return cameras;
+    }
+
 
 }