若依框架下Excel 导出自定义列属性只能下垃选择

最近一直忙着写业务代码，好久没更新博客，记录这几天用到的一个小功能吧

背景需求：项目需要Excel导出，但一些信息需要在Excel列表显示可读性强的中文，数据库存储对应的数据的的id或者数字，为了保持和表单录入一致性。比如下面实体的属性sex，数据库存储的对应是0和1，导出excel里面直接写入男、女，但是有些字段是动态的，比如说办公室房间号码roomNo字段，不可能将所有的都列在上面，因为是动态的随时增加或者减少，所以只要能做到将@Excel的属性combo动态赋值即可

需要导入的模板实体：

@Data
@ApiModel("员工基本信息主对象")
public class StaffMainImportTemplate extends BaseEntity {
    private static final long serialVersionUID = 1L;

    private Long id;

    @Excel(name = "员工编号")
    @ApiModelProperty(value = "员工编号")
    private String staffNo;

    @Excel(name = "员工姓名")
    @ApiModelProperty(value = "员工姓名")
    private String staffName;

    @Excel(name = "性别", readConverterExp = "1=男,0=女", combo = {"男", "女"})
    @ApiModelProperty(value = "性别;1-男，0-女")
    private String sex;

    @Excel(name = "办公室号")
    @ApiModelProperty(value = "办公室号")
    private String roomNo;
}

@Excel注解【为了节省博客排版空间删减了一些】

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.FIELD)
public @interface Excel {
    /**
     * 读取内容转表达式 (如: 0=男,1=女,2=未知)
     */
    public String readConverterExp() default "";

    /**
     * 设置只能选择不能输入的列内容.
     */
    public String[] combo() default {};
}

利用反射获取指定类的属性的注解

@ApiOperation(value = "下载模板")
@RequestMapping(value = "/downStaffInformationTemplate", method = RequestMethod.GET)
public AjaxResult downStaffInformationTemplate()throws Exception{
      // 设置房间
      List listRoom=buildingService.listRoom();
      setCombo("roomNo",listRoom);
      ExcelUtil util=new ExcelUtil(StaffMainImportTemplate.class);
      return util.importTemplateExcel("staffMessage");
}
// 给combo赋值
private void setCombo(String param,List list)throws Exception{
      // 通过反射 获取目标实体类的属性成员-即办公室号号字段
      Field file=StaffMainImportTemplate.class.getDeclaredField(param);
      // 获取该字段的上叫Excel的注解
      Excel annotation=file.getAnnotation(Excel.class);
      InvocationHandler h=Proxy.getInvocationHandler(annotation);
      Field hField=h.getClass().getDeclaredField("memberValues");
      // 设置私有可访问
      hField.setAccessible(true);
      Map memberValues=(Map)hField.get(h);
      // 集合转数组
      String[]combo=list.toArray(new String[list.size()]);
      // 修改属性值
      memberValues.put("combo",combo);
}

若依框架动态赋值地方

 调用链路如上，代码如下

public void setDataValidation(Excel attr, Row row, int column) {
        if (attr.name().indexOf("注：") >= 0) {
            sheet.setColumnWidth(column, 6000);
        } else {
            // 设置列宽
            sheet.setColumnWidth(column, (int) ((attr.width() + 0.72) * 256));
            row.setHeight((short) (attr.height() * 20));
        }
        // 如果设置了提示信息则鼠标放上去提示.
        if (StringUtils.isNotEmpty(attr.prompt())) {
            // 这里默认设了2-101列提示.
            setXSSFPrompt(sheet, "", attr.prompt(), 1, 100, column, column);
        }
        // 如果设置了combo属性则本列只能选择不能输入
        if (attr.combo().length > 0) {
            // 这里默认设了2-101列只能选择不能输入.
            setXSSFValidation(sheet, attr.combo(), 1, 100, column, column);
        }
    }