actualtcpconnection.java

High performance DB query

/*
 * @(#)$Id: ActualTCPConnection.java,v 1.4 2004/07/02 23:59:22 huebsch Exp $
 *
 * Copyright (c) 2001-2004 Regents of the University of California.
 * All rights reserved.
 *
 * This file is distributed under the terms in the attached BERKELEY-LICENSE
 * file. If you do not find these files, copies can be found by writing to:
 * Computer Science Division, Database Group, Universite of California,
 * 617 Soda Hall #1776, Berkeley, CA 94720-1776. Attention: Berkeley License
 *
 * Copyright (c) 2003-2004 Intel Corporation. All rights reserved.
 *
 * This file is distributed under the terms in the attached INTEL-LICENSE file.
 * If you do not find these files, copies can be found by writing to:
 * Intel Research Berkeley, 2150 Shattuck Avenue, Suite 1300,
 * Berkeley, CA, 94704.  Attention:  Intel License Inquiry.
 */


package runtime.services.network.tcp;

import java.io.ByteArrayInputStream;
import java.io.ObjectInputStream;
import java.net.InetSocketAddress;
import java.net.Socket;
import java.nio.ByteBuffer;
import java.nio.channels.SocketChannel;
import java.util.LinkedList;
import services.network.tcp.TCPClient;
import services.network.tcp.TCPConnection;

/**
 * Class ActualTCPConnection
 *
 */
public class ActualTCPConnection implements TCPConnection {

    private SocketChannel theChannel;
    private InetSocketAddress theRemoteAddress;
    private TCPClient theClient;
    private Socket theSocket;
    private LinkedList queuedData;
    private ByteArrayInputStream inByteStream;
    private ObjectInputStream inObjectStream;
    private int objectInputBufferSize;
    private boolean error;

    /**
     * Constructor ActualTCPConnection
     *
     * @param theChannel
     * @param theRemoteAddress
     * @param theClient
     * @param objectInputBufferSize
     */
    protected ActualTCPConnection(SocketChannel theChannel,
                                  InetSocketAddress theRemoteAddress,
                                  TCPClient theClient,
                                  int objectInputBufferSize) {
        this.theChannel = theChannel;
        this.theRemoteAddress = theRemoteAddress;
        this.theClient = theClient;
        this.theSocket = theChannel.socket();
        this.queuedData = new LinkedList();
        error = false;

        // Setup input stream handlers
        try {
            byte inBuffer[] = new byte[objectInputBufferSize];

            this.inByteStream = new ByteArrayInputStream(inBuffer);
            this.inObjectStream = new ObjectInputStream(inByteStream);
        } catch (Exception e) {}
    }

    /**
     * Method getRemoteIPAddress
     * @return
     */
    public InetSocketAddress getRemoteSocketAddress() {
        return (InetSocketAddress) theSocket.getRemoteSocketAddress();
    }

    /**
     * Method getLocalSocketAddress
     * @return
     */
    public InetSocketAddress getLocalSocketAddress() {
        return (InetSocketAddress) theSocket.getLocalSocketAddress();
    }

    /**
     * Method isClosed
     * @return
     */
    public boolean isClosed() {
        return theSocket.isClosed();
    }

    /**
     * Method isConnected
     * @return
     */
    public boolean isConnected() {
        return theChannel.isConnected();
    }

    /**
     * Method isConnectionPending
     * @return
     */
    public boolean isConnectionPending() {
        return theChannel.isConnectionPending();
    }

    /**
     * Method isError
     * @return
     */
    public boolean isError() {
        return error;
    }

    /**
     * Method getSocketChannel
     * @return
     */
    protected SocketChannel getSocketChannel() {
        return theChannel;
    }

    /**
     * Method getSocket
     * @return
     */
    protected Socket getSocket() {
        return theSocket;
    }

    /**
     * Method queueData
     *
     * @param data
     */
    protected void queueData(ByteBuffer data) {
        queuedData.add(data);
    }

    /**
     * Method requeueData
     *
     * @param data
     */
    protected void requeueData(ByteBuffer data) {
        queuedData.add(0, data);
    }

    /**
     * Method dequeueData
     * @return
     */
    protected ByteBuffer dequeueData() {
        if (queuedData.isEmpty()) {
            return null;
        }

        return (ByteBuffer) queuedData.removeFirst();
    }

    /**
     * Method getByteStream
     * @return
     */
    protected ByteArrayInputStream getByteStream() {
        return inByteStream;
    }

    /**
     * Method getInputStream
     * @return
     */
    protected ObjectInputStream getInputStream() {
        return inObjectStream;
    }

    /**
     * Method setErrorFlag
     *
     * @param value
     */
    protected void setErrorFlag(boolean value) {
        error = value;
    }

    /**
     * Method getClient
     * @return
     */
    protected TCPClient getClient() {
        return theClient;
    }
}