com.twitter.finagle.kestrel.java.ClientBase Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of finagle-kestrel_2.9.2 Show documentation
finagle-kestrel
The newest version!
package com.twitter.finagle.kestrel.java;

import java.util.Iterator;
import java.util.concurrent.Callable;

import scala.Function0;
import scala.Option;
import scala.collection.JavaConversions;
import scala.collection.immutable.Stream;

import org.jboss.netty.buffer.ChannelBuffer;

import com.twitter.concurrent.Offer;
import com.twitter.finagle.kestrel.ReadHandle;
import com.twitter.finagle.kestrel.protocol.Response;
import com.twitter.util.Duration;
import com.twitter.util.Function;
import com.twitter.util.Future;
import com.twitter.util.Time;
import com.twitter.util.Timer;

public class ClientBase extends com.twitter.finagle.kestrel.java.Client {
  com.twitter.finagle.kestrel.Client underlying;

  public ClientBase(com.twitter.finagle.kestrel.Client underlying) {
    this.underlying = underlying;
  }

  public Future get(String key, Duration waitFor) {
    Future> result = underlying.get(key, waitFor);
    return result.map(new Function, ChannelBuffer>() {
      public ChannelBuffer apply(Option value) {
        if (value.isDefined()) {
          return (ChannelBuffer)value.get();
        } else {
          return null;
        }
      }
    });
  }

  public Future set(String key, ChannelBuffer value, Time expiry) {
    return underlying.set(key, value, expiry);
  }

  public Future delete(String key) {
    return underlying.delete(key);
  }

  public Future flush(String key) {
    return underlying.delete(key);
  }

  public Future write(String queueName, Offer offer) {
    return underlying.write(queueName, offer);
  }

  public ReadHandle read(String queueName) {
    return underlying.read(queueName);
  }

  public ReadHandle readReliably(String queueName) {
    return underlying.readReliably(queueName);
  }

  public ReadHandle readReliably(String queueName, Timer timer, Callable> backoffs) {
    final Callable> backoffsFinal = backoffs;
    Function0> backoffsFunction = new com.twitter.util.Function0>() {
      public Stream apply() {
        try {
          return JavaConversions.asScalaIterator(backoffsFinal.call()).toStream();
        } catch (Exception e) {
          return Stream.empty();
        }
      }
    };

    return underlying.readReliably(queueName, timer, backoffsFunction);
  }

  public void close() {
    underlying.close();
  }
}