solana-streamer-sdk 0.3.3

A lightweight Rust library for real-time event streaming from Solana DEX trading programs. Supports PumpFun, PumpSwap, Bonk, and Raydium protocols with Yellowstone gRPC and ShredStream.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102

use std::collections::HashMap;

use prost_types::Timestamp;
use solana_sdk::{instruction::CompiledInstruction, pubkey::Pubkey};
use solana_transaction_status::UiCompiledInstruction;

use crate::streaming::event_parser::common::filter::EventTypeFilter;
use crate::streaming::event_parser::{
    core::traits::{EventParser, GenericEventParseConfig, GenericEventParser, UnifiedEvent},
    EventParserFactory, Protocol,
};

pub struct MutilEventParser {
    inner: GenericEventParser,
}

impl MutilEventParser {
    pub fn new(protocols: Vec<Protocol>, event_type_filter: Option<EventTypeFilter>) -> Self {
        let mut inner = GenericEventParser::new(vec![], vec![]);
        // Configure all event types
        for protocol in protocols {
            let parse = EventParserFactory::create_parser(protocol);

            // Merge inner_instruction_configs, append configurations to existing Vec
            for (key, configs) in parse.inner_instruction_configs() {
                let filtered_configs: Vec<GenericEventParseConfig> = configs.into_iter().filter(|config| {
                    event_type_filter.as_ref().map(|filter| filter.include.contains(&config.event_type)).unwrap_or(true)
                }).collect();
                inner.inner_instruction_configs.entry(key).or_insert_with(Vec::new).extend(filtered_configs);
            }

            // Merge instruction_configs, append configurations to existing Vec
            for (key, configs) in parse.instruction_configs() {
                let filtered_configs: Vec<GenericEventParseConfig> = configs.into_iter().filter(|config| {
                    event_type_filter.as_ref().map(|filter| filter.include.contains(&config.event_type)).unwrap_or(true)
                }).collect();
                inner.instruction_configs.entry(key).or_insert_with(Vec::new).extend(filtered_configs);
            }

            // Append program_ids (this is already appending)
            inner.program_ids.extend(parse.supported_program_ids().clone());
        }
        Self { inner }
    }
}

#[async_trait::async_trait]
impl EventParser for MutilEventParser {
    fn inner_instruction_configs(&self) -> HashMap<&'static str, Vec<GenericEventParseConfig>> {
        self.inner.inner_instruction_configs()
    }
    fn instruction_configs(&self) -> HashMap<Vec<u8>, Vec<GenericEventParseConfig>> {
        self.inner.instruction_configs()
    }
    fn parse_events_from_inner_instruction(
        &self,
        inner_instruction: &UiCompiledInstruction,
        signature: &str,
        slot: u64,
        block_time: Option<Timestamp>,
        program_received_time_ms: i64,
        index: String,
    ) -> Vec<Box<dyn UnifiedEvent>> {
        self.inner.parse_events_from_inner_instruction(
            inner_instruction,
            signature,
            slot,
            block_time,
            program_received_time_ms,
            index,
        )
    }

    fn parse_events_from_instruction(
        &self,
        instruction: &CompiledInstruction,
        accounts: &[Pubkey],
        signature: &str,
        slot: u64,
        block_time: Option<Timestamp>,
        program_received_time_ms: i64,
        index: String,
    ) -> Vec<Box<dyn UnifiedEvent>> {
        self.inner.parse_events_from_instruction(
            instruction,
            accounts,
            signature,
            slot,
            block_time,
            program_received_time_ms,
            index,
        )
    }

    fn should_handle(&self, program_id: &Pubkey) -> bool {
        self.inner.should_handle(program_id)
    }

    fn supported_program_ids(&self) -> Vec<Pubkey> {
        self.inner.supported_program_ids()
    }
}