hyperstack-interpreter 0.6.9

AST transformation runtime and VM for HyperStack streaming pipelines
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

use prost_types::Any;
use serde_json::Value;
use std::collections::HashMap;

pub type ProtoDecoder = fn(&[u8]) -> Result<(Value, String), Box<dyn std::error::Error>>;

#[derive(Debug)]
pub struct ProtoRouter {
    decoders: HashMap<String, ProtoDecoder>,
}

impl ProtoRouter {
    pub fn new() -> Self {
        ProtoRouter {
            decoders: HashMap::new(),
        }
    }

    pub fn register(&mut self, type_url: String, decoder: ProtoDecoder) {
        self.decoders.insert(type_url, decoder);
    }

    pub fn decode(&self, any: Any) -> Result<(Value, String), Box<dyn std::error::Error>> {
        let decoder = self
            .decoders
            .get(&any.type_url)
            .ok_or_else(|| format!("No decoder found for type_url: {}", any.type_url))?;

        decoder(&any.value)
    }
}

impl Default for ProtoRouter {
    fn default() -> Self {
        Self::new()
    }
}