| | |
|---|
| | | package com.ruoyi.framework.config; |
|---|
| | | |
|---|
| | | import cn.hutool.core.util.StrUtil; |
|---|
| | | import com.ruoyi.common.utils.StringUtils; |
|---|
| | | import cn.hutool.core.util.ObjectUtil; |
|---|
| | | import com.fasterxml.jackson.databind.ObjectMapper; |
|---|
| | | import com.ruoyi.framework.config.properties.RedissonProperties; |
|---|
| | | import com.ruoyi.framework.handler.KeyPrefixHandler; |
|---|
| | | 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.RedissonAutoConfigurationCustomizer; |
|---|
| | | 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.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.data.redis.core.script.DefaultRedisScript; |
|---|
| | | |
|---|
| | | import java.io.IOException; |
|---|
| | | import java.util.HashMap; |
|---|
| | | import java.util.List; |
|---|
| | | import java.util.Map; |
|---|
| | |
|---|
| | | @Slf4j |
|---|
| | | @Configuration |
|---|
| | | @EnableCaching |
|---|
| | | @EnableConfigurationProperties(RedissonProperties.class) |
|---|
| | | public class RedisConfig extends CachingConfigurerSupport { |
|---|
| | | |
|---|
| | | private static final String REDIS_PROTOCOL_PREFIX = "redis://"; |
|---|
| | | private static final String REDISS_PROTOCOL_PREFIX = "rediss://"; |
|---|
| | | @Autowired |
|---|
| | | private RedissonProperties redissonProperties; |
|---|
| | | |
|---|
| | | @Autowired |
|---|
| | | private RedisProperties redisProperties; |
|---|
| | | @Autowired |
|---|
| | | private ObjectMapper objectMapper; |
|---|
| | | |
|---|
| | | @Autowired |
|---|
| | | private RedissonProperties redissonProperties; |
|---|
| | | @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() |
|---|
| | | .setNameMapper(new KeyPrefixHandler(redissonProperties.getKeyPrefix()))//设置redis key前缀 |
|---|
| | | .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)) { |
|---|
| | | config.useClusterServers() |
|---|
| | | .setNameMapper(new KeyPrefixHandler(redissonProperties.getKeyPrefix()))//设置redis key前缀 |
|---|
| | | .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()); |
|---|
| | | } |
|---|
| | | log.info("初始化 redis 配置"); |
|---|
| | | }; |
|---|
| | | } |
|---|
| | | |
|---|
| | | @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()); |
|---|
| | | /** |
|---|
| | | * 整合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, new JsonJacksonCodec(objectMapper)); |
|---|
| | | } |
|---|
| | | |
|---|
| | | RedissonProperties.SingleServerConfig singleServerConfig = redissonProperties.getSingleServerConfig(); |
|---|
| | | // 使用单机模式 |
|---|
| | | 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()); |
|---|
| | | 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); |
|---|
| | | } |
|---|
| | | |
|---|
| | | @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" |
|---|
| | | */ |
|---|
| | | |
|---|
| | | } |
|---|
