org.jgroups.util.DefaultSocketFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jgroups
Reliable cluster communication toolkit
There is a newer version: 5.3.13.Final
package org.jgroups.util;

import java.io.IOException;
import java.net.DatagramSocket;
import java.net.InetAddress;
import java.net.MulticastSocket;
import java.net.ServerSocket;
import java.net.Socket;
import java.net.SocketAddress;
import java.net.SocketException;
import java.util.function.Consumer;

import javax.net.ServerSocketFactory;
import javax.net.ssl.SSLContext;

/**
 * Default implementation, ignores service names
 * @author Bela Ban
 */
public class DefaultSocketFactory implements SocketFactory {
    private final javax.net.SocketFactory socketFactory;
    private final ServerSocketFactory serverSocketFactory;
    private Consumer socketConfigurator = s -> {};
    private Consumer serverSocketConfigurator = s -> {};

    public DefaultSocketFactory() {
        this(javax.net.SocketFactory.getDefault(), ServerSocketFactory.getDefault());
    }

    public DefaultSocketFactory(SSLContext sslContext) {
        this.socketFactory = sslContext.getSocketFactory();
        this.serverSocketFactory = sslContext.getServerSocketFactory();
    }

    public DefaultSocketFactory(javax.net.SocketFactory socketFactory, ServerSocketFactory serverSocketFactory) {
        this.socketFactory = socketFactory;
        this.serverSocketFactory = serverSocketFactory;
    }

    public void setSocketConfigurator(Consumer socketConfigurator) {
        this.socketConfigurator = socketConfigurator;
    }

    public void setServerSocketConfigurator(Consumer serverSocketConfigurator) {
        this.serverSocketConfigurator = serverSocketConfigurator;
    }

    private Socket configureSocket(Socket socket) {
        socketConfigurator.accept(socket);
        return socket;
    }

    private ServerSocket configureSocket(ServerSocket socket) {
        serverSocketConfigurator.accept(socket);
        return socket;
    }

    @Override
    public Socket createSocket(String s) throws IOException {
        return configureSocket(socketFactory.createSocket());
    }

    @Override
    public Socket createSocket(String s, String host, int port) throws IOException {
        return configureSocket(socketFactory.createSocket(host, port));
    }

    @Override
    public Socket createSocket(String s, InetAddress host, int port) throws IOException {
        return configureSocket(socketFactory.createSocket(host, port));
    }

    @Override
    public Socket createSocket(String s, String host, int port, InetAddress localHost, int localPort) throws IOException {
        return configureSocket(socketFactory.createSocket(host, port, localHost, localPort));
    }

    @Override
    public Socket createSocket(String s, InetAddress host, int port, InetAddress localHost, int localPort) throws IOException {
        return configureSocket(socketFactory.createSocket(host, port, localHost, localPort));
    }

    @Override
    public ServerSocket createServerSocket(String s) throws IOException {
        return configureSocket(serverSocketFactory.createServerSocket());
    }

    @Override
    public ServerSocket createServerSocket(String s, int port) throws IOException {
        return configureSocket(serverSocketFactory.createServerSocket(port));
    }

    @Override
    public ServerSocket createServerSocket(String s, int port, int backlog) throws IOException {
        return configureSocket(serverSocketFactory.createServerSocket(port, backlog));
    }

    @Override
    public ServerSocket createServerSocket(String s, int port, int backlog, InetAddress bindAddress) throws IOException {
        return configureSocket(serverSocketFactory.createServerSocket(port, backlog, bindAddress));
    }

    public DatagramSocket createDatagramSocket(String service_name) throws SocketException {
        return new DatagramSocket();
    }

    public DatagramSocket createDatagramSocket(String service_name, SocketAddress bindaddr) throws SocketException {
        return new DatagramSocket(bindaddr);
    }

    public DatagramSocket createDatagramSocket(String service_name, int port) throws SocketException {
        return new DatagramSocket(port);
    }

    public DatagramSocket createDatagramSocket(String service_name, int port, InetAddress laddr) throws SocketException {
        return new DatagramSocket(port, laddr);
    }

    public MulticastSocket createMulticastSocket(String service_name) throws IOException {
        return new MulticastSocket();
    }

    public MulticastSocket createMulticastSocket(String service_name, int port) throws IOException {
        return new MulticastSocket(port);
    }

    public MulticastSocket createMulticastSocket(String service_name, SocketAddress bindaddr) throws IOException {
        return new MulticastSocket(bindaddr);
    }

    @Override
    public void close(Socket socket) throws IOException {
        Util.close(socket);
    }

    @Override
    public void close(ServerSocket serverSocket) throws IOException {
        Util.close(serverSocket);
    }

    @Override
    public void close(DatagramSocket datagramSocket) {
        Util.close(datagramSocket);
    }
}