trade/trade-drift-dlob
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #278 from drift-labs/jack/env-clients

Browse Source 
chore: update clients to be env vars
This commit is contained in:
jack
2024-11-21 12:24:30 -05:00
committed by GitHub
parent d43d2d2591 4b9b337a60
commit 3682cbf54b
4 changed files with 22 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/index.ts
View File

@@ -62,8 +62,14 @@ setGlobalDispatcher(
require('dotenv').config(); require('dotenv').config();
// Reading in Redis env vars const envClients = [];
const REDIS_CLIENTS = [RedisClientPrefix.DLOB, RedisClientPrefix.DLOB_HELIUS]; const clients = process.env.REDIS_CLIENT.trim()
.replace(/^\[|\]$/g, '')
.split(/\s*,\s*/);
clients.forEach((client) => envClients.push(RedisClientPrefix[client]));
const REDIS_CLIENTS = envClients.length ? envClients : [RedisClientPrefix.DLOB, RedisClientPrefix.DLOB_HELIUS];
console.log('Redis Clients:', REDIS_CLIENTS); console.log('Redis Clients:', REDIS_CLIENTS);
const driftEnv = (process.env.ENV || 'devnet') as DriftEnv; const driftEnv = (process.env.ENV || 'devnet') as DriftEnv;

4
src/publishers/tradesPublisher.ts
View File

@@ -36,7 +36,9 @@ setGlobalDispatcher(
require('dotenv').config(); require('dotenv').config();
const driftEnv = (process.env.ENV || 'devnet') as DriftEnv; const driftEnv = (process.env.ENV || 'devnet') as DriftEnv;
const commitHash = process.env.COMMIT; const commitHash = process.env.COMMIT;
const redisClientPrefix = RedisClientPrefix.DLOB_HELIUS; const REDIS_CLIENT = process.env.REDIS_CLIENT || 'DLOB';
console.log('Redis Clients:', REDIS_CLIENT);
const redisClientPrefix = RedisClientPrefix[REDIS_CLIENT];
//@ts-ignore //@ts-ignore
const sdkConfig = initialize({ env: process.env.ENV }); const sdkConfig = initialize({ env: process.env.ENV });

9
src/serverLite.ts
View File

@@ -38,7 +38,14 @@ setGlobalDispatcher(
require('dotenv').config(); require('dotenv').config();
// Reading in Redis env vars // Reading in Redis env vars
const REDIS_CLIENTS = [RedisClientPrefix.DLOB, RedisClientPrefix.DLOB_HELIUS]; const envClients = [];
const clients = process.env.REDIS_CLIENT.trim()
.replace(/^\[|\]$/g, '')
.split(/\s*,\s*/);
clients.forEach((client) => envClients.push(RedisClientPrefix[client]));
const REDIS_CLIENTS = envClients.length ? envClients : [RedisClientPrefix.DLOB, RedisClientPrefix.DLOB_HELIUS];
console.log('Redis Clients:', REDIS_CLIENTS); console.log('Redis Clients:', REDIS_CLIENTS);
const driftEnv = (process.env.ENV || 'devnet') as DriftEnv; const driftEnv = (process.env.ENV || 'devnet') as DriftEnv;

4
src/wsConnectionManager.ts
View File

@@ -34,7 +34,9 @@ const WS_PORT = process.env.WS_PORT || '3000';
console.log(`WS LISTENER PORT : ${WS_PORT}`); console.log(`WS LISTENER PORT : ${WS_PORT}`);
const MAX_BUFFERED_AMOUNT = 300000; const MAX_BUFFERED_AMOUNT = 300000;
const CHANNEL_PREFIX = RedisClientPrefix.DLOB; const REDIS_CLIENT = process.env.REDIS_CLIENT || 'DLOB';
const CHANNEL_PREFIX = RedisClientPrefix[REDIS_CLIENT];
console.log('Redis Clients:', REDIS_CLIENT);
const CHANNEL_PREFIX_HELIUS = RedisClientPrefix.DLOB_HELIUS; const CHANNEL_PREFIX_HELIUS = RedisClientPrefix.DLOB_HELIUS;
const sanitiseChannelForClient = (channel: string | undefined): string => { const sanitiseChannelForClient = (channel: string | undefined): string => {