@onebun/core 0.1.2 → 0.1.4

package/src/queue/adapters/memory.adapter.ts

@@ -0,0 +1,509 @@
+/**
+ * In-Memory Queue Adapter
+ *
+ * A simple in-process message bus. Useful for development, testing,
+ * and single-instance deployments where external dependencies are not needed.
+ */
+
+import type {
+  QueueAdapter,
+  QueueAdapterType,
+  QueueFeature,
+  QueueEvents,
+  Message,
+  MessageMetadata,
+  PublishOptions,
+  SubscribeOptions,
+  Subscription,
+  ScheduledJobOptions,
+  ScheduledJobInfo,
+  MessageHandler,
+} from '../types';
+
+import { createQueuePatternMatcher, type QueuePatternMatch } from '../pattern-matcher';
+import { QueueScheduler } from '../scheduler';
+
+// ============================================================================
+// Types
+// ============================================================================
+
+interface SubscriptionEntry {
+  pattern: string;
+  handler: MessageHandler;
+  options?: SubscribeOptions;
+  matcher: (topic: string) => QueuePatternMatch;
+  paused: boolean;
+}
+
+interface DelayedMessage {
+  pattern: string;
+  data: unknown;
+  options?: PublishOptions;
+  executeAt: number;
+}
+
+// ============================================================================
+// In-Memory Message Implementation
+// ============================================================================
+
+/**
+ * In-memory message implementation
+ */
+class InMemoryMessage<T> implements Message<T> {
+  id: string;
+  pattern: string;
+  data: T;
+  timestamp: number;
+  redelivered: boolean;
+  metadata: MessageMetadata;
+  attempt?: number;
+  maxAttempts?: number;
+
+  private acked = false;
+  private nacked = false;
+  private onAck?: () => void;
+  private onNack?: (requeue: boolean) => void;
+
+  constructor(
+    id: string,
+    pattern: string,
+    data: T,
+    metadata: MessageMetadata,
+    options?: {
+      redelivered?: boolean;
+      attempt?: number;
+      maxAttempts?: number;
+      onAck?: () => void;
+      onNack?: (requeue: boolean) => void;
+    },
+  ) {
+    this.id = id;
+    this.pattern = pattern;
+    this.data = data;
+    this.timestamp = Date.now();
+    this.metadata = metadata;
+    this.redelivered = options?.redelivered ?? false;
+    this.attempt = options?.attempt;
+    this.maxAttempts = options?.maxAttempts;
+    this.onAck = options?.onAck;
+    this.onNack = options?.onNack;
+  }
+
+  async ack(): Promise<void> {
+    if (this.acked || this.nacked) {
+      return;
+    }
+    this.acked = true;
+    this.onAck?.();
+  }
+
+  async nack(requeue = false): Promise<void> {
+    if (this.acked || this.nacked) {
+      return;
+    }
+    this.nacked = true;
+    this.onNack?.(requeue);
+  }
+}
+
+// ============================================================================
+// In-Memory Subscription Implementation
+// ============================================================================
+
+class InMemorySubscription implements Subscription {
+  private active = true;
+
+  constructor(
+    private readonly entry: SubscriptionEntry,
+    private readonly onUnsubscribe: () => void,
+  ) {}
+
+  async unsubscribe(): Promise<void> {
+    this.active = false;
+    this.onUnsubscribe();
+  }
+
+  pause(): void {
+    this.entry.paused = true;
+  }
+
+  resume(): void {
+    this.entry.paused = false;
+  }
+
+  get pattern(): string {
+    return this.entry.pattern;
+  }
+
+  get isActive(): boolean {
+    return this.active && !this.entry.paused;
+  }
+}
+
+// ============================================================================
+// In-Memory Queue Adapter
+// ============================================================================
+
+/**
+ * In-Memory Queue Adapter
+ *
+ * Implements a simple in-process message bus.
+ *
+ * Supported features:
+ * - Pattern subscriptions with wildcards
+ * - Delayed messages
+ * - Priority (via sorting)
+ * - Scheduled jobs
+ *
+ * Not supported (requires external storage):
+ * - Consumer groups (all handlers receive all messages)
+ * - Dead letter queues
+ * - Retry with persistence
+ *
+ * @example
+ * ```typescript
+ * const adapter = new InMemoryQueueAdapter();
+ * await adapter.connect();
+ *
+ * await adapter.subscribe('orders.*', async (message) => {
+ *   console.log('Received:', message.data);
+ * });
+ *
+ * await adapter.publish('orders.created', { orderId: 123 });
+ * ```
+ */
+export class InMemoryQueueAdapter implements QueueAdapter {
+  readonly name = 'memory';
+  readonly type: QueueAdapterType = 'memory';
+
+  private subscriptions: SubscriptionEntry[] = [];
+  private delayedMessages: DelayedMessage[] = [];
+  private messageIdCounter = 0;
+  private connected = false;
+  private scheduler: QueueScheduler | null = null;
+  private delayedInterval?: ReturnType<typeof setInterval>;
+
+  // Event handlers
+  private eventHandlers: Map<keyof QueueEvents, Set<(...args: unknown[]) => void>> = new Map();
+
+  // ============================================================================
+  // Lifecycle
+  // ============================================================================
+
+  async connect(): Promise<void> {
+    if (this.connected) {
+      return;
+    }
+
+    this.connected = true;
+    this.scheduler = new QueueScheduler(this);
+
+    // Start delayed message processor
+    this.delayedInterval = setInterval(() => this.processDelayedMessages(), 100);
+
+    // Emit ready event
+    this.emit('onReady');
+  }
+
+  async disconnect(): Promise<void> {
+    if (!this.connected) {
+      return;
+    }
+
+    // Stop scheduler
+    if (this.scheduler) {
+      this.scheduler.stop();
+      this.scheduler = null;
+    }
+
+    // Clear delayed interval
+    if (this.delayedInterval) {
+      clearInterval(this.delayedInterval);
+      this.delayedInterval = undefined;
+    }
+
+    // Clear all subscriptions
+    this.subscriptions = [];
+    this.delayedMessages = [];
+
+    this.connected = false;
+  }
+
+  isConnected(): boolean {
+    return this.connected;
+  }
+
+  // ============================================================================
+  // Publishing
+  // ============================================================================
+
+  async publish<T>(pattern: string, data: T, options?: PublishOptions): Promise<string> {
+    this.ensureConnected();
+
+    const messageId = options?.messageId ?? this.generateMessageId();
+
+    // Handle delayed messages
+    if (options?.delay && options.delay > 0) {
+      this.delayedMessages.push({
+        pattern,
+        data,
+        options: { ...options, messageId },
+        executeAt: Date.now() + options.delay,
+      });
+      // Sort by priority (higher first) and then by time
+      this.delayedMessages.sort((a, b) => {
+        const priorityA = a.options?.priority ?? 0;
+        const priorityB = b.options?.priority ?? 0;
+        if (priorityA !== priorityB) {
+          return priorityB - priorityA; // Higher priority first
+        }
+
+        return a.executeAt - b.executeAt;
+      });
+
+      return messageId;
+    }
+
+    // Dispatch immediately
+    await this.dispatch(pattern, data, messageId, options?.metadata);
+
+    return messageId;
+  }
+
+  async publishBatch<T>(
+    messages: Array<{ pattern: string; data: T; options?: PublishOptions }>,
+  ): Promise<string[]> {
+    const ids: string[] = [];
+
+    // Sort by priority if any
+    const sorted = [...messages].sort((a, b) => {
+      const priorityA = a.options?.priority ?? 0;
+      const priorityB = b.options?.priority ?? 0;
+
+      return priorityB - priorityA;
+    });
+
+    for (const msg of sorted) {
+      const id = await this.publish(msg.pattern, msg.data, msg.options);
+      ids.push(id);
+    }
+
+    return ids;
+  }
+
+  // ============================================================================
+  // Subscribing
+  // ============================================================================
+
+  async subscribe<T>(
+    pattern: string,
+    handler: MessageHandler<T>,
+    options?: SubscribeOptions,
+  ): Promise<Subscription> {
+    this.ensureConnected();
+
+    const entry: SubscriptionEntry = {
+      pattern,
+      handler: handler as MessageHandler,
+      options,
+      matcher: createQueuePatternMatcher(pattern),
+      paused: false,
+    };
+
+    this.subscriptions.push(entry);
+
+    const subscription = new InMemorySubscription(entry, () => {
+      const index = this.subscriptions.indexOf(entry);
+      if (index !== -1) {
+        this.subscriptions.splice(index, 1);
+      }
+    });
+
+    return subscription;
+  }
+
+  // ============================================================================
+  // Scheduled Jobs
+  // ============================================================================
+
+  async addScheduledJob(name: string, options: ScheduledJobOptions): Promise<void> {
+    this.ensureConnected();
+
+    if (!this.scheduler) {
+      throw new Error('Scheduler not initialized');
+    }
+
+    if (options.schedule.cron) {
+      this.scheduler.addCronJob(name, options.schedule.cron, options.pattern, () => options.data, {
+        metadata: options.metadata,
+        overlapStrategy: options.overlapStrategy,
+      });
+    } else if (options.schedule.every) {
+      this.scheduler.addIntervalJob(name, options.schedule.every, options.pattern, () => options.data, {
+        metadata: options.metadata,
+      });
+    }
+  }
+
+  async removeScheduledJob(name: string): Promise<boolean> {
+    this.ensureConnected();
+
+    if (!this.scheduler) {
+      return false;
+    }
+
+    return this.scheduler.removeJob(name);
+  }
+
+  async getScheduledJobs(): Promise<ScheduledJobInfo[]> {
+    this.ensureConnected();
+
+    if (!this.scheduler) {
+      return [];
+    }
+
+    return this.scheduler.getJobs();
+  }
+
+  // ============================================================================
+  // Features
+  // ============================================================================
+
+  supports(feature: QueueFeature): boolean {
+    switch (feature) {
+      case 'delayed-messages':
+      case 'priority':
+      case 'scheduled-jobs':
+      case 'pattern-subscriptions':
+        return true;
+      case 'consumer-groups':
+      case 'dead-letter-queue':
+      case 'retry':
+        return false;
+      default:
+        return false;
+    }
+  }
+
+  // ============================================================================
+  // Events
+  // ============================================================================
+
+  on<E extends keyof QueueEvents>(event: E, handler: NonNullable<QueueEvents[E]>): void {
+    if (!this.eventHandlers.has(event)) {
+      this.eventHandlers.set(event, new Set());
+    }
+    this.eventHandlers.get(event)!.add(handler as (...args: unknown[]) => void);
+  }
+
+  off<E extends keyof QueueEvents>(event: E, handler: NonNullable<QueueEvents[E]>): void {
+    const handlers = this.eventHandlers.get(event);
+    if (handlers) {
+      handlers.delete(handler as (...args: unknown[]) => void);
+    }
+  }
+
+  // ============================================================================
+  // Private Methods
+  // ============================================================================
+
+  private ensureConnected(): void {
+    if (!this.connected) {
+      throw new Error('InMemoryQueueAdapter not connected. Call connect() first.');
+    }
+  }
+
+  private generateMessageId(): string {
+    return `msg-${++this.messageIdCounter}-${Date.now()}`;
+  }
+
+  private async dispatch<T>(
+    pattern: string,
+    data: T,
+    messageId: string,
+    metadata?: Partial<MessageMetadata>,
+  ): Promise<void> {
+    const fullMetadata: MessageMetadata = {
+      headers: {},
+      ...metadata,
+    };
+
+    // Find all matching subscriptions
+    for (const entry of this.subscriptions) {
+      if (entry.paused) {
+        continue;
+      }
+
+      const match = entry.matcher(pattern);
+      if (!match.matched) {
+        continue;
+      }
+
+      const message = new InMemoryMessage<T>(messageId, pattern, data, fullMetadata, {
+        onNack: (requeue) => {
+          if (requeue) {
+            // Re-dispatch the message
+            setImmediate(() => {
+              this.dispatch(pattern, data, messageId, metadata);
+            });
+          }
+        },
+      });
+
+      // Emit received event
+      this.emit('onMessageReceived', message);
+
+      try {
+        await entry.handler(message);
+
+        // Auto-ack if not manual mode
+        if (entry.options?.ackMode !== 'manual') {
+          await message.ack();
+        }
+
+        // Emit processed event
+        this.emit('onMessageProcessed', message);
+      } catch (error) {
+        // Emit failed event
+        this.emit('onMessageFailed', message, error as Error);
+
+        // Auto-nack if not manual mode
+        if (entry.options?.ackMode !== 'manual') {
+          await message.nack(false);
+        }
+      }
+    }
+  }
+
+  private processDelayedMessages(): void {
+    const now = Date.now();
+
+    while (this.delayedMessages.length > 0 && this.delayedMessages[0].executeAt <= now) {
+      const delayed = this.delayedMessages.shift()!;
+      const messageId = delayed.options?.messageId ?? this.generateMessageId();
+
+      // Dispatch without delay
+      this.dispatch(delayed.pattern, delayed.data, messageId, delayed.options?.metadata);
+    }
+  }
+
+  private emit<E extends keyof QueueEvents>(event: E, ...args: unknown[]): void {
+    const handlers = this.eventHandlers.get(event);
+    if (handlers) {
+      for (const handler of handlers) {
+        try {
+          handler(...args);
+        } catch {
+          // Silently ignore event handler errors
+        }
+      }
+    }
+  }
+}
+
+/**
+ * Create an in-memory queue adapter
+ */
+export function createInMemoryQueueAdapter(): InMemoryQueueAdapter {
+  return new InMemoryQueueAdapter();
+}