若依代码生成自带导入功能

785 阅读1分钟

1 修改ruoyi-generator下resources下vm.java.controller

  • 导入
import java.io.InputStream;
import org.springframework.web.multipart.MultipartFile;
  • 新增导出模板接口
/**
     * 导出模板
     */
    @PostMapping("/importTemplate")
    public void importTemplate(HttpServletResponse response)
    {
        ExcelUtil<${ClassName}> util = new ExcelUtil<${ClassName}>(${ClassName}.class);
        util.importTemplateExcel(response, "${functionName}数据");
    }
  • 新增导入数据
    /**
     * 导入数据
     */
    @Log(title = "${functionName}", businessType = BusinessType.IMPORT)
    @PreAuthorize("@ss.hasPermi('${permissionPrefix}:import')")
    @PostMapping("/importData")
    public AjaxResult importData(MultipartFile file) throws Exception
    {
        ExcelUtil<${ClassName}> util = new ExcelUtil<${ClassName}>(${ClassName}.class);
        InputStream inputStream = file.getInputStream();
        List<${ClassName}> list = util.importExcel(inputStream );
        inputStream.close();
        int count = ${className}Service.batchInsert${ClassName}(list);
        return AjaxResult.success("导入成功" + count + "条信息!");
    }

2 修改ruoyi-generator下resources下vm.java.service

  • 新增方法
/**
     * 批量新增${functionName}
     *
     * @param ${className}s ${functionName}List
     * @return 结果
     */
    public int batchInsert${ClassName}(List<${ClassName}> ${className}s);

3 修改ruoyi-generator下resources下vm.java.serviceImpl

  • 导入
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.springframework.util.CollectionUtils;
  • 新增批量新增方法
/**
     * 批量新增${functionName}
     *
     * @param ${className}s ${functionName}List
     * @return 结果
     */
    @Override
    public int batchInsert${ClassName}(List<${ClassName}> ${className}s)
    {
        SqlSession sqlSession = sqlSessionFactory.openSession(ExecutorType.BATCH, false);
        int count = 0;
        if (!CollectionUtils.isEmpty(${className}s)) {
            try {
                for (int i = 0; i < ${className}s.size(); i++) {
                    int row = ${className}Mapper.insert${ClassName}(${className}s.get(i));
                    // 防止内存溢出,没100次提交一次,并清除缓存
                    boolean bool = (i >0 && i%3 == 0) || i == ${className}s.size() - 1;
                    if (bool){
                        sqlSession.commit();
                        sqlSession.clearCache();
                    }
                    count = i + 1;
                }
            }catch (Exception e){
                e.printStackTrace();
                // 没有提交的数据可以回滚
                sqlSession.rollback();
            }finally {
                sqlSession.close();
                return count;
            }
        }
        return count;
    }

4 修改vue下index.vue.vm文件

  • 新增方法
/** 下载模板操作 */
    importTemplate() {
      this.download('${moduleName}/${businessName}/importTemplate', {
      }, `template_#[[${new Date().getTime()}]]#.xlsx`)
    },
    /** 导入按钮操作 */
    handleImport() {
      this.upload.title = "${functionName}导入";
      this.upload.open = true;
    },
    // 文件上传中处理
    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();
    }
  • 新增data
// 导入参数
      upload: {
        // 是否显示弹出层(导入)
        open: false,
        // 弹出层标题(导入)
        title: "",
        // 是否禁用上传
        isUploading: false,
        // 设置上传的请求头部
        headers: { Authorization: "Bearer " + getToken() },
        // 上传的地址
        url: process.env.VUE_APP_BASE_API + "/${moduleName}/${businessName}/importData"
      },
  • 新增对话框
<!-- 导入对话框 -->
    <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">
          <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>
  • 导入token
import {getToken} from "@/utils/auth";