MqAspect.java

package com.yanzuoguang.cloud.aop;

import com.yanzuoguang.mq.plan.YzgMqConsumer;
import com.yanzuoguang.util.exception.ExceptionHelper;
import com.yanzuoguang.util.log.Log;
import com.yanzuoguang.util.vo.LogVo;
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.aspectj.lang.reflect.MethodSignature;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.stereotype.Component;

import java.lang.reflect.Method;

/**
 * LogsAspect（接口请求日志切面）
 *
 * @author: Kang
 * @time: 2018年04月25日 11:43
 */
@Aspect
@Component
public class MqAspect extends BaseRequestAspect {

    private static final String TAG = MqAspect.class.getSimpleName();

    /**
     * exec aop point aspect
     */
    @Pointcut("execution(* *..mq..*Consumer.*(..))")
    public void mqAspect() {
    }

    /**
     * 执行环形切面
     *
     * @param joinPoint
     * @return
     */
    @Around(value = "mqAspect()")
    public Object requestWebAround(ProceedingJoinPoint joinPoint) throws Throwable {
        boolean clear = isRabbit(joinPoint);
        LogVo log = null;
        long start = System.currentTimeMillis();
        String url = getMethodUrl(joinPoint);
        if (clear) {
            log = startLog(TAG, url, getRequestBody(joinPoint));
            clear = clear && log != null;
        }
        if (clear) {
            Log.threadBegin();
            start = requestLog(TAG, clear, joinPoint);
        }
        Object result = null;
        Exception ex = null;
        try {
            result = executeMethod(joinPoint);
            return result;
        } catch (Exception e) {
            // 消费消息出错
            result = ExceptionHelper.getError(e);
            ex = e;
            throw e;
        } finally {
            responseLog(log, TAG, url, clear, joinPoint, start, result, ex);
        }
    }

    private boolean isRabbit(ProceedingJoinPoint joinPoint) {
        boolean ret = false;
        if (!(joinPoint.getSignature() instanceof MethodSignature)) {
            return false;
        }
        MethodSignature methodSignature = (MethodSignature) joinPoint.getSignature();
        Method targetMethod = methodSignature.getMethod();
        if (YzgMqConsumer.class.equals(targetMethod.getDeclaringClass())) {
            ret = false;
        } else {
            RabbitListener annotation = targetMethod.getAnnotation(RabbitListener.class);
            if (annotation != null) {
                ret = true;
            }
            if (!ret) {
                Message message = getMessage(joinPoint.getArgs());
                if (message != null) {
                    ret = true;
                }
            }
        }
        return ret;
    }

    private Object executeMethod(ProceedingJoinPoint joinPoint) throws Throwable {
        return joinPoint.proceed();
    }
}