JedisPoolHealthCheck.java

package com.github.anastasop.dropwizard.jedis;

import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;

import com.yammer.metrics.core.HealthCheck;

public class JedisPoolHealthCheck extends HealthCheck {
	private JedisPool jedisPool;
	
	public JedisPoolHealthCheck(String name, JedisPool jedisPool) {
		super(name);
		this.jedisPool = jedisPool;
	}

	@Override
	protected Result check() throws Exception {
		Jedis jedis = jedisPool.getResource();
		try {
			String reply = jedis.ping();
			jedisPool.returnResource(jedis);
			if (!reply.equalsIgnoreCase("PONG")) {
				return Result.unhealthy("ping response is not pong");
			}
		} catch(Exception e) {
			jedisPool.returnBrokenResource(jedis);
			return Result.unhealthy(e.getMessage());
		}
		return Result.healthy();
	}
}