ghost-crab 0.10.0

ethereum indexer SDK 👻🦀
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

use crate::indexer::rpc_manager::Provider;
use crate::indexer::templates::TemplateManager;
use crate::latest_block_manager::LatestBlockManager;
use alloy::providers::Provider as AlloyProvider;
use alloy::rpc::types::eth::Block;
use alloy::rpc::types::eth::BlockNumberOrTag;
use alloy::transports::TransportError;
use async_trait::async_trait;
use ghost_crab_common::config::BlockHandlerConfig;
use ghost_crab_common::config::ExecutionMode;
use std::sync::Arc;
use std::time::Duration;

pub struct BlockContext {
    pub provider: Provider,
    pub templates: TemplateManager,
    pub block_number: u64,
}

impl BlockContext {
    pub async fn block(&self, hydrate: bool) -> Result<Option<Block>, TransportError> {
        self.provider
            .get_block_by_number(BlockNumberOrTag::Number(self.block_number), hydrate)
            .await
    }
}

pub type BlockHandlerInstance = Arc<Box<(dyn BlockHandler + Send + Sync)>>;

#[async_trait]
pub trait BlockHandler {
    async fn handle(&self, params: BlockContext);
    fn name(&self) -> String;
}

#[derive(Clone)]
pub struct ProcessBlocksInput {
    pub handler: BlockHandlerInstance,
    pub templates: TemplateManager,
    pub provider: Provider,
    pub config: BlockHandlerConfig,
}

pub async fn process_blocks(
    ProcessBlocksInput { handler, templates, provider, config }: ProcessBlocksInput,
) -> Result<(), TransportError> {
    let execution_mode = config.execution_mode.unwrap_or(ExecutionMode::Parallel);

    let mut current_block = config.start_block;
    let mut latest_block_manager =
        LatestBlockManager::new(provider.clone(), Duration::from_secs(10));

    loop {
        let latest_block = latest_block_manager.get().await?;

        if current_block >= latest_block {
            tokio::time::sleep(tokio::time::Duration::from_secs(5)).await;
            continue;
        }

        match execution_mode {
            ExecutionMode::Parallel => {
                let handler = handler.clone();
                let provider = provider.clone();
                let templates = templates.clone();

                tokio::spawn(async move {
                    handler
                        .handle(BlockContext { provider, templates, block_number: current_block })
                        .await;
                });
            }
            ExecutionMode::Serial => {
                let templates = templates.clone();
                let provider = provider.clone();
                let templates = templates.clone();

                handler
                    .handle(BlockContext { provider, templates, block_number: current_block })
                    .await;
            }
        }

        current_block += config.step;
    }
}