# update 更新导入 设备属性的功能

maibu-service/maibu-iot-service/src/main/java/com/maibu/controller/DeviceRunParamController.java

@@ -4,6 +4,11 @@ import java.util.List;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.TimeoutException;
 
+import com.maibu.annotation.Log;
+import com.maibu.core.domain.entity.SysUser;
+import com.maibu.enums.BusinessType;
+import com.maibu.utils.poi.ExcelUtil;
+import io.swagger.annotations.ApiOperation;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
@@ -18,6 +23,7 @@ import com.maibu.core.domain.AjaxResult;
 import com.maibu.core.page.TableDataInfo;
 import com.maibu.dto.BaseQueryDTO;
 import com.maibu.service.impl.DeviceRunParamService;
+import org.springframework.web.multipart.MultipartFile;
 
 @RestController
 @RequestMapping("/iot/deviceRunParam")
@@ -47,5 +53,12 @@ public class DeviceRunParamController extends BaseController {
     public AjaxResult delete(@RequestBody List<Long> ids) {
         return AjaxResult.success(service.delete(ids));
     }
-
+    
+    @PostMapping("/importData")
+    public AjaxResult importData(MultipartFile file) throws Exception {
+        ExcelUtil<DeviceRunParam> util = new ExcelUtil<DeviceRunParam>(DeviceRunParam.class);
+        List<DeviceRunParam> paramList = util.importExcel(file.getInputStream());
+        boolean message = service.importParam(paramList);
+        return success(message);
+    }
 }

maibu-service/maibu-iot-service/src/main/java/com/maibu/service/impl/DeviceRunParamService.java

@@ -1,6 +1,7 @@
 package com.maibu.service.impl;
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.maibu.annotation.Excel;
 import com.maibu.constant.Constant;
 import com.maibu.core.business.DeviceRunParam;
 import com.maibu.core.business.device.NettyDevice;
@@ -16,11 +17,16 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.util.CollectionUtils;
 
+import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.List;
+import java.util.Map;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
+import java.util.function.Function;
+import java.util.stream.Collectors;
 
 @Service
 @Slf4j
@@ -130,5 +136,56 @@ public class DeviceRunParamService {
         return i;
     }
 
+    public boolean importParam(List<DeviceRunParam> params) {
+        try {
+            if (!CollectionUtils.isEmpty(params)) {
+
+                List<String> snList = params.stream().map(DeviceRunParam::getDeviceId).collect(Collectors.toList());
+
+                if (!CollectionUtils.isEmpty(snList)) {
+                    LambdaQueryWrapper<DeviceRunParam> queryWrapper = new LambdaQueryWrapper<>();
+                    queryWrapper.in(DeviceRunParam::getDeviceId, snList);
+                    List<DeviceRunParam> oldList = deviceRunParamMapper.selectList(queryWrapper);
+                    Map<String, DeviceRunParam> map = oldList.stream()
+                            .collect(Collectors.toMap(
+                                    DeviceRunParam::getDeviceId,
+                                    Function.identity()
+                            ));
+                    List<DeviceRunParam> saveList = new ArrayList<>();
+                    params.forEach(x -> {
+                        DeviceRunParam deviceRunParam = null;
+                        if (!CollectionUtils.isEmpty(map)) {
+                            deviceRunParam = map.get(x.getDeviceId());
+                        }
+                        if (!StringUtils.isEmpty(x.getCapabilities())) {
+                            List<String> list = Arrays.stream(x.getCapabilities().split(","))
+                                    .map(String::trim)
+                                    .collect(Collectors.toList());
+                            x.setCapabilityList(list);
+                        }
+                        if (deviceRunParam != null) {
+                            deviceRunParam.setCapabilityList(x.getCapabilityList());
+                            deviceRunParam.setProductModel(x.getProductModel());
+                            deviceRunParam.setRtkMode(x.getRtkMode());
+                            deviceRunParam.setDeviceType(x.getDeviceType());
+                            deviceRunParam.setPowerType(x.getPowerType());
+                            deviceRunParam.setCameraNum(x.getCameraNum());
+                            deviceRunParam.setWheelType(x.getWheelType());
+                            deviceRunParam.setEngineType(x.getEngineType());
+                            deviceRunParam.setHasHost(x.getHasHost());
+                            saveList.add(deviceRunParam);
+                        } else {
+                            saveList.add(x);
+                        }
+                    });
+                    return deviceRunParamMapper.saveOrUpdateBatch(saveList);
+                }
+            }
+            return false;
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+    }
+
 }