Skip to main content

ergo_runtime/compute/implementations/const_number/
impl.rs

1use std::collections::HashMap;
2
3use crate::common::Value;
4use crate::compute::{ComputeError, ComputePrimitive, ComputePrimitiveManifest, PrimitiveState};
5
6use super::manifest::const_number_manifest;
7
8pub struct ConstNumber {
9    manifest: ComputePrimitiveManifest,
10}
11
12impl ConstNumber {
13    pub fn new() -> Self {
14        Self {
15            manifest: const_number_manifest(),
16        }
17    }
18}
19
20impl Default for ConstNumber {
21    fn default() -> Self {
22        Self::new()
23    }
24}
25
26impl ComputePrimitive for ConstNumber {
27    fn manifest(&self) -> &ComputePrimitiveManifest {
28        &self.manifest
29    }
30
31    fn compute(
32        &self,
33        _inputs: &HashMap<String, Value>,
34        parameters: &HashMap<String, Value>,
35        _state: Option<&mut PrimitiveState>,
36    ) -> Result<HashMap<String, Value>, ComputeError> {
37        let value = parameters
38            .get("value")
39            .and_then(|v| v.as_number())
40            .expect("missing required parameter 'value' for const_number");
41
42        Ok(HashMap::from([("value".to_string(), Value::Number(value))]))
43    }
44}