TimeLimitedFutureTask.java

package org.robotninjas.util;

import com.google.common.util.concurrent.AbstractFuture;
import com.google.common.util.concurrent.TimeLimiter;

import java.util.concurrent.Callable;
import java.util.concurrent.RunnableFuture;
import java.util.concurrent.TimeUnit;

public class TimeLimitedFutureTask<V> extends AbstractFuture<V> implements RunnableFuture<V> {

  final TimeLimiter limiter;
  final TimeUnit unit;
  final long duration;
  final Callable<V> op;

  public TimeLimitedFutureTask(TimeLimiter limiter, TimeUnit unit, long duration, Callable<V> op) {
    this.limiter = limiter;
    this.unit = unit;
    this.duration = duration;
    this.op = op;
  }

  @Override
  public void run() {
    try {
      final V result = limiter.callWithTimeout(op, duration, unit, true);
      set(result);
    } catch (Exception e) {
      setException(e);
    }
  }

}