@drift-labs/sdk-browser/src/accounts/bulkUserStatsSubscription.ts

SDK for Drift Protocol

import { UserStats } from '../userStats';
import { BulkAccountLoader } from './bulkAccountLoader';
import { PollingUserStatsAccountSubscriber } from './pollingUserStatsAccountSubscriber';

/**
 * @param userStats
 * @param accountLoader
 */
export async function bulkPollingUserStatsSubscribe(
	userStats: UserStats[],
	accountLoader: BulkAccountLoader
): Promise<void> {
	if (userStats.length === 0) {
		await accountLoader.load();
		return;
	}

	await Promise.all(
		userStats.map((userStat) => {
			return (
				userStat.accountSubscriber as PollingUserStatsAccountSubscriber
			).addToAccountLoader();
		})
	);

	await accountLoader.load();

	await Promise.all(
		userStats.map(async (userStat) => {
			return userStat.subscribe();
		})
	);
}