feat: optimize metadata indexing with roaring bitmaps for 90% memory reduction

Replace JavaScript Sets with hardware-accelerated RoaringBitmap32 for metadata indexes.

Key improvements:
- 1.4x average speedup, up to 3.3x on 10K entities
- 90% memory reduction (40 bytes/UUID → 4 bytes/int)
- Hardware-accelerated multi-field intersection via SIMD (AVX2/SSE4.2)
- EntityIdMapper for bidirectional UUID ↔ integer mapping
- Portable serialization format

Benchmark results (1,000 queries):
- 10K entities: 3.74ms → 1.14ms (3.3x faster, 90% memory savings)
- 100K entities: 2.60ms → 1.78ms (1.5x faster, 88% memory savings)

Implementation:
- Add EntityIdMapper class for UUID/int mapping with persistence
- Modify ChunkData to use Map<string, RoaringBitmap32>
- Add getIdsForMultipleFields() for fast bitmap intersection
- Include comprehensive tests (25 tests passing)
- Add performance benchmark comparing Set vs Roaring

Technical details:
- roaring@2.4.0 dependency
- Maintains backward compatibility
- All queries still return UUID strings
- Automatic persistence via storage adapter
This commit is contained in:
David Snelling 2025-10-13 16:39:06 -07:00
parent 84b657ac47
commit 2f6ab9559a
9 changed files with 2647 additions and 110 deletions

207
src/utils/entityIdMapper.ts Normal file
View file

@ -0,0 +1,207 @@
/**
* EntityIdMapper - Bidirectional mapping between UUID strings and integer IDs for roaring bitmaps
*
* Roaring bitmaps require 32-bit unsigned integers, but Brainy uses UUID strings as entity IDs.
* This class provides efficient bidirectional mapping with persistence support.
*
* Features:
* - O(1) lookup in both directions
* - Persistent storage via storage adapter
* - Atomic counter for next ID
* - Serialization/deserialization support
*
* @module utils/entityIdMapper
*/
import type { BaseStorage } from '../storage/baseStorage'
export interface EntityIdMapperOptions {
storage: BaseStorage
storageKey?: string
}
export interface EntityIdMapperData {
nextId: number
uuidToInt: Record<string, number>
intToUuid: Record<number, string>
}
/**
* Maps entity UUIDs to integer IDs for use with Roaring Bitmaps
*/
export class EntityIdMapper {
private storage: BaseStorage
private storageKey: string
// Bidirectional maps
private uuidToInt = new Map<string, number>()
private intToUuid = new Map<number, string>()
// Atomic counter for next ID
private nextId = 1
// Dirty flag for persistence
private dirty = false
constructor(options: EntityIdMapperOptions) {
this.storage = options.storage
this.storageKey = options.storageKey || 'brainy:entityIdMapper'
}
/**
* Initialize the mapper by loading from storage
*/
async init(): Promise<void> {
try {
const data = await this.storage.getMetadata(this.storageKey) as EntityIdMapperData | null
if (data) {
this.nextId = data.nextId
// Rebuild maps from serialized data
this.uuidToInt = new Map(Object.entries(data.uuidToInt).map(([k, v]) => [k, Number(v)]))
this.intToUuid = new Map(Object.entries(data.intToUuid).map(([k, v]) => [Number(k), v]))
}
} catch (error) {
// First time initialization - maps are empty, nextId = 1
}
}
/**
* Get integer ID for UUID, assigning a new ID if not exists
*/
getOrAssign(uuid: string): number {
const existing = this.uuidToInt.get(uuid)
if (existing !== undefined) {
return existing
}
// Assign new ID
const newId = this.nextId++
this.uuidToInt.set(uuid, newId)
this.intToUuid.set(newId, uuid)
this.dirty = true
return newId
}
/**
* Get UUID for integer ID
*/
getUuid(intId: number): string | undefined {
return this.intToUuid.get(intId)
}
/**
* Get integer ID for UUID (without assigning if not exists)
*/
getInt(uuid: string): number | undefined {
return this.uuidToInt.get(uuid)
}
/**
* Check if UUID has been assigned an integer ID
*/
has(uuid: string): boolean {
return this.uuidToInt.has(uuid)
}
/**
* Remove mapping for UUID
*/
remove(uuid: string): boolean {
const intId = this.uuidToInt.get(uuid)
if (intId === undefined) {
return false
}
this.uuidToInt.delete(uuid)
this.intToUuid.delete(intId)
this.dirty = true
return true
}
/**
* Get total number of mappings
*/
get size(): number {
return this.uuidToInt.size
}
/**
* Convert array of UUIDs to array of integers
*/
uuidsToInts(uuids: string[]): number[] {
return uuids.map(uuid => this.getOrAssign(uuid))
}
/**
* Convert array of integers to array of UUIDs
*/
intsToUuids(ints: number[]): string[] {
const result: string[] = []
for (const intId of ints) {
const uuid = this.intToUuid.get(intId)
if (uuid) {
result.push(uuid)
}
}
return result
}
/**
* Convert iterable of integers to array of UUIDs (for roaring bitmap iteration)
*/
intsIterableToUuids(ints: Iterable<number>): string[] {
const result: string[] = []
for (const intId of ints) {
const uuid = this.intToUuid.get(intId)
if (uuid) {
result.push(uuid)
}
}
return result
}
/**
* Flush mappings to storage
*/
async flush(): Promise<void> {
if (!this.dirty) {
return
}
// Convert maps to plain objects for serialization
const data: EntityIdMapperData = {
nextId: this.nextId,
uuidToInt: Object.fromEntries(this.uuidToInt),
intToUuid: Object.fromEntries(this.intToUuid)
}
await this.storage.saveMetadata(this.storageKey, data)
this.dirty = false
}
/**
* Clear all mappings
*/
async clear(): Promise<void> {
this.uuidToInt.clear()
this.intToUuid.clear()
this.nextId = 1
this.dirty = true
await this.flush()
}
/**
* Get statistics about the mapper
*/
getStats() {
return {
mappings: this.uuidToInt.size,
nextId: this.nextId,
dirty: this.dirty,
memoryEstimate: this.uuidToInt.size * (36 + 8 + 4 + 8) // uuid string + map overhead + int + map overhead
}
}
}