PooledRedisClient

public class PooledRedisClient {

    protected final Logger logger = LoggerFactory.getLogger(getClass());

    private static final int DEFAULT_RETRY_TIME = 3;

    private static final int MIN_RETRY_TIME = 0;

    private static final int MAX_RETRY_TIME = 5;

    private static final long REDIS_EXECUTE_WARN_TIME = 50;

    @Setter
    private JedisPool pool;

    public void run(Consumer<Jedis> command) {

        execute(jedis -> {
            command.accept(jedis);
            return null;
        });
    }

    public void runWithRetry(Consumer<Jedis> command) {
        runWithRetry(command, DEFAULT_RETRY_TIME);
    }

    public void runWithRetry(Consumer<Jedis> command, int retries) {
        executeWithRetry(jedis -> {
            command.accept(jedis);
            return null;
        }, retries);
    }

    public <T> T execute(Function<Jedis, T> command) {
        long startTime = System.currentTimeMillis();
        Jedis jedis = null;
        try {
            jedis = this.pool.getResource();
            return command.apply(jedis);
        } finally {
            if (jedis != null) {
                jedis.close();
            }
            long endTime = System.currentTimeMillis();
            long useTime = endTime - startTime;
            if (useTime > REDIS_EXECUTE_WARN_TIME) {
                logger.warn("pooled redis client execute time warn. useTime:[{}],command:[{}]", useTime, command);
            }
        }
    }

    public <T> T executeWithRetry(Function<Jedis, T> command) {
        return executeWithRetry(command, DEFAULT_RETRY_TIME);
    }

    public <T> T executeWithRetry(Function<Jedis, T> command, int retryTime) {
        if (retryTime <= MIN_RETRY_TIME || retryTime > MAX_RETRY_TIME) {
            retryTime = DEFAULT_RETRY_TIME;
        }
        int failTime = 0;

        T result = null;
        while (failTime < retryTime) {
            long startTime = System.currentTimeMillis();
            Jedis jedis = null;
            try {
                jedis = this.pool.getResource();
                logger.debug("pooled redis client execute start.command:[{}]", command);
                result = command.apply(jedis);
                break;
            } catch (RuntimeException e) {
                failTime++;
                if (failTime >= retryTime) {
                    throw e;
                }
            } finally {
                if (jedis != null) {
                    jedis.close();
                }
                long endTime = System.currentTimeMillis();
                long useTime = endTime - startTime;
                if (useTime > REDIS_EXECUTE_WARN_TIME) {
                    logger.warn("pooled redis client execute time warn. useTime:[{}],command:[{}]", useTime, command);
                }
            }
        }
        return result;
    }

    public JedisPool getPool() {
        return this.pool;
    }

}