97 lines
3.4 KiB
Java
97 lines
3.4 KiB
Java
|
package com.ruoyi.aspect;
|
|||
|
|
|
|||
|
|
import cn.hutool.core.collection.CollUtil;
|
|||
|
|
import cn.hutool.core.util.ReflectUtil;
|
|||
|
|
import cn.hutool.core.util.StrUtil;
|
|||
|
|
import com.ruoyi.cache.DictCache;
|
|||
|
|
import com.ruoyi.common.exception.ServiceException;
|
|||
|
|
import com.ruoyi.common.utils.bean.BeanValidators;
|
|||
|
|
import lombok.RequiredArgsConstructor;
|
|||
|
|
import org.aspectj.lang.ProceedingJoinPoint;
|
|||
|
|
import org.aspectj.lang.annotation.Around;
|
|||
|
|
import org.aspectj.lang.annotation.Aspect;
|
|||
|
|
import org.springframework.stereotype.Component;
|
|||
|
|
|
|||
|
|
import javax.validation.Validator;
|
|||
|
|
import java.lang.reflect.Field;
|
|||
|
|
import java.util.List;
|
|||
|
|
|
|||
|
|
@Aspect
|
|||
|
|
@Component
|
|||
|
|
@RequiredArgsConstructor
|
|||
|
|
public class ImportAspect {
|
|||
|
|
protected final Validator validator;
|
|||
|
|
|
|||
|
|
@Around("execution(* com.ruoyi.database.service.*.*import*(..))")
|
|||
|
|
public Object around(ProceedingJoinPoint joinPoint) throws Throwable {
|
|||
|
|
// 获取方法的入参
|
|||
|
|
Object[] args = joinPoint.getArgs();
|
|||
|
|
List<?> importList = (List<?>) args[0];
|
|||
|
|
|
|||
|
|
|
|||
|
|
if (CollUtil.isEmpty(importList)) {
|
|||
|
|
throw new ServiceException("导入数据为空");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取list表中类的类型
|
|||
|
|
Class<?> clazz = importList.get(0).getClass();
|
|||
|
|
|
|||
|
|
// 错误标记，判断是否有错误信息
|
|||
|
|
boolean errorFlag = false;
|
|||
|
|
// 遍历list表中的数据，回填字典值
|
|||
|
|
for (Object obj : importList) {
|
|||
|
|
|
|||
|
|
// 记录属性中存在区域编码相关的属性的个数，判断如何回填区域编码
|
|||
|
|
int codeCount = 0;
|
|||
|
|
// 获取类中的所有属性
|
|||
|
|
for (Field field : clazz.getDeclaredFields()) {
|
|||
|
|
// 判断属性中是否有Cn结尾的属性
|
|||
|
|
String fieldName = field.getName();
|
|||
|
|
if (StrUtil.endWithIgnoreCase(fieldName, "Cn") && !field.getName().equals("structuredCameraTypeCn")) {
|
|||
|
|
field.setAccessible(true);
|
|||
|
|
// 获取非Cn属性名
|
|||
|
|
String fieldName1 = StrUtil.subBefore(fieldName, "Cn", true);
|
|||
|
|
String dictType = StrUtil.toUnderlineCase(fieldName1);
|
|||
|
|
// 获取字典标签
|
|||
|
|
Object dictLabel = field.get(obj);
|
|||
|
|
int dictValue;
|
|||
|
|
try {
|
|||
|
|
// 获取字典值
|
|||
|
|
dictValue = Integer.parseInt(DictCache.getDictValue(StrUtil.format("{}-{}", dictType, dictLabel)));
|
|||
|
|
} catch (Exception e) {
|
|||
|
|
// 字典值不存在，跳过
|
|||
|
|
continue;
|
|||
|
|
}
|
|||
|
|
// 回填字典值
|
|||
|
|
ReflectUtil.setFieldValue(obj, fieldName1, dictValue);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
// // 判断属性中是否存在区域编码相关的属性
|
|||
|
|
// if (StrUtil.equalsAny(fieldName, "placeCode", "gridCode", "buildingCode", "unitCode", "homeCode")) {
|
|||
|
|
// codeCount++;
|
|||
|
|
// }
|
|||
|
|
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 校验数据，校验不通过则记录错误信息
|
|||
|
|
try {
|
|||
|
|
BeanValidators.validateWithException(validator, obj);
|
|||
|
|
} catch (Exception e) {
|
|||
|
|
ReflectUtil.setFieldValue(obj, "remark", e.getMessage());
|
|||
|
|
errorFlag = true;
|
|||
|
|
continue;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (errorFlag) {
|
|||
|
|
return importList;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 执行目标方法
|
|||
|
|
return joinPoint.proceed();
|
|||
|
|
}
|
|||
|
|
}
|