前端
<el-row :gutter="10" class="mb8">
<el-col :span="1.5">
<el-button
type="info"
plain
icon="el-icon-upload2"
size="mini"
@click="handleImport"
v-hasPermi="['ipdzgl:importuser:import']"
>导入</el-button>
</el-col>
</el-row>
<!-- 用户导入对话框 -->
<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" slot="tip">
<el-checkbox v-model="upload.updateSupport" /> 是否更新已经存在的用户数据
</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";
data() {
return {
upload: {
// 是否显示弹出层(用户导入)
open: false,
// 弹出层标题(用户导入)
title: "",
// 是否禁用上传
isUploading: false,
// 是否更新已经存在的用户数据
updateSupport:false,
// 设置上传的请求头部
headers: { Authorization: "Bearer " + getToken() },
// 上传的地址
// 例如 http://localhost:81/params/keyword 则 localurl : /params/keyword
localurl:'/params/keyword',
url: process.env.VUE_APP_BASE_API + '/params/keyword' + "/importData",
},
}
}
method(){
handleImport() {
this.upload.title = "xxxxxxxx名字";
this.upload.open = true;
},
/** 下载模板操作 */
importTemplate() {
this.download(this.upload.localurl + '/importTemplate', {
}, `importuser_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();
}
}
后端
Controller 层
// 上传模板
@Log(title = "用户IP地址注册", businessType = BusinessType.IMPORT)
@RequiresPermissions("[权限名]")
@PostMapping("/importData")
public AjaxResult importData(MultipartFile file, boolean updateSupport) throws Exception
{
ExcelUtil<实体类MOdel> util = new ExcelUtil<实体类MOdel>(实体类MOdel.class);
List<实体类MOdel> dataList = util.importExcel(file.getInputStream());
String message = registCustUserService.importData(dataList, updateSupport);
return AjaxResult.success(message);
}
// 下载模板
@PostMapping("/importTemplate")
public void importTemplate(HttpServletResponse response) throws IOException
{
ExcelUtil<实体类MOdel> util = new ExcelUtil<实体类MOdel>(实体类MOdel.class);
util.importTemplateExcel(response, "模板名");
}
Service层
public String importData(List<实体类Model> dataList, boolean updateSupport);
IMPL
@Override
public String importData(List<实体类Model> dataList, boolean updateSupport) {
int successNum = 0;
int failureNum = 0;
StringBuilder successMsg = new StringBuilder();
StringBuilder failureMsg = new StringBuilder();
if (StringUtils.isNull(dataList) || dataList.size() == 0)
{
throw new ServiceException("导入用户数据不能为空!");
}
for (实体类Model data : dataList)
{
try
{
//TODO 需要改当前方法下 select 语句
List isExist = mdRedlistMapper.selectMdRedlistList(data);
if(isExist.size() >1) {
//TODO 需要改成自己方法的插入语句
mdRedlistMapper.insertMdRedlist(data);
successMsg.append("<br/>" + successNum + data.toString() + " 导入成功");
successNum++;
}else {
if(updateSupport){
successNum++;
mdRedlistMapper.updateMdRedlist(data);
successMsg.append("<br/>" + failureNum + "、号码 " + data.getRednumber() + " 已经存在, 更新成功");
} else {
failureNum++;
failureMsg.append("<br/>" + failureNum + "、号码 " + data.getRednumber() + " 已经存在");
}
}
}
catch (Exception e)
{
failureNum++;
String msg = "<br/>" + data.toString() + " 导入失败:";
failureMsg.append(msg + e.getMessage());
log.error(msg, e);
}
}
if (failureNum > 0)
{
failureMsg.insert(0, "很抱歉,导入失败!共 " + failureNum + " 条数据格式不正确,错误如下:");
throw new ServiceException(failureMsg.toString());
}
else
{
successMsg.insert(0, "恭喜您,数据已全部导入成功!共 " + successNum + " 条,数据如下:");
}
return successMsg.toString();
}
}
下载导入模板文件
上面upload 中添加
localurl 注意
// 用户导入参数
upload: {
// 是否显示弹出层(用户导入)
open: false,
// 弹出层标题(用户导入)
title: '',
// 是否禁用上传
isUploading: false,
// 是否更新已经存在的用户数据
updateSupport: 0,
// 设置上传的请求头部
headers: { Authorization: 'Bearer ' + getToken() },
localurl: '/params/whitelist',
// 上传的地址
url: process.env.VUE_APP_BASE_API + '/params/whitelist/importData'
},
/** 下载模板操作 */
importTemplate() {
this.download(this.upload.localurl + '/importTemplate/download', {
}, `黑名单导入_${new Date().getTime()}.xlsx`)
},
@PostMapping("/template/download")
@ResponseBody
public void getApplicationTemplate(HttpServletRequest req, HttpServletResponse res) {
try {
//获取要下载的模板名称
String fileName = "xxxxx信息_template.xlsx";
res.setHeader("Content-Disposition", "attachment;fileName=" + fileName);
//通知客服文件的MIME类型
res.setContentType("application/vnd.ms-excel;charset=UTF-8");
//获取文件的路径 (如果使用此该行代码来实现的话会出现在本地测试没问题, 打Jar包部署后出现文件损坏的问题,所以不建议使用)
// File file = ResourceUtils.getFile("classpath:static/"+fileName);
InputStream input = Thread.currentThread().getContextClassLoader().
getResourceAsStream(("templates/" + fileName));
OutputStream out = res.getOutputStream();
int len;
byte[] b = new byte[2048];
while ((len = input.read(b)) != -1) {
out.write(b, 0, len);
}
res.setHeader("Content-Length", String.valueOf(input.available()));
input.close();
out.close();
} catch (Exception ex) {
logger.error(ex.getMessage());
}
}