lbqms.git - Gitblit

			@@ -1,7 +1,6 @@
			package com.ruoyi.framework.config;

			import cn.hutool.core.util.ObjectUtil;
			import cn.hutool.core.util.StrUtil;
			import com.ruoyi.common.utils.StringUtils;
			import com.ruoyi.framework.config.properties.RedissonProperties;
			import lombok.extern.slf4j.Slf4j;
			@@ -12,19 +11,18 @@
			import org.redisson.spring.cache.CacheConfig;
			import org.redisson.spring.cache.RedissonSpringCacheManager;
			import org.springframework.beans.factory.annotation.Autowired;
			import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
			import org.springframework.boot.autoconfigure.data.redis.RedisProperties;
			import org.springframework.cache.CacheManager;
			import org.springframework.cache.annotation.CachingConfigurerSupport;
			import org.springframework.cache.annotation.EnableCaching;
			import org.springframework.context.annotation.Bean;
			import org.springframework.context.annotation.Configuration;
			import org.springframework.data.redis.core.script.DefaultRedisScript;
			import org.springframework.context.annotation.Primary;

			import java.io.IOException;
			import java.util.HashMap;
			import java.util.List;
			import java.util.Map;
			import java.util.stream.Collectors;

			/**
			* redis配置
			@@ -36,121 +34,132 @@
			@EnableCaching
			public class RedisConfig extends CachingConfigurerSupport {

			private static final String REDIS_PROTOCOL_PREFIX = "redis://";
			private static final String REDISS_PROTOCOL_PREFIX = "rediss://";
			private static final String REDIS_PROTOCOL_PREFIX = "redis://";
			private static final String REDISS_PROTOCOL_PREFIX = "rediss://";

			@Autowired
			private RedisProperties redisProperties;
			@Autowired
			private RedisProperties redisProperties;

			@Autowired
			private RedissonProperties redissonProperties;
			@Autowired
			private RedissonProperties redissonProperties;

			@Bean(destroyMethod = "shutdown")
			@ConditionalOnMissingBean(RedissonClient.class)
			public RedissonClient redisson() throws IOException {
			String prefix = REDIS_PROTOCOL_PREFIX;
			if (redisProperties.isSsl()) {
			prefix = REDISS_PROTOCOL_PREFIX;
			}
			Config config = new Config();
			config.setThreads(redissonProperties.getThreads())
			.setNettyThreads(redissonProperties.getNettyThreads())
			.setCodec(JsonJacksonCodec.INSTANCE)
			.setTransportMode(redissonProperties.getTransportMode());
			@Primary
			@Bean(destroyMethod = "shutdown")
			public RedissonClient redisson() {
			String prefix = REDIS_PROTOCOL_PREFIX;
			if (redisProperties.isSsl()) {
			prefix = REDISS_PROTOCOL_PREFIX;
			}
			Config config = new Config();
			config.setThreads(redissonProperties.getThreads())
			.setNettyThreads(redissonProperties.getNettyThreads())
			.setCodec(JsonJacksonCodec.INSTANCE);

			RedissonProperties.SingleServerConfig singleServerConfig = redissonProperties.getSingleServerConfig();
			if (ObjectUtil.isNotNull(singleServerConfig)) {
			// 使用单机模式
			config.useSingleServer()
			.setAddress(prefix + redisProperties.getHost() + ":" + redisProperties.getPort())
			.setConnectTimeout(((Long) redisProperties.getTimeout().toMillis()).intValue())
			.setDatabase(redisProperties.getDatabase())
			.setPassword(StringUtils.isNotBlank(redisProperties.getPassword()) ? redisProperties.getPassword() : null)
			.setTimeout(singleServerConfig.getTimeout())
			.setRetryAttempts(singleServerConfig.getRetryAttempts())
			.setRetryInterval(singleServerConfig.getRetryInterval())
			.setSubscriptionsPerConnection(singleServerConfig.getSubscriptionsPerConnection())
			.setClientName(singleServerConfig.getClientName())
			.setIdleConnectionTimeout(singleServerConfig.getIdleConnectionTimeout())
			.setSubscriptionConnectionMinimumIdleSize(singleServerConfig.getSubscriptionConnectionMinimumIdleSize())
			.setSubscriptionConnectionPoolSize(singleServerConfig.getSubscriptionConnectionPoolSize())
			.setConnectionMinimumIdleSize(singleServerConfig.getConnectionMinimumIdleSize())
			.setConnectionPoolSize(singleServerConfig.getConnectionPoolSize())
			.setDnsMonitoringInterval(singleServerConfig.getDnsMonitoringInterval());
			}
			RedissonProperties.SingleServerConfig singleServerConfig = redissonProperties.getSingleServerConfig();
			if (ObjectUtil.isNotNull(singleServerConfig)) {
			// 使用单机模式
			config.useSingleServer()
			.setAddress(prefix + redisProperties.getHost() + ":" + redisProperties.getPort())
			.setConnectTimeout(((Long) redisProperties.getTimeout().toMillis()).intValue())
			.setDatabase(redisProperties.getDatabase())
			.setPassword(StringUtils.isNotBlank(redisProperties.getPassword()) ? redisProperties.getPassword() : null)
			.setTimeout(singleServerConfig.getTimeout())
			.setClientName(singleServerConfig.getClientName())
			.setIdleConnectionTimeout(singleServerConfig.getIdleConnectionTimeout())
			.setSubscriptionConnectionPoolSize(singleServerConfig.getSubscriptionConnectionPoolSize())
			.setConnectionMinimumIdleSize(singleServerConfig.getConnectionMinimumIdleSize())
			.setConnectionPoolSize(singleServerConfig.getConnectionPoolSize());
			}
			// 集群配置方式参考下方注释
			RedissonProperties.ClusterServersConfig clusterServersConfig = redissonProperties.getClusterServersConfig();
			if (ObjectUtil.isNotNull(clusterServersConfig)) {
			// 使用集群模式
			String finalPrefix = prefix;
			List<String> nodes = redisProperties.getCluster().getNodes()
			.stream()
			.map(node -> finalPrefix + node)
			.collect(Collectors.toList());

			RedissonProperties.ClusterServersConfig clusterServersConfig = redissonProperties.getClusterServersConfig();
			if (ObjectUtil.isNotNull(clusterServersConfig)) {
			// 使用集群模式
			config.useClusterServers()
			.setConnectTimeout(((Long) redisProperties.getTimeout().toMillis()).intValue())
			.setPassword(StringUtils.isNotBlank(redisProperties.getPassword()) ? redisProperties.getPassword() : null)
			.setTimeout(clusterServersConfig.getTimeout())
			.setRetryAttempts(clusterServersConfig.getRetryAttempts())
			.setRetryInterval(clusterServersConfig.getRetryInterval())
			.setSubscriptionsPerConnection(clusterServersConfig.getSubscriptionsPerConnection())
			.setClientName(clusterServersConfig.getClientName())
			.setIdleConnectionTimeout(clusterServersConfig.getIdleConnectionTimeout())
			.setPingConnectionInterval(clusterServersConfig.getPingConnectionInterval())
			.setSubscriptionConnectionMinimumIdleSize(clusterServersConfig.getSubscriptionConnectionMinimumIdleSize())
			.setSubscriptionConnectionPoolSize(clusterServersConfig.getSubscriptionConnectionPoolSize())
			.setMasterConnectionMinimumIdleSize(clusterServersConfig.getMasterConnectionMinimumIdleSize())
			.setMasterConnectionPoolSize(clusterServersConfig.getMasterConnectionPoolSize())
			.setSlaveConnectionMinimumIdleSize(clusterServersConfig.getSlaveConnectionMinimumIdleSize())
			.setSlaveConnectionPoolSize(clusterServersConfig.getSlaveConnectionPoolSize())
			.setDnsMonitoringInterval(clusterServersConfig.getDnsMonitoringInterval())
			.setFailedSlaveReconnectionInterval(clusterServersConfig.getFailedSlaveReconnectionInterval())
			.setScanInterval(clusterServersConfig.getScanInterval())
			.setReadMode(clusterServersConfig.getReadMode())
			.setSubscriptionMode(clusterServersConfig.getSubscriptionMode())
			.setNodeAddresses(redisProperties.getCluster().getNodes());
			}
			RedissonClient redissonClient = Redisson.create(config);
			log.info("初始化 redis 配置");
			return redissonClient;
			}
			config.useClusterServers()
			.setConnectTimeout(((Long) redisProperties.getTimeout().toMillis()).intValue())
			.setPassword(StringUtils.isNotBlank(redisProperties.getPassword()) ? redisProperties.getPassword() : null)
			.setTimeout(clusterServersConfig.getTimeout())
			.setClientName(clusterServersConfig.getClientName())
			.setIdleConnectionTimeout(clusterServersConfig.getIdleConnectionTimeout())
			.setSubscriptionConnectionPoolSize(clusterServersConfig.getSubscriptionConnectionPoolSize())
			.setMasterConnectionMinimumIdleSize(clusterServersConfig.getMasterConnectionMinimumIdleSize())
			.setMasterConnectionPoolSize(clusterServersConfig.getMasterConnectionPoolSize())
			.setSlaveConnectionMinimumIdleSize(clusterServersConfig.getSlaveConnectionMinimumIdleSize())
			.setSlaveConnectionPoolSize(clusterServersConfig.getSlaveConnectionPoolSize())
			.setReadMode(clusterServersConfig.getReadMode())
			.setSubscriptionMode(clusterServersConfig.getSubscriptionMode())
			.setNodeAddresses(nodes);
			}
			RedissonClient redissonClient = Redisson.create(config);
			log.info("初始化 redis 配置");
			return redissonClient;
			}

			/**
			* 整合spring-cache
			*/
			@Bean
			public CacheManager cacheManager(RedissonClient redissonClient) {
			List<RedissonProperties.CacheGroup> cacheGroup = redissonProperties.getCacheGroup();
			Map<String, CacheConfig> config = new HashMap<>();
			for (RedissonProperties.CacheGroup group : cacheGroup) {
			CacheConfig cacheConfig = new CacheConfig(group.getTtl(), group.getMaxIdleTime());
			cacheConfig.setMaxSize(group.getMaxSize());
			config.put(group.getGroupId(), cacheConfig);
			}
			return new RedissonSpringCacheManager(redissonClient, config, JsonJacksonCodec.INSTANCE);
			}
			/**
			* 整合spring-cache
			*/
			@Bean
			public CacheManager cacheManager(RedissonClient redissonClient) {
			List<RedissonProperties.CacheGroup> cacheGroup = redissonProperties.getCacheGroup();
			Map<String, CacheConfig> config = new HashMap<>();
			for (RedissonProperties.CacheGroup group : cacheGroup) {
			CacheConfig cacheConfig = new CacheConfig(group.getTtl(), group.getMaxIdleTime());
			cacheConfig.setMaxSize(group.getMaxSize());
			config.put(group.getGroupId(), cacheConfig);
			}
			return new RedissonSpringCacheManager(redissonClient, config, JsonJacksonCodec.INSTANCE);
			}

			@Bean
			public DefaultRedisScript<Long> limitScript() {
			DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();
			redisScript.setScriptText(limitScriptText());
			redisScript.setResultType(Long.class);
			return redisScript;
			}

			/**
			* 限流脚本
			*/
			private String limitScriptText() {
			return StrUtil.builder()
			.append("local key = KEYS[1]\n")
			.append("local count = tonumber(ARGV[1])\n")
			.append("local time = tonumber(ARGV[2])\n")
			.append("local current = redis.call('get', key);\n")
			.append("if current and tonumber(current) > count then\n")
			.append(" return current;\n")
			.append("end\n")
			.append("current = redis.call('incr', key)\n")
			.append("if tonumber(current) == 1 then\n")
			.append(" redis.call('expire', key, time)\n")
			.append("end\n")
			.append("return current;")
			.toString();
			}
			/**
			* redis集群配置 yml
			*
			* --- # redis 集群配置(单机与集群只能开启一个另一个需要注释掉)
			* spring:
			* redis:
			* cluster:
			* nodes:
			* - 192.168.0.100:6379
			* - 192.168.0.101:6379
			* - 192.168.0.102:6379
			* # 密码
			* password:
			* # 连接超时时间
			* timeout: 10s
			* # 是否开启ssl
			* ssl: false
			*
			* redisson:
			* # 线程池数量
			* threads: 16
			* # Netty线程池数量
			* nettyThreads: 32
			* # 集群配置
			* clusterServersConfig:
			* # 客户端名称
			* clientName: ${ruoyi.name}
			* # master最小空闲连接数
			* masterConnectionMinimumIdleSize: 32
			* # master连接池大小
			* masterConnectionPoolSize: 64
			* # slave最小空闲连接数
			* slaveConnectionMinimumIdleSize: 32
			* # slave连接池大小
			* slaveConnectionPoolSize: 64
			* # 连接空闲超时，单位：毫秒
			* idleConnectionTimeout: 10000
			* # 命令等待超时，单位：毫秒
			* timeout: 3000
			* # 发布和订阅连接池大小
			* subscriptionConnectionPoolSize: 50
			* # 读取模式
			* readMode: "SLAVE"
			* # 订阅模式
			* subscriptionMode: "MASTER"
			*/

			}