Package org.teiid.transport

Source Code of org.teiid.transport.ODBCClientInstance

/*
* JBoss, Home of Professional Open Source.
* See the COPYRIGHT.txt file distributed with this work for information
* regarding copyright ownership.  Some portions may be licensed
* to Red Hat, Inc. under one or more contributor license agreements.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
* 02110-1301 USA.
*/
package org.teiid.transport;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.util.Arrays;
import java.util.concurrent.ConcurrentLinkedQueue;

import org.teiid.core.util.ReflectionHelper;
import org.teiid.jdbc.TeiidDriver;
import org.teiid.logging.LogConstants;
import org.teiid.logging.LogManager;
import org.teiid.logging.MessageLevel;
import org.teiid.net.CommunicationException;
import org.teiid.net.socket.ObjectChannel;
import org.teiid.net.socket.ServiceInvocationStruct;
import org.teiid.odbc.ODBCClientRemote;
import org.teiid.odbc.ODBCServerRemote;
import org.teiid.odbc.ODBCServerRemoteImpl;
import org.teiid.transport.PgFrontendProtocol.PGRequest;

public class ODBCClientInstance implements ChannelListener{

  private ODBCClientRemote client;
  private ODBCServerRemoteImpl server;
  private ReflectionHelper serverProxy = new ReflectionHelper(ODBCServerRemote.class);
  private ConcurrentLinkedQueue<PGRequest> messageQueue = new ConcurrentLinkedQueue<PGRequest>();
 
  public ODBCClientInstance(final ObjectChannel channel, ODBCServerRemote.AuthenticationType authType, TeiidDriver driver) {
    this.client = (ODBCClientRemote)Proxy.newProxyInstance(this.getClass().getClassLoader(), new Class[] {ODBCClientRemote.class}, new InvocationHandler() {
      @Override
      public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        if (LogManager.isMessageToBeRecorded(LogConstants.CTX_ODBC, MessageLevel.TRACE)) {
          LogManager.logTrace(LogConstants.CTX_ODBC, "invoking client method:", method.getName(), Arrays.deepToString(args)); //$NON-NLS-1$
        }
        ServiceInvocationStruct message = new ServiceInvocationStruct(args, method.getName(),ODBCServerRemote.class);
        channel.write(message);
        return null;
      }
    });
    this.server = new ODBCServerRemoteImpl(this, authType, driver) {
      @Override
      protected synchronized void doneExecuting() {
        super.doneExecuting();
        while (!server.isExecuting()) {
          PGRequest request = messageQueue.poll();
          if (request == null) {
            break;
          }
              if (!server.isErrorOccurred() || request.struct.methodName.equals("sync")) { //$NON-NLS-1$
                processMessage(request.struct);
              }
        }
      }
    };
  }
 
  public ODBCClientRemote getClient() {
    return client;
  }
 
  @Override
  public void disconnected() {
    server.terminate();
  }

  @Override
  public void exceptionOccurred(Throwable t) {
    server.terminate();
  }
 
  @Override
  public void onConnection() throws CommunicationException {
  }

  @Override
  public void receivedMessage(Object msg) throws CommunicationException {
        if (msg instanceof PGRequest) {
          PGRequest request = (PGRequest)msg;
          synchronized (server) {
            if (server.isExecuting()) {
              //queue until done
              messageQueue.add(request);
              return;
            }
            if (server.isErrorOccurred() && !request.struct.methodName.equals("sync")) { //$NON-NLS-1$
              //discard until sync
              return;
            }
      }
            processMessage(request.struct);
        }
  }

  private void processMessage(ServiceInvocationStruct serviceStruct) {
    try {
      Method m = this.serverProxy.findBestMethodOnTarget(serviceStruct.methodName, serviceStruct.args);
      try {
        // since the postgres protocol can produce more than single response
        // objects to a request, all the methods are designed to return void.
        // and relies on client interface to build the responses.
        m.invoke(this.server, serviceStruct.args);
      } catch (InvocationTargetException e) {
        throw e.getCause();
      }   
    } catch (Throwable e) {
      this.client.errorOccurred(e);
    }
  }

}
TOP

Related Classes of org.teiid.transport.ODBCClientInstance

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.