scray.client.finagle.ScrayCombinedStatefulTServiceAdapter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scray-client-jdbc Show documentation

scray java code, which can be used from java and scala

There is a newer version: 1.1.2

package scray.client.finagle;

import java.sql.SQLException;

import scray.service.qmodel.thriftjava.ScrayTQuery;
import scray.service.qmodel.thriftjava.ScrayUUID;
import scray.service.qservice.thriftjava.ScrayCombinedStatefulTService;
import scray.service.qservice.thriftjava.ScrayTResultFrame;

import com.twitter.finagle.Thrift;
import com.twitter.util.Await;
import com.twitter.util.Duration;
import com.twitter.util.Future;

public class ScrayCombinedStatefulTServiceAdapter implements ScrayTServiceAdapter {

	private ScrayCombinedStatefulTService.FutureIface client;
	private String endpoint;

	public ScrayCombinedStatefulTService.FutureIface getClient() {
		// lazy init
		if (client == null) {
			client = Thrift.newIface(endpoint,
					ScrayCombinedStatefulTService.FutureIface.class);
		}
		return client;
	}

	public ScrayCombinedStatefulTServiceAdapter(String endpoint) {
		this.endpoint = endpoint;
	}

	public ScrayUUID query(ScrayTQuery query, int queryTimeout)
			throws SQLException {
		try {
			Future fuuid = getClient().query(query);
			return Await.result(fuuid, Duration.fromSeconds(queryTimeout));
		} catch (Exception e) {
			throw new SQLException(e);
		}
	}

	public ScrayTResultFrame getResults(ScrayUUID queryId, int queryTimeout)
			throws SQLException {
		try {
			Future fframe = getClient().getResults(queryId);
			ScrayTResultFrame frame = Await.result(fframe,
					Duration.fromSeconds(queryTimeout));
			return frame;
		} catch (Exception e) {
			throw new SQLException(e);
		}
	}
}