聊聊jedis连接池的预热

@Component
@Slf4j
public class JedisWarmUp implements CommandLineRunner {

    @Autowired
    JedisConnectionFactory jedisConnectionFactory;

    @Override
    public void run(String... args) throws Exception {
        GenericObjectPoolConfig poolConfig = jedisConnectionFactory.getPoolConfig();
        if (poolConfig == null) {
            return ;
        }

        try {
            DirectFieldAccessor accessor = new DirectFieldAccessor(jedisConnectionFactory);
            JedisPool jedisPool = (JedisPool) accessor.getPropertyValue("pool");
            if (jedisPool == null) {
                log.info("jedisConnectionFactory.pool is null");
                return;
            }

            int warmUpSize = poolConfig.getMaxIdle();
            warmUp(jedisPool, warmUpSize);

        } catch (Exception e) {
            log.error("warmup jedisPool error:{}", e.getMessage());
        }
    }

    private void warmUp(JedisPool jedisPool, int warmUpSize) {
        List<Jedis> warmUpList = new ArrayList<>(warmUpSize);
        for (int i=0; i< warmUpSize; i++) {
            Jedis jedis = jedisPool.getResource();
            jedis.ping();
            warmUpList.add(jedis);
        }
        warmUpList.stream().forEach(jedis -> jedis.close());
    }
}