示例:
针对若依的导入,其他方案也可借鉴
/**
* 导入
*
* @param file Excel文件
*/
@Override
@Transactional(rollbackFor = Exception.class)
public void importInfo(MultipartFile file) {
if (file == null || StrUtil.isBlank(file.getOriginalFilename())) {
throw new BizException("请上传xls、xlsx格式文件");
}
// 文件大小不能超过32M
if (file.getSize() > 32 * 1024 * 1024) {
throw new BizException("文件大小不能超过32MB");
}
InputStream inputStream;
byte[] fileBytes;
try {
fileBytes = file.getBytes();
inputStream = file.getInputStream();
} catch (IOException e) {
throw new BizException("读取文件异常");
}
if (fileBytes.length < 1) {
throw new BizException("文件不能为空");
}
// 读取
ExcelUtil<ImportVO> util = new ExcelUtil<>(ImportVO.class);
List<ImportVO> importVos = util.importExcel(inputStream);
if (CollectionUtil.isEmpty(importVos) || importVos.get(0) == null) {
throw new BizException("导入的数据不能为空");
}
StringBuilder stringBuilder = new StringBuilder();
Map<String, Integer> codeMap = new HashMap<>();
List<Bo> bos = new ArrayList<>();
// 校验
for (int i = 0; i < importVos.size(); i++) {
ImportVO importVo = importVos.get(i);
// 示例
if (StringUtils.isEmpty(importVO.getCode())) {
// 判空
stringBuilder.append("第" + (i + 2) + "行编码不能为空;");
} else {
if (codeMap.containsKey(importVO.getCode().trim())) {
// 重复
stringBuilder.append("第" + (codeMap.get(importVO.getCode().trim()) + 2) + "行与第" + (i + 2) + "行编码重复;");
} else {
// 查找
LambdaQueryWrapper<Bo> lambdaQueryWrapper = Wrappers.lambdaQuery();
lambdaQueryWrapper.eq(Bo::getCode, importVO.getCode().trim());
List<Bo> list = iService.list(lambdaQueryWrapper);
if (CollectionUtil.isEmpty(list)) {
// 不存在
stringBuilder.append("第" + (i + 2) + "行编码不存在;");
} else {
importVO.setCode(importVO.getCode().trim());
}
codeMap.put(importVO.getCode().trim(), i);
}
}
// 转为自己的保存类
Bo bo = BeanUtil.toBean(importVO, Bo.class);
bos.add(bo);
}
if (stringBuilder.toString().length() > 0) {
throw new RuntimeException(stringBuilder.toString());
}
stringBuilder = new StringBuilder();
// 保存
for (int i = 0; i < bos.size(); i++) {
Bo bo = bos.get(i);
try {
// 自己内部保存方法,可能有其他校验
insertByBo(bo);
} catch (Exception e) {
stringBuilder.append("第" + (i + 2) + "行" + e.getMessage());
}
}
if (stringBuilder.toString().length() > 0) {
throw new RuntimeException(stringBuilder.toString());
}
}
本次教程到这里就结束了,希望大家多多关注支持(首席摸鱼师 微信同号),持续跟踪最新文章吧~