Skip to content

Commit

Permalink
sdk: fix event subscriber log config type
Browse files Browse the repository at this point in the history
  • Loading branch information
wphan committed Sep 18, 2024
1 parent 8df2e3d commit beb0be4
Showing 1 changed file with 17 additions and 11 deletions.
28 changes: 17 additions & 11 deletions sdk/src/events/eventSubscriber.ts
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@ import {
LogProvider,
EventSubscriberEvents,
WebSocketLogProviderConfig,
PollingLogProviderConfig,
EventsServerLogProviderConfig,
LogProviderType,
StreamingLogProviderConfig,
Expand Down Expand Up @@ -54,32 +53,39 @@ export class EventSubscriber {
}

private initializeLogProvider(subscribe = false) {
const logProviderConfig = this.options.logProviderConfig;

if (this.currentProviderType === 'websocket') {
const logProviderConfig = this.options
.logProviderConfig as WebSocketLogProviderConfig;
this.logProvider = new WebSocketLogProvider(
// @ts-ignore
this.connection,
this.address,
this.options.commitment,
logProviderConfig.resubTimeoutMs
(
this.options.logProviderConfig as WebSocketLogProviderConfig
).resubTimeoutMs
);
} else if (this.currentProviderType === 'polling') {
const logProviderConfig = this.options
.logProviderConfig as PollingLogProviderConfig;
const frequency =
'frequency' in logProviderConfig
? logProviderConfig.frequency
: (logProviderConfig as StreamingLogProviderConfig).fallbackFrequency;
const batchSize =
'batchSize' in logProviderConfig
? logProviderConfig.batchSize
: (logProviderConfig as StreamingLogProviderConfig).fallbackBatchSize;

this.logProvider = new PollingLogProvider(
// @ts-ignore
this.connection,
this.address,
this.options.commitment,
logProviderConfig.frequency,
logProviderConfig.batchSize
frequency,
batchSize
);
} else if (this.currentProviderType === 'events-server') {
const logProviderConfig = this.options
.logProviderConfig as EventsServerLogProviderConfig;
this.logProvider = new EventsServerLogProvider(
logProviderConfig.url,
(logProviderConfig as EventsServerLogProviderConfig).url,
this.options.eventTypes,
this.options.address ? this.options.address.toString() : undefined
);
Expand Down

0 comments on commit beb0be4

Please sign in to comment.