SocketChannelFactory.java

package water.network;

import water.H2OSecurityManager;

import java.io.IOException;
import java.nio.channels.ByteChannel;
import java.nio.channels.SocketChannel;

/**
 * Creates either a raw or an SSL/TLS wrapped socket depending on
 * the node's configuration. All sockets used in the application should be
 * created using this class.
 */
public class SocketChannelFactory {

    private volatile static SocketChannelFactory INSTANCE;

    private H2OSecurityManager sm;

    private SocketChannelFactory(H2OSecurityManager sm) {
        this.sm = sm;
    }

    public ByteChannel serverChannel(ByteChannel channel) throws IOException {
        if (sm.securityEnabled && !(channel instanceof SSLSocketChannel)) {
            return sm.wrapServerChannel((SocketChannel) channel);
        } else {
            return channel;
        }
    }

    public ByteChannel clientChannel(ByteChannel channel, String host, int port) throws IOException {
        if (sm.securityEnabled && !(channel instanceof SSLSocketChannel)) {
            return sm.wrapClientChannel((SocketChannel) channel, host, port);
        } else {
            return channel;
        }
    }

    public static SocketChannelFactory instance(H2OSecurityManager sm) {
        if (null == INSTANCE) {
            synchronized (SocketChannelFactory.class) {
                if (null == INSTANCE) {
                    INSTANCE = new SocketChannelFactory(sm);
                }
            }
        }
        return INSTANCE;
    }

}