boki/stock-bot
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

standartized OperationTracker. need to test it all out now

Browse source
This commit is contained in:
Boki 2025-07-01 11:15:33 -04:00
parent f78558224f
commit 680b5fd2ae
21 changed files with 2112 additions and 752 deletions
Download patch file Download diff file Expand all files Collapse all files

29
apps/stock/data-ingestion/src/handlers/qm/actions/intraday.action.ts
View file

@ -5,23 +5,8 @@
import type { ExecutionContext } from '@stock-bot/handlers';
import type { QMHandler } from '../qm.handler';
import { QM_CONFIG, QM_SESSION_IDS } from '../shared/config';
import { QMOperationTracker } from '../shared/operation-tracker';
import { QMSessionManager } from '../shared/session-manager';
// Cache tracker instance
let operationTracker: QMOperationTracker | null = null;
/**
* Get or initialize the operation tracker
*/
async function getOperationTracker(handler: QMHandler): Promise<QMOperationTracker> {
if (!operationTracker) {
const { initializeQMOperations } = await import('../shared/operation-registry');
operationTracker = await initializeQMOperations(handler.mongodb, handler.logger);
}
return operationTracker;
}
/**
* Update intraday bars for a single symbol
* This handles both initial crawl and incremental updates
@ -107,8 +92,7 @@ export async function updateIntradayBars(
);
// Update operation tracking
const tracker = await getOperationTracker(this);
await tracker.updateSymbolOperation(qmSearchCode, 'intraday_bars', {
await this.operationRegistry.updateOperation('qm', qmSearchCode, 'intraday_bars', {
status: 'success',
lastRecordDate: targetDate,
recordCount: barsData.length
@ -134,8 +118,7 @@ export async function updateIntradayBars(
this.logger.info('No intraday data for date', { symbol, date: targetDate });
// Still update operation tracking as successful (no data is a valid result)
const tracker = await getOperationTracker(this);
await tracker.updateSymbolOperation(qmSearchCode, 'intraday_bars', {
await this.operationRegistry.updateOperation('qm', qmSearchCode, 'intraday_bars', {
status: 'success',
lastRecordDate: targetDate,
recordCount: 0
@ -164,8 +147,7 @@ export async function updateIntradayBars(
});
// Update operation tracking for failure
const tracker = await getOperationTracker(this);
await tracker.updateSymbolOperation(qmSearchCode, 'intraday_bars', {
await this.operationRegistry.updateOperation('qm', qmSearchCode, 'intraday_bars', {
status: 'failure'
});
@ -196,7 +178,6 @@ export async function scheduleIntradayUpdates(
errors: number;
}> {
const { limit = 50, mode = 'update', forceUpdate = false } = input;
const tracker = await getOperationTracker(this);
this.logger.info('Scheduling intraday updates', { limit, mode, forceUpdate });
@ -205,12 +186,12 @@ export async function scheduleIntradayUpdates(
if (mode === 'crawl') {
// Get symbols that need historical crawl
symbolsToProcess = await tracker.getSymbolsForIntradayCrawl('intraday_bars', {
symbolsToProcess = await this.operationRegistry.getSymbolsForCrawl('qm', 'intraday_bars', {
limit
});
} else {
// Get symbols that need regular updates
const staleSymbols = await tracker.getStaleSymbols('intraday_bars', {
const staleSymbols = await this.operationRegistry.getStaleSymbols('qm', 'intraday_bars', {
minHoursSinceRun: forceUpdate ? 0 : 1, // Hourly updates
limit
});