use crate::{
json_value::JsonValue,
processor::{Context, Process, ProcessDecision, Result, Titles},
};
pub struct Merger {
data: Vec<JsonValue>,
next: Box<dyn Process>,
titles: Option<Titles>,
}
impl Merger {
pub fn create_process(next: Box<dyn Process>) -> Box<dyn Process> {
Box::new(Merger {
data: Vec::new(),
next,
titles: None,
})
}
}
impl Process for Merger {
fn complete(&mut self) -> Result<()> {
let mut data = Vec::new();
for value in &self.data {
let value = value.clone();
data.push(value);
}
let value = data.into();
let context = Context::new_with_no_context(value);
self.data.clear();
self.next.process(context)?;
Ok(())
}
fn process(&mut self, context: Context) -> Result<ProcessDecision> {
let value = context.build();
self.data.push(value);
Ok(ProcessDecision::Continue)
}
fn start(&mut self, titles_so_far: Titles) -> Result<()> {
self.titles = Some(titles_so_far);
self.next.start(Titles::default())
}
}
#[cfg(test)]
mod tests {
use std::cell::RefCell;
use std::ops::Deref;
use std::rc::Rc;
use super::*;
use crate::json_value::JsonValue;
use crate::processor::{Context, Titles};
#[test]
fn start_will_remove_the_title() -> Result<()> {
struct Next(Rc<RefCell<bool>>);
let data = Rc::new(RefCell::new(false));
let one = Rc::new("one".into());
let two = Rc::new("two".into());
let titles = Titles::default().with_title(&one).with_title(&two);
impl Process for Next {
fn complete(&mut self) -> Result<()> {
Ok(())
}
fn process(&mut self, _: Context) -> Result<ProcessDecision> {
Ok(ProcessDecision::Continue)
}
fn start(&mut self, titles: Titles) -> Result<()> {
assert_eq!(titles.len(), 0);
*self.0.borrow_mut() = true;
Ok(())
}
}
let next = Box::new(Next(data.clone()));
let mut merger = Merger::create_process(next);
merger.start(titles)?;
let binding = data.borrow();
let data = &*binding;
assert_eq!(data, &true);
Ok(())
}
#[test]
fn complete_will_complete_with_the_correct_values() -> Result<()> {
struct Next {
data: Rc<RefCell<Option<JsonValue>>>,
}
let data = Rc::new(RefCell::new(Option::None));
impl Process for Next {
fn complete(&mut self) -> Result<()> {
Ok(())
}
fn process(&mut self, context: Context) -> Result<ProcessDecision> {
let input = context.input().deref().clone();
assert!(self.data.borrow().is_none());
*self.data.borrow_mut() = Some(input);
Ok(ProcessDecision::Continue)
}
fn start(&mut self, _: Titles) -> Result<()> {
Ok(())
}
}
let next = Box::new(Next { data: data.clone() });
let one = Rc::new("one".into());
let two = Rc::new("two".into());
let titles = Titles::default().with_title(&one).with_title(&two);
let mut merger = Merger::create_process(next);
merger.start(titles)?;
let context = Context::new_with_no_context("one".into())
.with_result(&one, Some((1).into()))
.with_result(&two, Some((2).into()));
merger.process(context)?;
let context = Context::new_with_no_context("one".into())
.with_result(&one, Some((4).into()))
.with_result(&two, Some((6).into()));
merger.process(context)?;
let context = Context::new_with_no_context("three".into())
.with_result(&one, Some((3).into()))
.with_result(&two, Some((4).into()));
merger.process(context)?;
merger.complete()?;
let binding = data.borrow();
let data = binding.deref().clone().unwrap();
let data = format!("{data}");
assert_eq!(
data,
r#"[{"one": 1, "two": 2}, {"one": 4, "two": 6}, {"one": 3, "two": 4}]"#
);
Ok(())
}
}