兰宝车间质量管理系统
blame | 历史 | 补丁 | 提交 | 提交对比 | 忽略空白
抓蛙师
2022-07-14 ce7536df9f82833044c9cbc479f0de42de2beffe 
 ruoyi-framework/src/main/java/com/ruoyi/framework/config/RedisConfig.java


@@ -2,28 +2,25 @@




import cn.hutool.core.util.ObjectUtil;


import com.fasterxml.jackson.databind.ObjectMapper;


import com.ruoyi.common.utils.StringUtils;


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.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配置


@@ -33,13 +30,8 @@


@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 RedisProperties redisProperties;




    @Autowired


    private RedissonProperties redissonProperties;


@@ -47,61 +39,42 @@


    @Autowired


    private ObjectMapper objectMapper;




    @Primary


    @Bean(destroyMethod = "shutdown")


    public RedissonClient redisson(RedisProperties redisProperties, RedissonProperties redissonProperties, ObjectMapper objectMapper) {


        String prefix = REDIS_PROTOCOL_PREFIX;


        if (redisProperties.isSsl()) {


            prefix = REDISS_PROTOCOL_PREFIX;


        }


        Config config = new 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)


                .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)


                .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;


    @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 配置");


        };


    }




    /**