net.gdface.thrift.ClientFactory Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of common-thrifty Show documentation
Show all versions of common-thrifty Show documentation
common tools for Thrift/Swift
// ______________________________________________________
// Generated by sql2java - https://github.com/10km/sql2java-2-6-7 (custom branch)
// modified by guyadong from
// sql2java original version https://sourceforge.net/projects/sql2java/
// JDBC driver used at code generation time: com.mysql.jdbc.Driver
// template: client.factory.vm
// ______________________________________________________
package net.gdface.thrift;
import java.lang.reflect.Field;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Executor;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
import com.google.common.base.Function;
import com.google.common.base.Throwables;
import com.google.common.net.HostAndPort;
import com.google.common.util.concurrent.FutureCallback;
import com.google.common.util.concurrent.ListenableFuture;
import com.google.common.util.concurrent.MoreExecutors;
import com.microsoft.thrifty.protocol.BinaryProtocol;
import com.microsoft.thrifty.protocol.Protocol;
import com.microsoft.thrifty.service.AsyncClientBase;
import com.microsoft.thrifty.transport.SocketTransport;
import static com.google.common.net.HostAndPort.fromParts;
import static com.google.common.net.HostAndPort.fromString;
import static com.google.common.base.Preconditions.*;
/**
* Factory class for creating client instance of IFaceLog
* Example:
*
* // get a asynchronous instance
* IFaceLogClientAsync client = ClientFactory.builder()
* .setHostAndPort("127.0.0.1",9090)
* .setTimeout(10,TimeUnit.SECONDS)
* .build();
*
* @author guyadong
*
*/
public class ClientFactory {
private HostAndPort hostAndPort;
/** the timeout value to be used for socket reading in milliseconds */
private long readTimeout = 1000;
/** the timeout value to be used for socket connection in milliseconds */
private long connectTimeout = 500;
private Executor executor = MoreExecutors.directExecutor();
/**
* 接口实例代理函数对象,
* 如果提供了该函数对象(不为null),则在创建接口实例时调用该函数，
* 将thrift/swift创建的接口实例转为代理实例,
*/
private Function decorator = null;
protected ClientFactory() {
}
/**
* set all timeout arguments
* @param time
* @param unit
* @return
* @see #setConnectTimeout(long time,TimeUnit unit)
* @see #setReadTimeout(long time,TimeUnit unit)
*/
public ClientFactory setTimeout(long time,TimeUnit unit){
setConnectTimeout(time,unit);
setReadTimeout(time,unit);
return this;
}
public ClientFactory setConnectTimeout(long connectTimeout,TimeUnit unit) {
this.connectTimeout = unit.toMillis(connectTimeout);
return this;
}
public ClientFactory setReadTimeout(long readTimeout,TimeUnit unit) {
this.readTimeout = unit.toMillis(readTimeout);
return this;
}
public ClientFactory setHostAndPort(HostAndPort hostAndPort) {
if(null == this.hostAndPort){
synchronized(this){
if(null == this.hostAndPort){
this.hostAndPort = checkNotNull(hostAndPort,"hostAndPort must not be null");
}
}
}else{
throw new IllegalStateException("the memeber hostAndPort be initialized always");
}
return this;
}
public ClientFactory setHostAndPort(String host,int port) {
return setHostAndPort(fromParts(host, port));
}
public ClientFactory setHostAndPort(String host) {
return setHostAndPort(fromString(host));
}
public synchronized ClientFactory setExecutor(Executor executor) {
this.executor = checkNotNull(executor,"executor is null");
return this;
}
public Executor getExecutor() {
return executor;
}
public HostAndPort getHostAndPort(){
return checkNotNull(this.hostAndPort,"hostAndPort is null");
}
@SuppressWarnings("unchecked")
public ClientFactory setDecorator(Function decorator) {
this.decorator = (Function) decorator;
return this;
}
/**
* @param stubClass
* @param closeListener
* @return instance of {@link net.gdface.facelog.client.thrift.IFaceLogClient}
*/
public T applyInstance(Class stubClass,AsyncClientBase.Listener closeListener) {
try {
SocketTransport transport =
new SocketTransport.Builder(hostAndPort.getHost(),hostAndPort.getPort())
.connectTimeout((int) connectTimeout)
.readTimeout((int) readTimeout).build();
transport.connect();
Protocol protocol = new BinaryProtocol(transport);
/* force set private field 'strictWrite' to true */
Field field = BinaryProtocol.class.getDeclaredField("strictWrite");
field.setAccessible(true);
field.set(protocol, true);
return (T)stubClass.getConstructor(Protocol.class,AsyncClientBase.Listener.class).newInstance(protocol,closeListener);
} catch (Exception e) {
Throwables.throwIfUnchecked(e);
throw new RuntimeException(e);
}
}
public static ClientFactory builder() {
return new ClientFactory();
}
/**
* 构造{@code interfaceClass}实例
* thriftyImplClass或interfaceClass为null时,假设destClass为thrifty 异步实例类型
* @param 接口类
* @param 返回实例类型
* @param 基于thrifty实现接口类的实例类型
* @param interfaceClass
* @param thriftyImplClass
* @param destClass 返回的实例类型，如果interfaceClass和thriftyImplClass为null,必须有参数为{@link ClientFactory}的构造函数
* 否则必须有参数类型为interfaceClass的构造函数
* @return 返回 {@code destClass }实例
*/
@SuppressWarnings("unchecked")
public O build(Class interfaceClass, Class thriftyImplClass,Class destClass){
try {
checkNotNull(destClass,"destClass is null");
if(interfaceClass == null || thriftyImplClass == null){
// destClass 为异步模式实例
return destClass.getDeclaredConstructor(ClientFactory.class).newInstance(this);
}
T instance =thriftyImplClass.getDeclaredConstructor(ClientFactory.class).newInstance(this);
if(decorator !=null){
instance = (T) decorator.apply(instance);
}
return destClass.getDeclaredConstructor(interfaceClass).newInstance(instance);
} catch (Exception e) {
Throwables.throwIfUnchecked(e);
throw new RuntimeException(e);
}
}
/**
* 构造{@code interfaceClass}实例
* {@link #build(Class, Class, Class)}的简化版本，当thriftImplClass只实现了一个接口时，自动推断接口类型
* NOTE从1.1.17版本以后，该方法第一个参数为thrift实现的服务接口的类，不再是服务接口类
* @param thriftyImplClass
* @param destClass
* @return
* @see #build(Class, Class, Class)
*/
@SuppressWarnings("unchecked")
public O build(Class thriftyImplClass,Class destClass){
checkArgument(thriftyImplClass != null);
checkArgument(!thriftyImplClass.isInterface(),
"%s must not be a interface,must a class implemented service interface",thriftyImplClass.getName());
checkArgument(thriftyImplClass.getInterfaces().length ==1,
"can't determines interface class from %s",thriftyImplClass.getName());
Class interfaceClass = (Class) thriftyImplClass.getInterfaces()[0];
return build(interfaceClass,thriftyImplClass,destClass);
}
/**
* 创建异步实例
* @param destClass 返回的实例类型，必须有参数为{@link ClientFactory}的构造函数
* @return
* @see #build(Class, Class, Class)
*/
public O buildAsync(Class destClass){
return build(null,null,destClass);
}
/**
* 测试thrifty服务连接
* {@code timeoutMills}>0时设置连接超时参数
* @param host 主机名
* @param port 端口号
* @param timeoutMills 指定连接超时(毫秒),<=0使用默认值
* @return 连接成功返回{@code true},否则返回{@code false}
*/
public static final boolean testConnect(String host,int port,long timeoutMills){
ClientFactory clientFactory = ClientFactory.builder().setHostAndPort(host,port);
if(timeoutMills > 0){
clientFactory.setTimeout(timeoutMills, TimeUnit.MILLISECONDS);
}
return clientFactory.testConnect();
}
/**
* 测试指定连接是否有效
* @return 连接有效返回{@code true},否则返回{@code false}
*/
public boolean testConnect(){
try {
SocketTransport transport =
new SocketTransport.Builder(hostAndPort.getHost(),hostAndPort.getPort())
.connectTimeout((int) connectTimeout)
.readTimeout((int) readTimeout).build();
transport.connect();
transport.close();
return true;
} catch (Exception e) {
}
return false;
}
public void addCallback(
final ListenableFuture future,
final FutureCallback callback) {
ThriftUtils.addCallback(future, callback, getExecutor());
}
@Override
public String toString() {
StringBuilder builder = new StringBuilder();
builder.append("ClientFactory [hostAndPort=");
builder.append(hostAndPort);
builder.append(", readTimeout=");
builder.append(readTimeout);
builder.append(", connectTimeout=");
builder.append(connectTimeout);
builder.append(", executor=");
builder.append(executor);
builder.append("]");
return builder.toString();
}
public class ListenableFutureDecorator implements ListenableFuture{
private final ListenableFuture future;
public ListenableFutureDecorator(ListenableFuture future) {
this.future = checkNotNull(future,"future is null");
}
@Override
public boolean cancel(boolean mayInterruptIfRunning) {
return future.cancel(mayInterruptIfRunning);
}
@Override
public V get() throws InterruptedException, ExecutionException {
return future.get();
}
@Override
public V get(long timeout, TimeUnit unit) throws InterruptedException, ExecutionException, TimeoutException {
return future.get(timeout, unit);
}
@Override
public boolean isCancelled() {
return future.isCancelled();
}
@Override
public boolean isDone() {
return future.isDone();
}
@Override
public void addListener(Runnable listener, Executor executor) {
future.addListener(listener, executor);
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy