feat: conditional check on using ws account subscriber + unused

Author: LukasDeco

sdk/src/accounts/webSocketDriftClientAccountSubscriberV2.ts

@@ -132,9 +132,6 @@ export class WebSocketDriftClientAccountSubscriberV2
 			this.subscriptionPromiseResolver = res;
 		});
 
-		// Profile public key generation
-		const pubkeyStartTime = performance.now();
-
 		const [perpMarketAccountPubkeys, spotMarketAccountPubkeys] =
 			await Promise.all([
 				Promise.all(
@@ -148,8 +145,6 @@ export class WebSocketDriftClientAccountSubscriberV2
 					)
 				),
 			]);
-		const pubkeyEndTime = performance.now();
-		const pubkeyDuration = pubkeyEndTime - pubkeyStartTime;
 
 		// Profile findAllMarketsAndOracles if needed
 		let findAllMarketsDuration = 0;
@@ -292,12 +287,7 @@ export class WebSocketDriftClientAccountSubscriberV2
 					subscribeToOraclesEndTime - subscribeToOraclesStartTime;
 				return duration;
 			})(),
-			(async () => {
-				const stateFetchStartTime = performance.now();
-				await this.stateAccountSubscriber.fetch();
-				const stateFetchEndTime = performance.now();
-				const stateFetchDuration = stateFetchEndTime - stateFetchStartTime;
-			})(),
+			this.stateAccountSubscriber.fetch(),
 		]);
 
 		const initialPerpMarketDataFromLatestData = new Map(

sdk/src/driftClient.ts

@@ -193,8 +193,8 @@ import { getOracleId } from './oracles/oracleId';
 import { SignedMsgOrderParams } from './types';
 import { sha256 } from '@noble/hashes/sha256';
 import { getOracleConfidenceFromMMOracleData } from './oracles/utils';
-import { WebSocketDriftClientAccountSubscriberV2 } from './accounts/webSocketDriftClientAccountSubscriberV2';
 import { Commitment } from 'gill';
+import { WebSocketDriftClientAccountSubscriber } from '@drift-labs/sdk';
 
 type RemainingAccountParams = {
 	userAccounts: UserAccount[];
@@ -439,7 +439,10 @@ export class DriftClient {
 				}
 			);
 		} else {
-			this.accountSubscriber = new WebSocketDriftClientAccountSubscriberV2(
+			const accountSubscriberClass =
+				config.accountSubscription?.driftClientAccountSubscriber ??
+				WebSocketDriftClientAccountSubscriber;
+			this.accountSubscriber = new accountSubscriberClass(
 				this.program,
 				config.perpMarketIndexes ?? [],
 				config.spotMarketIndexes ?? [],