From 586f4cfe50bb771ae5fb16d7df0d0b17c43e7e37 Mon Sep 17 00:00:00 2001
From: theo-s68 <194042188+theo-s68@users.noreply.github.com>
Date: Wed, 22 Oct 2025 11:23:10 +0700
Subject: [PATCH] add disconnect to StreamConnection

---
 .../adapter/stream/StreamConnectionInterface.java         | 2 ++
 .../adapter/stream/StreamConnectionPoolWrapper.java       | 8 ++++++++
 2 files changed, 10 insertions(+)

diff --git a/clients/common/src/main/java/com/binance/connector/client/common/websocket/adapter/stream/StreamConnectionInterface.java b/clients/common/src/main/java/com/binance/connector/client/common/websocket/adapter/stream/StreamConnectionInterface.java
index 984aab88..b87bc7fd 100644
--- a/clients/common/src/main/java/com/binance/connector/client/common/websocket/adapter/stream/StreamConnectionInterface.java
+++ b/clients/common/src/main/java/com/binance/connector/client/common/websocket/adapter/stream/StreamConnectionInterface.java
@@ -16,6 +16,8 @@ Map<String, StreamBlockingQueue<String>> subscribe(
 
     void connect();
 
+    void disconnect();
+
     boolean isConnected();
 
     void setUserAgent(String userAgent);
diff --git a/clients/common/src/main/java/com/binance/connector/client/common/websocket/adapter/stream/StreamConnectionPoolWrapper.java b/clients/common/src/main/java/com/binance/connector/client/common/websocket/adapter/stream/StreamConnectionPoolWrapper.java
index e1543c86..432ca77f 100644
--- a/clients/common/src/main/java/com/binance/connector/client/common/websocket/adapter/stream/StreamConnectionPoolWrapper.java
+++ b/clients/common/src/main/java/com/binance/connector/client/common/websocket/adapter/stream/StreamConnectionPoolWrapper.java
@@ -84,6 +84,14 @@ public void connect() {
         isConnected = true;
     }
 
+    @Override
+    public void disconnect() {
+        for (ConnectionWrapper connectionWrapper : connectionList) {
+            connectionWrapper.disconnect();
+        }
+        isConnected = false;
+    }
+
     public Map<String, StreamBlockingQueue<String>> subscribe(
             RequestWrapperDTO<Set<String>, Object> requestWrapperDTO) {
         Set<String> params = requestWrapperDTO.getParams();