From 80b6cbfc9c861469146318d0b3dd5f8b8b525b8a Mon Sep 17 00:00:00 2001 From: xiejun <xiejun@vci-tech.com> Date: 星期五, 01 十一月 2024 15:11:19 +0800 Subject: [PATCH] Revert "集成获取mdm分发通用数据格式接口集成" --- 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