fix: oracle string map not initted grpc v2 (#1936)

* fix: oracle string map not initted grpc v2

* fix: prettier

Author: LukasDeco

sdk/scripts/client-test.ts

@@ -104,6 +104,12 @@ async function initializeGrpcDriftClientV2() {
 					'Perp market data market index:',
 					perpMarketData?.marketIndex
 				);
+				const oracle = driftClient.getOracleDataForPerpMarket(data.marketIndex);
+				const mmOracle = driftClient.getMMOracleDataForPerpMarket(
+					data.marketIndex
+				);
+				console.log('Perp oracle price:', oracle.price.toString());
+				console.log('Perp MM oracle price:', mmOracle.price.toString());
 				perpMarketUpdateCount++;
 				if (
 					perpMarketUpdateCount >= 10 &&
@@ -127,6 +133,8 @@ async function initializeGrpcDriftClientV2() {
 					'Spot market data market index:',
 					spotMarketData?.marketIndex
 				);
+				const oracle = driftClient.getOracleDataForSpotMarket(data.marketIndex);
+				console.log('Spot oracle price:', oracle.price.toString());
 				spotMarketUpdateCount++;
 				if (
 					perpMarketUpdateCount >= 10 &&
@@ -175,6 +183,21 @@ async function initializeGrpcDriftClientV2() {
 	await driftClient.subscribe();
 	console.log('DriftClient initialized and listening for updates.');
 
+	for (const marketIndex of config.perpMarketIndexes) {
+		const oracle = driftClient.getOracleDataForPerpMarket(marketIndex);
+		const mmOracle = driftClient.getMMOracleDataForPerpMarket(marketIndex);
+		console.log('Initial perp oracle price:', oracle.price.toString());
+		console.log('Initial perp MM oracle price:', mmOracle.price.toString());
+	}
+
+	for (const marketIndex of config.spotMarketIndexes) {
+		const oracle = driftClient.getOracleDataForSpotMarket(marketIndex);
+		console.log('Initial spot oracle price:', oracle.price.toString());
+	}
+
+	const stateAccount = driftClient.getStateAccount();
+	console.log('Initial state account:', stateAccount.toString());
+
 	await updatePromise;
 	console.log('Received required number of updates.');
 }

sdk/src/accounts/grpcDriftClientAccountSubscriberV2.ts

@@ -149,6 +149,58 @@ export class grpcDriftClientAccountSubscriberV2 extends WebSocketDriftClientAcco
 		);
 	}
 
+	override async setPerpOracleMap() {
+		const perpMarketsMap = this.perpMarketsSubscriber?.getAccountDataMap();
+		const perpMarkets = Array.from(perpMarketsMap.values());
+		const addOraclePromises = [];
+		for (const perpMarket of perpMarkets) {
+			if (!perpMarket || !perpMarket.data) {
+				continue;
+			}
+			const perpMarketAccount = perpMarket.data;
+			const perpMarketIndex = perpMarketAccount.marketIndex;
+			const oracle = perpMarketAccount.amm.oracle;
+			const oracleId = getOracleId(oracle, perpMarket.data.amm.oracleSource);
+			if (!this.oracleSubscribers.has(oracleId)) {
+				addOraclePromises.push(
+					this.addOracle({
+						publicKey: oracle,
+						source: perpMarket.data.amm.oracleSource,
+					})
+				);
+			}
+			this.perpOracleMap.set(perpMarketIndex, oracle);
+			this.perpOracleStringMap.set(perpMarketIndex, oracleId);
+		}
+		await Promise.all(addOraclePromises);
+	}
+
+	override async setSpotOracleMap() {
+		const spotMarketsMap = this.spotMarketsSubscriber?.getAccountDataMap();
+		const spotMarkets = Array.from(spotMarketsMap.values());
+		const addOraclePromises = [];
+		for (const spotMarket of spotMarkets) {
+			if (!spotMarket || !spotMarket.data) {
+				continue;
+			}
+			const spotMarketAccount = spotMarket.data;
+			const spotMarketIndex = spotMarketAccount.marketIndex;
+			const oracle = spotMarketAccount.oracle;
+			const oracleId = getOracleId(oracle, spotMarketAccount.oracleSource);
+			if (!this.oracleSubscribers.has(oracleId)) {
+				addOraclePromises.push(
+					this.addOracle({
+						publicKey: oracle,
+						source: spotMarketAccount.oracleSource,
+					})
+				);
+			}
+			this.spotOracleMap.set(spotMarketIndex, oracle);
+			this.spotOracleStringMap.set(spotMarketIndex, oracleId);
+		}
+		await Promise.all(addOraclePromises);
+	}
+
 	override async subscribeToPerpMarketAccounts(): Promise<boolean> {
 		const perpMarketIndexToAccountPubkeys: Array<[number, PublicKey]> =
 			await Promise.all(

sdk/src/accounts/grpcMultiAccountSubscriber.ts

@@ -101,6 +101,10 @@ export class grpcMultiAccountSubscriber<T> {
 		return this.dataMap.get(accountPubkey.toBase58());
 	}
 
+	getAccountDataMap(): Map<string, DataAndSlot<T>> {
+		return this.dataMap;
+	}
+
 	async subscribe(
 		accounts: PublicKey[],
 		onChange: (