梁俊星：存放地点增加导入功能

lab-admin/pom.xml

@@ -74,6 +74,11 @@
             <version>5.7.17</version>
         </dependency>
 
+        <dependency>
+            <groupId>com.lab</groupId>
+            <artifactId>lab-common</artifactId>
+        </dependency>
+
     </dependencies>
 
     <build>

lab-admin/src/main/java/com/ruoyi/asset/controller/TbAssetController.java

@@ -20,6 +20,7 @@ import com.ruoyi.asset.domain.TbAsset;
 import com.ruoyi.asset.service.ITbAssetService;
 import com.ruoyi.common.utils.poi.ExcelUtil;
 import com.ruoyi.common.core.page.TableDataInfo;
+import org.springframework.web.multipart.MultipartFile;
 
 /**
  * 资产信息Controller
@@ -60,6 +61,19 @@ public class TbAssetController extends BaseController
     }
 
     /**
+     * 导入
+     * @return
+     */
+    @Log(title = "资产信息", businessType = BusinessType.IMPORT)
+    @PreAuthorize("@ss.hasPermi('asset:asset:add')")
+    @PostMapping("/importData")
+    public void importData(HttpServletResponse response)
+    {
+        ExcelUtil<TbAsset> util = new ExcelUtil<TbAsset>(TbAsset.class);
+        util.importTemplateExcel(response, "资产信息数据");
+    }
+
+    /**
      * 获取资产信息详细信息
      */
     @PreAuthorize("@ss.hasPermi('asset:asset:query')")

lab-admin/src/main/java/com/ruoyi/asset/controller/TbPlaceController.java

@@ -2,6 +2,8 @@ package com.ruoyi.asset.controller;
 
 import java.util.List;
 import javax.servlet.http.HttpServletResponse;
+
+import com.ruoyi.asset.domain.TbAsset;
 import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
@@ -20,6 +22,7 @@ import com.ruoyi.asset.domain.TbPlace;
 import com.ruoyi.asset.service.ITbPlaceService;
 import com.ruoyi.common.utils.poi.ExcelUtil;
 import com.ruoyi.common.core.page.TableDataInfo;
+import org.springframework.web.multipart.MultipartFile;
 
 /**
  * 地点Controller
@@ -60,6 +63,28 @@ public class TbPlaceController extends BaseController
     }
 
     /**
+     * 导入
+     * @return
+     */
+    @Log(title = "地点信息", businessType = BusinessType.IMPORT)
+    @PreAuthorize("@ss.hasPermi('asset:place:add')")
+    @PostMapping("/importData")
+    public AjaxResult importData(MultipartFile file) throws Exception
+    {
+        ExcelUtil<TbPlace> util = new ExcelUtil<TbPlace>(TbPlace.class);
+        List<TbPlace> list = util.importExcel(file.getInputStream());
+        boolean flag = tbPlaceService.saveBatch(list);
+        return AjaxResult.success(flag);
+    }
+
+    @PostMapping("/importTemplate")
+    public void importTemplate(HttpServletResponse response)
+    {
+        ExcelUtil<TbPlace> util = new ExcelUtil<TbPlace>(TbPlace.class);
+        util.importTemplateExcel(response, "地点数据");
+    }
+
+    /**
      * 获取地点详细信息
      */
     @PreAuthorize("@ss.hasPermi('asset:place:query')")

lab-admin/src/main/java/com/ruoyi/asset/mapper/TbPlaceMapper.java

@@ -62,4 +62,6 @@ public interface TbPlaceMapper extends BaseMapper<TbPlace>
      * @return 结果
      */
     public int deleteTbPlaceByIds(Long[] ids);
+
+    int replace(TbPlace place);
 }

lab-admin/src/main/java/com/ruoyi/asset/service/ITbPlaceService.java

@@ -67,4 +67,6 @@ public interface ITbPlaceService extends IService<TbPlace>
      * @return 地点信息集合
      */
     public List<TbPlace> selectTbPlaceAll();
+
+    boolean saveBatch(List<TbPlace> list);
 }

lab-admin/src/main/java/com/ruoyi/asset/service/impl/TbPlaceServiceImpl.java

@@ -102,4 +102,14 @@ public class TbPlaceServiceImpl extends ServiceImpl<TbPlaceMapper, TbPlace> impl
     public List<TbPlace> selectTbPlaceAll() {
         return list();
     }
+
+    @Override
+    public boolean saveBatch(List<TbPlace> list) {
+        for (TbPlace place : list){
+            tbPlaceMapper.replace(place);
+        }
+        return true;
+    }
+
+
 }

lab-admin/src/main/resources/mapper/asset/TbPlaceMapper.xml

@@ -58,4 +58,16 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
             #{id}
         </foreach>
     </delete>
+
+    <insert id="replace" parameterType="TbPlace">
+        replace into tb_place
+        <trim prefix="(" suffix=")" suffixOverrides=",">
+            <if test="name != null">name,</if>
+            <if test="purpose != null">purpose,</if>
+        </trim>
+        <trim prefix="values (" suffix=")" suffixOverrides=",">
+            <if test="name != null">#{name},</if>
+            <if test="purpose != null">#{purpose},</if>
+        </trim>
+    </insert>
 </mapper>

lab-ui/src/views/asset/place/index.vue

@@ -58,6 +58,16 @@
           v-hasPermi="['asset:place:export']"
         >导出</el-button>
       </el-col>
+      <el-col :span="1.5">
+        <el-button
+          type="primary"
+          plain
+          icon="el-icon-plus"
+          size="mini"
+          @click="handleImport"
+          v-hasPermi="['asset:place:add']"
+        >导入</el-button>
+      </el-col>
       <right-toolbar :showSearch.sync="showSearch" @queryTable="getList"></right-toolbar>
     </el-row>
 
@@ -85,7 +95,7 @@
         </template>
       </el-table-column>
     </el-table>
-    
+
     <pagination
       v-show="total>0"
       :total="total"
@@ -109,11 +119,42 @@
         <el-button @click="cancel">取 消</el-button>
       </div>
     </el-dialog>
+
+        <!-- 导入对话框 -->
+        <el-dialog :title="upload.title" :visible.sync="upload.open" width="400px" append-to-body>
+      <el-upload
+        ref="upload"
+        :limit="1"
+        accept=".xlsx, .xls"
+        :headers="upload.headers"
+        :action="upload.url + '?updateSupport=' + upload.updateSupport"
+        :disabled="upload.isUploading"
+        :on-progress="handleFileUploadProgress"
+        :on-success="handleFileSuccess"
+        :auto-upload="false"
+        drag
+      >
+        <i class="el-icon-upload"></i>
+        <div class="el-upload__text">将文件拖到此处，或<em>点击上传</em></div>
+        <div class="el-upload__tip text-center" slot="tip">
+          <div class="el-upload__tip" slot="tip">
+            <el-checkbox v-model="upload.updateSupport" /> 是否更新已经存在的用户数据
+          </div>
+          <span>仅允许导入xls、xlsx格式文件。</span>
+          <el-link type="primary" :underline="false" style="font-size:12px;vertical-align: baseline;" @click="importTemplate">下载模板</el-link>
+        </div>
+      </el-upload>
+      <div slot="footer" class="dialog-footer">
+        <el-button type="primary" @click="submitFileForm">确 定</el-button>
+        <el-button @click="upload.open = false">取 消</el-button>
+      </div>
+    </el-dialog>
   </div>
 </template>
 
 <script>
 import { listPlace, getPlace, delPlace, addPlace, updatePlace } from "@/api/asset/place";
+import {getToken} from "@/utils/auth";
 
 export default {
   name: "Place",
@@ -151,6 +192,21 @@ export default {
         name: [
           { required: true, message: "地点名称不能为空", trigger: "blur" }
         ],
+      },
+      // 用户导入参数
+      upload: {
+        // 是否显示弹出层
+        open: false,
+        // 弹出层标题
+        title: "",
+        // 是否禁用上传
+        isUploading: false,
+        // 是否更新已经存在的用户数据
+        updateSupport: 0,
+        // 设置上传的请求头部
+        headers: { Authorization: "Bearer " + getToken() },
+        // 上传的地址
+        url: process.env.VUE_APP_BASE_API + "/asset/place/importData"
       }
     };
   },
@@ -203,6 +259,32 @@ export default {
       this.open = true;
       this.title = "添加地点";
     },
+    /** 导入按钮操作 */
+    handleImport() {
+      this.upload.title = "地点导入";
+      this.upload.open = true;
+    },
+        /** 下载模板操作 */
+        importTemplate() {
+      this.download('asset/place/importTemplate', {
+      }, `place_template_${new Date().getTime()}.xlsx`)
+    },
+        // 文件上传中处理
+        handleFileUploadProgress(event, file, fileList) {
+      this.upload.isUploading = true;
+    },
+    // 文件上传成功处理
+    handleFileSuccess(response, file, fileList) {
+      this.upload.open = false;
+      this.upload.isUploading = false;
+      this.$refs.upload.clearFiles();
+      this.$alert("<div style='overflow: auto;overflow-x: hidden;max-height: 70vh;padding: 10px 20px 0;'>" + response.msg + "</div>", "导入结果", { dangerouslyUseHTMLString: true });
+      this.getList();
+    },
+    // 提交上传文件
+    submitFileForm() {
+      this.$refs.upload.submit();
+    },
     /** 修改按钮操作 */
     handleUpdate(row) {
       this.reset();