本文内容:spring 中如何使用注解实现面向切面编程,以及如何使用自定义注解。
一个场景
比如用户登录,每个请求发起之前都会判断用户是否登录,如果每个请求都去判断一次,那就重复地做了很多事情,只要是有重复的地方,就有优化的空间。现在就把重复的地方抽取出来,暂且称之为 “拦截器”,然后每次请求之前就先经过 “拦截器”,这个编程的思想就可以称之为面向切面编程。AOP(Aspect Oriented Program)
最典型的应用就是事务管理和权限验证,还有日志统计,下文中的案例就是接口执行时间的统计。
spring 中使用 AOP(基于注解)
不得不说注解是个很巧妙的设计,使用很少量的信息描述数据,这类数据称之为元数据,描述数据的数据。关于注解的理解,这里有个传送门:http://www.importnew.com/10294.html
下面的案例是在 springBoot 中进行的,直观地感受一下如何使用注解完成 AOP。
@Service
public class UserService {
public void getUser() {
//To do something
System.out.println("getUser() has been called");
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
切面是这样定义的:
@Component
@Aspect
public class LoggerAspect {
/**
* getUser()执行之前执行
*/
@Before("execution(* com.springboot.demo.service.UserService.getUser(..))")
public void callBefore() {
System.out.println("before call method");
System.out.println("begin........................");
}
/**
* getUser()执行之后执行
*/
@After("execution(* com.springboot.demo.service.UserService.getUser(..))")
public void callAfter() {
System.out.println("after call method");
System.out.println("end..............................");
}
}
来个单元测试验证一下:
@RunWith(SpringRunner.class)
@SpringBootTest
public class UserServiceTest {
@Autowired
private UserService userService;
@Test
public void getUserTest() {
userService.getUser();
}
}
具体案例
假如有以下的业务场景: UserService 业务类中有个 getUser() 这个方法,现在想统计一下这个方法的执行时间,可能需要测试这个接口的性能。通常做法是方法开始时获取系统当前时间,然后方法结束时获取当前时间,最后 excuteTime=endTime-startTime。
如果现在不仅是这个方法需要统计,还有 getUserByName()、getUserById() 需要统计,上述的方法明显很笨了。
使用 AOP 怎么解决? 抽取公共部分为一个切面,方法执行前记录时间,然后执行目标方法,最后,目标方法执行完成之后再获取一次系统时间。
具体实现如下:在 LoggerAspect 中再写一个方法,记录 getUser() 方法的执行时间。
/**
* 记录执行时间
* @param point 切点
* @return
* @throws Throwable
*/
@Around("execution(* com.springboot.demo.service.UserService.getUser(..))")
public Object getMethodExecuteTime(ProceedingJoinPoint point) throws Throwable {
System.out.println("---------------getMethodExecuteTime------------------");
long startTime = System.currentTimeMillis();
//调用目标方法
Object result = point.proceed();
long endTime = System.currentTimeMillis();
long executeTime = endTime - startTime;
System.out.println("executeTime=" + executeTime + "------------------");
return result;
}
@Around 将目标方法再次封装,控制了它的调用时机,以此来记录 getUser() 的执行时间。但是好像并没有达到记录 UserService 中的多个方法的执行时间的目的。
@Around(“execution(* com.springboot.demo.service.UserService.getUser(..))”)
其中指定了切点是 getUser() 这个方法,这里的表达式很丰富,可以设置为:
- com.springboot.demo.service.UserService.*(..)
表示 UserService 中的每一个方法都是切点,甚至可以是这样:
- com.springboot.demo.service..(..)
表示 service 包下的所有类的所有方法都是切点,但是这样很明显不够灵活,如果能自定义地控制就更好了。
自定义一个注解
如果用一个注解标注某个方法需要记录其执行时间,岂不是更加优雅。
/**
* @Description 标注某个方法需要记录执行时间
* @Author YaoQi
* @Date 2018/7/6 15:51
*/
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface Logger {
String value() default "";
}
注解是用来描述数据的,上面的这个注解的意思是:这个注解将作用于方法,并且在运行时有效。但是这样只是标注了,如何读取这个标注的信息?
在 LoggerAspect 中加入这一个方法:
/**
* @param point
* @return
* @throws Throwable
*/
@Around("@annotation(com.springboot.demo.annotation.Logger)")
public Object getMethodExecuteTimeForLogger(ProceedingJoinPoint point) throws Throwable {
System.out.println("---------------getMethodExecuteTime------------------");
long startTime = System.currentTimeMillis();
Object result = point.proceed();
long endTime = System.currentTimeMillis();
long executeTime = endTime - startTime;
System.out.println("executeTime=" + executeTime + "------------------");
return result;
}
哪个方法需要记录执行时间就将 @Logger 放在对应的方法上:
@Logger
public void getUser() {
System.out.println("getUser() has been called");
}