Subversion Repositories SmartDukaan

package in.shop2020.thrift.clients;

import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;

import org.apache.thrift.TException;
import org.apache.thrift.transport.TTransportException;

import in.shop2020.test.TestService.Client;
import in.shop2020.utils.Logger;


public class TestClient extends GenericServiceClient {
        
        private Client client = null;
        
        public TestClient(String clientIdentifier,
                        String hostConfigKey, String portConfigKey) throws Exception {
                super(clientIdentifier, hostConfigKey, portConfigKey);
                client = new Client(protocol);
                
        }
        
        public TestClient() throws Exception{
                this(TestClient.class.getSimpleName(), "localhost", "9009");
                if(transport.isOpen()){
                        Logger.log("Transport was already open", this);
                }
                try {
                        transport.open();
                        Logger.log("Transport opened", this);
                } catch (TTransportException e) {
                        Logger.log("Encountered exception while open transport "+ e, this);
                }
        }
        
        @PostConstruct
        private void openTransport(){
                if(transport.isOpen()){
                        Logger.log("Transport was already open", this);
                }
                try {
                        transport.open();
                } catch (TTransportException e) {
                        Logger.log("Encountered exception while open transport "+ e, this);
                }
        }
        
        @PreDestroy
        private void closeTransport(){
                if(transport != null && transport.isOpen()){
                        Logger.log("Closing transport :", this);
                        transport.close();
                }
        }
        
        public Client getClient(){
                return client;
        }

        @Override
        public void closeSession() {
                try {
                        client.closeSession();
                } catch (TException e) {
                        e.printStackTrace();
                }
        }
        
}