From 94ea57deacaa8d0593b3c0a7ec28bda6e53bb437 Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Sun, 26 Sep 2021 20:11:04 -0400
Subject: [PATCH] close method

---
 src/twitch/websocket_client.rs | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/twitch/websocket_client.rs b/src/twitch/websocket_client.rs
index d4e0f7a4f2..b8ece637eb 100644
--- a/src/twitch/websocket_client.rs
+++ b/src/twitch/websocket_client.rs
@@ -167,6 +167,18 @@ impl TwitchWebsocketClient {
     Ok(())
   }
 
+  pub async fn close(&mut self) -> AnyhowResult<()> {
+    let mut socket = match self.socket {
+      None => return Ok(()), // Already closed.
+      Some(ref mut socket) => socket,
+    };
+
+    socket.close(None).await?;
+    self.socket = None;
+
+    Ok(())
+  }
+
   pub fn is_connected(&self) -> bool {
     // TODO: This probably isn't accurate
     self.socket.is_some()