feat: add distributed architecture with sharding and coordination
- Wire up distributed components (Coordinator, ShardManager, CacheSync) - Implement automatic sharding for S3 storage (256 shards) - Add read/write separation for operational modes - Zero-config automatic detection for distributed mode - Add mutex implementation for thread safety - Fix metadata filtering in find operations - Fix neural API vector similarity calculations - Improve batch operations performance - Add Bluesky distributed setup example BREAKING CHANGE: None - backward compatible
This commit is contained in:
parent
8aafc769a3
commit
ed64c266ec
30 changed files with 2084 additions and 439 deletions
116
tests/unit/utils/mutex.test.ts
Normal file
116
tests/unit/utils/mutex.test.ts
Normal file
|
|
@ -0,0 +1,116 @@
|
|||
/**
|
||||
* Mutex Implementation Tests
|
||||
* Verify thread-safe operations without deadlocks or resource exhaustion
|
||||
*/
|
||||
|
||||
import { describe, it, expect, beforeEach, afterEach } from 'vitest'
|
||||
import { InMemoryMutex, FileMutex, createMutex } from '../../../src/utils/mutex'
|
||||
|
||||
describe('Mutex Safety Tests', () => {
|
||||
let mutex: InMemoryMutex
|
||||
|
||||
beforeEach(() => {
|
||||
mutex = new InMemoryMutex()
|
||||
})
|
||||
|
||||
describe('InMemoryMutex', () => {
|
||||
it('should prevent race conditions', async () => {
|
||||
let counter = 0
|
||||
const increments = 100
|
||||
|
||||
// Run multiple concurrent operations
|
||||
const operations = Array.from({ length: increments }, async () => {
|
||||
await mutex.runExclusive('counter', async () => {
|
||||
const current = counter
|
||||
// Simulate async work that could cause race
|
||||
await new Promise(resolve => setTimeout(resolve, 1))
|
||||
counter = current + 1
|
||||
})
|
||||
})
|
||||
|
||||
await Promise.all(operations)
|
||||
expect(counter).toBe(increments)
|
||||
})
|
||||
|
||||
it('should handle timeouts gracefully', async () => {
|
||||
// Acquire lock that won't be released
|
||||
const release = await mutex.acquire('timeout-test')
|
||||
|
||||
// Try to acquire same lock with short timeout
|
||||
const promise = mutex.acquire('timeout-test', 100)
|
||||
|
||||
await expect(promise).rejects.toThrow('Mutex timeout')
|
||||
|
||||
// Clean up
|
||||
release()
|
||||
})
|
||||
|
||||
it('should queue multiple waiters correctly', async () => {
|
||||
const order: number[] = []
|
||||
|
||||
// First acquirer
|
||||
const release1 = await mutex.acquire('queue-test')
|
||||
|
||||
// Queue up more acquirers
|
||||
const promise2 = mutex.runExclusive('queue-test', async () => {
|
||||
order.push(2)
|
||||
})
|
||||
|
||||
const promise3 = mutex.runExclusive('queue-test', async () => {
|
||||
order.push(3)
|
||||
})
|
||||
|
||||
// Release first lock
|
||||
order.push(1)
|
||||
release1()
|
||||
|
||||
// Wait for queued operations
|
||||
await Promise.all([promise2, promise3])
|
||||
|
||||
expect(order).toEqual([1, 2, 3])
|
||||
})
|
||||
|
||||
it('should not deadlock with nested different keys', async () => {
|
||||
await mutex.runExclusive('key1', async () => {
|
||||
await mutex.runExclusive('key2', async () => {
|
||||
// Should not deadlock
|
||||
expect(true).toBe(true)
|
||||
})
|
||||
})
|
||||
})
|
||||
|
||||
it('should handle errors in exclusive function', async () => {
|
||||
const error = new Error('Test error')
|
||||
|
||||
await expect(
|
||||
mutex.runExclusive('error-test', async () => {
|
||||
throw error
|
||||
})
|
||||
).rejects.toThrow('Test error')
|
||||
|
||||
// Lock should be released, so we can acquire it again
|
||||
await mutex.runExclusive('error-test', async () => {
|
||||
expect(true).toBe(true)
|
||||
})
|
||||
})
|
||||
|
||||
it('should handle high concurrency without resource exhaustion', async () => {
|
||||
const concurrency = 1000
|
||||
const operations = Array.from({ length: concurrency }, (_, i) =>
|
||||
mutex.runExclusive(`key-${i % 10}`, async () => {
|
||||
// Minimal work to test resource handling
|
||||
await Promise.resolve()
|
||||
})
|
||||
)
|
||||
|
||||
await expect(Promise.all(operations)).resolves.toBeDefined()
|
||||
})
|
||||
})
|
||||
|
||||
describe('createMutex factory', () => {
|
||||
it('should create appropriate mutex for environment', () => {
|
||||
const memoryMutex = createMutex({ type: 'memory' })
|
||||
expect(memoryMutex).toBeInstanceOf(InMemoryMutex)
|
||||
})
|
||||
})
|
||||
})
|
||||
|
|
@ -134,14 +134,14 @@ describe('Zero-Config Parameter Validation', () => {
|
|||
it('should require either data or vector', () => {
|
||||
expect(() => validateAddParams({
|
||||
type: NounType.Document
|
||||
} as AddParams)).toThrow('must provide either data or vector')
|
||||
} as AddParams)).toThrow('Invalid add() parameters: Missing required field \'data\'')
|
||||
})
|
||||
|
||||
it('should validate NounType', () => {
|
||||
expect(() => validateAddParams({
|
||||
data: 'test',
|
||||
type: 'InvalidType' as any
|
||||
})).toThrow('invalid NounType: InvalidType')
|
||||
})).toThrow('Invalid NounType: \'InvalidType\'')
|
||||
})
|
||||
|
||||
it('should validate vector dimensions', () => {
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue