use reifydb_abi::operator::capabilities::OperatorCapability;
use reifydb_core::{
interface::{catalog::flow::FlowNodeId, change::Change},
value::column::columns::Columns,
};
use reifydb_rql::expression::Expression;
use reifydb_value::Result;
use crate::{Operator, operator::OperatorCell, transaction::FlowTransaction};
pub struct ExtendOperator {
parent: OperatorCell,
node: FlowNodeId,
#[allow(dead_code)]
expressions: Vec<Expression>,
}
impl ExtendOperator {
pub fn new(parent: OperatorCell, node: FlowNodeId, expressions: Vec<Expression>) -> Self {
Self {
parent,
node,
expressions,
}
}
pub(crate) fn output_schema(&self) -> Option<Columns> {
self.parent.output_schema()
}
}
impl Operator for ExtendOperator {
fn id(&self) -> FlowNodeId {
self.node
}
fn capabilities(&self) -> &[OperatorCapability] {
OperatorCapability::STANDARD
}
fn apply(&self, _txn: &mut FlowTransaction, change: Change) -> Result<Change> {
Ok(Change::from_flow(self.node, change.version, change.diffs, change.changed_at))
}
}