diff --git a/src/main/java/net/openhft/chronicle/map/WiredChronicleMapStatelessClientBuilder.java b/src/main/java/net/openhft/chronicle/map/WiredChronicleMapStatelessClientBuilder.java index cb800e3e1..1b9f4db51 100644 --- a/src/main/java/net/openhft/chronicle/map/WiredChronicleMapStatelessClientBuilder.java +++ b/src/main/java/net/openhft/chronicle/map/WiredChronicleMapStatelessClientBuilder.java @@ -40,14 +40,6 @@ public final class WiredChronicleMapStatelessClientBuilder implements private byte localIdentifier; private short channelID; - public WiredStatelessClientTcpConnectionHub hub() { - return hub; - } - - public void hub(WiredStatelessClientTcpConnectionHub hub) { - this.hub = hub; - } - public WiredChronicleMapStatelessClientBuilder(InetSocketAddress remoteAddress, Class keyClass, Class valueClass, short channelID) { this.keyClass = keyClass; this.valueClass = valueClass; @@ -155,4 +147,16 @@ public final class WiredChronicleMapStatelessClientBuilder implements public short channelID() { return channelID; } + + + + public WiredStatelessClientTcpConnectionHub hub() { + return hub; + } + + public WiredChronicleMapStatelessClientBuilder hub(WiredStatelessClientTcpConnectionHub hub) { + this.hub = hub; + return this; + } + } diff --git a/src/test/java/net/openhft/chronicle/wire/TestWire.java b/src/test/java/net/openhft/chronicle/wire/TestWire.java index b16c79e03..2b2f74ea8 100644 --- a/src/test/java/net/openhft/chronicle/wire/TestWire.java +++ b/src/test/java/net/openhft/chronicle/wire/TestWire.java @@ -77,11 +77,13 @@ public class TestWire { ((WiredStatelessChronicleMap) statelessMap).createChannel((short) 2); - ChronicleMap map2 = new WiredChronicleMapStatelessClientBuilder( + WiredChronicleMapStatelessClientBuilder localhost = new WiredChronicleMapStatelessClientBuilder<>( new InetSocketAddress("localhost", SERVER_PORT), String.class, String.class, - (short) 2).create(); + (short) 2).hub(builder.hub()); + + ChronicleMap map2 = localhost.create(); map2.put("hello", "world");