Struct PrimaryKeyIndex 

pub struct PrimaryKeyIndex { /* private fields */ }

Thread-safe primary key deduplication index.

Sync dedup in the hot path: BloomFilter::may_contain(), FxHashSet::contains(), and TextDictReader::ordinal() are all sync.

Interior mutability for the mutable state (bloom + uncommitted set) is behind parking_lot::Mutex. The committed readers are only mutated via &mut self methods (commit/abort path), so no lock is needed for them.

Implementations

impl PrimaryKeyIndex

pub fn new( field: Field, readers: Vec<Arc<SegmentReader>>, snapshot: SegmentSnapshot, ) -> Self

Create a new PrimaryKeyIndex by scanning committed segments.

Iterates each reader’s fast-field text dictionary to populate the bloom filter with all existing primary key values. The snapshot keeps ref counts alive so segments aren’t deleted while we hold readers.

pub fn check_and_insert(&self, doc: &Document) -> Result<()>

Check whether a document’s primary key is unique, and if so, register it.

Returns Ok(()) if the key is new (inserted into bloom + uncommitted set). Returns Err(DuplicatePrimaryKey) if the key already exists. Returns Err(Document) if the primary key field is missing or empty.

pub fn refresh( &mut self, new_readers: Vec<Arc<SegmentReader>>, snapshot: SegmentSnapshot, )

Refresh after commit: replace committed readers and clear uncommitted set.

The new readers include the just-committed segments, so their text dictionaries already contain the previously-uncommitted keys. The snapshot keeps ref counts alive so segments aren’t deleted while we hold readers to them.

pub fn rollback_uncommitted_key(&self, doc: &Document)

Roll back an uncommitted key registration (e.g. when channel send fails after check_and_insert succeeded). Bloom may retain the key but that only causes harmless false positives, never missed duplicates.

pub fn clear_uncommitted(&mut self)

Clear uncommitted keys (e.g. on abort). Bloom may retain stale entries but that only causes harmless false positives (extra committed-segment lookups), never missed duplicates.