lsm/sa-token
1
0
Fork 0
mirror of https://gitee.com/dromara/sa-token.git synced 2025-05-06 13:47:56 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

!208 alone-redis 尝试添加集群模式

Browse Source 
Merge pull request !208 from AppleOfGray/dev
This commit is contained in:
孔明 2023-01-06 15:46:28 +00:00 committed by Gitee
commit 46a98b8ad5
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 32 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
sa-token-plugin/sa-token-alone-redis/src/main/java/cn/dev33/satoken/dao/alone/SaAloneRedisInject.java
View File

@ -9,8 +9,7 @@ import org.springframework.boot.context.properties.bind.Binder;
import org.springframework.context.EnvironmentAware; import org.springframework.context.EnvironmentAware;
import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Configuration;
import org.springframework.core.env.Environment; import org.springframework.core.env.Environment;
import org.springframework.data.redis.connection.RedisPassword; import org.springframework.data.redis.connection.*;
import org.springframework.data.redis.connection.RedisStandaloneConfiguration;
import org.springframework.data.redis.connection.lettuce.LettuceClientConfiguration; import org.springframework.data.redis.connection.lettuce.LettuceClientConfiguration;
import org.springframework.data.redis.connection.lettuce.LettuceConnectionFactory; import org.springframework.data.redis.connection.lettuce.LettuceConnectionFactory;
import org.springframework.data.redis.connection.lettuce.LettucePoolingClientConfiguration; import org.springframework.data.redis.connection.lettuce.LettucePoolingClientConfiguration;
@ -22,6 +21,9 @@ import cn.dev33.satoken.dao.SaTokenDaoRedisFastjson;
import cn.dev33.satoken.dao.SaTokenDaoRedisFastjson2; import cn.dev33.satoken.dao.SaTokenDaoRedisFastjson2;
import cn.dev33.satoken.dao.SaTokenDaoRedisJackson; import cn.dev33.satoken.dao.SaTokenDaoRedisJackson;
import java.util.List;
import java.util.stream.Collectors;
/** /**
* 为 SaTokenDao 单独设置Redis连接信息 * 为 SaTokenDao 单独设置Redis连接信息
* @author kong * @author kong
@ -59,14 +61,33 @@ public class SaAloneRedisInject implements EnvironmentAware{
// 获取cfg对象 // 获取cfg对象
RedisProperties cfg = Binder.get(environment).bind(ALONE_PREFIX, RedisProperties.class).get(); RedisProperties cfg = Binder.get(environment).bind(ALONE_PREFIX, RedisProperties.class).get();
// 1. Redis配置 // 1. Redis配置
RedisStandaloneConfiguration redisConfig = new RedisStandaloneConfiguration(); RedisConfiguration redisAloneConfig;
redisConfig.setHostName(cfg.getHost()); if (cfg.getCluster().getNodes().size() > 0) {
redisConfig.setPort(cfg.getPort()); // 集群模式
redisConfig.setDatabase(cfg.getDatabase()); RedisClusterConfiguration redisClusterConfig = new RedisClusterConfiguration();
redisConfig.setPassword(RedisPassword.of(cfg.getPassword())); redisClusterConfig.setUsername(cfg.getUsername());
redisClusterConfig.setPassword(cfg.getPassword());
RedisProperties.Cluster cluster = cfg.getCluster();
List<RedisNode> serverList = cluster.getNodes().stream().map(node -> {
String[] ipAndPort = node.split(":");
return new RedisNode(ipAndPort[0].trim(), Integer.parseInt(ipAndPort[1]));
}).collect(Collectors.toList());
redisClusterConfig.setClusterNodes(serverList);
redisClusterConfig.setMaxRedirects(cluster.getMaxRedirects());
redisAloneConfig = redisClusterConfig;
} else {
// 单体模式
RedisStandaloneConfiguration redisConfig = new RedisStandaloneConfiguration();
redisConfig.setHostName(cfg.getHost());
redisConfig.setPort(cfg.getPort());
redisConfig.setDatabase(cfg.getDatabase());
redisConfig.setPassword(RedisPassword.of(cfg.getPassword()));
redisConfig.setDatabase(cfg.getDatabase());
redisAloneConfig = redisConfig;
}
// 2. 连接池配置 // 2. 连接池配置
GenericObjectPoolConfig poolConfig = new GenericObjectPoolConfig(); GenericObjectPoolConfig poolConfig = new GenericObjectPoolConfig();
// pool配置 // pool配置
@ -93,7 +114,7 @@ public class SaAloneRedisInject implements EnvironmentAware{
} }
// 创建Factory对象 // 创建Factory对象
LettuceClientConfiguration clientConfig = builder.poolConfig(poolConfig).build(); LettuceClientConfiguration clientConfig = builder.poolConfig(poolConfig).build();
LettuceConnectionFactory factory = new LettuceConnectionFactory(redisConfig, clientConfig); LettuceConnectionFactory factory = new LettuceConnectionFactory(redisAloneConfig, clientConfig);
factory.afterPropertiesSet(); factory.afterPropertiesSet();
// 3. 开始初始化 SaTokenDao // 3. 开始初始化 SaTokenDao