From 4696fd707f1c723970eac62d333588b51dffbd2b Mon Sep 17 00:00:00 2001
From: Crunchy <3114200645@qq.com>
Date: 星期四, 04 七月 2024 12:37:16 +0800
Subject: [PATCH] 检验对象导入完成

---
 cnas-server/src/main/java/com/yuanchu/mom/service/impl/DeviceServiceImpl.java |   44 ++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 42 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 9b7ce3c..1a4aa31 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,19 +1,24 @@
 package com.yuanchu.mom.service.impl;
 
 import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
 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.dto.DeviceDto;
+import com.yuanchu.mom.dto.DeviceDto1;
 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;
@@ -28,6 +33,8 @@
     private GetLook getLook;
 
     private DeviceMapper deviceMapper;
+
+    private StructureItemParameterMapper structureItemParameterMapper;
 
     @Override
     public Map<String, Object> selectDeviceParameter(Page page, DeviceDto itemParameter) {
@@ -76,8 +83,41 @@
     }
 
     @Override
-    public List<Device> selectDeviceByCategory(String category) {
-        return deviceMapper.selectList(Wrappers.<Device>lambdaQuery().eq(Device::getLargeCategory, category).eq(Device::getDeviceStatus, 0));
+    public List<Device> selectDeviceByCategory(String inspectionItem, String inspectionItemSubclass) {
+        Integer id;
+        try {
+            if (ObjectUtils.isEmpty(inspectionItemSubclass)) {
+                id = structureItemParameterMapper.selectOne(Wrappers.<StructureItemParameter>lambdaQuery()
+                        .eq(StructureItemParameter::getInspectionItem, inspectionItem)
+                        .last("limit 1").select(StructureItemParameter::getId)).getId();
+            }else {
+                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) {
+            String[] ids = device.getInsProductIds().split(",");
+            for (String i : ids) {
+                if (i.equals(id + "")) {
+                    devices2.add(device);
+                    break;
+                }
+            }
+        }
+        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