@tanstack/db 0.3.2 → 0.4.0

package/src/collection/indexes.ts

@@ -0,0 +1,172 @@
+import { IndexProxy, LazyIndexWrapper } from "../indexes/lazy-index"
+import {
+  createSingleRowRefProxy,
+  toExpression,
+} from "../query/builder/ref-proxy"
+import { BTreeIndex } from "../indexes/btree-index"
+import type { StandardSchemaV1 } from "@standard-schema/spec"
+import type { BaseIndex, IndexResolver } from "../indexes/base-index"
+import type { ChangeMessage } from "../types"
+import type { IndexOptions } from "../indexes/index-options"
+import type { SingleRowRefProxy } from "../query/builder/ref-proxy"
+import type { CollectionLifecycleManager } from "./lifecycle"
+import type { CollectionStateManager } from "./state"
+
+export class CollectionIndexesManager<
+  TOutput extends object = Record<string, unknown>,
+  TKey extends string | number = string | number,
+  TSchema extends StandardSchemaV1 = StandardSchemaV1,
+  TInput extends object = TOutput,
+> {
+  private lifecycle!: CollectionLifecycleManager<TOutput, TKey, TSchema, TInput>
+  private state!: CollectionStateManager<TOutput, TKey, TSchema, TInput>
+
+  public lazyIndexes = new Map<number, LazyIndexWrapper<TKey>>()
+  public resolvedIndexes = new Map<number, BaseIndex<TKey>>()
+  public isIndexesResolved = false
+  public indexCounter = 0
+
+  constructor() {}
+
+  setDeps(deps: {
+    state: CollectionStateManager<TOutput, TKey, TSchema, TInput>
+    lifecycle: CollectionLifecycleManager<TOutput, TKey, TSchema, TInput>
+  }) {
+    this.state = deps.state
+    this.lifecycle = deps.lifecycle
+  }
+
+  /**
+   * Creates an index on a collection for faster queries.
+   */
+  public createIndex<TResolver extends IndexResolver<TKey> = typeof BTreeIndex>(
+    indexCallback: (row: SingleRowRefProxy<TOutput>) => any,
+    config: IndexOptions<TResolver> = {}
+  ): IndexProxy<TKey> {
+    this.lifecycle.validateCollectionUsable(`createIndex`)
+
+    const indexId = ++this.indexCounter
+    const singleRowRefProxy = createSingleRowRefProxy<TOutput>()
+    const indexExpression = indexCallback(singleRowRefProxy)
+    const expression = toExpression(indexExpression)
+
+    // Default to BTreeIndex if no type specified
+    const resolver = config.indexType ?? (BTreeIndex as unknown as TResolver)
+
+    // Create lazy wrapper
+    const lazyIndex = new LazyIndexWrapper<TKey>(
+      indexId,
+      expression,
+      config.name,
+      resolver,
+      config.options,
+      this.state.entries()
+    )
+
+    this.lazyIndexes.set(indexId, lazyIndex)
+
+    // For BTreeIndex, resolve immediately and synchronously
+    if ((resolver as unknown) === BTreeIndex) {
+      try {
+        const resolvedIndex = lazyIndex.getResolved()
+        this.resolvedIndexes.set(indexId, resolvedIndex)
+      } catch (error) {
+        console.warn(`Failed to resolve BTreeIndex:`, error)
+      }
+    } else if (typeof resolver === `function` && resolver.prototype) {
+      // Other synchronous constructors - resolve immediately
+      try {
+        const resolvedIndex = lazyIndex.getResolved()
+        this.resolvedIndexes.set(indexId, resolvedIndex)
+      } catch {
+        // Fallback to async resolution
+        this.resolveSingleIndex(indexId, lazyIndex).catch((error) => {
+          console.warn(`Failed to resolve single index:`, error)
+        })
+      }
+    } else if (this.isIndexesResolved) {
+      // Async loader but indexes are already resolved - resolve this one
+      this.resolveSingleIndex(indexId, lazyIndex).catch((error) => {
+        console.warn(`Failed to resolve single index:`, error)
+      })
+    }
+
+    return new IndexProxy(indexId, lazyIndex)
+  }
+
+  /**
+   * Resolve all lazy indexes (called when collection first syncs)
+   */
+  public async resolveAllIndexes(): Promise<void> {
+    if (this.isIndexesResolved) return
+
+    const resolutionPromises = Array.from(this.lazyIndexes.entries()).map(
+      async ([indexId, lazyIndex]) => {
+        const resolvedIndex = await lazyIndex.resolve()
+
+        // Build index with current data
+        resolvedIndex.build(this.state.entries())
+
+        this.resolvedIndexes.set(indexId, resolvedIndex)
+        return { indexId, resolvedIndex }
+      }
+    )
+
+    await Promise.all(resolutionPromises)
+    this.isIndexesResolved = true
+  }
+
+  /**
+   * Resolve a single index immediately
+   */
+  private async resolveSingleIndex(
+    indexId: number,
+    lazyIndex: LazyIndexWrapper<TKey>
+  ): Promise<BaseIndex<TKey>> {
+    const resolvedIndex = await lazyIndex.resolve()
+    resolvedIndex.build(this.state.entries())
+    this.resolvedIndexes.set(indexId, resolvedIndex)
+    return resolvedIndex
+  }
+
+  /**
+   * Get resolved indexes for query optimization
+   */
+  get indexes(): Map<number, BaseIndex<TKey>> {
+    return this.resolvedIndexes
+  }
+
+  /**
+   * Updates all indexes when the collection changes
+   */
+  public updateIndexes(changes: Array<ChangeMessage<TOutput, TKey>>): void {
+    for (const index of this.resolvedIndexes.values()) {
+      for (const change of changes) {
+        switch (change.type) {
+          case `insert`:
+            index.add(change.key, change.value)
+            break
+          case `update`:
+            if (change.previousValue) {
+              index.update(change.key, change.previousValue, change.value)
+            } else {
+              index.add(change.key, change.value)
+            }
+            break
+          case `delete`:
+            index.remove(change.key, change.value)
+            break
+        }
+      }
+    }
+  }
+
+  /**
+   * Clean up the collection by stopping sync and clearing data
+   * This can be called manually or automatically by garbage collection
+   */
+  public cleanup(): void {
+    this.lazyIndexes.clear()
+    this.resolvedIndexes.clear()
+  }
+}

package/src/collection/lifecycle.ts

@@ -0,0 +1,221 @@
+import {
+  CollectionInErrorStateError,
+  InvalidCollectionStatusTransitionError,
+} from "../errors"
+import type { StandardSchemaV1 } from "@standard-schema/spec"
+import type { CollectionConfig, CollectionStatus } from "../types"
+import type { CollectionEventsManager } from "./events"
+import type { CollectionIndexesManager } from "./indexes"
+import type { CollectionChangesManager } from "./changes"
+import type { CollectionSyncManager } from "./sync"
+import type { CollectionStateManager } from "./state"
+
+export class CollectionLifecycleManager<
+  TOutput extends object = Record<string, unknown>,
+  TKey extends string | number = string | number,
+  TSchema extends StandardSchemaV1 = StandardSchemaV1,
+  TInput extends object = TOutput,
+> {
+  private config: CollectionConfig<TOutput, TKey, TSchema>
+  private id: string
+  private indexes!: CollectionIndexesManager<TOutput, TKey, TSchema, TInput>
+  private events!: CollectionEventsManager
+  private changes!: CollectionChangesManager<TOutput, TKey, TSchema, TInput>
+  private sync!: CollectionSyncManager<TOutput, TKey, TSchema, TInput>
+  private state!: CollectionStateManager<TOutput, TKey, TSchema, TInput>
+
+  public status: CollectionStatus = `idle`
+  public hasBeenReady = false
+  public hasReceivedFirstCommit = false
+  public onFirstReadyCallbacks: Array<() => void> = []
+  public gcTimeoutId: ReturnType<typeof setTimeout> | null = null
+
+  /**
+   * Creates a new CollectionLifecycleManager instance
+   */
+  constructor(config: CollectionConfig<TOutput, TKey, TSchema>, id: string) {
+    this.config = config
+    this.id = id
+  }
+
+  setDeps(deps: {
+    indexes: CollectionIndexesManager<TOutput, TKey, TSchema, TInput>
+    events: CollectionEventsManager
+    changes: CollectionChangesManager<TOutput, TKey, TSchema, TInput>
+    sync: CollectionSyncManager<TOutput, TKey, TSchema, TInput>
+    state: CollectionStateManager<TOutput, TKey, TSchema, TInput>
+  }) {
+    this.indexes = deps.indexes
+    this.events = deps.events
+    this.changes = deps.changes
+    this.sync = deps.sync
+    this.state = deps.state
+  }
+
+  /**
+   * Validates state transitions to prevent invalid status changes
+   */
+  public validateStatusTransition(
+    from: CollectionStatus,
+    to: CollectionStatus
+  ): void {
+    if (from === to) {
+      // Allow same state transitions
+      return
+    }
+    const validTransitions: Record<
+      CollectionStatus,
+      Array<CollectionStatus>
+    > = {
+      idle: [`loading`, `error`, `cleaned-up`],
+      loading: [`initialCommit`, `ready`, `error`, `cleaned-up`],
+      initialCommit: [`ready`, `error`, `cleaned-up`],
+      ready: [`cleaned-up`, `error`],
+      error: [`cleaned-up`, `idle`],
+      "cleaned-up": [`loading`, `error`],
+    }
+
+    if (!validTransitions[from].includes(to)) {
+      throw new InvalidCollectionStatusTransitionError(from, to, this.id)
+    }
+  }
+
+  /**
+   * Safely update the collection status with validation
+   * @private
+   */
+  public setStatus(newStatus: CollectionStatus): void {
+    this.validateStatusTransition(this.status, newStatus)
+    const previousStatus = this.status
+    this.status = newStatus
+
+    // Resolve indexes when collection becomes ready
+    if (newStatus === `ready` && !this.indexes.isIndexesResolved) {
+      // Resolve indexes asynchronously without blocking
+      this.indexes.resolveAllIndexes().catch((error) => {
+        console.warn(`Failed to resolve indexes:`, error)
+      })
+    }
+
+    // Emit event
+    this.events.emitStatusChange(newStatus, previousStatus)
+  }
+
+  /**
+   * Validates that the collection is in a usable state for data operations
+   * @private
+   */
+  public validateCollectionUsable(operation: string): void {
+    switch (this.status) {
+      case `error`:
+        throw new CollectionInErrorStateError(operation, this.id)
+      case `cleaned-up`:
+        // Automatically restart the collection when operations are called on cleaned-up collections
+        this.sync.startSync()
+        break
+    }
+  }
+
+  /**
+   * Mark the collection as ready for use
+   * This is called by sync implementations to explicitly signal that the collection is ready,
+   * providing a more intuitive alternative to using commits for readiness signaling
+   * @private - Should only be called by sync implementations
+   */
+  public markReady(): void {
+    // Can transition to ready from loading or initialCommit states
+    if (this.status === `loading` || this.status === `initialCommit`) {
+      this.setStatus(`ready`)
+
+      // Call any registered first ready callbacks (only on first time becoming ready)
+      if (!this.hasBeenReady) {
+        this.hasBeenReady = true
+
+        // Also mark as having received first commit for backwards compatibility
+        if (!this.hasReceivedFirstCommit) {
+          this.hasReceivedFirstCommit = true
+        }
+
+        const callbacks = [...this.onFirstReadyCallbacks]
+        this.onFirstReadyCallbacks = []
+        callbacks.forEach((callback) => callback())
+      }
+    }
+
+    // Always notify dependents when markReady is called, after status is set
+    // This ensures live queries get notified when their dependencies become ready
+    if (this.changes.changeSubscriptions.size > 0) {
+      this.changes.emitEmptyReadyEvent()
+    }
+  }
+
+  /**
+   * Start the garbage collection timer
+   * Called when the collection becomes inactive (no subscribers)
+   */
+  public startGCTimer(): void {
+    if (this.gcTimeoutId) {
+      clearTimeout(this.gcTimeoutId)
+    }
+
+    const gcTime = this.config.gcTime ?? 300000 // 5 minutes default
+
+    // If gcTime is 0, GC is disabled
+    if (gcTime === 0) {
+      return
+    }
+
+    this.gcTimeoutId = setTimeout(() => {
+      if (this.changes.activeSubscribersCount === 0) {
+        // We call the main collection cleanup, not just the one for the
+        // lifecycle manager
+        this.cleanup()
+      }
+    }, gcTime)
+  }
+
+  /**
+   * Cancel the garbage collection timer
+   * Called when the collection becomes active again
+   */
+  public cancelGCTimer(): void {
+    if (this.gcTimeoutId) {
+      clearTimeout(this.gcTimeoutId)
+      this.gcTimeoutId = null
+    }
+  }
+
+  /**
+   * Register a callback to be executed when the collection first becomes ready
+   * Useful for preloading collections
+   * @param callback Function to call when the collection first becomes ready
+   */
+  public onFirstReady(callback: () => void): void {
+    // If already ready, call immediately
+    if (this.hasBeenReady) {
+      callback()
+      return
+    }
+
+    this.onFirstReadyCallbacks.push(callback)
+  }
+
+  public cleanup(): void {
+    this.events.cleanup()
+    this.sync.cleanup()
+    this.state.cleanup()
+    this.changes.cleanup()
+    this.indexes.cleanup()
+
+    if (this.gcTimeoutId) {
+      clearTimeout(this.gcTimeoutId)
+      this.gcTimeoutId = null
+    }
+
+    this.hasBeenReady = false
+    this.onFirstReadyCallbacks = []
+
+    // Set status to cleaned-up
+    this.setStatus(`cleaned-up`)
+  }
+}