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

//! A filter that maps records by calling custom WASM plugins

use gasket::framework::*;
use serde::Deserialize;

use crate::framework::*;

#[derive(Stage)]
#[stage(name = "filter-wasm", unit = "ChainEvent", worker = "Worker")]
pub struct Stage {
    pub input: FilterInputPort,
    pub output: FilterOutputPort,

    plugin: extism::Plugin,

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

impl Stage {
    fn map_record(&mut self, r: Record) -> Result<Vec<Record>, Error> {
        let extism::convert::Json::<serde_json::Value>(output) = match r {
            Record::CborBlock(x) => self.plugin.call("map_cbor_block", x).unwrap(),
            Record::CborTx(x) => self.plugin.call("map_cbor_tx", x).unwrap(),
            Record::ParsedTx(x) => self
                .plugin
                .call("map_u5c_tx", extism::convert::Json(x))
                .unwrap(),
            Record::ParsedBlock(x) => self
                .plugin
                .call("map_u5c_block", extism::convert::Json(x))
                .unwrap(),
            Record::GenericJson(x) => self
                .plugin
                .call("map_json", extism::convert::Json(x))
                .unwrap(),
            Record::OuraV1Event(x) => self
                .plugin
                .call("map_json", extism::convert::Json(x))
                .unwrap(),
        };

        let output = match output {
            serde_json::Value::Null => vec![],
            serde_json::Value::Array(x) => x.into_iter().map(Record::GenericJson).collect(),
            x => vec![Record::GenericJson(x)],
        };

        Ok(output)
    }
}

#[derive(Default)]
pub struct Worker;

impl From<&Stage> for Worker {
    fn from(_: &Stage) -> Self {
        Self
    }
}

#[async_trait::async_trait(?Send)]
impl gasket::framework::Worker<Stage> for Worker {
    async fn bootstrap(_: &Stage) -> Result<Self, WorkerError> {
        Ok(Default::default())
    }

    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 output = unit
            .clone()
            .try_map_record_to_many(|x| stage.map_record(x))
            .or_panic()?;

        for unit in output {
            stage.output.send(unit.clone().into()).await.or_panic()?;
            stage.ops_count.inc(1);
        }

        Ok(())
    }
}

#[derive(Default, Deserialize)]
pub struct Config {
    pub path: String,
}

impl Config {
    pub fn bootstrapper(self, _ctx: &Context) -> Result<Stage, Error> {
        let wasm = extism::Wasm::file(self.path);
        let manifest = extism::Manifest::new([wasm]);
        let plugin = extism::Plugin::new(&manifest, [], true).map_err(Error::custom)?;

        Ok(Stage {
            input: Default::default(),
            output: Default::default(),
            ops_count: Default::default(),
            plugin,
        })
    }
}