Add forgotten WebSocketSecureNetworkModule changes

Signed-off-by: James Tomson <[email protected]>
eclipse-paho · May 27, 2022 · e1a8b5c · e1a8b5c
1 parent f6681ea
commit e1a8b5c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/.../java/org/eclipse/paho/client/mqttv3/internal/websocket/WebSocketSecureNetworkModule.java b/.../java/org/eclipse/paho/client/mqttv3/internal/websocket/WebSocketSecureNetworkModule.java
@@ -24,6 +24,7 @@
 import java.util.Properties;
 import javax.net.ssl.SSLSocketFactory;
 
+import org.eclipse.paho.client.mqttv3.IMqttDns;
 import org.eclipse.paho.client.mqttv3.MqttException;
 import org.eclipse.paho.client.mqttv3.internal.SSLNetworkModule;
 import org.eclipse.paho.client.mqttv3.logging.Logger;
@@ -49,8 +50,8 @@ public class WebSocketSecureNetworkModule extends SSLNetworkModule{
 	 */
 	private ByteArrayOutputStream outputStream = new ExtendedByteArrayOutputStream(this);
 
-	public WebSocketSecureNetworkModule(SSLSocketFactory factory, String uri, String host, int port, String clientId, Properties customWebSocketHeaders) {
-		super(factory, host, port, clientId);
+	public WebSocketSecureNetworkModule(SSLSocketFactory factory, String uri, String host, int port, String clientId, IMqttDns dns, Properties customWebSocketHeaders) {
+		super(factory, host, port, clientId, dns);
 		this.uri = uri;
 		this.host = host;
 		this.port = port;

diff --git a/...rg/eclipse/paho/client/mqttv3/internal/websocket/WebSocketSecureNetworkModuleFactory.java b/...rg/eclipse/paho/client/mqttv3/internal/websocket/WebSocketSecureNetworkModuleFactory.java
@@ -64,7 +64,7 @@ public NetworkModule createNetworkModule(URI brokerUri, MqttConnectOptions optio
 
 		// Create the network module...
 		WebSocketSecureNetworkModule netModule = new WebSocketSecureNetworkModule((SSLSocketFactory) factory,
-				brokerUri.toString(), host, port, clientId, options.getCustomWebSocketHeaders());
+				brokerUri.toString(), host, port, clientId, options.getDns(), options.getCustomWebSocketHeaders());
 		netModule.setSSLhandshakeTimeout(options.getConnectionTimeout());
 		netModule.setSSLHostnameVerifier(options.getSSLHostnameVerifier());
 		netModule.setHttpsHostnameVerificationEnabled(options.isHttpsHostnameVerificationEnabled());