oura 2.2.0

The tail of Cardano
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

use crossterm::style::{Color, Print, Stylize};
use crossterm::ExecutableCommand;
use gasket::framework::*;
use serde::Deserialize;
use std::io::{stdout, Stdout};

use crate::framework::*;

mod format;
mod throttle;

use format::*;
use throttle::Throttle;

pub struct Worker {
    stdout: Stdout,
    throttle: Throttle,
}

impl Worker {
    fn compute_terminal_width(&self, wrap: bool) -> Option<usize> {
        if !wrap {
            return None;
        }

        if let Ok((x, _y)) = crossterm::terminal::size() {
            return Some(x as usize);
        }

        None
    }
}

#[async_trait::async_trait(?Send)]
impl gasket::framework::Worker<Stage> for Worker {
    async fn bootstrap(stage: &Stage) -> Result<Self, WorkerError> {
        let mut stdout = stdout();

        stdout
            .execute(Print(
                "Oura terminal output started, waiting for chain data\n".with(Color::DarkGrey),
            ))
            .or_panic()?;

        let worker = Self {
            stdout,
            throttle: stage.config.throttle_min_span_millis.into(),
        };

        Ok(worker)
    }

    async fn schedule(
        &mut self,
        stage: &mut Stage,
    ) -> Result<WorkSchedule<ChainEvent>, WorkerError> {
        let msg = stage.input.recv().await.or_panic()?;
        Ok(WorkSchedule::Unit(msg.payload))
    }

    async fn execute(&mut self, unit: &ChainEvent, stage: &mut Stage) -> Result<(), WorkerError> {
        let width = self.compute_terminal_width(stage.config.wrap.unwrap_or_default());

        let point = unit.point().clone();

        let line = match unit {
            ChainEvent::Apply(_, record) => {
                LogLine::handle(record, width, &stage.config.adahandle_policy)
            }
            ChainEvent::Undo(_, record) => {
                LogLine::handle(record, width, &stage.config.adahandle_policy)
            }
            ChainEvent::Reset(point) => LogLine::reset(point.clone()),
        };

        self.throttle.wait_turn();
        self.stdout.execute(Print(line)).or_panic()?;

        stage.latest_block.set(point.slot_or_default() as i64);
        stage.cursor.send(point.into()).await.or_panic()?;

        Ok(())
    }
}

#[derive(Stage)]
#[stage(name = "filter", unit = "ChainEvent", worker = "Worker")]
pub struct Stage {
    config: Config,

    pub cursor: SinkCursorPort,
    pub input: MapperInputPort,

    #[metric]
    latest_block: gasket::metrics::Gauge,

    #[metric]
    ops_count: gasket::metrics::Counter,
}

#[derive(Default, Debug, Deserialize)]
pub struct Config {
    pub throttle_min_span_millis: Option<u64>,
    pub wrap: Option<bool>,
    pub adahandle_policy: Option<String>,
}

impl Config {
    pub fn bootstrapper(self, _: &Context) -> Result<Stage, Error> {
        let stage = Stage {
            config: self,
            ops_count: Default::default(),
            latest_block: Default::default(),
            input: Default::default(),
            cursor: Default::default(),
        };

        Ok(stage)
    }
}