From 112a87d32e19b47b9c92534900d9cdd02846bdb3 Mon Sep 17 00:00:00 2001
From: value <z1292839451@163.com>
Date: 星期三, 24 四月 2024 18:17:25 +0800
Subject: [PATCH] 完善检验流程

---
 cnas-server/src/main/java/com/yuanchu/mom/service/impl/DeviceServiceImpl.java |   41 +++++++++++++++++++++++++++++++++++++++--
 1 files changed, 39 insertions(+), 2 deletions(-)

diff --git a/cnas-server/src/main/java/com/yuanchu/mom/service/impl/DeviceServiceImpl.java b/cnas-server/src/main/java/com/yuanchu/mom/service/impl/DeviceServiceImpl.java
index 0b055d1..3286e6b 100644
--- a/cnas-server/src/main/java/com/yuanchu/mom/service/impl/DeviceServiceImpl.java
+++ b/cnas-server/src/main/java/com/yuanchu/mom/service/impl/DeviceServiceImpl.java
@@ -1,16 +1,21 @@
 package com.yuanchu.mom.service.impl;
 
+import com.alibaba.fastjson.JSON;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.yuanchu.mom.common.GetLook;
 import com.yuanchu.mom.common.PrintChina;
 import com.yuanchu.mom.mapper.DeviceMapper;
+import com.yuanchu.mom.mapper.StructureItemParameterMapper;
 import com.yuanchu.mom.pojo.Device;
+import com.yuanchu.mom.pojo.StructureItemParameter;
 import com.yuanchu.mom.service.DeviceService;
 import com.yuanchu.mom.utils.QueryWrappers;
 import lombok.AllArgsConstructor;
 import org.springframework.stereotype.Service;
 
+import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -26,13 +31,15 @@
 
     private DeviceMapper deviceMapper;
 
+    private StructureItemParameterMapper structureItemParameterMapper;
+
     @Override
     public Map<String, Object> selectDeviceParameter(Page page, Device itemParameter) {
         Map<String, Object> map = new HashMap<>();
         map.put("head", PrintChina.printChina(Device.class));
         Map<String, Integer> map1 = getLook.selectPowerByMethodAndUserId("selectDeviceParameter");
         if (map1.get("look") == 1) itemParameter.setCreateUser(map1.get("userId"));
-        map.put("body", deviceMapper.selectDeviceParameter(page, QueryWrappers.queryWrappers(itemParameter)));
+        map.put("body", deviceMapper.selectPage(page, QueryWrappers.queryWrappers(itemParameter)));
         return map;
     }
 
@@ -53,7 +60,7 @@
 
     @Override
     public List<Device> selectEquipmentOverview() {
-        return deviceMapper.selectEquipmentOverview();
+        return deviceMapper.selectEquipmentOverview(new Page(1, 10), QueryWrappers.queryWrappers(new Device()));
     }
 
     @Override
@@ -66,5 +73,35 @@
         return deviceMapper.search(status, deviceName, specificationModel, largeCategory);
     }
 
+    @Override
+    public List<Device> selectDevicePrincipal() {
+        return deviceMapper.selectDevicePrincipal();
+    }
+
+    @Override
+    public List<Device> selectDeviceByCategory(String inspectionItem, String inspectionItemSubclass) {
+        Integer id;
+        try {
+            id = structureItemParameterMapper.selectOne(Wrappers.<StructureItemParameter>lambdaQuery().eq(StructureItemParameter::getInspectionItem, inspectionItem).eq(StructureItemParameter::getInspectionItemSubclass, inspectionItemSubclass).last("limit 1").select(StructureItemParameter::getId)).getId();
+        } catch (Exception e) {
+            return null;
+        }
+        List<Device> devices = deviceMapper.selectList(Wrappers.<Device>lambdaQuery().eq(Device::getDeviceStatus, 0).isNotNull(Device::getInsProductIds));
+        List<Device> devices2 = new ArrayList<>();
+        for (Device device : devices) {
+            List<Integer> ids = JSON.parseArray(device.getInsProductIds(), Integer.class);
+            for (Integer i : ids) {
+                if (i == id) {
+                    devices2.add(device);
+                }
+            }
+        }
+        return devices2;
+    }
+
+    @Override
+    public Device selectDeviceByCode(String code) {
+        return deviceMapper.selectOne(Wrappers.<Device>lambdaQuery().eq(Device::getFactoryNo, code).last("limit 1"));
+    }
 }
 

--
Gitblit v1.9.3