使用aop切面，自动替换为字典数据 | 我的日常分享

使用aop切面，自动替换为字典数据

发表于2022-12-19更新于2022-12-19字数统计464阅读时长5分

使用aop切面，自动替换为字典数据

image-20221219000331111

一、代码

Dict.java

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
package cn.yuencode.oisp.admin.annotation;

import java.lang.annotation.*;

/**
 * 被此注解修饰的方法中返回的List集合中的对象中被DictValue注解修饰
 * 的字段值，将被替换为字典数据标签值
 * @author jiaxiaoyu
 * @date 2022/12/18
 */
@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Inherited
public @interface Dict {

}

DictValue.java

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
package cn.yuencode.oisp.admin.annotation;

import java.lang.annotation.*;

/**
 * 被此注解修饰的字段，其值将替换为字典数据标签值，需要结合注解Dict使用
 * @author jiaxiaoyu
 * @date 2022/12/18
 */
@Target({ElementType.FIELD})
@Retention(RetentionPolicy.RUNTIME)
@Inherited
public @interface DictValue {
    /**
     * 字典标签数据键值 默认空字符串，使用字段值作为字段标签数据键值，
     * 如果不为空，则使用value作为字段标签数据键值。
     */
    String value() default "";

    /**
     * 字典类型
     */
    String dictType();
}

DictAspect.java

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
package cn.yuencode.oisp.admin.aop;

import cn.yuencode.oisp.admin.annotation.DictValue;
import cn.yuencode.oisp.common.security.util.DictUtils;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.stereotype.Component;

import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * 字典切面
 * @author jiaxiaoyu
 * @date 2022/12/18
 */
@Aspect
@Component
@Slf4j
public class DictAspect {
    @Pointcut("@annotation(cn.yuencode.oisp.admin.annotation.Dict)")
    public void dictPointCut() {
    }

    @Around("dictPointCut()")
    public Object aroundMethod(ProceedingJoinPoint pjd) throws Throwable {
        Object result = null;
        try {
            result = pjd.proceed();
            List r = (ArrayList) result;
            for (Object obj : r) {
                Arrays.stream(obj.getClass().getDeclaredFields())
                        .forEach(field -> {
                            Object fieldValue = getFieldValue(obj,field);
                            if (field.isAnnotationPresent(DictValue.class) && field.getType().equals(String.class)) {
                                String dictType = field.getAnnotation(DictValue.class).dictType();
                                String value = field.getAnnotation(DictValue.class).value();
                                // 获取字典标签值
                                String dictLabel = DictUtils.getDictLabel(dictType, value.equals("") ? (String) fieldValue : value);
                                setFieldValue(obj, field, dictLabel);
                            }
                        });
            }
        } catch (Throwable e) {
            throw e;
        }
        return result;
    }
    private Object getFieldValue(Object obj, Field field) {
        try {
            field.setAccessible(true);
            return field.get(obj);
        } catch (Exception e) {
        }
        return null;
    }

    private void setFieldValue(Object obj, Field field, Object value) {
        field.setAccessible(true);
        try {
            field.set(obj, value);
        } catch (IllegalAccessException e) {
            throw new RuntimeException(e);
        }
    }
}

二、效果

结合Excel导出使用

image-20221219001055526