Spring AOP 返回结果注入业务数据

 一、数据返回后切面注入业务值

package com.test.aspect;

import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;

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.beans.factory.annotation.Autowired;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Component;

import java.util.*;

/**
 * @Author Andy
 * @Date 2021/9/7
 */
@Aspect
@Component
@Slf4j
public class GiftAspect {

    private final static String FILED_GIFT_PROD_ID = "giftProdId";
    private final static String FILED_COUPON_TYPE = "couponType";
    private final static String FILED_GIFT_PRO_DTO = "giftProDto ";

    @Autowired
    private final ProductMapper productMapper;

    @Pointcut("@annotation(com.test.aspect.annotation.AutoGiftCoupon)")
    public void giftCouponPointCut() {

    }

    @Around("giftCouponPointCut()")
    public Object doAround(ProceedingJoinPoint pjp) throws Throwable {
        long time1 = System.currentTimeMillis();
        Object result = pjp.proceed();
        long time2 = System.currentTimeMillis();
        log.debug("获取JSON数据 耗时：" + (time2 - time1) + "ms");
        long start = System.currentTimeMillis();
        this.parseCoupon(result);
        long end = System.currentTimeMillis();
        log.debug("解析注入JSON数据  耗时" + (end - start) + "ms");
        return result;
    }

    /**
     * 解析赠品券
     *
     * @param response
     */
    private void parseCoupon(Object response) {
        if (response instanceof ResponseEntity) {
            Object result = ((ResponseEntity)response).getBody();
            if (result instanceof IPage || result instanceof List) {
                List<Object> list;
                if (result instanceof IPage) {
                    list = ((IPage)result).getRecords();
                } else {
                    list = (List<Object>)result;
                }
                List<JSONObject> items = new ArrayList<>();
                Set<Long> set = new HashSet<>();
                for (Object record : list) {
                    ObjectMapper mapper = new ObjectMapper();
                    String json = "{}";
                    try {
                        //解决@JsonFormat注解解析不了的问题详见SysAnnouncement类的@JsonFormat
                        json = mapper.writeValueAsString(record);
                    } catch (JsonProcessingException e) {
                        log.error("json解析失败" + e.getMessage(), e);
                    }
                    JSONObject item = JSONObject.parseObject(json);
                 /*   自定义字段注解获取数据，用来填充
                 for (Field field : getAllFields(record)) {
                        //update-end--Author:scott  -- Date:20190603 ----for：解决继承实体字段无法翻译问题------
                        if (field.getAnnotation(Dict.class) != null) {
                            String code = field.getAnnotation(Dict.class).dicCode();
                            String text = field.getAnnotation(Dict.class).dicText();
                            String table = field.getAnnotation(Dict.class).dictTable();
                            String key = String.valueOf(item.get(field.getName()));
                            //翻译字典值对应的txt
                            String textValue = translateDictValue(code, text, table, key);
                            item.put(field.getName() + CommonConstant.DICT_TEXT_SUFFIX, textValue);
                        }
                    }*/

                    Integer couponType = item.getInteger(FILED_COUPON_TYPE);
                    if (CouponType.C2G.value().equals(couponType)) {
                        set.add(item.getLong(FILED_GIFT_PROD_ID));
                    }
                    items.add(item);
                }
                ((IPage) ((ResponseEntity) response).getBody()).setRecords(items);
                Map<Long, GiftProDto> map = getGiftProDtoMap(set);
                items.forEach(item -> {
                    Integer couponType = item.getInteger(FILED_COUPON_TYPE);
                    if (CouponType.C2G.value().equals(couponType)) {
                        Long giftProdId = item.getLong(FILED_GIFT_PROD_ID);
                        if (giftProdId != null) {
                            item.put(FILED_GIFT_PRO_DTO, map.get(giftProdId));
                        }
                    }
                });

            } else {
                ObjectMapper mapper = new ObjectMapper();
                String json = "{}";
                try {
                    //解决@JsonFormat注解解析不了的问题详见SysAnnouncement类的@JsonFormat
                    json = mapper.writeValueAsString(result);
                } catch (JsonProcessingException e) {
                    log.error("json解析失败" + e.getMessage(), e);
                }
                JSONObject item = JSONObject.parseObject(json);
                Integer couponType = item.getInteger(FILED_COUPON_TYPE);
                if (CouponType.C2G.value().equals(couponType)) {
                    Long giftProdId = item.getLong(FILED_GIFT_PROD_ID);
                    if (giftProdId != null) {
                        GiftProDto giftProDto = productMapper.getGiftProInfo(giftProdId);
                        item.put(FILED_GIFT_PRO_DTO, giftProDto);
                    }
                }

            }
        }

    }

    private Map<Long, GiftProDto> getGiftProDtoMap(Set<Long> set) {
        Map<Long, GiftProDto> map = new HashMap<>();
        set.forEach(item -> {
            GiftProDto giftProDto = productMapper.getGiftProInfo(item);
            map.put(item, giftProDto);
        });
        return map;
    }

}

二、注解声明类 ，只需要在接口上加上注解即可

package com.test.aspect.annotation;

import java.lang.annotation.*;

/**
 * @Author Andy
 * @Date 2021/9/7
 */
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface AutoGiftCoupon {
    /**
     * 赠品券标识
     *
     * @return
     */
    String value() default "";
}

三、利用反射获取字段值

/**
	 * 获取类的所有属性，包括父类
	 * 
	 * @param object
	 * @return
	 */
	public static Field[] getAllFields(Object object) {
		Class<?> clazz = object.getClass();
		List<Field> fieldList = new ArrayList<>();
		while (clazz != null) {
			fieldList.addAll(new ArrayList<>(Arrays.asList(clazz.getDeclaredFields())));
			clazz = clazz.getSuperclass();
		}
		Field[] fields = new Field[fieldList.size()];
		fieldList.toArray(fields);
		return fields;
	}