46 lines
1.5 KiB
TypeScript
46 lines
1.5 KiB
TypeScript
import { TradovateClient } from './tradovate-class';
|
|
import { getFirms } from './db';
|
|
|
|
// Use global to persist the client pool across HMR reloads in dev mode
|
|
const g = global as typeof globalThis & {
|
|
__tradovateClients?: Map<number, TradovateClient>;
|
|
__tradovateClientsInitialized?: boolean;
|
|
};
|
|
|
|
function ensureMap(): Map<number, TradovateClient> {
|
|
if (!g.__tradovateClients) {
|
|
g.__tradovateClients = new Map();
|
|
}
|
|
return g.__tradovateClients;
|
|
}
|
|
|
|
export function initClient(id: number, username: string, password: string, firmName: string): TradovateClient {
|
|
const map = ensureMap();
|
|
const client = new TradovateClient(username, password, async () => {
|
|
console.log(`[${firmName}] sync complete — ${client.accountList.length} account(s)`);
|
|
});
|
|
map.set(id, client);
|
|
return client;
|
|
}
|
|
|
|
export function removeClient(id: number): void {
|
|
ensureMap().delete(id);
|
|
}
|
|
|
|
export function getClients(): Map<number, TradovateClient> {
|
|
const map = ensureMap();
|
|
if (!g.__tradovateClientsInitialized) {
|
|
g.__tradovateClientsInitialized = true;
|
|
try {
|
|
const firms = getFirms();
|
|
for (const firm of firms) {
|
|
initClient(firm.id, firm.username, firm.password, firm.name);
|
|
}
|
|
console.log(`[clients] Initialized ${firms.length} Tradovate client(s)`);
|
|
} catch (err) {
|
|
console.error('[clients] Failed to initialize clients', err);
|
|
}
|
|
}
|
|
return map;
|
|
}
|