kingfisher-bin 2.8.0

MongoDB's blazingly fast and accurate secret scanning and validation tool
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

use std::{cell::RefCell, sync::Arc};

use kingfisher_vectorscan::{BlockDatabase, BlockScanner};
use thread_local::ThreadLocal; // external crate ✔

pub struct ScannerPool {
    vsdb: Arc<BlockDatabase>, // keep DB alive
    pool: ThreadLocal<RefCell<BlockScanner<'static>>>,
}

impl ScannerPool {
    pub fn new(vsdb: Arc<BlockDatabase>) -> Self {
        // Initially empty; each Rayon worker builds its own scanner on first use
        Self { vsdb, pool: ThreadLocal::new() }
    }

    #[inline]
    pub fn with<F, R>(&self, f: F) -> R
    where
        F: FnOnce(&mut BlockScanner<'_>) -> R,
    {
        // get_or creates exactly one scratch arena for *this* thread
        let cell = self.pool.get_or(|| {
            RefCell::new(
                BlockScanner::new(unsafe { &*(self.vsdb.as_ref() as *const _) })
                    .expect("Vectorscan scratch alloc"),
            )
        });
        f(&mut cell.borrow_mut())
    }
}