Spring Boot AOP 日志切面实现

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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
@Aspect
@Component
public class LogAspect {

	private static final Map<Class<?>, Logger> loggerHolder = new ConcurrentHashMap<Class<?>, Logger>();

	@Pointcut(value = "@annotation(com.linghit.common.log.annotation.Loggable)")
	public void log() {
	}

	/**
	 * 
	 * around:根据日志注解类型为方法调用记录日志 <br/>
	 */
	@Around("log()")
	public Object around(ProceedingJoinPoint joinPoint) throws Throwable {
		AbstractLogBean logBean = null;

		Method method = MethodSignature.class.cast(joinPoint.getSignature())
				.getMethod();
		Annotation[] annotations = method.getAnnotations();
		for (Annotation annotation : annotations) {
			if (annotation instanceof Loggable) {
				Loggable loggable = Loggable.class.cast(annotation);
				logBean = getLogBean(loggable.value(), joinPoint);
				break;
			}
		}

		Object retVal = null;
		if (null == logBean) {
			retVal = joinPoint.proceed(joinPoint.getArgs());
		} else {
			logBean.setEventName(StringUtils.isBlank(logBean.getEventName()) ? method
					.getName() : logBean.getEventName() + "_"
					+ method.getName());
			logBean.setRequest(joinPoint.getArgs());
			try {
				retVal = joinPoint.proceed(joinPoint.getArgs());
			} catch (Exception e) {
				retVal = Resp.createError(RespCode.BUSINESS_INVALID,
						"service.fail", "服务失败");

				logBean.setE(e);

			} finally {

				logBean.setResponse(retVal);

				Logger logger = getLogger(joinPoint.getTarget().getClass());
				LogUtils.log(logger, logBean);
			}
		}

		return retVal;

	}

	private AbstractLogBean getLogBean(DataAccessType type, JoinPoint joinPoint) {
		AbstractLogBean logBean = null;
		switch (type.getValue()) {
		case "MySQL":
			logBean = DataAccessLogBean.newDataAccessMysqlLogBean();
			break;
		case "Http":
			logBean = DataAccessLogBean.newDataAccessHttpLogBean();
			break;
		case "Redis":
			logBean = DataAccessLogBean.newDataAccessRedisLogBean();
			break;
		default:
			logBean = new ServiceAccessLogBean(joinPoint.getTarget().getClass()
					.getSimpleName(), DataAccessType.DUBBO.getValue(),
					RpcContextUtils.getClientIp(),
					RpcContextUtils.getLocalAddress());
		}
		return logBean;
	}

	private Logger getLogger(Class<?> clazz) {
		if (!loggerHolder.containsKey(clazz)) {
			loggerHolder.put(clazz, LoggerFactory.getLogger(clazz));
		}
		return loggerHolder.get(clazz);
	}
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
@Target({ ElementType.METHOD })
@Retention(RetentionPolicy.RUNTIME)
@Inherited
@Documented
public @interface Loggable {

	/**
	 * Name of the logType in which the logging takes place.
	 * <p>
	 * May be used to determine the target cache (or caches), matching the
	 * qualifier value (or the bean name(s)) of (a) specific bean definition.
	 */
	DataAccessType value() default DataAccessType.DUBBO;
}

1
2
@EnableAspectJAutoProxy(proxyTargetClass = true)
@ComponentScan("com.linghit.ocs.zhanxing.service.handler.LogAspect")

1
2
@Loggable
public void test{}