package eu.dnetlib.openaire.thrift;

import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.protocol.TProtocol;
import org.apache.thrift.transport.TSocket;
import org.apache.thrift.transport.TTransport;
import org.apache.thrift.transport.TTransportException;

import eu.dnetlib.openaire.thrift.OpenAIREConnector;

public class ThriftClient {

	public OpenAIREConnector.Client getConnector() throws TTransportException {
		return getConnector("localhost", 7911);
	}

	public OpenAIREConnector.Client getConnector(String host, int port) throws TTransportException {
		TTransport transport = new TSocket(host, port);
		transport.open();

		TProtocol protocol = new  TBinaryProtocol(transport);
		
		return new OpenAIREConnector.Client(protocol);
	}
}