@kbve/droid 0.0.6 → 0.1.1

package/src/lib/gateway/WorkerPool.ts

@@ -0,0 +1,304 @@
+// Worker pool manager with round-robin routing for DB operations
+
+import type { WorkerMessage, WorkerResponse } from './types';
+
+export interface WorkerPoolConfig {
+	size: number;
+	workerUrl: string | URL;
+}
+
+/**
+ * WorkerPool: Manages a pool of workers with round-robin load balancing
+ *
+ * Features:
+ * - Fixed-size pool (default: 3 workers)
+ * - Round-robin routing for database operations
+ * - Request/response tracking with UUIDs
+ * - Automatic error handling with exponential backoff retries
+ */
+export class WorkerPool {
+	private workers: Worker[] = [];
+	private poolSize: number;
+	private roundRobinIndex = 0;
+	private pending = new Map<
+		string,
+		{
+			resolve: (value: unknown) => void;
+			reject: (reason?: unknown) => void;
+			workerId: number;
+		}
+	>();
+	private workerUrl: string | URL;
+	private workerRetryCount = new Map<number, number>();
+	private workerRetryTimers = new Map<
+		number,
+		ReturnType<typeof setTimeout>
+	>();
+
+	private readonly MAX_RETRY_ATTEMPTS = 3;
+	private readonly BASE_RETRY_DELAY_MS = 1000;
+	private readonly MAX_RETRY_DELAY_MS = 8000;
+	private readonly REQUEST_TIMEOUT_MS = 30000;
+
+	constructor(config: WorkerPoolConfig) {
+		this.poolSize = config.size;
+		this.workerUrl = config.workerUrl;
+	}
+
+	/**
+	 * Initialize the worker pool
+	 */
+	async init(): Promise<void> {
+		console.log(
+			`[WorkerPool] Initializing pool with ${this.poolSize} workers`,
+		);
+
+		for (let i = 0; i < this.poolSize; i++) {
+			try {
+				const worker = new Worker(this.workerUrl, {
+					type: 'module',
+					name: `db-worker-${i}`,
+				});
+
+				worker.onmessage = (e: MessageEvent) =>
+					this.handleMessage(e.data, i);
+				worker.onerror = (err) => this.handleError(err, i);
+
+				this.workers.push(worker);
+				console.log(`[WorkerPool] Worker ${i} initialized`);
+			} catch (err) {
+				console.error(
+					`[WorkerPool] Failed to create worker ${i}:`,
+					err,
+				);
+				throw err;
+			}
+		}
+
+		// Wait for all workers to be ready
+		await Promise.all(
+			this.workers.map((_worker, i) =>
+				this.sendToWorker(i, {
+					id: crypto.randomUUID(),
+					type: 'ping',
+					payload: {},
+				}),
+			),
+		);
+
+		console.log('[WorkerPool] All workers ready');
+	}
+
+	private getNextWorker(): number {
+		const workerId = this.roundRobinIndex;
+		this.roundRobinIndex = (this.roundRobinIndex + 1) % this.poolSize;
+		return workerId;
+	}
+
+	private sendToWorker<T>(
+		workerId: number,
+		message: WorkerMessage,
+	): Promise<T> {
+		return new Promise<T>((resolve, reject) => {
+			const worker = this.workers[workerId];
+			if (!worker) {
+				return reject(new Error(`Worker ${workerId} not available`));
+			}
+
+			const timer = setTimeout(() => {
+				if (this.pending.has(message.id)) {
+					this.pending.delete(message.id);
+					reject(
+						new Error(
+							`Worker ${workerId} request ${message.type} timed out after ${this.REQUEST_TIMEOUT_MS}ms`,
+						),
+					);
+				}
+			}, this.REQUEST_TIMEOUT_MS);
+
+			this.pending.set(message.id, {
+				resolve: (value: unknown) => {
+					clearTimeout(timer);
+					resolve(value as T);
+				},
+				reject: (err: unknown) => {
+					clearTimeout(timer);
+					reject(err);
+				},
+				workerId,
+			});
+
+			try {
+				worker.postMessage(message);
+			} catch (err) {
+				clearTimeout(timer);
+				this.pending.delete(message.id);
+				reject(err);
+			}
+		});
+	}
+
+	/**
+	 * Send message to next available worker (round-robin)
+	 */
+	send<T>(type: string, payload?: unknown): Promise<T> {
+		const workerId = this.getNextWorker();
+		const id = crypto.randomUUID();
+		const message: WorkerMessage = { id, type, payload };
+
+		return this.sendToWorker<T>(workerId, message);
+	}
+
+	private handleMessage(response: WorkerResponse, workerId: number) {
+		const { id, ok, data, error } = response;
+
+		if (!id) {
+			console.warn(
+				`[WorkerPool] Worker ${workerId} sent message without ID:`,
+				response,
+			);
+			return;
+		}
+
+		const pending = this.pending.get(id);
+		if (!pending) {
+			console.warn(
+				`[WorkerPool] Received response for unknown request ${id}`,
+			);
+			return;
+		}
+
+		this.pending.delete(id);
+
+		if (ok) {
+			pending.resolve(data);
+		} else {
+			pending.reject(new Error(error || 'Worker request failed'));
+		}
+	}
+
+	private handleError(error: ErrorEvent, workerId: number) {
+		console.error(`[WorkerPool] Worker ${workerId} error:`, error);
+
+		const failedRequests: string[] = [];
+		for (const [id, pending] of this.pending.entries()) {
+			if (pending.workerId === workerId) {
+				pending.reject(
+					new Error(`Worker ${workerId} crashed: ${error.message}`),
+				);
+				failedRequests.push(id);
+			}
+		}
+
+		failedRequests.forEach((id) => this.pending.delete(id));
+
+		this.scheduleWorkerRecreation(workerId);
+	}
+
+	private scheduleWorkerRecreation(workerId: number) {
+		const retryCount = this.workerRetryCount.get(workerId) || 0;
+
+		if (retryCount >= this.MAX_RETRY_ATTEMPTS) {
+			console.error(
+				`[WorkerPool] Worker ${workerId} has failed ${retryCount} times. Giving up.`,
+			);
+			this.workerRetryCount.delete(workerId);
+			return;
+		}
+
+		const delay = Math.min(
+			this.BASE_RETRY_DELAY_MS * Math.pow(2, retryCount),
+			this.MAX_RETRY_DELAY_MS,
+		);
+
+		console.log(
+			`[WorkerPool] Scheduling worker ${workerId} recreation in ${delay}ms (attempt ${retryCount + 1}/${this.MAX_RETRY_ATTEMPTS})`,
+		);
+
+		const existingTimer = this.workerRetryTimers.get(workerId);
+		if (existingTimer) {
+			clearTimeout(existingTimer);
+		}
+
+		const timer = setTimeout(() => {
+			this.workerRetryTimers.delete(workerId);
+			this.recreateWorker(workerId);
+		}, delay);
+
+		this.workerRetryTimers.set(workerId, timer);
+		this.workerRetryCount.set(workerId, retryCount + 1);
+	}
+
+	private async recreateWorker(workerId: number) {
+		const retryCount = this.workerRetryCount.get(workerId) || 0;
+		console.log(
+			`[WorkerPool] Attempting to recreate worker ${workerId} (attempt ${retryCount}/${this.MAX_RETRY_ATTEMPTS})`,
+		);
+
+		try {
+			const worker = new Worker(this.workerUrl, {
+				type: 'module',
+				name: `db-worker-${workerId}`,
+			});
+
+			worker.onmessage = (e: MessageEvent) =>
+				this.handleMessage(e.data, workerId);
+			worker.onerror = (err) => this.handleError(err, workerId);
+
+			this.workers[workerId] = worker;
+
+			await this.sendToWorker(workerId, {
+				id: crypto.randomUUID(),
+				type: 'ping',
+				payload: {},
+			});
+
+			console.log(
+				`[WorkerPool] Worker ${workerId} recreated successfully`,
+			);
+			this.workerRetryCount.delete(workerId);
+		} catch (err) {
+			console.error(
+				`[WorkerPool] Failed to recreate worker ${workerId}:`,
+				err,
+			);
+		}
+	}
+
+	/**
+	 * Terminate all workers
+	 */
+	terminate() {
+		console.log('[WorkerPool] Terminating all workers');
+
+		this.workerRetryTimers.forEach((timer) => clearTimeout(timer));
+		this.workerRetryTimers.clear();
+		this.workerRetryCount.clear();
+
+		this.workers.forEach((worker, i) => {
+			try {
+				worker.terminate();
+			} catch (err) {
+				console.error(
+					`[WorkerPool] Error terminating worker ${i}:`,
+					err,
+				);
+			}
+		});
+
+		this.workers = [];
+		this.pending.clear();
+	}
+
+	/**
+	 * Get pool statistics
+	 */
+	getStats() {
+		return {
+			size: this.poolSize,
+			activeWorkers: this.workers.length,
+			pendingRequests: this.pending.size,
+			nextWorker: this.roundRobinIndex,
+		};
+	}
+}

package/src/lib/gateway/capabilities.ts

@@ -0,0 +1,80 @@
+// Browser capability detection for strategy selection
+
+import type { BrowserCapabilities, StrategyType } from './types';
+
+/**
+ * Detect browser capabilities for worker and communication support
+ */
+export function detectCapabilities(): BrowserCapabilities {
+  // SSR guard
+  if (typeof window === 'undefined') {
+    return {
+      hasSharedWorker: false,
+      hasWorker: false,
+      hasBroadcastChannel: false,
+      isAndroid: false,
+      isSafari: false,
+    };
+  }
+
+  const userAgent = navigator.userAgent || '';
+
+  return {
+    hasSharedWorker: typeof SharedWorker !== 'undefined',
+    hasWorker: typeof Worker !== 'undefined',
+    hasBroadcastChannel: typeof BroadcastChannel !== 'undefined',
+    isAndroid: /Android/i.test(userAgent),
+    isSafari: /Safari/i.test(userAgent) && !/Chrome/i.test(userAgent),
+  };
+}
+
+/**
+ * Select the best strategy based on browser capabilities
+ *
+ * Priority:
+ * 1. SharedWorker + Worker Pool (desktop Chrome, Firefox, Edge)
+ * 2. WebWorker Pool only (Android, Safari, or no SharedWorker)
+ * 3. Direct/Main thread (emergency fallback)
+ */
+export function selectStrategy(capabilities?: BrowserCapabilities): StrategyType {
+  const caps = capabilities || detectCapabilities();
+
+  // Desktop with SharedWorker support (best performance)
+  if (caps.hasSharedWorker && caps.hasBroadcastChannel && !caps.isAndroid) {
+    return 'shared-worker';
+  }
+
+  // Android or browsers without SharedWorker (still use worker pool)
+  if (caps.hasWorker && caps.hasBroadcastChannel) {
+    return 'web-worker';
+  }
+
+  // Fallback to main thread (no workers or BroadcastChannel)
+  return 'direct';
+}
+
+/**
+ * Log detected capabilities for debugging
+ */
+export function logCapabilities(capabilities: BrowserCapabilities) {
+  console.log('[SupabaseGateway] Browser capabilities:', {
+    SharedWorker: capabilities.hasSharedWorker ? 'yes' : 'no',
+    Worker: capabilities.hasWorker ? 'yes' : 'no',
+    BroadcastChannel: capabilities.hasBroadcastChannel ? 'yes' : 'no',
+    Platform: capabilities.isAndroid ? 'Android' : capabilities.isSafari ? 'Safari' : 'Desktop',
+  });
+}
+
+/**
+ * Get a human-readable description of the selected strategy
+ */
+export function getStrategyDescription(strategy: StrategyType): string {
+  switch (strategy) {
+    case 'shared-worker':
+      return 'SharedWorker + DB Worker Pool (optimal for desktop)';
+    case 'web-worker':
+      return 'WebSocket Worker + DB Worker Pool (optimized for mobile)';
+    case 'direct':
+      return 'Direct/Main thread (compatibility fallback)';
+  }
+}

package/src/lib/gateway/index.ts

@@ -0,0 +1,15 @@
+// Main exports for Supabase Gateway
+
+export { SupabaseGateway } from './SupabaseGateway';
+export { WorkerPool } from './WorkerPool';
+export { getWorkerCommunication, WorkerCommunication } from './WorkerCommunication';
+export {
+  detectCapabilities,
+  selectStrategy,
+  logCapabilities,
+  getStrategyDescription
+} from './capabilities';
+export * from './types';
+export { SharedWorkerStrategy } from './strategies/SharedWorkerStrategy';
+export { WebWorkerStrategy } from './strategies/WebWorkerStrategy';
+export { DirectStrategy } from './strategies/DirectStrategy';