1 
mod block_subscription;
2 
mod grpc_stream;
3 

            

        

            

                4
            

            

                
            

            

                
use anyhow::Result;

            

        

            

                5
            

            

                
            

            

                
pub use {block_subscription::RpcIngester, grpc_stream::GrpcIngester};

            

        

            

                6
            

            

                
            

            

                

            

        

            

                7
            

            

                
            

            

                
use {

            

        

            

                8
            

            

                
            

            

                
    crate::types::BonsolInstruction, solana_sdk::pubkey::Pubkey,

            

        

            

                9
            

            

                
            

            

                
    tokio::sync::mpsc::UnboundedReceiver,

            

        

            

                10
            

            

                
            

            

                
};

            

        

            

                11
            

            

                
            

            

                

            

        

            

                12
            

            

                
            

            

                
pub type TxChannel = UnboundedReceiver<Vec<BonsolInstruction>>;

            

        

            

                13
            

            

                
            

            

                

            

        

            

                14
            

            

                
            

            

                
#[allow(dead_code)]

            

        

            

                15
            

            

                
            

            

                
#[derive(Debug, thiserror::Error)]

            

        

            

                16
            

            

                
            

            

                
pub enum IngestErrorType {

            

        

            

                17
            

            

                
            

            

                
    #[error("RPC Error")]

            

        

            

                18
            

            

                
            

            

                
    RpcError,

            

        

            

                19
            

            

                
            

            

                
    #[error("I/O Error")]

            

        

            

                20
            

            

                
            

            

                
    IoError,

            

        

            

                21
            

            

                
            

            

                
}

            

        

            

                22
            

            

                
            

            

                

            

        

            

                23
            

            

                
            

            

                
#[derive(Debug, thiserror::Error)]

            

        

            

                24
            

            

                
            

            

                
#[error("IngestError: {code} - {message}")]

            

        

            

                25
            

            

                
            

            

                
pub struct IngestError {

            

        

            

                26
            

            

                
            

            

                
    pub code: IngestErrorType,

            

        

            

                27
            

            

                
            

            

                
    pub message: String,

            

        

            

                28
            

            

                
            

            

                
}

            

        

            

                29
            

            

                
            

            

                

            

        

            

                30
            

            

                
            

            

                
pub type IngesterResult = Result<(), IngestError>;

            

        

            

                31
            

            

                
            

            

                
pub trait Ingester {

            

        

            

                32
            

            

                
            

            

                
    fn start(&mut self, program: Pubkey) -> Result<TxChannel>;

            

        

            

                33
            

            

                
            

            

                

            

        

            

                34
            

            

                
            

            

                
    fn stop(&mut self) -> Result<()>;

            

        

            

                35
            

            

                
            

            

                
}

            

        

            

                36
            

            

                
            

            

                

            

        

            

                37
            

            

                
            

            

                
#[cfg(test)]

            

        

            

                38
            

            

                
            

            

                
mod test {

            

        

            

                39
            

            

                
            

            

                

            

        

            

                40
            

            

                
            

            

                
    #[test]

            

        

            

                41
            

            

                1
            

            

                
    fn test_ingest_error_type_display() {

            

        

            

                42
            

            

                1
            

            

                
        let rpc_error = super::IngestErrorType::RpcError;

            

        

            

                43
            

            

                1
            

            

                
        let io_error = super::IngestErrorType::IoError;

            

        

            

                44
            

            

                1
            

            

                

            

        

            

                45
            

            

                1
            

            

                
        assert_eq!(rpc_error.to_string(), "RPC Error");

            

        

            

                46
            

            

                1
            

            

                
        assert_eq!(io_error.to_string(), "I/O Error");

            

        

            

                47
            

            

                1
            

            

                
    }

            

        

            

                48
            

            

                
            

            

                

            

        

            

                49
            

            

                
            

            

                
    #[test]

            

        

            

                50
            

            

                1
            

            

                
    fn test_ingest_error_display() {

            

        

            

                51
            

            

                1
            

            

                
        let rpc_error = super::IngestError {

            

        

            

                52
            

            

                1
            

            

                
            code: super::IngestErrorType::RpcError,

            

        

            

                53
            

            

                1
            

            

                
            message: "RPC failed".to_string(),

            

        

            

                54
            

            

                1
            

            

                
        };

            

        

            

                55
            

            

                1
            

            

                

            

        

            

                56
            

            

                1
            

            

                
        let io_error = super::IngestError {

            

        

            

                57
            

            

                1
            

            

                
            code: super::IngestErrorType::IoError,

            

        

            

                58
            

            

                1
            

            

                
            message: "I/O failed".to_string(),

            

        

            

                59
            

            

                1
            

            

                
        };

            

        

            

                60
            

            

                1
            

            

                

            

        

            

                61
            

            

                1
            

            

                
        assert_eq!(rpc_error.to_string(), "IngestError: RPC Error - RPC failed");

            

        

            

                62
            

            

                1
            

            

                
        assert_eq!(io_error.to_string(), "IngestError: I/O Error - I/O failed");

            

        

            

                63
            

            

                1
            

            

                
    }

            

        

            

                64
            

            

                
            

            

                
}