|
|
|
@ -2,31 +2,23 @@ package com.ruoyi.common.redis.config;
|
|
|
|
|
|
|
|
|
|
import cn.hutool.core.util.ObjectUtil;
|
|
|
|
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
|
|
|
|
import com.ruoyi.common.core.utils.StringUtils;
|
|
|
|
|
import com.ruoyi.common.redis.config.properties.RedissonProperties;
|
|
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
|
|
|
import org.redisson.Redisson;
|
|
|
|
|
import org.redisson.api.RedissonClient;
|
|
|
|
|
import org.redisson.codec.JsonJacksonCodec;
|
|
|
|
|
import org.redisson.config.Config;
|
|
|
|
|
import org.redisson.spring.cache.CacheConfig;
|
|
|
|
|
import org.redisson.spring.cache.RedissonSpringCacheManager;
|
|
|
|
|
import org.redisson.spring.starter.RedissonAutoConfiguration;
|
|
|
|
|
import org.redisson.spring.starter.RedissonAutoConfigurationCustomizer;
|
|
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
|
|
import org.springframework.boot.autoconfigure.AutoConfigureBefore;
|
|
|
|
|
import org.springframework.boot.autoconfigure.data.redis.RedisProperties;
|
|
|
|
|
import org.springframework.boot.context.properties.EnableConfigurationProperties;
|
|
|
|
|
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.context.annotation.Primary;
|
|
|
|
|
|
|
|
|
|
import java.util.HashMap;
|
|
|
|
|
import java.util.List;
|
|
|
|
|
import java.util.Map;
|
|
|
|
|
import java.util.stream.Collectors;
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* redis配置
|
|
|
|
@ -34,64 +26,37 @@ import java.util.stream.Collectors;
|
|
|
|
|
* @author Lion Li
|
|
|
|
|
*/
|
|
|
|
|
@Slf4j
|
|
|
|
|
@AutoConfigureBefore(RedissonAutoConfiguration.class)
|
|
|
|
|
@EnableConfigurationProperties(RedissonProperties.class)
|
|
|
|
|
@Configuration
|
|
|
|
|
@EnableCaching
|
|
|
|
|
@EnableConfigurationProperties(RedissonProperties.class)
|
|
|
|
|
public class RedisConfiguration extends CachingConfigurerSupport {
|
|
|
|
|
|
|
|
|
|
private static final String REDIS_PROTOCOL_PREFIX = "redis://";
|
|
|
|
|
private static final String REDISS_PROTOCOL_PREFIX = "rediss://";
|
|
|
|
|
|
|
|
|
|
@Autowired
|
|
|
|
|
private RedisProperties redisProperties;
|
|
|
|
|
|
|
|
|
|
@Autowired
|
|
|
|
|
private RedissonProperties redissonProperties;
|
|
|
|
|
|
|
|
|
|
@Autowired
|
|
|
|
|
private ObjectMapper objectMapper;
|
|
|
|
|
|
|
|
|
|
@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())
|
|
|
|
|
@Bean
|
|
|
|
|
public RedissonAutoConfigurationCustomizer redissonCustomizer() {
|
|
|
|
|
return config -> {
|
|
|
|
|
config.setThreads(redissonProperties.getThreads())
|
|
|
|
|
.setNettyThreads(redissonProperties.getNettyThreads())
|
|
|
|
|
.setCodec(new JsonJacksonCodec(objectMapper));
|
|
|
|
|
|
|
|
|
|
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)
|
|
|
|
|
RedissonProperties.SingleServerConfig singleServerConfig = redissonProperties.getSingleServerConfig();
|
|
|
|
|
if (ObjectUtil.isNotNull(singleServerConfig)) {
|
|
|
|
|
// 使用单机模式
|
|
|
|
|
config.useSingleServer()
|
|
|
|
|
.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());
|
|
|
|
|
|
|
|
|
|
config.useClusterServers()
|
|
|
|
|
.setConnectTimeout(((Long) redisProperties.getTimeout().toMillis()).intValue())
|
|
|
|
|
.setPassword(StringUtils.isNotBlank(redisProperties.getPassword()) ? redisProperties.getPassword() : null)
|
|
|
|
|
}
|
|
|
|
|
// 集群配置方式 参考下方注释
|
|
|
|
|
RedissonProperties.ClusterServersConfig clusterServersConfig = redissonProperties.getClusterServersConfig();
|
|
|
|
|
if (ObjectUtil.isNotNull(clusterServersConfig)) {
|
|
|
|
|
config.useClusterServers()
|
|
|
|
|
.setTimeout(clusterServersConfig.getTimeout())
|
|
|
|
|
.setClientName(clusterServersConfig.getClientName())
|
|
|
|
|
.setIdleConnectionTimeout(clusterServersConfig.getIdleConnectionTimeout())
|
|
|
|
@ -101,12 +66,10 @@ public class RedisConfiguration extends CachingConfigurerSupport {
|
|
|
|
|
.setSlaveConnectionMinimumIdleSize(clusterServersConfig.getSlaveConnectionMinimumIdleSize())
|
|
|
|
|
.setSlaveConnectionPoolSize(clusterServersConfig.getSlaveConnectionPoolSize())
|
|
|
|
|
.setReadMode(clusterServersConfig.getReadMode())
|
|
|
|
|
.setSubscriptionMode(clusterServersConfig.getSubscriptionMode())
|
|
|
|
|
.setNodeAddresses(nodes);
|
|
|
|
|
}
|
|
|
|
|
RedissonClient redissonClient = Redisson.create(config);
|
|
|
|
|
log.info("初始化 redis 配置");
|
|
|
|
|
return redissonClient;
|
|
|
|
|
.setSubscriptionMode(clusterServersConfig.getSubscriptionMode());
|
|
|
|
|
}
|
|
|
|
|
log.info("初始化 redis 配置");
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|