From 9b4433fddf5b401edb0aace8a404ac733b122702 Mon Sep 17 00:00:00 2001 From: 田源 <tianyuan@vci-tech.com> Date: 星期四, 03 四月 2025 14:35:02 +0800 Subject: [PATCH] 添加非密字段显示 --- Source/BladeX-Tool/blade-starter-redis/src/main/java/org/springblade/core/redis/lock/RedisLockAspect.java | 103 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 103 insertions(+), 0 deletions(-) diff --git a/Source/BladeX-Tool/blade-starter-redis/src/main/java/org/springblade/core/redis/lock/RedisLockAspect.java b/Source/BladeX-Tool/blade-starter-redis/src/main/java/org/springblade/core/redis/lock/RedisLockAspect.java new file mode 100644 index 0000000..c0a03f7 --- /dev/null +++ b/Source/BladeX-Tool/blade-starter-redis/src/main/java/org/springblade/core/redis/lock/RedisLockAspect.java @@ -0,0 +1,103 @@ +/* + * Copyright (c) 2018-2028, DreamLu All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * Redistributions of source code must retain the above copyright notice, + * this list of conditions and the following disclaimer. + * Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * Neither the name of the dreamlu.net developer nor the names of its + * contributors may be used to endorse or promote products derived from + * this software without specific prior written permission. + * Author: DreamLu 鍗㈡槬姊� (596392912@qq.com) + */ + +package org.springblade.core.redis.lock; + +import lombok.RequiredArgsConstructor; +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.springblade.core.tool.spel.BladeExpressionEvaluator; +import org.springblade.core.tool.utils.CharPool; +import org.springblade.core.tool.utils.StringUtil; +import org.springframework.beans.BeansException; +import org.springframework.context.ApplicationContext; +import org.springframework.context.ApplicationContextAware; +import org.springframework.context.expression.AnnotatedElementKey; +import org.springframework.expression.EvaluationContext; +import org.springframework.util.Assert; + +import java.lang.reflect.Method; +import java.util.concurrent.TimeUnit; + +/** + * redis 鍒嗗竷寮忛攣 + * + * @author L.cm + */ +@Aspect +@RequiredArgsConstructor +public class RedisLockAspect implements ApplicationContextAware { + + /** + * 琛ㄨ揪寮忓鐞� + */ + private static final BladeExpressionEvaluator EVALUATOR = new BladeExpressionEvaluator(); + /** + * redis 闄愭祦鏈嶅姟 + */ + private final RedisLockClient redisLockClient; + private ApplicationContext applicationContext; + + /** + * AOP 鐜垏 娉ㄨВ @RedisLock + */ + @Around("@annotation(redisLock)") + public Object aroundRedisLock(ProceedingJoinPoint point, RedisLock redisLock) { + String lockName = redisLock.value(); + Assert.hasText(lockName, "@RedisLock value must have length; it must not be null or empty"); + // el 琛ㄨ揪寮� + String lockParam = redisLock.param(); + // 琛ㄨ揪寮忎笉涓虹┖ + String lockKey; + if (StringUtil.isNotBlank(lockParam)) { + String evalAsText = evalLockParam(point, lockParam); + lockKey = lockName + CharPool.COLON + evalAsText; + } else { + lockKey = lockName; + } + LockType lockType = redisLock.type(); + long waitTime = redisLock.waitTime(); + long leaseTime = redisLock.leaseTime(); + TimeUnit timeUnit = redisLock.timeUnit(); + return redisLockClient.lock(lockKey, lockType, waitTime, leaseTime, timeUnit, point::proceed); + } + + /** + * 璁$畻鍙傛暟琛ㄨ揪寮� + * + * @param point ProceedingJoinPoint + * @param lockParam lockParam + * @return 缁撴灉 + */ + private String evalLockParam(ProceedingJoinPoint point, String lockParam) { + MethodSignature ms = (MethodSignature) point.getSignature(); + Method method = ms.getMethod(); + Object[] args = point.getArgs(); + Object target = point.getTarget(); + Class<?> targetClass = target.getClass(); + EvaluationContext context = EVALUATOR.createContext(method, args, target, targetClass, applicationContext); + AnnotatedElementKey elementKey = new AnnotatedElementKey(method, targetClass); + return EVALUATOR.evalAsText(lockParam, elementKey, context); + } + + @Override + public void setApplicationContext(ApplicationContext applicationContext) throws BeansException { + this.applicationContext = applicationContext; + } +} -- Gitblit v1.9.3