Skip to main content

ergo_runtime/source/implementations/number/
impl.rs

1use std::collections::HashMap;
2
3use crate::common::Value;
4use crate::runtime::ExecutionContext;
5use crate::source::{ParameterValue, SourcePrimitive, SourcePrimitiveManifest};
6
7use super::manifest::number_source_manifest;
8
9pub struct NumberSource {
10    manifest: SourcePrimitiveManifest,
11}
12
13impl NumberSource {
14    pub fn new() -> Self {
15        Self {
16            manifest: number_source_manifest(),
17        }
18    }
19}
20
21impl Default for NumberSource {
22    fn default() -> Self {
23        Self::new()
24    }
25}
26
27impl SourcePrimitive for NumberSource {
28    fn manifest(&self) -> &SourcePrimitiveManifest {
29        &self.manifest
30    }
31
32    fn produce(
33        &self,
34        parameters: &HashMap<String, ParameterValue>,
35        _ctx: &ExecutionContext,
36    ) -> HashMap<String, Value> {
37        let value = parameters
38            .get("value")
39            .and_then(|v| match v {
40                ParameterValue::Number(n) => Some(*n),
41                ParameterValue::Int(i) => Some(*i as f64),
42                _ => None,
43            })
44            .expect("missing required parameter 'value' for number_source");
45
46        HashMap::from([("value".to_string(), Value::Number(value))])
47    }
48}