guang
2024-06-14 93ac75952d5d6f64256434b99c8da612fa339263
提交 | 用户 | age
da6731 1 package com.hx.redisson.register;
C 2
3 import com.fasterxml.jackson.annotation.JsonAutoDetect;
4 import com.fasterxml.jackson.annotation.PropertyAccessor;
5 import com.fasterxml.jackson.databind.ObjectMapper;
6 import com.fasterxml.jackson.databind.jsontype.impl.LaissezFaireSubTypeValidator;
7 import com.hx.redisson.entity.RedissondbConfigEntity;
848054 8 import com.hx.redisson.manager.RedisManager;
da6731 9 import org.apache.commons.collections.CollectionUtils;
C 10 import org.redisson.Redisson;
11 import org.redisson.api.RedissonClient;
12 import org.redisson.client.codec.StringCodec;
13 import org.redisson.config.Config;
14 import org.redisson.config.SingleServerConfig;
15 import org.redisson.spring.data.connection.RedissonConnectionFactory;
16 import org.slf4j.Logger;
17 import org.slf4j.LoggerFactory;
18 import org.springframework.beans.factory.config.AutowireCapableBeanFactory;
19 import org.springframework.beans.factory.support.BeanDefinitionRegistry;
20 import org.springframework.beans.factory.support.GenericBeanDefinition;
21 import org.springframework.boot.context.properties.bind.Binder;
22 import org.springframework.context.EnvironmentAware;
23 import org.springframework.context.annotation.Configuration;
24 import org.springframework.context.annotation.ImportBeanDefinitionRegistrar;
25 import org.springframework.core.env.Environment;
26 import org.springframework.core.type.AnnotationMetadata;
27 import org.springframework.data.redis.connection.RedisConnectionFactory;
28 import org.springframework.data.redis.core.RedisTemplate;
29 import org.springframework.data.redis.serializer.Jackson2JsonRedisSerializer;
30 import org.springframework.data.redis.serializer.RedisSerializer;
31 import org.springframework.data.redis.serializer.StringRedisSerializer;
32
33 import java.util.List;
34
35 /**
36  * <h1>redistemplate初始化</h1>
37  * <p>
38  * 作用:
39  * <p>
40  * 读取系统配置,系统启动时,读取redis 的配置,初始化所有的redistemplate
41  * 并动态注册为bean
42  *
43  * @author gengzi
44  * @date 2021年1月5日22:16:29
45  */
46 @Configuration
47 public class RedisRegister implements EnvironmentAware, ImportBeanDefinitionRegistrar {
48
49     private Logger logger = LoggerFactory.getLogger(RedisRegister.class);
50
51     /**用于获取环境配置*/
52     private Environment environment;
53     /**用于绑定对象*/
54     private Binder binder;
55
56     /**
57      * 设置环境
58      *
59      * @param environment
60      */
61     @Override
62     public void setEnvironment(Environment environment) {
63         this.environment = environment;
64         this.binder = Binder.get(environment);
65     }
66
67     /**
68      * 注册bean
69      *
70      * @param importingClassMetadata
71      * @param registry
72      */
73     @Override
74     public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) {
75         logger.info("《《《动态注册bean开始》》》");
76         RedissondbConfigEntity redissondb;
77         try {
78             redissondb = binder.bind("redissondb", RedissondbConfigEntity.class).get();
79         } catch (Exception e) {
80             logger.error("读取redissondb环境配置失败", e);
81             return;
82         }
83         List<Integer> databases = redissondb.getDatabases();
84         if (CollectionUtils.isNotEmpty(databases)) {
85             databases.forEach(db -> {
86                 // 单机模式,集群只能使用db0
87                 Config config = new Config();
88                 config.setCodec(StringCodec.INSTANCE);
89                 SingleServerConfig singleConfig = config.useSingleServer();
90                 singleConfig.setAddress("redis://"+redissondb.getHost()+":"+redissondb.getPort());
91                 singleConfig.setPassword(redissondb.getPassword());
92                 singleConfig.setDatabase(db);
93
94                 System.out.println("getHost:"+singleConfig.getAddress());
95                 System.out.println("getHost:"+singleConfig.getPassword());
96                 RedissonClient redissonClient = Redisson.create(config);
848054 97
da6731 98                 // 构造RedissonConnectionFactory
C 99                 RedissonConnectionFactory redisConnectionFactory = new RedissonConnectionFactory(redissonClient);
100                 // bean定义
101                 GenericBeanDefinition redisTemplate = new GenericBeanDefinition();
102                 // 设置bean 的类型
103                 redisTemplate.setBeanClass(RedisTemplate.class);
104                 // 设置自动注入的形式,根据名称
105                 redisTemplate.setAutowireMode(AutowireCapableBeanFactory.AUTOWIRE_BY_NAME);
106                 // redisTemplate 的属性配置
107                 redisTemplate(redisTemplate, redisConnectionFactory);
108                 // 注册Bean
109                 registry.registerBeanDefinition("redisTemplate" + db, redisTemplate);
848054 110
F 111                 RedisManager.redissonTemplateMap.put(db+"",redissonClient);
da6731 112             });
C 113         }
114         logger.info("《《《动态注册bean结束》》》");
115
116     }
117
118     /**
119      * redisTemplate 的属性配置
120      *
121      * @param redisTemplate          泛型bean
122      * @param redisConnectionFactory 连接工厂
123      * @return
124      */
125     public GenericBeanDefinition redisTemplate(GenericBeanDefinition redisTemplate, RedisConnectionFactory redisConnectionFactory) {
126         RedisSerializer<String> stringRedisSerializer = new StringRedisSerializer();
127         Jackson2JsonRedisSerializer jackson2JsonRedisSerializer = new Jackson2JsonRedisSerializer(Object.class);
128         // 解决查询缓存转换异常的问题
129         ObjectMapper om = new ObjectMapper();
130         om.setVisibility(PropertyAccessor.ALL, JsonAutoDetect.Visibility.ANY);
131         om.activateDefaultTyping(LaissezFaireSubTypeValidator.instance,
132                 ObjectMapper.DefaultTyping.NON_FINAL);
133         jackson2JsonRedisSerializer.setObjectMapper(om);
134         // key采用String的序列化方式,value采用json序列化方式
135         // 通过方法设置属性值
136         redisTemplate.getPropertyValues().add("connectionFactory", redisConnectionFactory);
137         redisTemplate.getPropertyValues().add("keySerializer", stringRedisSerializer);
138         redisTemplate.getPropertyValues().add("hashKeySerializer", stringRedisSerializer);
139         redisTemplate.getPropertyValues().add("valueSerializer", jackson2JsonRedisSerializer);
140         redisTemplate.getPropertyValues().add("hashValueSerializer", jackson2JsonRedisSerializer);
141
142         return redisTemplate;
143     }
144 }
145