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

added a smart queue manager and moved proxy logic to proxy manager to make handler just schedule a call to it

Browse source
This commit is contained in:
Boki 2025-06-23 10:45:06 -04:00
parent da1c52a841
commit e7c0fe2798
19 changed files with 903 additions and 231 deletions
Download patch file Download diff file Expand all files Collapse all files

21
libs/core/di/src/container/builder.ts
View file

@ -1,4 +1,4 @@
import { createContainer, InjectionMode, type AwilixContainer } from 'awilix';
import { createContainer, InjectionMode, asFunction, type AwilixContainer } from 'awilix';
import type { AppConfig as StockBotAppConfig } from '@stock-bot/config';
import { appConfigSchema, type AppConfig } from '../config/schemas';
import {
@ -100,7 +100,7 @@ export class ServiceContainerBuilder {
influxPort: 9009,
database: 'questdb',
}) : undefined,
proxy: this.options.enableProxy ? (config.proxy || { cachePrefix: 'proxy:', ttl: 3600 }) : undefined,
proxy: this.options.enableProxy ? (config.proxy || { enabled: false, cachePrefix: 'proxy:', ttl: 3600 }) : undefined,
browser: this.options.enableBrowser ? (config.browser || { headless: true, timeout: 30000 }) : undefined,
queue: this.options.enableQueue ? (config.queue || {
enabled: true,
@ -127,11 +127,12 @@ export class ServiceContainerBuilder {
// Register service container aggregate
container.register({
serviceContainer: asFunction(({
config, logger, cache, proxyManager, browser,
config, logger, cache, globalCache, proxyManager, browser,
queueManager, mongoClient, postgresClient, questdbClient
}) => ({
logger,
cache,
globalCache,
proxy: proxyManager, // Map proxyManager to proxy
browser,
queue: queueManager, // Map queueManager to queue
@ -181,10 +182,14 @@ export class ServiceContainerBuilder {
} : undefined,
queue: stockBotConfig.queue,
browser: stockBotConfig.browser,
proxy: stockBotConfig.proxy,
proxy: stockBotConfig.proxy ? {
...{
enabled: false,
cachePrefix: 'proxy:',
ttl: 3600,
},
...stockBotConfig.proxy
} : undefined,
};
}
}
// Add missing import
import { asFunction } from 'awilix';
}

13
libs/core/di/src/container/types.ts
View file

@ -1,13 +1,13 @@
import type { Browser } from '@stock-bot/browser';
import type { CacheProvider } from '@stock-bot/cache';
import type { IServiceContainer } from '@stock-bot/handlers';
import type { Logger } from '@stock-bot/logger';
import type { AppConfig } from '../config/schemas';
import type { CacheProvider } from '@stock-bot/cache';
import type { ProxyManager } from '@stock-bot/proxy';
import type { Browser } from '@stock-bot/browser';
import type { QueueManager } from '@stock-bot/queue';
import type { MongoDBClient } from '@stock-bot/mongodb';
import type { PostgreSQLClient } from '@stock-bot/postgres';
import type { ProxyManager } from '@stock-bot/proxy';
import type { QuestDBClient } from '@stock-bot/questdb';
import type { SmartQueueManager } from '@stock-bot/queue';
import type { AppConfig } from '../config/schemas';
export interface ServiceDefinitions {
// Configuration
@ -16,9 +16,10 @@ export interface ServiceDefinitions {
// Core services
cache: CacheProvider | null;
globalCache: CacheProvider | null;
proxyManager: ProxyManager | null;
browser: Browser;
queueManager: QueueManager | null;
queueManager: SmartQueueManager | null;
// Database clients
mongoClient: MongoDBClient | null;