From bc672038e88a24d9f0f7127cd45c656a15c231a2 Mon Sep 17 00:00:00 2001
From: XiaoRuby <3114200645@qq.com>
Date: 星期三, 09 八月 2023 09:10:49 +0800
Subject: [PATCH] MOM系统-8-9

---
 standard-server/src/main/java/com/yuanchu/mom/service/impl/DeviceServiceImpl.java |   24 ++++++++++++++++++++++++
 1 files changed, 24 insertions(+), 0 deletions(-)

diff --git a/standard-server/src/main/java/com/yuanchu/mom/service/impl/DeviceServiceImpl.java b/standard-server/src/main/java/com/yuanchu/mom/service/impl/DeviceServiceImpl.java
index 629fb2c..e04e90f 100644
--- a/standard-server/src/main/java/com/yuanchu/mom/service/impl/DeviceServiceImpl.java
+++ b/standard-server/src/main/java/com/yuanchu/mom/service/impl/DeviceServiceImpl.java
@@ -2,10 +2,13 @@
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
 import com.yuanchu.mom.pojo.Device;
 import com.yuanchu.mom.mapper.DeviceMapper;
+import com.yuanchu.mom.pojo.RawInsProduct;
 import com.yuanchu.mom.service.DeviceService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.yuanchu.mom.service.RawInsProductService;
 import com.yuanchu.mom.utils.MyUtil;
 import org.springframework.stereotype.Service;
 
@@ -28,6 +31,9 @@
 
     @Resource
     private DeviceMapper deviceMapper;
+
+    @Resource
+    private RawInsProductService rawInsProductService;
 
     @Override
     public Map<String, Object> selectTechnology(String deviceGroup) {
@@ -56,4 +62,22 @@
         Device device = deviceMapper.selectById(id);
        return device.getName();
     }
+
+    @Override
+    public Boolean addDevice(Device device) {
+        int insert = deviceMapper.insert(device);
+        if (insert == 1){
+            LambdaUpdateWrapper<RawInsProduct> updateWrapper = new LambdaUpdateWrapper<>();
+            updateWrapper.eq(RawInsProduct::getId, device.getRawInsProductId());
+            updateWrapper.set(RawInsProduct::getDeviceId, device.getId());
+            return rawInsProductService.update(updateWrapper);
+        }
+
+        return false;
+    }
+
+    @Override
+    public List<Map<String, Object>> deviceTwoTree(Integer type) {
+        return deviceMapper.deviceTwoTree(type);
+    }
 }

--
Gitblit v1.9.3