reifydb-sub-flow 0.5.0

Flow subsystem for stream processing and data flows
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

// SPDX-License-Identifier: Apache-2.0
// Copyright (c) 2025 ReifyDB

use reifydb_abi::operator::capabilities::CAPABILITY_ALL_STANDARD;
use reifydb_core::{
	interface::{
		catalog::{flow::FlowNodeId, series::Series},
		change::Change,
	},
	value::column::{ColumnWithName, buffer::ColumnBuffer, columns::Columns},
};
use reifydb_type::{
	Result,
	fragment::Fragment,
	value::{row_number::RowNumber, r#type::Type},
};

use crate::{Operator, transaction::FlowTransaction};

pub struct PrimitiveSeriesOperator {
	node: FlowNodeId,
	series: Series,
}

impl PrimitiveSeriesOperator {
	pub fn new(node: FlowNodeId, series: Series) -> Self {
		Self {
			node,
			series,
		}
	}
}

impl Operator for PrimitiveSeriesOperator {
	fn id(&self) -> FlowNodeId {
		self.node
	}

	fn capabilities(&self) -> u32 {
		CAPABILITY_ALL_STANDARD
	}

	fn apply(&self, _txn: &mut FlowTransaction, change: Change) -> Result<Change> {
		Ok(Change::from_flow(self.node, change.version, change.diffs, change.changed_at))
	}

	fn pull(&self, _txn: &mut FlowTransaction, rows: &[RowNumber]) -> Result<Columns> {
		if rows.is_empty() {
			return Ok(self.empty_columns());
		}

		Ok(self.empty_columns())
	}
}

impl PrimitiveSeriesOperator {
	fn empty_columns(&self) -> Columns {
		let mut columns = Vec::with_capacity(1 + self.series.columns.len());

		columns.push(ColumnWithName {
			name: Fragment::internal("timestamp"),
			data: ColumnBuffer::with_capacity(Type::Int8, 0),
		});

		for col in &self.series.columns {
			columns.push(ColumnWithName {
				name: Fragment::internal(&col.name),
				data: ColumnBuffer::with_capacity(col.constraint.get_type(), 0),
			});
		}

		Columns::new(columns)
	}
}