【Spring】AOP——使用@around实现面向切面的方法增强

工作业务中，有大量分布式加锁的重复代码，存在两个问题，一是代码重复率高，二是容易产生霰弹式修改，使用注解和AOP可以实现代码复用，简化分布式锁加锁和解锁流程。 @around注解是AspectJ框架提供的，允许我们在目标方法的执行前后进行代码增强。下面通过一个示例来介绍如何使用@around注解以及自定义注解实现加解锁（ReenTrantLock）简化。

改造前原代码很经典的桥段，初始化10个线程，每个线程分别执行m++1000次，若想得到10000的结果，则必须加锁同步，加锁则需要在执行m++前进行加锁，在finally语句块中执行解锁操作，因此可以通过@around注解实现简化。

@Service
@Slf4j
public class HelloService {private int m = 0;private static int nums = 0;public ReentrantLock lock;// spel表达式private static final String LOCK_KEY = "'asset_tenant_lock_prefix:' +" + "T(org.example.Common.ContextHelper).getTenantInfo().getOrgId()";@AutowiredHelloService helloService;public String doSomeSum() throws InterruptedException {nums++;log.info("调用次数：{}", nums);m = 0;CountDownLatch countDownLatch = new CountDownLatch(10);lock = new ReentrantLock();for (int i = 0; i < 10; i++) {Thread thread = new Thread(() -> {for (int j = 0; j < 1000; j++) {helloService.increM();}countDownLatch.countDown();});thread.start();}countDownLatch.await();return String.valueOf(m);}@DistributeLock(key = LOCK_KEY)public void increM() {lock.lock();try {m++;} finally {lock.unlock();}}
}

2. 改造第一步，创建一个注解注解名为DistributeLock，有一个熟悉key，（这个例子是为了模仿使用Redis分布式锁）

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface DistributeLock {String key();
}

3.改造第二步，定义一个切面类切面类中，定义一个增强方法useDistributeLock，参数joinPoint是切入点，distributeLock是切入点形参，用于传入key。这里我们key使用的是Spel表达式，使用SpelExpressionParser可以求得最终的key值，joinPoint.proceed()执行的则是上文提到的原方法的执行内容，m++。

@Component
@Slf4j
public class DistributeLockAspect {private static ReentrantLock lock = new ReentrantLock();@Around("@annotation(distributeLock)")public Object useDistributeLock(ProceedingJoinPoint joinPoint, DistributeLock distributeLock) throws Throwable {lock.lock();org.springframework.expression.ExpressionParser parser= new SpelExpressionParser();String lockKey = (String) parser.parseExpression(distributeLock.key()).getValue();log.info(lockKey);try {return joinPoint.proceed();} finally {lock.unlock();}}
}

4.改造第三步，使用使用很简单，只需要在需要增强的方法上加上我们创建的注解，并且给注解的属性Key一个字符串形式的Spel表达式即可实现加解锁操作。代码行数大大减少。

public class HelloService {private int m = 0;private static int nums = 0;public ReentrantLock lock;// spel表达式private static final String LOCK_KEY = "'asset_tenant_lock_prefix:' +" + "T(org.example.Common.ContextHelper).getTenantInfo().getOrgId()";@AutowiredHelloService helloService;public String doSomeSum() throws InterruptedException {nums++;log.info("调用次数：{}", nums);m = 0;CountDownLatch countDownLatch = new CountDownLatch(10);lock = new ReentrantLock();for (int i = 0; i < 10; i++) {Thread thread = new Thread(() -> {for (int j = 0; j < 1000; j++) {helloService.increM();}countDownLatch.countDown();});thread.start();}countDownLatch.await();return String.valueOf(m);}@DistributeLock(key = LOCK_KEY)public void increM() {m++;}
}

5.最终测试结果最终的测试结果是10000，说明加锁成功