最近一直忙着写业务代码,好久没更新博客,记录这几天用到的一个小功能吧
背景需求:项目需要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<String> listRoom=buildingService.listRoom();
setCombo("roomNo",listRoom);
ExcelUtil<StaffMainImportTemplate> util=new ExcelUtil<StaffMainImportTemplate>(StaffMainImportTemplate.class);
return util.importTemplateExcel("staffMessage");
}
// 给combo赋值
private void setCombo(String param,List<String> 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);
}
}
|