feat: add close method & implement drop

sleeyax · Jul 5, 2023 · 49075b0 · 49075b0
1 parent faf31c2
commit 49075b0
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -99,9 +99,9 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
                 });
 
                 tokio::spawn(async move {
-                    socket
-                        .emit("discordUser", RpcClient::default().get_user())
-                        .ok();
+                    let rpc_client = RpcClient::default();
+                    socket.emit("discordUser", rpc_client.get_user()).ok();
+                    drop(rpc_client);
                 });
             }
         })

diff --git a/src/rpc_client.rs b/src/rpc_client.rs
@@ -158,10 +158,21 @@ impl RpcClient {
         info!("client {}: get user", &self.client.get_client_id());
         &self.user
     }
+
+    pub fn close(&mut self) -> bool {
+        info!("client {}: close", &self.client.get_client_id());
+        self.client.disconnect().is_ok()
+    }
 }
 
 impl Default for RpcClient {
     fn default() -> Self {
         Self::new(DEFAULT_CLIENT_ID.into())
     }
 }
+
+impl Drop for RpcClient {
+    fn drop(&mut self) {
+        self.close();
+    }
+}