业务:批量导入不同用户的档案信息,excel格式为:信息用户(独一无二)- 姓名 - 性别 - …
数据库档案字段值表为:信息用户id-档案字段id-档案字段值
已有插入方法:add(String username, Integer[] fieldIds, String[] fieldValues),传入信息用户、档案字段id、档案字段值,其中信息用户username将用来查询得到数据库中对应的信息用户id进行操作;fieldIds与fieldValues按数组中的下标一一对应。
效果图:
思路:
我是这样想的,先把excel中的一行行数据返回成可以操作的形式。
首先声明一个Map<String, Object> resultMap = new HashMap<>(),用于放入执行结果code以及数据rowList。
其中
code是代表执行结果,如文件值为空/格式有误情况;
rowList是excel返回成的一行一行的数据,即一个个字符串数组List<String[]> rowList = new ArrayList<>();
/**
* 将excel文件返回成Map
*
* @param file
* @return 返回Map<String, Object> key有两种, ①code为状态码;②rowList为数据集合
*/
public static Map<String, Object> excelToMap(MultipartFile file) {
Map<String, Object> resultMap = new HashMap<>();
List<String[]> rowList = new ArrayList<>();
try {
String suffer = file.getOriginalFilename().substring(file.getOriginalFilename().indexOf(".") + 1);
if ("xls".equals(suffer) || "xlsx".equals(suffer)) {
InputStream in = file.getInputStream();
HSSFWorkbook workbook = new HSSFWorkbook(in);
// 得到sheet
HSSFSheet sheet = workbook.getSheetAt(0);
if (sheet != null) {
// 得到每行的值
// getLastCellNum比列标大1,行标&列标都以0开始。
int size = sheet.getRow(0).getLastCellNum();
for (int i = 0; i <= sheet.getLastRowNum(); i++) {
HSSFRow row = sheet.getRow(i);
String[] cellArr = new String[size];
// 得到一行每列的值
for (int j = 0; j < size; j++) {
HSSFCell cell = row.getCell(j);
if (cell == null || cell.getCellTypeEnum() == CellType.BLANK) {
cellArr[j] = new String(" ".getBytes("UTF-8"));
continue;
}
cell.setCellType(CellType.STRING);//单元格类型统一设置为String,便于取出
String cellValueStr = new String(cell.getStringCellValue().getBytes("UTF-8"));
cellArr[j] = cellValueStr;
}
rowList.add(cellArr);
resultMap.put("rowList", rowList);
}
resultMap.put("code", 200);
return resultMap;
}
//文件值为空
resultMap.put("code", 0);
return resultMap;
}
//文件格式有误
resultMap.put("code", -1);
return resultMap;
} catch (Exception e) {
logger.info(e.toString());
//文件格式有误
resultMap.put("code", -1);
return resultMap;
}
}
然后判断执行结果code,如果成功则接着获取rowList。
将rowList中的第一行即标题进行遍历并在数据库中查询,得到其对应的档案字段id,返回成Integer[] fieldIds;再将rowList中的其他行进行遍历,得到 要插入的值 ,返回成String[] cellArr,其中cellArr[0]为信息用户username。并将cellArr数组里其余的值的编码格式转换成utf-8,返回成String[] fieldValues。
最后一行一行用前面说的add(String username, Integer[] fieldIds, String[] fieldValues)方法进行添加操作。
/**
* 批量导入信息数据
*/
@Transactional(propagation = Propagation.REQUIRED)
@PostMapping("/batchAdd")
@ResponseBody
public AjaxResult batchAdd(@RequestParam(value = "file", required = false) MultipartFile file) {
try {
Map<String, Object> excelToMap = ExcelUtils.excelToMap(file);//将excel转成map
Integer code = (Integer) excelToMap.get("code");//自动拆箱与int比大小
if (code == 200) {
List<String[]> rowList = (List<String[]>) excelToMap.get("rowList");
//判断第一个单元格值是否为信息用户
String firstCell = rowList.get(0)[0];
if (firstCell.equals("信息用户")) {
List<String[]> rowList = (List<String[]>) excelToMap.get("rowList");
String[] firstRow = rowList.get(0);//获取到第一行,即excel标题行
Integer[] fieldIds = new Integer[(firstRow.length - 1)];//将excel标题行的value遍历,获取数据库中对应字段id
for (int i = 1; i < firstRow.length; i++) {
String cellValue = firstRow[i];
QueryWrapper<Field> queryWrapper = new QueryWrapper<>();
queryWrapper.eq("field_name", cellValue);
Integer id = fieldService.list(queryWrapper).get(0).getId();
fieldIds[(i - 1)] = id;
}
//获得username以及fieldValues
Integer resultCode;//导入后返回code
int sum = 0;//总需导入数
int successNum = 0;//成功导入数
//int quitNum = 0;//跳过导入数
int failNum = 0;//失败导入数
for (int i = 1; i < rowList.size(); i++) {
String[] cellArr = rowList.get(i);
//判断username是否为空
String username = cellArr[0];
if (username.trim().length() != 0) {
sum = sum + 1;
String[] fieldValues = new String[(firstRow.length - 1)];
for (int j = 1; j < cellArr.length; j++) {
if (cellArr[j] == null) {
continue;
}
fieldValues[(j - 1)] = new String(encoder.encode(cellArr[j].getBytes("UTF-8")), StandardCharsets.UTF_8);//编码转换成utf-8
}
//获得成功导入数以及跳过数
resultCode = add(username, fieldIds, fieldValues).getCode();
switch (resultCode) {
case 200:
successNum = successNum + 1;
break;
default:
failNum = failNum + 1;
}
} else continue;
}
String message = "总需导入用户数:" + sum + "</br>导入成功用户数:" + successNum + " </br>导入失败用户数:" + failNum + "。";
return success(message);
}
}
return error("导入文件有误,请检查文件格式!");
} catch (Exception e) {
logger.info(e.toString());
return error();
}
}
然后前端
<form class="form-horizontal m">
<div class="form-group">
<label class="col-sm-3 control-label ">Excel文件:</label>
<div class="col-sm-8">
<input class="" type="file" id="file" name="file"
accept="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet, application/vnd.ms-excel"
/>
</div>
</div>
<button type="button" class="btn btn-primary" onclick="uploadFile()">提交</button>
</form>
function uploadFile() {
let formData = new FormData();
let file = $('#file')[0].files[0];
if (typeof (file) != "undefined") {
$.modal.msgLoading("处理中,请稍后...");
formData.append("file", file);
$.ajax({
cache: true,
type: "POST",
url: rootPath + "/xxxController/batchAdd",
data: formData,
processData: false,
contentType: false,
dataType: "json",
async: false,
error: function (request) {
$.modal.alertError("上传文件有误");
},
success: function (data) {
let content = data.message;
if (data.code == 200) {
layer.alert(content, {
icon: 1,
title: "系统提示",
btn: ["确认"],
btnclass: ["btn btn-primary"],
}, function () {
$.modal.close();
$.table.refresh(window.rootPath + "/UserController/list");
});
} else {
$.modal.alertError(data.message);
}
}
});
}
}
else {
$.modal.alertError("未上传Excel文件");
}
}
消息弹出框用的是若依,
附上若依操作手册:https://www.w3cschool.cn/ruoyi/ruoyi-8cnv311f.html