balius-runtime 0.5.2

Runtime for run Headless dApps on UTxO-based blockchains
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

use std::sync::Arc;

use tokio::sync::Mutex;

use crate::{metrics::Metrics, wit::balius::app::ledger as wit};

pub mod mock;
pub mod u5c;

pub use wit::{Host as CustomLedger, LedgerError, TxoRef, Utxo, UtxoPage, UtxoPattern};

#[async_trait::async_trait]
pub trait LedgerProvider {
    async fn read_utxos(
        &mut self,
        refs: Vec<wit::TxoRef>,
    ) -> Result<Vec<wit::Utxo>, wit::LedgerError>;
    async fn search_utxos(
        &mut self,
        pattern: wit::UtxoPattern,
        start: Option<String>,
        max_items: u32,
    ) -> Result<wit::UtxoPage, wit::LedgerError>;
    async fn read_params(&mut self) -> Result<wit::Json, wit::LedgerError>;
}

#[derive(Clone)]
#[allow(clippy::large_enum_variant)]
pub enum Ledger {
    Mock(mock::Ledger),
    U5C(u5c::Ledger),
    Custom(Arc<Mutex<dyn LedgerProvider + Send + Sync>>),
}

impl From<mock::Ledger> for Ledger {
    fn from(ledger: mock::Ledger) -> Self {
        Ledger::Mock(ledger)
    }
}

impl From<u5c::Ledger> for Ledger {
    fn from(ledger: u5c::Ledger) -> Self {
        Ledger::U5C(ledger)
    }
}

#[derive(Clone)]
pub struct LedgerHost {
    worker_id: String,
    ledger: Ledger,
    metrics: Arc<Metrics>,
}
impl LedgerHost {
    pub fn new(worker_id: &str, ledger: &Ledger, metrics: &Arc<Metrics>) -> Self {
        Self {
            worker_id: worker_id.to_string(),
            ledger: ledger.clone(),
            metrics: metrics.clone(),
        }
    }
}

impl wit::Host for LedgerHost {
    async fn read_utxos(
        &mut self,
        refs: Vec<wit::TxoRef>,
    ) -> Result<Vec<wit::Utxo>, wit::LedgerError> {
        self.metrics.ledger_read_utxos(&self.worker_id);
        match &mut self.ledger {
            Ledger::Mock(ledger) => ledger.read_utxos(refs).await,
            Ledger::U5C(ledger) => ledger.read_utxos(refs).await,
            Ledger::Custom(ledger) => {
                let mut lock = ledger.lock().await;
                lock.read_utxos(refs).await
            }
        }
    }

    async fn search_utxos(
        &mut self,
        pattern: wit::UtxoPattern,
        start: Option<String>,
        max_items: u32,
    ) -> Result<wit::UtxoPage, wit::LedgerError> {
        self.metrics.ledger_search_utxos(&self.worker_id);
        match &mut self.ledger {
            Ledger::Mock(ledger) => ledger.search_utxos(pattern, start, max_items).await,
            Ledger::U5C(ledger) => ledger.search_utxos(pattern, start, max_items).await,
            Ledger::Custom(ledger) => {
                let mut lock = ledger.lock().await;
                lock.search_utxos(pattern, start, max_items).await
            }
        }
    }

    async fn read_params(&mut self) -> Result<wit::Json, wit::LedgerError> {
        self.metrics.ledger_read_params(&self.worker_id);
        match &mut self.ledger {
            Ledger::Mock(ledger) => ledger.read_params().await,
            Ledger::U5C(ledger) => ledger.read_params().await,
            Ledger::Custom(ledger) => {
                let mut lock = ledger.lock().await;
                lock.read_params().await
            }
        }
    }
}