增加导入功能

kgraph-graph/src/main/java/com/kgraph/graph/suport/controller/TBuildFlowController.java

@@ -19,7 +19,8 @@ import com.kgraph.common.enums.BusinessType;;
 import com.kgraph.graph.suport.domain.TBuildFlow;
 import com.kgraph.graph.suport.service.ITBuildFlowService;
 import com.kgraph.common.utils.poi.ExcelUtil;;
-import com.kgraph.common.core.page.TableDataInfo;;
+import com.kgraph.common.core.page.TableDataInfo;
+import org.springframework.web.multipart.MultipartFile;;
 
 /**
  * 构建流程Controller
@@ -110,4 +111,22 @@ public class TBuildFlowController extends BaseController
     {
         return success(tBuildFlowService.getFlow(subtaskId));
     }
+
+    @Log(title = "排故流程导入", businessType = BusinessType.IMPORT)
+    @PreAuthorize("@ss.hasPermi('showInfo:breakdown:add')")
+    @PostMapping("/importData")
+    public AjaxResult importData(MultipartFile file, boolean updateSupport) throws Exception
+    {
+        ExcelUtil<TBuildFlow> util = new ExcelUtil<>(TBuildFlow.class);
+        List<TBuildFlow> TBuildFlowList = util.importExcel(file.getInputStream());
+        String message = tBuildFlowService.excelImport(TBuildFlowList);
+        return AjaxResult.success(message);
+    }
+
+    @PostMapping("/importTemplate")
+    public void importTemplate(HttpServletResponse response)
+    {
+        ExcelUtil<TBuildFlow> util = new ExcelUtil<>(TBuildFlow.class);
+        util.importTemplateExcel(response, "排故流程");
+    }
 }

kgraph-graph/src/main/java/com/kgraph/graph/suport/controller/TCheckPiontController.java

@@ -2,6 +2,7 @@ package com.kgraph.graph.suport.controller;
 
 import java.util.List;
 import javax.servlet.http.HttpServletResponse;
+
 import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
@@ -19,7 +20,8 @@ import com.kgraph.common.enums.BusinessType;;
 import com.kgraph.graph.suport.domain.TCheckPiont;
 import com.kgraph.graph.suport.service.ITCheckPiontService;
 import com.kgraph.common.utils.poi.ExcelUtil;;
-import com.kgraph.common.core.page.TableDataInfo;;
+import com.kgraph.common.core.page.TableDataInfo;
+import org.springframework.web.multipart.MultipartFile;;
 
 /**
  * 检查点管理Controller
@@ -101,4 +103,22 @@ public class TCheckPiontController extends BaseController
     {
         return toAjax(tCheckPiontService.deleteTCheckPiontByIds(ids));
     }
+
+    @Log(title = "排故流程导入", businessType = BusinessType.IMPORT)
+    @PreAuthorize("@ss.hasPermi('showInfo:breakdown:add')")
+    @PostMapping("/importData")
+    public AjaxResult importData(MultipartFile file, boolean updateSupport) throws Exception
+    {
+        ExcelUtil<TCheckPiont> util = new ExcelUtil<>(TCheckPiont.class);
+        List<TCheckPiont> TCheckPiontList = util.importExcel(file.getInputStream());
+        String message = tCheckPiontService.excelImport(TCheckPiontList);
+        return AjaxResult.success(message);
+    }
+
+    @PostMapping("/importTemplate")
+    public void importTemplate(HttpServletResponse response)
+    {
+        ExcelUtil<TCheckPiont> util = new ExcelUtil<>(TCheckPiont.class);
+        util.importTemplateExcel(response, "检查点管理");
+    }
 }

kgraph-graph/src/main/java/com/kgraph/graph/suport/domain/TBuildFlow.java

@@ -29,7 +29,7 @@ public class TBuildFlow extends BaseEntity
     private String errorAppearance;
 
     /** 故障编码 */
-    @Excel(name = "故障编码")
+    /*@Excel(name = "故障编码")*/
     private String errorCode;
 
     /** 流程数据 */
@@ -37,6 +37,6 @@ public class TBuildFlow extends BaseEntity
     private String flowEncode;
 
     /** 流程树json */
-    @Excel(name = "流程树json")
+    /*@Excel(name = "流程树json")*/
     private String flowTreeJson;
 }

kgraph-graph/src/main/java/com/kgraph/graph/suport/service/ITBuildFlowService.java

@@ -62,4 +62,6 @@ public interface ITBuildFlowService
     public int deleteTBuildFlowById(Long id);
 
     Map getFlow(Long subtaskId);
+
+    String excelImport(List<TBuildFlow> tBuildFlowList);
 }

kgraph-graph/src/main/java/com/kgraph/graph/suport/service/ITCheckPiontService.java

@@ -58,4 +58,6 @@ public interface ITCheckPiontService
      * @return 结果
      */
     public int deleteTCheckPiontById(Long id);
+
+    String excelImport(List<TCheckPiont> tCheckPiontList);
 }

kgraph-graph/src/main/java/com/kgraph/graph/suport/service/impl/TBuildFlowServiceImpl.java

@@ -118,4 +118,16 @@ public class TBuildFlowServiceImpl implements ITBuildFlowService
     public Map getFlow(Long subtaskId) {
         return tBuildFlowMapper.getFlow(subtaskId);
     }
+
+    @Override
+    public String excelImport(List<TBuildFlow> TBuildFlowList) {
+        for (TBuildFlow tbuildflow : TBuildFlowList) {
+            if (tbuildflow.getId() == null) {
+                tBuildFlowMapper.insertTBuildFlow(tbuildflow);
+            } else {
+                tBuildFlowMapper.updateTBuildFlow(tbuildflow);
+            }
+        }
+        return "上传成功";
+    }
 }

kgraph-graph/src/main/java/com/kgraph/graph/suport/service/impl/TCheckPiontServiceImpl.java

@@ -2,6 +2,7 @@ package com.kgraph.graph.suport.service.impl;
 
 import java.util.List;
 import com.kgraph.common.utils.DateUtils;;
+import com.kgraph.graph.suport.domain.TBuildFlow;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import com.kgraph.graph.suport.mapper.TCheckPiontMapper;
@@ -93,4 +94,16 @@ public class TCheckPiontServiceImpl implements ITCheckPiontService
     {
         return tCheckPiontMapper.deleteTCheckPiontById(id);
     }
+
+    @Override
+    public String excelImport(List<TCheckPiont> TCheckPiontList) {
+        for (TCheckPiont tcheckpiont : TCheckPiontList) {
+            if (tcheckpiont.getId() == null) {
+                tCheckPiontMapper.insertTCheckPiont(tcheckpiont);
+            } else {
+                tCheckPiontMapper.updateTCheckPiont(tcheckpiont);
+            }
+        }
+        return "上传成功";
+    }
 }