TransportConfig to allow defining the address mapper.

Author: matyasberry

transport-parent/transport-api/src/main/java/io/scalecube/cluster/transport/api/TransportConfig.java

@@ -1,6 +1,8 @@
 package io.scalecube.cluster.transport.api;
 
+import io.scalecube.net.Address;
 import java.util.StringJoiner;
+import java.util.function.Function;
 import reactor.core.Exceptions;
 
 public final class TransportConfig implements Cloneable {
@@ -20,6 +22,7 @@ public final class TransportConfig implements Cloneable {
   private MessageCodec messageCodec = MessageCodec.INSTANCE;
   private int maxFrameLength = 2 * 1024 * 1024; // 2 MB
   private TransportFactory transportFactory;
+  private Function<Address, Address> addressMapper = Function.identity();
 
   public TransportConfig() {}
 
@@ -134,6 +137,22 @@ public TransportConfig maxFrameLength(int maxFrameLength) {
     return t;
   }
 
+  /**
+   * Setter for {@code addressMapper}.
+   *
+   * @param addressMapper address mapper
+   * @return new {@code TransportConfig} instance
+   */
+  public TransportConfig addressMapper(Function<Address, Address> addressMapper) {
+    TransportConfig t = clone();
+    t.addressMapper = addressMapper;
+    return t;
+  }
+
+  public Function<Address, Address> addressMapper() {
+    return addressMapper;
+  }
+
   public TransportFactory transportFactory() {
     return transportFactory;
   }
@@ -168,6 +187,7 @@ public String toString() {
         .add("messageCodec=" + messageCodec)
         .add("maxFrameLength=" + maxFrameLength)
         .add("transportFactory=" + transportFactory)
+        .add("addressMapper=" + addressMapper)
         .toString();
   }
 }

transport-parent/transport-netty/src/main/java/io/scalecube/transport/netty/TransportImpl.java

@@ -59,18 +59,7 @@ public final class TransportImpl implements Transport {
   private final Function<Address, Address> addressMapper;
 
   /**
-   * Constructor with cofig as parameter.
-   *
-   * @param messageCodec message codec
-   * @param receiver transport receiver part
-   * @param sender transport sender part
-   */
-  public TransportImpl(MessageCodec messageCodec, Receiver receiver, Sender sender) {
-    this(messageCodec, receiver, sender, Function.identity());
-  }
-
-  /**
-   * Constructor with cofig as parameter.
+   * Constructor with config as parameter.
    *
    * @param messageCodec message codec
    * @param receiver transport receiver part

transport-parent/transport-netty/src/main/java/io/scalecube/transport/netty/tcp/TcpTransportFactory.java

@@ -9,6 +9,10 @@ public final class TcpTransportFactory implements TransportFactory {
 
   @Override
   public Transport createTransport(TransportConfig config) {
-    return new TransportImpl(config.messageCodec(), new TcpReceiver(config), new TcpSender(config));
+    return new TransportImpl(
+        config.messageCodec(),
+        new TcpReceiver(config),
+        new TcpSender(config),
+        config.addressMapper());
   }
 }

transport-parent/transport-netty/src/main/java/io/scalecube/transport/netty/websocket/WebsocketTransportFactory.java

@@ -10,6 +10,9 @@ public final class WebsocketTransportFactory implements TransportFactory {
   @Override
   public Transport createTransport(TransportConfig config) {
     return new TransportImpl(
-        config.messageCodec(), new WebsocketReceiver(config), new WebsocketSender(config));
+        config.messageCodec(),
+        new WebsocketReceiver(config),
+        new WebsocketSender(config),
+        config.addressMapper());
   }
 }