当前位置：首页 > article >正文

若依框架修改模板，添加通过excel导入数据功能

article 2025/8/26 19:45:08

版本：我后端使用的是RuoYi-Vue-fast版本，前端是RuoYi-Vue3

需求:

我需要每个侧边栏功能都需要具有导入excel功能，但是若依只有用户才具备，我需要代码生成的每个功能都拥有导入功能。

每次生成一个一个改实在是太麻烦了。索性直接修改生成模板

1.后端修改

首先找到自动生成的内容，目录是src/main/resources/vm/java，我们需要改的是下面框起来的三个类。

在controller.java.vm中添加：

  /*** 导入${functionName}数据*/@PreAuthorize("@ss.hasPermi('${permissionPrefix}:import')")@Log(title = "${functionName}", businessType = BusinessType.IMPORT)@PostMapping("/importData")public AjaxResult importData(MultipartFile file, boolean updateSupport) throws Exception{ExcelUtil<${ClassName}> util = new ExcelUtil<${ClassName}>(${ClassName}.class);List<${ClassName}> ${className}List = util.importExcel(file.getInputStream());String operName = getUsername();String message = ${className}Service.import${ClassName}(${className}List, updateSupport, operName);return success(message);}/*** 下载${functionName}导入模板*/@PostMapping("/importTemplate")public void importTemplate(HttpServletResponse response){ExcelUtil<${ClassName}> util = new ExcelUtil<${ClassName}>(${ClassName}.class);util.importTemplateExcel(response, "${functionName}");}

在service.java.vm中添加：

/*** 导入${functionName}数据** @param ${className}List ${functionName}数据列表* @param isUpdateSupport 是否更新支持，如果已存在，则进行更新数据* @param operName 操作用户* @return 结果*/public String import${ClassName}(List<${ClassName}> ${className}List, Boolean isUpdateSupport, String operName);

在serviceImpl.java.vm中添加：

在顶部添加

@Autowiredprotected Validator validator;

 /*** 导入${functionName}数据** @param ${className}List ${functionName}数据列表* @param isUpdateSupport 是否更新支持，如果已存在，则进行更新数据* @param operName 操作用户* @return 结果*/@Overridepublic String import${ClassName}(List<${ClassName}> ${className}List, Boolean isUpdateSupport, String operName){if (StringUtils.isNull(${className}List) || ${className}List.size() == 0){throw new ServiceException("导入数据不能为空！");}int successNum = 0;int failureNum = 0;StringBuilder successMsg = new StringBuilder();StringBuilder failureMsg = new StringBuilder();for (${ClassName} ${className} : ${className}List){try{// 验证是否存在这个数据${ClassName} exist${ClassName} = ${className}Mapper.select${ClassName}ById(${className}.get${pkColumn.capJavaField}());if (StringUtils.isNull(exist${ClassName})){BeanValidators.validateWithException(validator, ${className});${className}.setCreateBy(operName);this.insert${ClassName}(${className});successNum++;successMsg.append("<br/>" + successNum + "、数据 " + ${className}.get${pkColumn.capJavaField}() + " 导入成功");}else if (isUpdateSupport){BeanValidators.validateWithException(validator, ${className});${className}.setUpdateBy(operName);this.update${ClassName}(${className});successNum++;successMsg.append("<br/>" + successNum + "、数据 " + ${className}.get${pkColumn.capJavaField}() + " 更新成功");}else{failureNum++;failureMsg.append("<br/>" + failureNum + "、数据 " + ${className}.get${pkColumn.capJavaField}() + " 已存在");}}catch (Exception e){failureNum++;String msg = "<br/>" + failureNum + "、数据 " + ${className}.get${pkColumn.capJavaField}() + " 导入失败：";failureMsg.append(msg + e.getMessage());}}if (failureNum > 0){failureMsg.insert(0, "很抱歉，导入失败！共 " + failureNum + " 条数据格式不正确，错误如下：");throw new ServiceException(failureMsg.toString());}else{successMsg.insert(0, "恭喜您，数据已全部导入成功！共 " + successNum + " 条，数据如下：");}return successMsg.toString();}