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

Updated subscribers to use polling

Browse Source
This commit is contained in:
Luke Steyn
2023-05-31 23:05:57 +09:00
parent 92f78dd540
commit a86a1f0e3e
1 changed files with 26 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
src/index.ts
View File

@@ -220,28 +220,32 @@ const endpointResponseTimeHistogram = meter.createHistogram(
const getPhoenixSubscriber = ( const getPhoenixSubscriber = (
driftClient: DriftClient, driftClient: DriftClient,
marketConfig: SpotMarketConfig marketConfig: SpotMarketConfig,
accountLoader: BulkAccountLoader
) => { ) => {
return new PhoenixSubscriber({ return new PhoenixSubscriber({
connection: driftClient.connection, connection: driftClient.connection,
programId: new PublicKey(sdkConfig.PHOENIX), programId: new PublicKey(sdkConfig.PHOENIX),
marketAddress: marketConfig.phoenixMarket, marketAddress: marketConfig.phoenixMarket,
accountSubscription: { accountSubscription: {
type: "websocket", type: "polling",
accountLoader,
}, },
}); });
}; };
const getSerumSubscriber = ( const getSerumSubscriber = (
driftClient: DriftClient, driftClient: DriftClient,
marketConfig: SpotMarketConfig marketConfig: SpotMarketConfig,
accountLoader: BulkAccountLoader
) => { ) => {
return new SerumSubscriber({ return new SerumSubscriber({
connection: driftClient.connection, connection: driftClient.connection,
programId: new PublicKey(sdkConfig.SERUM_V3), programId: new PublicKey(sdkConfig.SERUM_V3),
marketAddress: marketConfig.serumMarket, marketAddress: marketConfig.serumMarket,
accountSubscription: { accountSubscription: {
type: "websocket", type: "polling",
accountLoader,
}, },
}); });
}; };
@@ -255,7 +259,10 @@ type SubscriberLookup = {
let MARKET_SUBSCRIBERS: SubscriberLookup = {}; let MARKET_SUBSCRIBERS: SubscriberLookup = {};
const initializeAllMarketSubscribers = async (driftClient: DriftClient) => { const initializeAllMarketSubscribers = async (
driftClient: DriftClient,
bulkAccountLoader: BulkAccountLoader
) => {
const markets: SubscriberLookup = {}; const markets: SubscriberLookup = {};
for (const market of sdkConfig.SPOT_MARKETS) { for (const market of sdkConfig.SPOT_MARKETS) {
@@ -265,13 +272,21 @@ const initializeAllMarketSubscribers = async (driftClient: DriftClient) => {
}; };
if (market.phoenixMarket) { if (market.phoenixMarket) {
const phoenixSubscriber = getPhoenixSubscriber(driftClient, market); const phoenixSubscriber = getPhoenixSubscriber(
driftClient,
market,
bulkAccountLoader
);
await phoenixSubscriber.subscribe(); await phoenixSubscriber.subscribe();
markets[market.marketIndex].phoenix = phoenixSubscriber; markets[market.marketIndex].phoenix = phoenixSubscriber;
} }
if (market.serumMarket) { if (market.serumMarket) {
const serumSubscriber = getSerumSubscriber(driftClient, market); const serumSubscriber = getSerumSubscriber(
driftClient,
market,
bulkAccountLoader
);
await serumSubscriber.subscribe(); await serumSubscriber.subscribe();
markets[market.marketIndex].serum = serumSubscriber; markets[market.marketIndex].serum = serumSubscriber;
} }
@@ -377,7 +392,10 @@ const main = async () => {
}); });
}); });
MARKET_SUBSCRIBERS = await initializeAllMarketSubscribers(driftClient); MARKET_SUBSCRIBERS = await initializeAllMarketSubscribers(
driftClient,
bulkAccountLoader
);
// start http server listening to /health endpoint using http package // start http server listening to /health endpoint using http package
app.get("/health", handleResponseTime, async (req, res, next) => { app.get("/health", handleResponseTime, async (req, res, next) => {