zksync_protobuf_config 0.1.0

Protobuf deserialization for ZKsync configs
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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131

use anyhow::Context as _;
use zksync_config::configs;
use zksync_protobuf::{
    repr::{read_required_repr, ProtoRepr},
    required,
};

use crate::proto::database as proto;

impl proto::MerkleTreeMode {
    fn new(x: &configs::database::MerkleTreeMode) -> Self {
        use configs::database::MerkleTreeMode as From;
        match x {
            From::Full => Self::Full,
            From::Lightweight => Self::Lightweight,
        }
    }

    fn parse(&self) -> configs::database::MerkleTreeMode {
        use configs::database::MerkleTreeMode as To;
        match self {
            Self::Full => To::Full,
            Self::Lightweight => To::Lightweight,
        }
    }
}

impl ProtoRepr for proto::MerkleTree {
    type Type = configs::database::MerkleTreeConfig;
    fn read(&self) -> anyhow::Result<Self::Type> {
        Ok(Self::Type {
            path: required(&self.path).context("path")?.clone(),
            mode: required(&self.mode)
                .and_then(|x| Ok(proto::MerkleTreeMode::try_from(*x)?))
                .context("mode")?
                .parse(),
            multi_get_chunk_size: required(&self.multi_get_chunk_size)
                .and_then(|x| Ok((*x).try_into()?))
                .context("multi_get_chunk_size")?,
            block_cache_size_mb: required(&self.block_cache_size_mb)
                .and_then(|x| Ok((*x).try_into()?))
                .context("block_cache_size_mb")?,
            memtable_capacity_mb: required(&self.memtable_capacity_mb)
                .and_then(|x| Ok((*x).try_into()?))
                .context("memtable_capacity_mb")?,
            stalled_writes_timeout_sec: *required(&self.stalled_writes_timeout_sec)
                .context("stalled_writes_timeout_sec")?,
            max_l1_batches_per_iter: required(&self.max_l1_batches_per_iter)
                .and_then(|x| Ok((*x).try_into()?))
                .context("max_l1_batches_per_iter")?,
        })
    }

    fn build(this: &Self::Type) -> Self {
        Self {
            path: Some(this.path.clone()),
            mode: Some(proto::MerkleTreeMode::new(&this.mode).into()),
            multi_get_chunk_size: Some(this.multi_get_chunk_size.try_into().unwrap()),
            block_cache_size_mb: Some(this.block_cache_size_mb.try_into().unwrap()),
            memtable_capacity_mb: Some(this.memtable_capacity_mb.try_into().unwrap()),
            stalled_writes_timeout_sec: Some(this.stalled_writes_timeout_sec),
            max_l1_batches_per_iter: Some(this.max_l1_batches_per_iter.try_into().unwrap()),
        }
    }
}

impl ProtoRepr for proto::Db {
    type Type = configs::database::DBConfig;

    fn read(&self) -> anyhow::Result<Self::Type> {
        Ok(Self::Type {
            state_keeper_db_path: required(&self.state_keeper_db_path)
                .context("state_keeper_db_path")?
                .clone(),
            merkle_tree: read_required_repr(&self.merkle_tree).context("merkle_tree")?,
            experimental: self
                .experimental
                .as_ref()
                .map(ProtoRepr::read)
                .transpose()
                .context("experimental")?
                .unwrap_or_default(),
        })
    }

    fn build(this: &Self::Type) -> Self {
        Self {
            state_keeper_db_path: Some(this.state_keeper_db_path.clone()),
            merkle_tree: Some(ProtoRepr::build(&this.merkle_tree)),
            experimental: Some(ProtoRepr::build(&this.experimental)),
        }
    }
}

impl ProtoRepr for proto::Postgres {
    type Type = configs::database::PostgresConfig;

    fn read(&self) -> anyhow::Result<Self::Type> {
        let (test_server_url, test_prover_url) = self
            .test
            .as_ref()
            .map(|x| (x.server_url.clone(), x.prover_url.clone()))
            .unwrap_or_default();

        Ok(Self::Type {
            max_connections: self.max_connections,
            max_connections_master: self.max_connections_master,
            acquire_timeout_sec: self.acquire_timeout_sec,
            statement_timeout_sec: self.statement_timeout_sec,
            long_connection_threshold_ms: self.long_connection_threshold_ms,
            slow_query_threshold_ms: self.slow_query_threshold_ms,
            test_server_url,
            test_prover_url,
        })
    }

    fn build(this: &Self::Type) -> Self {
        Self {
            max_connections: this.max_connections,
            max_connections_master: this.max_connections_master,
            acquire_timeout_sec: this.acquire_timeout_sec,
            statement_timeout_sec: this.statement_timeout_sec,
            long_connection_threshold_ms: this.long_connection_threshold_ms,
            slow_query_threshold_ms: this.slow_query_threshold_ms,
            test: Some(proto::TestDatabase {
                server_url: this.test_server_url.clone(),
                prover_url: this.test_prover_url.clone(),
            }),
        }
    }
}