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

use serde::{Deserialize, Serialize};
use std::num::NonZeroU64;
use url::Url;

use crate::ColumnMapping;

#[derive(Default, Debug, Clone, Deserialize, Serialize)]
pub struct ClientConfig {
    pub url: Option<Url>,
    pub bearer_token: Option<String>,
    pub http_req_timeout_millis: Option<NonZeroU64>,
    pub max_num_retries: Option<usize>,
    pub retry_backoff_ms: Option<u64>,
    pub retry_base_ms: Option<u64>,
    pub retry_ceiling_ms: Option<u64>,
}

#[derive(Default, Debug, Clone, Serialize, Deserialize)]
pub struct StreamConfig {
    pub column_mapping: Option<ColumnMapping>,
    pub event_signature: Option<String>,
    #[serde(default)]
    pub hex_output: HexOutput,
    // Initial batch size. Size would be adjusted based on response size during execution.
    pub batch_size: Option<u64>,
    // Maximum batch size that could be used during dynamic adjustment.
    pub max_batch_size: Option<u64>,
    // Minimum batch size that could be used during dynamic adjustment.
    pub min_batch_size: Option<u64>,
    pub concurrency: Option<usize>,
    pub max_num_blocks: Option<usize>,
    pub max_num_transactions: Option<usize>,
    pub max_num_logs: Option<usize>,
    pub max_num_traces: Option<usize>,
    // Size of a response in bytes from which step size will be lowered
    pub response_bytes_ceiling: Option<u64>,
    // Size of a response in bytes from which step size will be increased
    pub response_bytes_floor: Option<u64>,
}

#[derive(Clone, Copy, Debug, Serialize, Deserialize)]
pub enum HexOutput {
    NoEncode,
    Prefixed,
    NonPrefixed,
}

impl Default for HexOutput {
    fn default() -> Self {
        Self::NoEncode
    }
}