Skip to content

Commit

Permalink
Add config to disable account streaming (#55)
Browse files Browse the repository at this point in the history
Adds a config to disable account streaming
  • Loading branch information
buffalu authored Nov 9, 2024
1 parent e4e196f commit f4f5e30
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion server/src/geyser_grpc_plugin.rs
Original file line number Diff line number Diff line change
Expand Up @@ -57,6 +57,7 @@ pub struct PluginData {

is_startup_completed: AtomicBool,
ignore_startup_updates: bool,
account_data_notifications_enabled: bool,
}

#[derive(Default)]
Expand Down Expand Up @@ -99,6 +100,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 PluginConfig {
Expand Down Expand Up @@ -198,6 +200,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");

Expand Down Expand Up @@ -519,7 +522,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 {
Expand Down

0 comments on commit f4f5e30

Please sign in to comment.