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.ExecutorType;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.springframework.util.CollectionUtils;
@Autowired
private SqlSessionFactory sqlSessionFactory;
/**
* 批量新增${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();
}
// 导入参数
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>
import {getToken} from "@/utils/auth";
<el-col :span="1.5">
<el-button
type="warning"
plain
icon="el-icon-upload2"
size="mini"
@click="handleImport"
v-hasPermi="['${moduleName}:${businessName}:import']"
>导入</el-button>
</el-col>