package cn.besteam.common.aspect;
|
|
import cn.besteam.common.aspect.annotation.LockApi;
|
import com.baomidou.lock.LockExecutor;
|
import com.baomidou.lock.LockInfo;
|
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.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
|
import java.lang.reflect.Method;
|
|
/**
|
* 系统日志，切面处理类
|
*
|
* @author chenqilong
|
* @since 2019/4/19
|
*/
|
@Aspect
|
@Component
|
public class LockApiAspect {
|
|
@Autowired
|
private LockExecutor lockExecutor;
|
|
private LockKeyGenerator lockKeyGenerator = new LockKeyGenerator();
|
|
@Pointcut("@annotation(cn.besteam.common.aspect.annotation.LockApi)")
|
public void lockPointCut() {
|
|
}
|
|
@Around("lockPointCut()")
|
public Object around(ProceedingJoinPoint point) throws Throwable {
|
|
LockInfo lockInfo = null;
|
try {
|
MethodSignature signature = (MethodSignature) point.getSignature();
|
Method method = signature.getMethod();
|
LockApi lockApi = method.getAnnotation(LockApi.class);
|
String keyName = lockKeyGenerator.getKeyName(point, lockApi);
|
lockInfo = lockExecutor.tryLock(keyName, lockApi.expire(), lockApi.tryTimeout());
|
if (null != lockInfo) {
|
return point.proceed();
|
}
|
return null;
|
} finally {
|
if (null != lockInfo) {
|
lockExecutor.releaseLock(lockInfo);
|
}
|
}
|
}
|
|
}
|
