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

use crate::{
    csv_headers::HeadersParsed,
    csv_parse_result::{CsvByteRecordWithHash, CsvLeftRightParseResult},
    diff_result::{DiffByteRecordFirstRow, DiffByteRecordsIterator},
};
use crossbeam_channel::{Receiver, Sender};

pub struct CsvHashReceiverStreamComparer {
    receiver: Receiver<CsvLeftRightParseResult<CsvByteRecordWithHash>>,
    sender_csv_records_recycle: Sender<csv::ByteRecord>,
    headers: HeadersParsed,
}

impl CsvHashReceiverStreamComparer {
    pub(crate) fn new(
        receiver: Receiver<CsvLeftRightParseResult<CsvByteRecordWithHash>>,
        sender_csv_records_recycle: Sender<csv::ByteRecord>,
    ) -> Self {
        Self {
            receiver,
            sender_csv_records_recycle,
            headers: Default::default(),
        }
    }
    pub fn recv_hashes_and_compare(self) -> DiffByteRecordsIterator {
        DiffByteRecordFirstRow::new(self.receiver, self.sender_csv_records_recycle, self.headers)
            .into_diff_byte_record_iter()
    }
    pub(crate) fn headers(mut self, headers: HeadersParsed) -> Self {
        self.headers = headers;
        self
    }
}