Jedis with ARM

ClosableJedis.java

public class CloseableJedis implements AutoCloseable {
  private final Jedis delegate;
  private final JedisPool pool;
  private boolean broken = false;

  public CloseableJedis(JedisPool pool) {
    this.pool = pool;
    delegate = pool.getResource();
  }

  public void setBroken() {
    broken = true;
  }

  public Jedis get() {
    return delegate;
  }

  @Override
  public void close() throws JedisException {
    if (broken) {
      pool.returnBrokenResource(delegate);
    } else {
      pool.returnResource(delegate);
    }
  }
}

Utils.Java

public class Utils {
  public static <T> T withJedis(JedisPool pool, final Function<Jedis, T> func) {
    try (CloseableJedis connection = new CloseableJedis(pool)) {
      return new Function<CloseableJedis, T>() {
        @Override
        public T apply(CloseableJedis input) {
          try {
            return func.apply(input.get());
          } catch (JedisConnectionException e) {
            input.setBroken();
            throw e;
          }
        }
      }.apply(connection);
    }
  }
}