From a3f0ebb5beabb475879009cb76878106377e1291 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=83=AD=E6=89=BF?= <guocheng@xinyingpower.com>
Date: Thu, 31 Oct 2024 11:45:23 +0800
Subject: [PATCH] =?UTF-8?q?feat:=E7=9B=91=E6=B5=8B=E8=AE=BE=E5=A4=87?=
 =?UTF-8?q?=E6=9F=A5=E8=AF=A2=E5=85=A8=E9=83=A8=E5=A2=9E=E5=8A=A0typeId?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../cac/controller/NSensorController.java     |  7 ++++---
 .../com/xydl/cac/service/NSensorService.java  |  2 +-
 .../cac/service/impl/NSensorServiceImpl.java  | 19 ++++++++++++-------
 3 files changed, 17 insertions(+), 11 deletions(-)

diff --git a/src/main/java/com/xydl/cac/controller/NSensorController.java b/src/main/java/com/xydl/cac/controller/NSensorController.java
index 65e76ac..2ddb0fb 100644
--- a/src/main/java/com/xydl/cac/controller/NSensorController.java
+++ b/src/main/java/com/xydl/cac/controller/NSensorController.java
@@ -62,8 +62,9 @@ public class NSensorController extends BasicController {
 
     @GetMapping("listAll")
     @ApiOperation("查询全部")
-    public Response<List<NSensor>> listAll(@ApiParam("主设备id") @RequestParam(value = "zsbid", required = false) Integer zsbid) throws Exception {
-        List<NSensor> result = service.listAll(zsbid);
+    public Response<List<NSensor>> listAll(@ApiParam("主设备id") @RequestParam(value = "zsbid", required = false) Integer zsbid,
+                                           @ApiParam("类型") @RequestParam(value = "typeId", required = false) Integer typeId) throws Exception {
+        List<NSensor> result = service.listAll(zsbid , typeId);
         return Response.success(result);
     }
 
@@ -190,7 +191,7 @@ public class NSensorController extends BasicController {
     public void exportLatest(@ApiParam("主设备id") @RequestParam(value = "zsbid", required = false) Integer zsbid, HttpServletResponse response) throws Exception {
         response.setContentType("application/vnd.ms-excel");
         response.setCharacterEncoding("utf-8");
-        List<NSensor> result = service.listAll(zsbid);
+        List<NSensor> result = service.listAll(zsbid , null);
 
         Map<Integer, List<NSensor>> map = new HashMap<>();
         for (NSensor nSensor : result) {
diff --git a/src/main/java/com/xydl/cac/service/NSensorService.java b/src/main/java/com/xydl/cac/service/NSensorService.java
index e34eccf..04c51b2 100644
--- a/src/main/java/com/xydl/cac/service/NSensorService.java
+++ b/src/main/java/com/xydl/cac/service/NSensorService.java
@@ -11,7 +11,7 @@ import java.util.List;
 import java.util.Map;
 
 public interface NSensorService {
-    List<NSensor> listAll(Integer zsbid) throws Exception;
+    List<NSensor> listAll(Integer zsbid, Integer typeId) throws Exception;
 
     Page<NSensor> list(Integer typeId, int pageNum, int pageSize) throws Exception;
 
diff --git a/src/main/java/com/xydl/cac/service/impl/NSensorServiceImpl.java b/src/main/java/com/xydl/cac/service/impl/NSensorServiceImpl.java
index 065b714..bdb38ee 100644
--- a/src/main/java/com/xydl/cac/service/impl/NSensorServiceImpl.java
+++ b/src/main/java/com/xydl/cac/service/impl/NSensorServiceImpl.java
@@ -51,13 +51,18 @@ public class NSensorServiceImpl implements NSensorService {
     I2relationRepository i2relationRepository;
 
     @Override
-    public List<NSensor> listAll(Integer zsbid) throws Exception {
-        List<NSensor> list;
-        if (zsbid == null) {
-            list = repository.findAll();
-        } else {
-            list = repository.findByZsbId(zsbid);
-        }
+    public List<NSensor> listAll(Integer zsbid, Integer typeId) throws Exception {
+        Specification<NSensor> specification = (root, query, builder) -> {
+            Predicate predicate = builder.conjunction();
+            if (zsbid != null) {
+                predicate.getExpressions().add(builder.equal(root.get("zsbId"), zsbid));
+            }
+            if (typeId != null) {
+                predicate.getExpressions().add(builder.equal(root.get("typeId"), typeId));
+            }
+            return predicate;
+        };
+        List<NSensor> list = repository.findAll(specification);
         this.fillOtherNames(list);
         return list;
     }