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

use std::sync::Arc;
use tokio::sync::RwLock;
use crate::Config;

pub mod synced_height;
pub mod block;
pub mod tx;
pub mod address_index;
pub mod utxo;
pub mod utxo_server;
pub mod rich_list;

pub use synced_height::SyncedHeightDB;
pub use block::BlockDB;
pub use tx::TxDB;
pub use address_index::AddressIndexDB;
pub use utxo::UtxoDB;
pub use utxo_server::UtxoServer;
pub use rich_list::RichList;

#[derive(Debug, Clone)]
pub struct Database {
    pub coin: String,
    pub config: Config,
    pub synced_height_db: Arc<RwLock<SyncedHeightDB>>,
    pub block_db: Arc<RwLock<BlockDB>>,
    pub tx_db: Arc<RwLock<TxDB>>,
    pub addr_index_db: Arc<RwLock<AddressIndexDB>>,
    pub utxo_server: Arc<RwLock<UtxoServer>>,
    pub rich_list: Arc<RwLock<RichList>>,
}

impl Database {
    pub fn new(coin: &str, config: &Config) -> Self {
        Self {
            coin: coin.to_string(),
            config: (*config).clone(),
            synced_height_db: Arc::new(RwLock::new(SyncedHeightDB::new(coin))),
            block_db        : Arc::new(RwLock::new(BlockDB::new(coin, false))),
            tx_db           : Arc::new(RwLock::new(TxDB::new(coin, false))),
            addr_index_db   : Arc::new(RwLock::new(AddressIndexDB::new(coin, false))),
            utxo_server     : Arc::new(RwLock::new(UtxoServer::new())),
            rich_list       : Arc::new(RwLock::new(RichList::new())),
        }
    }
}