// This file is part of line_cardinality. Copyright © 2024 line_cardinality contributors.
// line_cardinality is licensed under the GNU GPL v3.0 or any later version. See LICENSE file for full text.

use hashbrown::HashMap;

use bstr::ByteSlice;
use cfg_if::cfg_if;

use line_cardinality::CountUnique;

cfg_if! {
    if #[cfg(feature = "ahash")] {
        use ahash::RandomState;
    } else {
        use std::hash::RandomState;
    }
}

pub struct Processor {
    map: HashMap<String, (), RandomState>,
    count: usize,
}

impl Default for Processor {
    fn default() -> Self {
        Self {
            map: HashMap::with_hasher(super::init_hasher_state()),
            count: 0,
        }
    }
}

impl CountUnique for Processor {
    fn count_line(&mut self, line: &[u8]) {
        let line = line.to_str().unwrap();
        self.map.raw_entry_mut()
            .from_key(line)
            .or_insert_with(|| {
                self.count += 1;
                (line.to_string(), ())
            });
    }

    fn count(&self) -> usize {
        self.count
    }

    fn reset(&mut self) {
        unimplemented!("not used in benches")
    }
}