forked from 201206030/novel
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
ac1628a
commit 9bd95d3
Showing
8 changed files
with
208 additions
and
22 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
21 changes: 21 additions & 0 deletions
21
src/main/java/io/github/xxyopen/novel/core/annotation/Key.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,21 @@ | ||
| package io.github.xxyopen.novel.core.annotation; | ||
|
|
||
| import java.lang.annotation.Documented; | ||
| import java.lang.annotation.ElementType; | ||
| import java.lang.annotation.Retention; | ||
| import java.lang.annotation.Target; | ||
|
|
||
| import static java.lang.annotation.RetentionPolicy.RUNTIME; | ||
|
|
||
| /** | ||
| * 分布式锁-Key 注解 | ||
| * | ||
| * @author xiongxiaoyang | ||
| * @date 2022/6/20 | ||
| */ | ||
| @Documented | ||
| @Retention(RUNTIME) | ||
| @Target({ElementType.METHOD, ElementType.FIELD, ElementType.PARAMETER}) | ||
| public @interface Key { | ||
| String expr() default ""; | ||
| } |
31 changes: 31 additions & 0 deletions
31
src/main/java/io/github/xxyopen/novel/core/annotation/Lock.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,31 @@ | ||
| package io.github.xxyopen.novel.core.annotation; | ||
|
|
||
| import io.github.xxyopen.novel.core.common.constant.ErrorCodeEnum; | ||
|
|
||
| import java.lang.annotation.Documented; | ||
| import java.lang.annotation.Retention; | ||
| import java.lang.annotation.Target; | ||
|
|
||
| import static java.lang.annotation.ElementType.METHOD; | ||
| import static java.lang.annotation.RetentionPolicy.RUNTIME; | ||
|
|
||
| /** | ||
| * 分布式锁 注解 | ||
| * | ||
| * @author xiongxiaoyang | ||
| * @date 2022/6/20 | ||
| */ | ||
| @Documented | ||
| @Retention(RUNTIME) | ||
| @Target(METHOD) | ||
| public @interface Lock { | ||
|
|
||
| String prefix(); | ||
|
|
||
| boolean isWait() default false; | ||
|
|
||
| long waitTime() default 3L; | ||
|
|
||
| ErrorCodeEnum failCode() default ErrorCodeEnum.OK; | ||
|
|
||
| } |
81 changes: 81 additions & 0 deletions
81
src/main/java/io/github/xxyopen/novel/core/aspect/LockAspect.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,81 @@ | ||
| package io.github.xxyopen.novel.core.aspect; | ||
|
|
||
| import io.github.xxyopen.novel.core.annotation.Lock; | ||
| import io.github.xxyopen.novel.core.common.exception.BusinessException; | ||
| import lombok.SneakyThrows; | ||
| import org.aspectj.lang.ProceedingJoinPoint; | ||
| import org.aspectj.lang.annotation.Around; | ||
| import org.aspectj.lang.annotation.Aspect; | ||
| import org.aspectj.lang.reflect.MethodSignature; | ||
| import org.redisson.api.RLock; | ||
| import org.redisson.api.RedissonClient; | ||
| import org.springframework.expression.Expression; | ||
| import org.springframework.expression.ExpressionParser; | ||
| import org.springframework.expression.common.TemplateParserContext; | ||
| import org.springframework.expression.spel.standard.SpelExpressionParser; | ||
| import org.springframework.stereotype.Component; | ||
|
|
||
| import java.lang.reflect.Method; | ||
| import java.lang.reflect.Parameter; | ||
| import java.util.Objects; | ||
| import java.util.concurrent.TimeUnit; | ||
|
|
||
| /** | ||
| * 分布式锁 切面 | ||
| * | ||
| * @author xiongxiaoyang | ||
| * @date 2022/6/20 | ||
| */ | ||
| @Aspect | ||
| @Component | ||
| public record LockAspect(RedissonClient redissonClient) { | ||
|
|
||
| private static final String KEY_PREFIX = "Lock"; | ||
|
|
||
| private static final String KEY_SEPARATOR = "::"; | ||
|
|
||
| @Around(value = "@annotation(io.github.xxyopen.novel.core.annotation.Lock)") | ||
| @SneakyThrows | ||
| public Object doAround(ProceedingJoinPoint joinPoint) { | ||
| MethodSignature methodSignature = (MethodSignature) joinPoint.getSignature(); | ||
| Method targetMethod = methodSignature.getMethod(); | ||
| Lock lock = targetMethod.getAnnotation(Lock.class); | ||
| String lockKey = KEY_PREFIX + buildLockKey(lock.prefix(), targetMethod, | ||
| joinPoint.getArgs()); | ||
| RLock rLock = redissonClient.getLock(lockKey); | ||
| if (lock.isWait() ? rLock.tryLock(lock.waitTime(), TimeUnit.SECONDS) : rLock.tryLock()) { | ||
| try { | ||
| return joinPoint.proceed(); | ||
| } finally { | ||
| rLock.unlock(); | ||
| } | ||
| } | ||
| throw new BusinessException(lock.failCode()); | ||
| } | ||
|
|
||
| private String buildLockKey(String prefix, Method method, Object[] args) { | ||
| StringBuilder builder = new StringBuilder(); | ||
| if (Objects.nonNull(prefix) && !prefix.isEmpty()) { | ||
| builder.append(KEY_SEPARATOR).append(prefix); | ||
| } | ||
| Parameter[] parameters = method.getParameters(); | ||
| for (int i = 0; i < parameters.length; i++) { | ||
| builder.append(KEY_SEPARATOR); | ||
| if (parameters[i].isAnnotationPresent(io.github.xxyopen.novel.core.annotation.Key.class)) { | ||
| io.github.xxyopen.novel.core.annotation.Key key = parameters[i].getAnnotation(io.github.xxyopen.novel.core.annotation.Key.class); | ||
| builder.append(parseKeyExpr(key.expr(), args[i])); | ||
| } | ||
| } | ||
| return builder.toString(); | ||
| } | ||
|
|
||
| private String parseKeyExpr(String expr, Object arg) { | ||
| if (Objects.isNull(expr) || expr.isEmpty()) { | ||
| return arg.toString(); | ||
| } | ||
| ExpressionParser parser = new SpelExpressionParser(); | ||
| Expression expression = parser.parseExpression(expr, new TemplateParserContext()); | ||
| return expression.getValue(arg, String.class); | ||
| } | ||
|
|
||
| } |
26 changes: 26 additions & 0 deletions
26
src/main/java/io/github/xxyopen/novel/core/config/RedissonConfig.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,26 @@ | ||
| package io.github.xxyopen.novel.core.config; | ||
|
|
||
| import lombok.SneakyThrows; | ||
| import org.redisson.Redisson; | ||
| import org.redisson.api.RedissonClient; | ||
| import org.redisson.config.Config; | ||
| import org.springframework.context.annotation.Bean; | ||
| import org.springframework.context.annotation.Configuration; | ||
|
|
||
| /** | ||
| * Redisson 配置类 | ||
| * | ||
| * @author xiongxiaoyang | ||
| * @date 2022/6/20 | ||
| */ | ||
| @Configuration | ||
| public class RedissonConfig { | ||
|
|
||
| @Bean | ||
| @SneakyThrows | ||
| public RedissonClient redissonClient(){ | ||
| Config config = Config.fromYAML(getClass().getResource("/redisson.yml")); | ||
| return Redisson.create(config); | ||
| } | ||
|
|
||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,3 @@ | ||
| singleServerConfig: | ||
| address: "redis://127.0.0.1:6379" | ||
| password: 123456 |