From 7098a5f688e71de535654f417ae15dffc7d96189 Mon Sep 17 00:00:00 2001
From: buffalu <85544055+buffalu@users.noreply.github.com>
Date: Sat, 9 Nov 2024 17:02:59 -0600
Subject: [PATCH] Add config to disable account streaming (#55)

Adds a config to disable account streaming
---
 server/src/geyser_grpc_plugin.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/server/src/geyser_grpc_plugin.rs b/server/src/geyser_grpc_plugin.rs
index e18f1f7..426e043 100644
--- a/server/src/geyser_grpc_plugin.rs
+++ b/server/src/geyser_grpc_plugin.rs
@@ -52,6 +52,7 @@ pub struct PluginData {
 
     is_startup_completed: AtomicBool,
     ignore_startup_updates: bool,
+    account_data_notifications_enabled: bool,
 }
 
 #[derive(Default)]
@@ -75,6 +76,7 @@ pub struct PluginConfig {
     pub block_update_buffer_size: usize,
     pub transaction_update_buffer_size: usize,
     pub skip_startup_stream: Option<bool>,
+    pub account_data_notifications_enabled: Option<bool>,
 }
 
 impl GeyserPlugin for GeyserGrpcPlugin {
@@ -161,6 +163,7 @@ impl GeyserPlugin for GeyserGrpcPlugin {
             is_startup_completed: AtomicBool::new(false),
             // don't skip startup to keep backwards compatability
             ignore_startup_updates: config.skip_startup_stream.unwrap_or(false),
+            account_data_notifications_enabled: config.account_data_notifications_enabled.unwrap_or(true),
         });
         info!("plugin data initialized");
 
@@ -482,7 +485,7 @@ impl GeyserPlugin for GeyserGrpcPlugin {
     }
 
     fn account_data_notifications_enabled(&self) -> bool {
-        true
+        self.data.as_ref().map(|d| d.account_data_notifications_enabled).unwrap_or(true)
     }
 
     fn transaction_notifications_enabled(&self) -> bool {