scray.client.test.FinagleTestClient 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.test;

import java.nio.ByteBuffer;
import java.util.HashMap;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;

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

import scala.runtime.BoxedUnit;
import scray.service.qmodel.thriftjava.ScrayTColumnInfo;
import scray.service.qmodel.thriftjava.ScrayTQuery;
import scray.service.qmodel.thriftjava.ScrayTQueryInfo;
import scray.service.qmodel.thriftjava.ScrayTRow;
import scray.service.qmodel.thriftjava.ScrayTTableInfo;
import scray.service.qmodel.thriftjava.ScrayUUID;
import scray.service.qservice.thriftjava.ScrayStatelessTService;
import scray.service.qservice.thriftjava.ScrayTResultFrame;

public class FinagleTestClient {

	public static void main(String[] args) throws Exception {

		ScrayStatelessTService.ServiceIface client = Thrift.client().newIface(
				"localhost:18181", ScrayStatelessTService.ServiceToClient.class);

		// prepare a query
		List clist = new LinkedList();

		for (int i = 1; i < 3; i++) {
			clist.add(new ScrayTColumnInfo("col" + i));
		}

		ScrayTTableInfo tinfo = new ScrayTTableInfo("foo", "bar", "baz");
	
		
		ScrayTQueryInfo qinfo = new ScrayTQueryInfo("bla", tinfo, clist);
		ScrayTQuery query = new ScrayTQuery(qinfo,
				new HashMap(),
				"SELECT col1, col2 FROM @baz");

		Future quid = client.query(query).onSuccess(
				new Function() {
					@Override
					public BoxedUnit apply(ScrayUUID response) {
						System.out.println("Received response: " + response);
						return null;
					}
				});

		ScrayUUID quidok = Await.result(quid);

		boolean hasNextFrame = true;
		int pagecount = -1;
		int rowcount = 0;

		while (hasNextFrame) {

			pagecount += 1;

			Future frame = client.getResults(quidok,
					pagecount).onSuccess(
					new Function() {
						@Override
						public BoxedUnit apply(ScrayTResultFrame response) {
							System.out
									.println("Received response: " + response);
							return null;
						}
					});

			ScrayTResultFrame frameok = Await.result(frame);

			List rows = frameok.getRows();

			for (Iterator iterator = rows.iterator(); iterator
					.hasNext();) {
				ScrayTRow scrayTRow = (ScrayTRow) iterator.next();
				if (!scrayTRow.isSetColumns()) {
					hasNextFrame = false;
				} else {
					rowcount++;
					System.out.println("Row (" + rowcount + ") : " + scrayTRow);
				}
			}

		}
	}
}