设备批量导入接口

src/main/java/com/sky/ioc/controller/device/DeviceController.java

@@ -9,6 +9,9 @@ import io.swagger.annotations.ApiOperation;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.IOException;
 
 @Api(tags = "设备管理--设备")
 @Slf4j
@@ -81,4 +84,10 @@ public class DeviceController {
     public ReturnMsg getDeviceAlarm(String alarm_type,@RequestParam(defaultValue = "1") Integer page,@RequestParam(defaultValue = "20") Integer page_size) {
         return deviceMessageService.getList(alarm_type,page,page_size);
     }
+
+    @ApiOperation("获取设备告警列表")
+    @PostMapping(value = "/import_device")
+    public ReturnMsg importDevice(MultipartFile file,String group_id) throws IOException {
+        return deviceTypeService.importDeviceType(file,group_id);
+    }
 }

src/main/java/com/sky/ioc/entity/domain/device/SecurityDevice.java

@@ -1,7 +1,11 @@
 package com.sky.ioc.entity.domain.device;
 
+import com.baomidou.mybatisplus.annotation.IdType;
+import com.baomidou.mybatisplus.annotation.TableId;
 import com.baomidou.mybatisplus.annotation.TableName;
+import io.swagger.models.auth.In;
 import lombok.Data;
+import nonapi.io.github.classgraph.json.Id;
 
 @Data
 @TableName("energy_device")
@@ -16,5 +20,5 @@ public class SecurityDevice {
     private String gateway_code;//所属单位ID')
     private String gateway_ip;//网关服务器')
     private String usage;//用途')
-    private String status;//设备状态（1-开启）')
+    private Integer status;//设备状态（1-开启）')
 }

src/main/java/com/sky/ioc/service/device/DeviceTypeService.java

@@ -1,6 +1,9 @@
 package com.sky.ioc.service.device;
 
 import com.sky.ioc.tool.ReturnMsg;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.IOException;
 
 public interface DeviceTypeService {
 
@@ -16,5 +19,5 @@ public interface DeviceTypeService {
     ReturnMsg addDeviceType(String name);
     ReturnMsg getDeviceType(String name);
 
-    ReturnMsg importDeviceType(String group);
+    ReturnMsg importDeviceType(MultipartFile file,String group_id) throws IOException;
 }

src/main/java/com/sky/ioc/service/device/impl/DeviceTypeServiceImpl.java

@@ -6,7 +6,6 @@ import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
 import com.github.pagehelper.PageHelper;
 import com.github.pagehelper.PageInfo;
-import com.sky.ioc.entity.domain.device.DeviceMessage;
 import com.sky.ioc.entity.domain.device.DeviceType;
 import com.sky.ioc.entity.domain.device.SecurityDevice;
 import com.sky.ioc.mapper.device.DeviceTypeMapper;
@@ -14,10 +13,21 @@ import com.sky.ioc.mapper.device.SecurityDeviceMapper;
 import com.sky.ioc.service.device.DeviceTypeService;
 import com.sky.ioc.tool.ReturnMsg;
 import org.apache.commons.lang3.StringUtils;
+import org.apache.poi.hssf.usermodel.HSSFWorkbook;
+import org.apache.poi.ss.usermodel.Cell;
+import org.apache.poi.ss.usermodel.Row;
+import org.apache.poi.ss.usermodel.Sheet;
+import org.apache.poi.ss.usermodel.Workbook;
+import org.apache.poi.xssf.usermodel.XSSFWorkbook;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.web.multipart.MultipartFile;
 
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.ArrayList;
 import java.util.List;
+import java.util.Map;
 import java.util.stream.Collectors;
 
 @Service
@@ -110,8 +120,65 @@ public class DeviceTypeServiceImpl implements DeviceTypeService {
     }
 
     @Override
-    public ReturnMsg importDeviceType(String group) {
-        return null;
+    public ReturnMsg importDeviceType(MultipartFile file,String group_id) throws IOException {
+        String fileName = file.getOriginalFilename();//文件名称
+        String suffixName = fileName.substring(fileName.lastIndexOf("."));//文件后缀名
+        //创建上传的工作表
+        InputStream inputStream = file.getInputStream();
+        Workbook workbook = null;
+        if(suffixName.equals(".xls")){
+            workbook = new HSSFWorkbook(inputStream);
+        }else if(suffixName.equals(".xlsx")){
+            workbook = new XSSFWorkbook(inputStream);
+        }else{
+            return ReturnMsg.fail("文件格式不正确");
+        }
+        if(workbook!=null){
+            //读取数据
+            Sheet sheet = null; //表
+            Row row = null;//行
+            Cell cell = null;//单元格
+            List<Map<String,Object>> data = new ArrayList<>();//存放读取的内容
+            int number = workbook.getNumberOfSheets();
+            for (int i = 0; i < number ; i++) { //遍历sheets
+                sheet = workbook.getSheetAt(i);
+                int rownum = sheet.getLastRowNum()+1;
+                for (int j = 1; j < rownum; j++) {//遍历sheet的row 忽略第一行表头
+                    //表格内容
+                    row = sheet.getRow(j);
+                    if (row==null){
+                        continue;
+                    }
+                    String id = row.getCell(0).toString();
+                    String device_name = row.getCell(1).toString();
+                    String device_id = row.getCell(2).toString();
+                    double device_type = row.getCell(3).getNumericCellValue();
+                    String device_location = row.getCell(4).toString();
+                    String loop_detail = row.getCell(5).toString();
+                    String gateway_code = row.getCell(6).toString();
+                    String gateway_ip = row.getCell(7).toString();
+                    String usage = row.getCell(8).toString();
+                    double status = row.getCell(9).getNumericCellValue();
+                    SecurityDevice securityDevice = new SecurityDevice();
+                    securityDevice.setDevice_id(device_id);
+                    securityDevice.setDevice_name(device_name);
+                    securityDevice.setDevice_type((int) device_type);
+                    securityDevice.setDevice_location(device_location);
+                    securityDevice.setLoop_detail(loop_detail);
+                    securityDevice.setGateway_code(gateway_code);
+                    securityDevice.setGateway_ip(gateway_ip);
+                    securityDevice.setUsage(usage);
+                    securityDevice.setStatus((int) status);
+                    securityDeviceMapper.insert(securityDevice);
+                }
+
+            }
+
+            return  ReturnMsg.ok(data);
+        }else{
+            return ReturnMsg.fail("Excel工作簿为空");
+        }
+
     }
 
 

src/main/java/com/sky/ioc/tool/ExcelUtils.java

@@ -23,7 +23,6 @@ public class ExcelUtils {
     /**
      * 导出excel文件数据
      * @param sheetName
-     * @param column
      * @param data
      * @param request
      * @param response