/*
* Copyright 2014 CyberVision, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
/**
*
*/
package org.kaaproject.kaa.server.common.thrift.util;
import org.apache.thrift.TException;
import org.apache.thrift.TServiceClient;
import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.protocol.TMultiplexedProtocol;
import org.apache.thrift.protocol.TProtocol;
import org.apache.thrift.transport.TSocket;
import org.apache.thrift.transport.TTransport;
import org.kaaproject.kaa.server.common.thrift.KaaThriftService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.lang.reflect.Constructor;
import java.lang.reflect.InvocationTargetException;
// TODO: Auto-generated Javadoc
/**
* The Class ThriftClient.
*
* @param <T> the generic type
* @author Andrey Panasenko apanasenko@cybervisiontech.com
*/
public class ThriftClient<T extends TServiceClient> implements Runnable {
/**
* The Constant LOG.
*/
private static final Logger LOG = LoggerFactory
.getLogger(ThriftClient.class);
/**
* The endpoint host.
*/
private String endpointHost;
/**
* The endpoint port.
*/
private int endpointPort;
/**
* The t class.
*/
private Class<T> classT;
/**
* The t constructor.
*/
private Constructor<T> constructorT;
/**
* The client.
*/
private T client;
/**
* The transport.
*/
private TTransport transport;
/**
* The activity.
*/
private ThriftActivity<T> activity;
/**
* The Constructor.
*
* @param endpointHost the endpoint host
* @param endpointPort the endpoint port
* @param kaaThriftService the kaa thrift service
* @param clazz the clazz
* @throws NoSuchMethodException the no such method exception
* @throws SecurityException the security exception
* @throws InstantiationException the instantiation exception
* @throws IllegalAccessException the illegal access exception
* @throws IllegalArgumentException the illegal argument exception
* @throws InvocationTargetException the invocation target exception
*/
public ThriftClient(String endpointHost, int endpointPort, KaaThriftService kaaThriftService,
Class<T> clazz)
throws NoSuchMethodException,
InstantiationException,
IllegalAccessException,
InvocationTargetException {
this.classT = clazz;
this.endpointHost = endpointHost;
this.endpointPort = endpointPort;
constructorT = classT.getConstructor(TProtocol.class, TProtocol.class);
transport = new TSocket(endpointHost, endpointPort);
LOG.debug("ThriftClient sokcet to " + endpointHost + ":" + endpointPort + " created.");
TProtocol protocol = new TBinaryProtocol(transport);
TMultiplexedProtocol mp = new TMultiplexedProtocol(protocol, kaaThriftService.getServiceName());
client = constructorT.newInstance(mp, mp);
LOG.debug("ThriftClient new Client to " + endpointHost + ":" + endpointPort + " created.");
}
/**
* Sets the thrift activity.
*
* @param activity the thrift activity
*/
public void setThriftActivity(ThriftActivity<T> activity) {
this.activity = activity;
}
/* (non-Javadoc)
* @see java.lang.Runnable#run()
*/
@Override
public void run() {
try {
LOG.debug("Call Thrift server ["
+ endpointHost + ":" + endpointPort + "]");
transport.open();
if (activity != null) {
invoke(activity);
LOG.debug("Successfuly invoke Thrift server ["
+ endpointHost + ":" + endpointPort + "]");
activity.isSuccess(true);
} else {
LOG.error(
"Error - Activity not set while invoke thrift object "
+ endpointHost + ":" + endpointPort);
}
} catch (TException
| IllegalArgumentException
| SecurityException ex) {
LOG.error(
"Unexpected error occurred while invoke thrift object "
+ endpointHost + ":" + endpointPort,
ex);
if (activity != null) {
activity.isSuccess(false);
}
} finally {
transport.close();
}
}
/**
* Gets the client.
*
* @return the client
*/
public T getClient() {
return client;
}
/**
* Invoke.
*
* @param activity the activity
*/
public void invoke(ThriftActivity<T> activity) {
activity.doInTemplate(client);
}
}