SpringBoot实现冷数据预热

最新推荐文章于 2024-07-23 20:56:23 发布

想象之中丶

最新推荐文章于 2024-07-23 20:56:23 发布

阅读量219

点赞数

本文链接：https://blog.csdn.net/weixin_58267543/article/details/132440076

版权

SpringBoot 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

SpringBoot实现冷数据预热功能

实现原理

实现ServletContextListener接口，重写contextInitialized()方法，当SpringContext初始化完成后，会调用该方法
在该方法里实现业务逻辑

示例

定义冷数据预热业务类，实现ServletContextListener

import com.alibaba.fastjson.JSONObject;
import gk.springboot.preheat.dao.UserDao;
import gk.springboot.preheat.model.UserInfo;
import org.springframework.boot.autoconfigure.AutoConfigureAfter;
import org.springframework.data.redis.core.Cursor;
import org.springframework.data.redis.core.RedisCallback;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.ScanOptions;
import org.springframework.data.redis.hash.Jackson2HashMapper;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import javax.servlet.ServletContextEvent;
import javax.servlet.ServletContextListener;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

@Component
// 由于配置了redisTemplate，故考虑需要在该配置后加载，防止redisTemplate配置未生效
@AutoConfigureAfter(value = RedisTemplateConfig.class)
public class PreHeatUserToRedis implements ServletContextListener {
    @Resource
    private RedisTemplate<String, Object> redisTemplate;
    @Resource
    private UserDao userDao;

    public static final String USER_KEY_PREFIX = "user:";

    @Override
    public void contextInitialized(ServletContextEvent sce) {
        // 先将原来数据清空（即使用unlink删除key）
        /*Set<String> keys = redisTemplate.keys(USER_KEY_PREFIX + "*");
        redisTemplate.unlink(keys);*/

        // 如果keys *命令被禁用，考虑使用scan命令，将user:*匹配到的key取出，然后删除
        // keys *命令当redis库中有很多key时，会阻塞redis，导致其他业务请求redis等待时间较长
        // 使用scan命令类似于limit查询，每次只查询N个符合pattern的key，不会造成redis阻塞
        Set<String> keys = redisTemplate.execute((RedisCallback<Set<String>>) connection -> {
            Set<String> keysTmp = new HashSet<>();
            Cursor<byte[]> cursor = connection.scan(ScanOptions.scanOptions().match(USER_KEY_PREFIX + "*").count(1000).build());
            while (cursor.hasNext()) {
                keysTmp.add(new String(cursor.next()));
            }
            return keysTmp;
        });
        redisTemplate.unlink(keys); // 删除

        // 数据库查询数据
        List<UserInfo> users = userDao.getAllUserInfo();
        // 使用 Spring-redis提供的Jackson2HashMapper将对象转Map类，
        // 快速将对象转成Map（具体可参考：https://blog.csdn.net/qq_16159433/article/details/121876883）
        users.forEach(u -> redisTemplate.opsForHash().putAll(
                USER_KEY_PREFIX + u.getId(),
                // 这里使用fastJSON转也可以
                JSONObject.parseObject(JSONObject.toJSONString(u)).getInnerMap()
                // new Jackson2HashMapper(true).toHash(u))
                )
        );
    }
}

为了解决Spring中redisTemplate写数据出现乱码问题，需要重新对redisTemplate重新配置序列化对象

import com.alibaba.fastjson.support.spring.GenericFastJsonRedisSerializer;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.serializer.StringRedisSerializer;

@Configuration
public class RedisTemplateConfig {
    @Bean(value = "redisTemplate")
    public RedisTemplate<String, Object> redisTemplate(RedisConnectionFactory connectionFactory) {
        // 重新配置 RedisTemplate中的序列对象，解决默认序列化对象出现乱码问题
        RedisTemplate<String, Object> redisTemplate = new RedisTemplate<>();
        redisTemplate.setKeySerializer(new StringRedisSerializer());
        // 用的是fastJSON
        redisTemplate.setValueSerializer(new GenericFastJsonRedisSerializer());
        redisTemplate.setHashKeySerializer(new StringRedisSerializer());
        redisTemplate.setHashValueSerializer(new GenericFastJsonRedisSerializer());
        redisTemplate.setConnectionFactory(connectionFactory);
        return redisTemplate;
    }
}