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/config/RateLimiterAutoConfiguration.java | 64 ++++++++++++++++++++++++++++++++ 1 files changed, 64 insertions(+), 0 deletions(-) diff --git a/Source/BladeX-Tool/blade-starter-redis/src/main/java/org/springblade/core/redis/config/RateLimiterAutoConfiguration.java b/Source/BladeX-Tool/blade-starter-redis/src/main/java/org/springblade/core/redis/config/RateLimiterAutoConfiguration.java new file mode 100644 index 0000000..01e203d --- /dev/null +++ b/Source/BladeX-Tool/blade-starter-redis/src/main/java/org/springblade/core/redis/config/RateLimiterAutoConfiguration.java @@ -0,0 +1,64 @@ +/* + * 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.config; + +import org.springblade.core.redis.ratelimiter.RedisRateLimiterAspect; +import org.springblade.core.redis.ratelimiter.RedisRateLimiterClient; +import org.springframework.boot.autoconfigure.AutoConfiguration; +import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean; +import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; +import org.springframework.context.annotation.Bean; +import org.springframework.core.env.Environment; +import org.springframework.core.io.ClassPathResource; +import org.springframework.data.redis.core.StringRedisTemplate; +import org.springframework.data.redis.core.script.DefaultRedisScript; +import org.springframework.data.redis.core.script.RedisScript; +import org.springframework.scripting.support.ResourceScriptSource; + +import java.util.List; + +/** + * 鍩轰簬 redis 鐨勫垎甯冨紡闄愭祦鑷姩閰嶇疆 + * + * @author L.cm + */ +@AutoConfiguration +@ConditionalOnProperty(value = "blade.redis.rate-limiter.enabled", havingValue = "true") +public class RateLimiterAutoConfiguration { + + @SuppressWarnings("unchecked") + private RedisScript<List<Long>> redisRateLimiterScript() { + DefaultRedisScript redisScript = new DefaultRedisScript<>(); + redisScript.setScriptSource(new ResourceScriptSource(new ClassPathResource("META-INF/scripts/blade_rate_limiter.lua"))); + redisScript.setResultType(List.class); + return redisScript; + } + + @Bean + @ConditionalOnMissingBean + public RedisRateLimiterClient redisRateLimiter(StringRedisTemplate redisTemplate, Environment environment) { + RedisScript<List<Long>> redisRateLimiterScript = redisRateLimiterScript(); + return new RedisRateLimiterClient(redisTemplate, redisRateLimiterScript, environment); + } + + @Bean + @ConditionalOnMissingBean + public RedisRateLimiterAspect redisRateLimiterAspect(RedisRateLimiterClient rateLimiterClient) { + return new RedisRateLimiterAspect(rateLimiterClient); + } +} -- Gitblit v1.9.3