feat: arisu unsub

hopinc · Oct 2, 2023 · 43fc370 · 43fc370
1 parent 18f9f14
commit 43fc370
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 0 deletions.
diff --git a/src/utils/arisu/mod.rs b/src/utils/arisu/mod.rs
@@ -75,6 +75,22 @@ impl ArisuClient {
     }
 
     #[allow(dead_code)]
+    pub async fn unsubscribe_logs(&self) -> Result<()> {
+        if !self.logs_requested.load(Relaxed) {
+            return Ok(());
+        }
+
+        self.tx.send(json!({
+            "op": OpCode::UnsubscribeLogs,
+        }))?;
+
+        while self.logs_requested.load(Relaxed) {
+            tokio::time::sleep(tokio::time::Duration::from_millis(100)).await;
+        }
+
+        Ok(())
+    }
+
     pub async fn request_metrics(&self) -> Result<()> {
         if self.metrics_requested.load(Relaxed) {
             return Ok(());
@@ -90,6 +106,23 @@ impl ArisuClient {
 
         Ok(())
     }
+
+    #[allow(dead_code)]
+    pub async fn unsubscribe_metrics(&self) -> Result<()> {
+        if !self.metrics_requested.load(Relaxed) {
+            return Ok(());
+        }
+
+        self.tx.send(json!({
+            "op": OpCode::UnsubscribeMetrics,
+        }))?;
+
+        while self.metrics_requested.load(Relaxed) {
+            tokio::time::sleep(tokio::time::Duration::from_millis(100)).await;
+        }
+
+        Ok(())
+    }
 }
 
 impl Stream for ArisuClient {

diff --git a/src/utils/arisu/types.rs b/src/utils/arisu/types.rs
@@ -25,6 +25,8 @@ pub enum OpCode {
     Metrics,
     RequestLogs,
     RequestLogsAck,
+    UnsubscribeLogs,
+    UnsubscribeMetrics,
 }
 
 #[derive(Debug, Clone, PartialEq, Eq)]