scray.client.finagle.ScrayStatelessTServiceAdapter 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

The newest version!

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.ScrayStatelessTService;
import scray.service.qservice.thriftjava.ScrayStatelessTService.ServiceIface;
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 ScrayStatelessTServiceAdapter implements ScrayTServiceAdapter {

	// adapter keeps paging state
	private int pageIndex = 0;

	protected ScrayStatelessTService.ServiceIface client;
	private String endpoint;

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

	public ScrayStatelessTServiceAdapter(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,
					pageIndex);
			ScrayTResultFrame frame = Await.result(fframe, Duration.fromSeconds(queryTimeout));
			pageIndex += 1;
			return frame;
		} catch (Exception e) {
			throw new SQLException(e);
		}
	}
}