Support Unix Domain Sockets via jnr-unixsocket

JAVA-163

Author: joankaradimov

build.gradle

@@ -20,6 +20,7 @@ apply plugin: 'eclipse'
 apply plugin: 'idea'
 
 def configDir = new File(rootDir, 'config')
+ext.jnrUnixsocketVersion = '0.18'
 ext.nettyVersion = '4.1.17.Final'
 ext.snappyVersion = '1.1.4'
 

driver-core/build.gradle

@@ -29,6 +29,7 @@ clirr {
 dependencies {
     compile project(':bson')
 
+    compile "com.github.jnr:jnr-unixsocket:$jnrUnixsocketVersion", optional
     compile "io.netty:netty-buffer:$nettyVersion", optional
     compile "io.netty:netty-transport:$nettyVersion", optional
     compile "io.netty:netty-handler:$nettyVersion", optional

driver-core/src/main/com/mongodb/ServerAddress.java

@@ -19,9 +19,13 @@
 import com.mongodb.annotations.Immutable;
 import com.mongodb.lang.Nullable;
 
+import jnr.unixsocket.UnixSocketAddress;
+
+import java.io.File;
 import java.io.Serializable;
 import java.net.InetAddress;
 import java.net.InetSocketAddress;
+import java.net.SocketAddress;
 import java.net.UnknownHostException;
 
 /**
@@ -33,6 +37,7 @@ public class ServerAddress implements Serializable {
 
     private final String host;
     private final int port;
+    private final SocketAddress address;
 
     /**
      * Creates a ServerAddress with default host and port
@@ -56,7 +61,7 @@ public ServerAddress(@Nullable final String host) {
      * @param inetAddress host address
      */
     public ServerAddress(final InetAddress inetAddress) {
-        this(inetAddress.getHostName(), defaultPort());
+        this(inetAddress.getHostName(), defaultPort(), new InetSocketAddress(inetAddress, defaultPort()));
     }
 
     /**
@@ -66,7 +71,7 @@ public ServerAddress(final InetAddress inetAddress) {
      * @param port        mongod port
      */
     public ServerAddress(final InetAddress inetAddress, final int port) {
-        this(inetAddress.getHostName(), port);
+        this(inetAddress.getHostName(), port, new InetSocketAddress(inetAddress, port));
     }
 
     /**
@@ -75,7 +80,38 @@ public ServerAddress(final InetAddress inetAddress, final int port) {
      * @param inetSocketAddress inet socket address containing hostname and port
      */
     public ServerAddress(final InetSocketAddress inetSocketAddress) {
-        this(inetSocketAddress.getAddress(), inetSocketAddress.getPort());
+        this(inetSocketAddress.getHostName(), inetSocketAddress.getPort(), inetSocketAddress);
+    }
+
+    /**
+     * Creates a ServerAddress
+     *
+     * @param serverAddress an instance to be shallow-copied
+     */
+    public ServerAddress(final ServerAddress serverAddress) {
+        this(serverAddress.host, serverAddress.port, serverAddress.address);
+    }
+
+    /**
+     * Creates a ServerAddress
+     *
+     * @param path the file used for the Unix domain socket
+     */
+    public ServerAddress(final File path) {
+        this(path.toString(), 0, new UnixSocketAddress(path));
+    }
+
+    /**
+     * Creates a ServerAddress - intended for internal usage
+     *
+     * @param host hostname
+     * @param port mongod port
+     * @param address an instance of socket address or `null`
+     */
+    protected ServerAddress(final String host, final int port, final SocketAddress address) {
+        this.host = host;
+        this.port = port;
+        this.address = address;
     }
 
     /**
@@ -125,6 +161,7 @@ public ServerAddress(@Nullable final String host, final int port) {
         }
         this.host = hostToUse.toLowerCase();
         this.port = portToUse;
+        this.address = null;
     }
 
     @Override
@@ -179,7 +216,10 @@ public int getPort() {
      *
      * @return socket address
      */
-    public InetSocketAddress getSocketAddress() {
+    public SocketAddress getSocketAddress() {
+        if (address != null) {
+            return address;
+        }
         try {
             return new InetSocketAddress(InetAddress.getByName(host), port);
         } catch (UnknownHostException e) {

driver-core/src/main/com/mongodb/connection/ClusterSettings.java

@@ -144,7 +144,7 @@ public Builder hosts(final List<ServerAddress> hosts) {
             Set<ServerAddress> hostsSet = new LinkedHashSet<ServerAddress>(hosts.size());
             for (ServerAddress host : hosts) {
                 notNull("host", host);
-                hostsSet.add(new ServerAddress(host.getHost(), host.getPort()));
+                hostsSet.add(new ServerAddress(host));
             }
             this.hosts = unmodifiableList(new ArrayList<ServerAddress>(hostsSet));
             return this;

driver-core/src/main/com/mongodb/connection/SocketChannelStream.java

@@ -19,6 +19,8 @@
 import com.mongodb.MongoSocketOpenException;
 import com.mongodb.MongoSocketReadException;
 import com.mongodb.ServerAddress;
+import jnr.unixsocket.UnixSocketAddress;
+import jnr.unixsocket.UnixSocketChannel;
 import org.bson.ByteBuf;
 
 import java.io.IOException;
@@ -48,8 +50,12 @@ class SocketChannelStream implements Stream {
     @Override
     public void open() throws IOException {
         try {
-            socketChannel = SocketChannel.open();
-            SocketStreamHelper.initialize(socketChannel.socket(), address, settings, sslSettings);
+            if (address.getSocketAddress() instanceof UnixSocketAddress) {
+                socketChannel = UnixSocketChannel.open((UnixSocketAddress) address.getSocketAddress());
+            } else {
+                socketChannel = SocketChannel.open();
+                SocketStreamHelper.initialize(socketChannel.socket(), address, settings, sslSettings);
+            }
         } catch (IOException e) {
             close();
             throw new MongoSocketOpenException("Exception opening socket", getAddress(), e);

driver-core/src/main/com/mongodb/connection/SocketStreamFactory.java

@@ -24,6 +24,8 @@
 import javax.net.ssl.SSLContext;
 import java.security.NoSuchAlgorithmException;
 
+import jnr.unixsocket.UnixSocketAddress;
+
 import static com.mongodb.assertions.Assertions.notNull;
 
 /**
@@ -63,7 +65,9 @@ public SocketStreamFactory(final SocketSettings settings, final SslSettings sslS
     @Override
     public Stream create(final ServerAddress serverAddress) {
         Stream stream;
-        if (socketFactory != null) {
+        if (serverAddress.getSocketAddress() instanceof UnixSocketAddress) {
+            stream = new SocketChannelStream(serverAddress, settings, sslSettings, bufferProvider);
+        } else if (socketFactory != null) {
             stream = new SocketStream(serverAddress, settings, sslSettings, socketFactory, bufferProvider);
         } else if (sslSettings.isEnabled()) {
             stream = new SocketStream(serverAddress, settings, sslSettings, getSslContext().getSocketFactory(), bufferProvider);

mongo-java-driver/build.gradle

@@ -21,6 +21,7 @@ archivesBaseName = 'mongo-java-driver'
 
 // dependencies copied from driver-core
 dependencies {
+    compile "com.github.jnr:jnr-unixsocket:$jnrUnixsocketVersion", optional
     compile "io.netty:netty-buffer:$nettyVersion", optional
     compile "io.netty:netty-transport:$nettyVersion", optional
     compile "io.netty:netty-handler:$nettyVersion", optional