Parker
4 years ago
parent
a8ae3e0b1d
commit
43243e24d3
@ -0,0 +1,23 @@
|
||||
package org.opsli.common.annotation;
|
||||
|
||||
|
||||
import java.lang.annotation.*;
|
||||
|
||||
/**
|
||||
* @BelongsProject: opsli-boot
|
||||
* @BelongsPackage: org.opsli.common.annotation
|
||||
* @Author: Parker
|
||||
* @CreateTime: 2020-09-16 16:36
|
||||
* @Description: 字典标示
|
||||
|
||||
*
|
||||
*/
|
||||
@Retention(RetentionPolicy.RUNTIME)
|
||||
@Target(ElementType.FIELD)
|
||||
@Documented
|
||||
public @interface DictType {
|
||||
|
||||
/** 字典类型 code */
|
||||
String value();
|
||||
|
||||
}
|
||||
@ -0,0 +1,25 @@
|
||||
package org.opsli.common.enums;
|
||||
|
||||
|
||||
/**
|
||||
* @BelongsProject: opsli-boot
|
||||
* @BelongsPackage: org.opsli.common.enums
|
||||
* @Author: Parker
|
||||
* @CreateTime: 2020-09-17 23:40
|
||||
* @Description: Excel 操作类型
|
||||
*/
|
||||
public enum ExcelOperate {
|
||||
|
||||
/**
|
||||
* 读
|
||||
*/
|
||||
READ,
|
||||
|
||||
/**
|
||||
* 写
|
||||
*/
|
||||
WRITE;
|
||||
|
||||
;
|
||||
|
||||
}
|
||||
@ -0,0 +1,155 @@
|
||||
package org.opsli.core.utils;
|
||||
|
||||
import cn.hutool.core.util.ReflectUtil;
|
||||
import com.alibaba.excel.support.ExcelTypeEnum;
|
||||
import com.google.common.collect.Maps;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.apache.commons.lang3.StringUtils;
|
||||
import org.opsli.api.wrapper.system.dict.DictModel;
|
||||
import org.opsli.common.annotation.DictType;
|
||||
import org.opsli.common.enums.ExcelOperate;
|
||||
import org.opsli.plugins.excel.ExcelPlugin;
|
||||
import org.opsli.plugins.excel.exception.ExcelPluginException;
|
||||
import org.springframework.stereotype.Component;
|
||||
import org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
import java.lang.reflect.Field;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* @BelongsProject: opsli-boot
|
||||
* @BelongsPackage: org.opsli.core.utils
|
||||
* @Author: Parker
|
||||
* @CreateTime: 2020-09-23 21:53
|
||||
* @Description: ExcelUtil
|
||||
*/
|
||||
@Slf4j
|
||||
@Component
|
||||
public class ExcelUtil extends ExcelPlugin {
|
||||
|
||||
@Override
|
||||
public <T> List<T> readExcel(MultipartFile excel, Class<T> rowModel) throws ExcelPluginException {
|
||||
List<T> ts = super.readExcel(excel, rowModel);
|
||||
// 处理数据
|
||||
return this.handleDict(ts, rowModel, ExcelOperate.READ);
|
||||
}
|
||||
|
||||
@Override
|
||||
public <T> List<T> readExcel(MultipartFile excel, Class<T> rowModel, String sheetName) throws ExcelPluginException {
|
||||
List<T> ts = super.readExcel(excel, rowModel, sheetName);
|
||||
// 处理数据
|
||||
return this.handleDict(ts, rowModel, ExcelOperate.READ);
|
||||
}
|
||||
|
||||
@Override
|
||||
public <T> List<T> readExcel(MultipartFile excel, Class<T> rowModel, String sheetName, int headLineNum) throws ExcelPluginException {
|
||||
List<T> ts = super.readExcel(excel, rowModel, sheetName, headLineNum);
|
||||
// 处理数据
|
||||
return this.handleDict(ts, rowModel, ExcelOperate.READ);
|
||||
}
|
||||
|
||||
@Override
|
||||
public <T> void writeExcel(HttpServletResponse response, List<T> list, String fileName, String sheetName, Class<T> classType, ExcelTypeEnum excelTypeEnum) throws ExcelPluginException {
|
||||
// 处理数据
|
||||
List<T> ts = this.handleDict(list, classType, ExcelOperate.WRITE);
|
||||
super.writeExcel(response, ts, fileName, sheetName, classType, excelTypeEnum);
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理字典
|
||||
* @param datas
|
||||
* @param typeClazz
|
||||
* @param operate
|
||||
* @param <T>
|
||||
* @return
|
||||
*/
|
||||
private <T> List<T> handleDict(List<T> datas, Class<T> typeClazz, ExcelOperate operate){
|
||||
|
||||
// 空处理
|
||||
if(datas == null || datas.size() == 0){
|
||||
return datas;
|
||||
}
|
||||
|
||||
// 字段名 - 字典code
|
||||
Map<String,String> fieldAndTypeCode = Maps.newHashMap();
|
||||
// 字典code - 字典值
|
||||
Map<String,List<DictModel>> typeCodeAndValue = Maps.newHashMap();
|
||||
|
||||
Field[] fields = ReflectUtil.getFields(typeClazz);
|
||||
for (Field field : fields) {
|
||||
DictType dictType = field.getAnnotation(DictType.class);
|
||||
if(dictType != null){
|
||||
String typeCode = dictType.value();
|
||||
fieldAndTypeCode.put(field.getName(), typeCode);
|
||||
}
|
||||
}
|
||||
|
||||
// 如果没有字典 则直接返回
|
||||
if(fieldAndTypeCode.size() == 0){
|
||||
return datas;
|
||||
}
|
||||
|
||||
// 取Redis 值
|
||||
for (Map.Entry<String, String> entry : fieldAndTypeCode.entrySet()) {
|
||||
String typeCode = entry.getValue();
|
||||
List<DictModel> dictList = DictUtil.getDictList(typeCode);
|
||||
// 如果字典 List 为空 则走下一个
|
||||
if(dictList == null || dictList.size() == 0) continue;
|
||||
typeCodeAndValue.put(typeCode, dictList);
|
||||
}
|
||||
|
||||
// 数据字典赋值
|
||||
for (T data : datas) {
|
||||
for (Map.Entry<String, String> entry : fieldAndTypeCode.entrySet()) {
|
||||
try {
|
||||
String fieldName = entry.getKey();
|
||||
String typeCode = entry.getValue();
|
||||
String fieldValue = (String) ReflectUtil.getFieldValue(data, fieldName);
|
||||
List<DictModel> dictModels = typeCodeAndValue.get(typeCode);
|
||||
// 匹配字典
|
||||
String dictVal = this.matchingDict(dictModels, fieldValue, operate);
|
||||
if(StringUtils.isEmpty(dictVal)){
|
||||
continue;
|
||||
}
|
||||
// 赋值
|
||||
ReflectUtil.setFieldValue(data, fieldName, dictVal);
|
||||
}catch (Exception e){
|
||||
log.error(e.getMessage(), e);
|
||||
}
|
||||
}
|
||||
}
|
||||
return datas;
|
||||
}
|
||||
|
||||
/**
|
||||
* 匹配字典数据
|
||||
* @param dictModels
|
||||
* @param fieldValue
|
||||
* @param operate
|
||||
* @return
|
||||
*/
|
||||
private String matchingDict(List<DictModel> dictModels, String fieldValue, ExcelOperate operate){
|
||||
String val = "";
|
||||
for (DictModel dictModel : dictModels) {
|
||||
// 读操作
|
||||
if(ExcelOperate.READ == operate){
|
||||
// 判断 Excel 读入 字典名称是否与 当前字典匹配
|
||||
if(dictModel.getDictName().equals(fieldValue)){
|
||||
val = dictModel.getDictValue();
|
||||
break;
|
||||
}
|
||||
}
|
||||
// 写操作
|
||||
else if(ExcelOperate.WRITE == operate){
|
||||
// 判断 Excel 写出 字典值是否与 当前字典匹配
|
||||
if(dictModel.getDictValue().equals(fieldValue)){
|
||||
val = dictModel.getDictName();
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
return val;
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue