fixed pipeline handlers

apps/stock/data-pipeline/src/handlers/exchanges/exchanges.handler.ts

@@ -1,74 +1,111 @@
-import { getLogger } from '@stock-bot/logger';
-import { handlerRegistry, createJobHandler, type HandlerConfig, type ScheduledJobConfig } from '@stock-bot/queue';
-import type { IServiceContainer } from '@stock-bot/handlers';
-import { exchangeOperations } from './operations';
-
-const logger = getLogger('exchanges-handler');
-
-const HANDLER_NAME = 'exchanges';
-
-const exchangesHandlerConfig: HandlerConfig = {
-  concurrency: 1,
-  maxAttempts: 3,
-  scheduledJobs: [
-    {
-      operation: 'sync-all-exchanges',
-      cronPattern: '0 0 * * 0', // Weekly on Sunday at midnight
-      payload: { clearFirst: true },
-      priority: 10,
-      immediately: false,
-    } as ScheduledJobConfig,
-    {
-      operation: 'sync-qm-exchanges',
-      cronPattern: '0 1 * * *', // Daily at 1 AM
-      payload: {},
-      priority: 5,
-      immediately: false,
-    } as ScheduledJobConfig,
-    {
-      operation: 'sync-ib-exchanges',
-      cronPattern: '0 3 * * *', // Daily at 3 AM
-      payload: {},
-      priority: 3,
-      immediately: false,
-    } as ScheduledJobConfig,
-    {
-      operation: 'sync-qm-provider-mappings',
-      cronPattern: '0 3 * * *', // Daily at 3 AM
-      payload: {},
-      priority: 7,
-      immediately: false,
-    } as ScheduledJobConfig,
-  ],
-  operations: {
-    'sync-all-exchanges': exchangeOperations.syncAllExchanges,
-    'sync-qm-exchanges': exchangeOperations.syncQMExchanges,
-    'sync-ib-exchanges': exchangeOperations.syncIBExchanges,
-    'sync-qm-provider-mappings': exchangeOperations.syncQMProviderMappings,
-    'clear-postgresql-data': exchangeOperations.clearPostgreSQLData,
-    'get-exchange-stats': exchangeOperations.getExchangeStats,
-    'get-provider-mapping-stats': exchangeOperations.getProviderMappingStats,
-    'enhanced-sync-status': exchangeOperations['enhanced-sync-status'],
-  },
-};
-
-export function initializeExchangesHandler(container: IServiceContainer) {
-  logger.info('Registering exchanges handler...');
-
-  // Update operations to use container
-  const containerAwareOperations = Object.entries(exchangeOperations).reduce((acc, [key, operation]) => {
-    acc[key] = createJobHandler(async (payload: any) => {
-      return operation(payload, container);
-    });
-    return acc;
-  }, {} as Record<string, any>);
-
-  const exchangesHandlerConfigWithContainer: HandlerConfig = {
-    ...exchangesHandlerConfig,
-    operations: containerAwareOperations,
-  };
-
-  handlerRegistry.register(HANDLER_NAME, exchangesHandlerConfigWithContainer);
-  logger.info('Exchanges handler registered successfully');
-}
-
+import {
+  BaseHandler,
+  Handler,
+  Operation,
+  ScheduledOperation,
+  type IServiceContainer,
+} from '@stock-bot/handlers';
+import { clearPostgreSQLData } from './operations/clear-postgresql-data.operations';
+import { getSyncStatus } from './operations/enhanced-sync-status.operations';
+import { getExchangeStats } from './operations/exchange-stats.operations';
+import { getProviderMappingStats } from './operations/provider-mapping-stats.operations';
+import { syncQMExchanges } from './operations/qm-exchanges.operations';
+import { syncAllExchanges } from './operations/sync-all-exchanges.operations';
+import { syncIBExchanges } from './operations/sync-ib-exchanges.operations';
+import { syncQMProviderMappings } from './operations/sync-qm-provider-mappings.operations';
+
+@Handler('exchanges')
+export class ExchangesHandler extends BaseHandler {
+  constructor(services: IServiceContainer) {
+    super(services);
+  }
+
+  /**
+   * Sync all exchanges - weekly full sync
+   */
+  @Operation('sync-all-exchanges')
+  @ScheduledOperation('sync-all-exchanges', '0 0 * * 0', {
+    priority: 10,
+    description: 'Weekly full exchange sync on Sunday at midnight',
+  })
+  async syncAllExchanges(payload?: { clearFirst?: boolean }): Promise<unknown> {
+    const finalPayload = payload || { clearFirst: true };
+    this.log('info', 'Starting sync of all exchanges', finalPayload);
+    return syncAllExchanges(finalPayload, this.services);
+  }
+
+  /**
+   * Sync exchanges from QuestionsAndMethods
+   */
+  @Operation('sync-qm-exchanges')
+  @ScheduledOperation('sync-qm-exchanges', '0 1 * * *', {
+    priority: 5,
+    description: 'Daily sync of QM exchanges at 1 AM',
+  })
+  async syncQMExchanges(): Promise<unknown> {
+    this.log('info', 'Starting QM exchanges sync...');
+    return syncQMExchanges({}, this.services);
+  }
+
+  /**
+   * Sync exchanges from Interactive Brokers
+   */
+  @Operation('sync-ib-exchanges')
+  @ScheduledOperation('sync-ib-exchanges', '0 3 * * *', {
+    priority: 3,
+    description: 'Daily sync of IB exchanges at 3 AM',
+  })
+  async syncIBExchanges(): Promise<unknown> {
+    this.log('info', 'Starting IB exchanges sync...');
+    return syncIBExchanges({}, this.services);
+  }
+
+  /**
+   * Sync provider mappings from QuestionsAndMethods
+   */
+  @Operation('sync-qm-provider-mappings')
+  @ScheduledOperation('sync-qm-provider-mappings', '0 3 * * *', {
+    priority: 7,
+    description: 'Daily sync of QM provider mappings at 3 AM',
+  })
+  async syncQMProviderMappings(): Promise<unknown> {
+    this.log('info', 'Starting QM provider mappings sync...');
+    return syncQMProviderMappings({}, this.services);
+  }
+
+  /**
+   * Clear PostgreSQL data - maintenance operation
+   */
+  @Operation('clear-postgresql-data')
+  async clearPostgreSQLData(payload: { type?: 'exchanges' | 'provider_mappings' | 'all' }): Promise<unknown> {
+    this.log('warn', 'Clearing PostgreSQL data', payload);
+    return clearPostgreSQLData(payload, this.services);
+  }
+
+  /**
+   * Get exchange statistics
+   */
+  @Operation('get-exchange-stats')
+  async getExchangeStats(): Promise<unknown> {
+    this.log('info', 'Getting exchange statistics...');
+    return getExchangeStats({}, this.services);
+  }
+
+  /**
+   * Get provider mapping statistics
+   */
+  @Operation('get-provider-mapping-stats')
+  async getProviderMappingStats(): Promise<unknown> {
+    this.log('info', 'Getting provider mapping statistics...');
+    return getProviderMappingStats({}, this.services);
+  }
+
+  /**
+   * Get enhanced sync status
+   */
+  @Operation('enhanced-sync-status')
+  async getEnhancedSyncStatus(): Promise<unknown> {
+    this.log('info', 'Getting enhanced sync status...');
+    return getSyncStatus({}, this.services);
+  }
+}