com.mirth.connect.connectors.tcp.StateAwareSocket.java Source code

Introduction

Here is the source code for com.mirth.connect.connectors.tcp.StateAwareSocket.java

Source

/*
 * Copyright (c) Mirth Corporation. All rights reserved.
 * 
 * http://www.mirthcorp.com
 * 
 * The software in this package is published under the terms of the MPL license a copy of which has
 * been included with this distribution in the LICENSE.txt file.
 */

package com.mirth.connect.connectors.tcp;

import java.io.BufferedInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.InetAddress;
import java.net.Socket;
import java.net.SocketException;
import java.net.SocketImpl;

import org.apache.commons.lang3.StringUtils;

public class StateAwareSocket extends Socket implements StateAwareSocketInterface {

    protected BufferedInputStream bis = null;

    public StateAwareSocket() {
        super();
    }

    public StateAwareSocket(String host, int port) throws Exception {
        super(host, port);
    }

    public StateAwareSocket(InetAddress address, int port) throws Exception {
        super(address, port);
    }

    public StateAwareSocket(SocketImpl impl) throws SocketException {
        super(impl);
    }

    @Override
    public InputStream getInputStream() throws IOException {
        if (bis == null) {
            bis = new BufferedInputStream(super.getInputStream());
        }
        return bis;
    }

    /**
     * The only (portable) way in Java to detect that the remote host has closed the connection is
     * to attempt to read from the connection and see if you get -1. We use the mark() and reset()
     * feature of BufferedInputStream to nondestructively peek into the stream to check for this.
     * Warning: since we've started consuming data, anyone reading from this socket must now use our
     * BIS and not create their own from getInputStream().
     * 
     * @return true if the remote end has closed its side of this socket
     */
    @Override
    public boolean remoteSideHasClosed() throws IOException {
        if (isClosed()) {
            return true;
        }
        int oldTimeout;
        try {
            oldTimeout = getSoTimeout();
        } catch (IOException e) {
            if (StringUtils.containsIgnoreCase(e.getMessage(), "Socket Closed")) {
                return true;
            }
            throw e;
        }
        setSoTimeout(100);
        getInputStream().mark(1);
        try {
            return bis.read() == -1;
        } catch (IOException e) {
            return false;
        } finally {
            try {
                bis.reset();
            } catch (IOException e) {
            }
            try {
                setSoTimeout(oldTimeout);
            } catch (SocketException e) {
            }
        }
    }
}