EasyPoi導入驗證功能

1准備好要導入的Excel，注意Excel的標題要和domain中的 @Excel(name = "標題")一樣

 

 

 

 

 

1 導入驗證包支持

<!-- JSR 303 規范驗證包 -->
<dependency>
  <groupId>org.hibernate</groupId>
  <artifactId>hibernate-validator</artifactId>
  <version>5.2.4.Final</version>
</dependency>

　　

2 domain中添加驗證方法

@Entity
@Table(name = "employee")
public class Employee extends BaseDomain {

    @Excel(name = "名字")
    @NotNull(message = "用戶名不能空")
    private String username;
    private String password;
    @Excel(name = "年齡")
    @Max(value = 100)
    @Min(value = 18)
    private Integer age;
    @Excel(name = "郵箱",width = 20)
    @NotNull
    private String email;
    ...
}

3 自定義驗證

• 實現IExcelVerifyHandler
• 把這個類交給Spring管理(千萬不要忘了讓Spring去掃描到它)

  /**
   * 自定義驗證(我們會在這里做唯一性的驗證)
   */
  @Component
  public class EmployeeExcelVerifyHandler implements IExcelVerifyHandler<Employee> {
  
      @Autowired
      private IEmployeeService employeeService;
      /**
       *
       * ExcelVerifyHandlerResult
       *   suceess :代表驗證成功還是失敗(如果用戶名重復，就代表失敗)
       *   msg:失敗的原因
       */
      @Override
      public ExcelVerifyHandlerResult verifyHandler(Employee employee) {
  
          ExcelVerifyHandlerResult result = new ExcelVerifyHandlerResult(true);
          //如果根據用戶名獲取到用戶，代表這個用戶已經存在
          Employee tempEmp = employeeService.findByUsername(employee.getUsername());
          if(tempEmp!=null){
              result.setSuccess(false);
              result.setMsg("用戶名重復");
          }
  
          return  result;
      }
  }

  3 實現驗證功能

• @Controller
  @RequestMapping("/import")
  public class ImportController extends BaseController {
  
      @Autowired
      private IEmployeeService employeeService;
      @Autowired
      private IDepartmentService departmentService;
      @Autowired
      private EmployeeExcelVerifyHandler employeeExcelVerifyHandler;
  
      @RequestMapping("/index")
      public String index(){
          return "import";
      }
  
  
      @RequestMapping("/employeeXlsx")
      public String employeeXlsx(MultipartFile empFile, HttpServletResponse response) throws Exception {
          //一.使用EasyPoi獲取文件數據
          ImportParams params = new ImportParams();
          params.setHeadRows(1);
          params.setNeedVerfiy(true); //設置驗證支持
          params.setVerifyHandler(employeeExcelVerifyHandler); //設置一個驗證處理器
  
          //二.獲取excel中的數據，封裝成了一個結果對象(有很多東西)
          ExcelImportResult<Employee> result = ExcelImportUtil.importExcelMore(
                  empFile.getInputStream(),
                  Employee.class, params);
          //三.獲到正確的數據，並把它們保存到數據庫
          List<Employee> list = result.getList();
          list.forEach(e->{
              e.setPassword("123");
              Department dept = departmentService.findByName(e.getDepartment().getName());
              e.setDepartment(dept);
              employeeService.save(e);
          });
          //四.如果有錯誤，把錯誤數據返回到前台(讓前台下載一個錯誤的excel)
         //4.1判斷是否有錯誤
          if(result.isVerfiyFail()){
              //4.2拿到錯誤的文件薄
              Workbook failWorkbook = result.getFailWorkbook();
  
              //把這個文件導出
              response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"); //mime類型
              response.setHeader("Content-disposition", "attachment;filename=error.xlsx"); //告訴瀏覽下載的是一個附件，名字叫做error.xlsx
              response.setHeader("Pragma", "No-cache");//設置不要緩存
              OutputStream ouputStream = response.getOutputStream();
              failWorkbook.write(ouputStream);
              ouputStream.flush();
              ouputStream.close();
          }
  
          return "import";
      }
  
  
  }