sdk: add retry logic for polling account subscribers (#75)

Author: crispheaney

sdk/src/accounts/pollingClearingHouseAccountSubscriber.ts

@@ -85,8 +85,14 @@ export class PollingClearingHouseAccountSubscriber
 
 		await this.updateAccountsToPoll();
 		await this.addToAccountLoader();
-		await this.fetch();
-		const subscriptionSucceeded = this.didSubscriptionSucceed();
+
+		let subscriptionSucceeded = false;
+		let retries = 0;
+		while (!subscriptionSucceeded && retries < 5) {
+			await this.fetch();
+			subscriptionSucceeded = this.didSubscriptionSucceed();
+			retries++;
+		}
 
 		if (subscriptionSucceeded) {
 			this.eventEmitter.emit('update');
@@ -237,6 +243,10 @@ export class PollingClearingHouseAccountSubscriber
 					// @ts-ignore
 					this.eventEmitter.emit(accountToPoll.eventType, account);
 					this.eventEmitter.emit('update');
+
+					if (!this.isSubscribed) {
+						this.isSubscribed = this.didSubscriptionSucceed();
+					}
 				}
 			);
 		}

sdk/src/accounts/pollingOracleSubscriber.ts

@@ -37,11 +37,21 @@ export class PollingOracleSubscriber implements OracleSubscriber {
 		}
 
 		this.addToAccountLoader();
-		await this.fetch();
-		this.eventEmitter.emit('update');
 
-		this.isSubscribed = true;
-		return true;
+		let subscriptionSucceeded = false;
+		let retries = 0;
+		while (!subscriptionSucceeded && retries < 5) {
+			await this.fetch();
+			subscriptionSucceeded = this.didSubscriptionSucceed();
+			retries++;
+		}
+
+		if (subscriptionSucceeded) {
+			this.eventEmitter.emit('update');
+		}
+
+		this.isSubscribed = subscriptionSucceeded;
+		return subscriptionSucceeded;
 	}
 
 	addToAccountLoader(): void {
@@ -100,4 +110,8 @@ export class PollingOracleSubscriber implements OracleSubscriber {
 		this.assertIsSubscribed();
 		return this.oraclePriceData;
 	}
+
+	didSubscriptionSucceed(): boolean {
+		return !!this.oraclePriceData;
+	}
 }

sdk/src/accounts/pollingTokenAccountSubscriber.ts

@@ -36,11 +36,20 @@ export class PollingTokenAccountSubscriber implements TokenAccountSubscriber {
 		}
 
 		this.addToAccountLoader();
-		await this.fetch();
-		this.eventEmitter.emit('update');
+		let subscriptionSucceeded = false;
+		let retries = 0;
+		while (!subscriptionSucceeded && retries < 5) {
+			await this.fetch();
+			subscriptionSucceeded = this.didSubscriptionSucceed();
+			retries++;
+		}
+
+		if (subscriptionSucceeded) {
+			this.eventEmitter.emit('update');
+		}
 
-		this.isSubscribed = true;
-		return true;
+		this.isSubscribed = subscriptionSucceeded;
+		return subscriptionSucceeded;
 	}
 
 	addToAccountLoader(): void {
@@ -96,4 +105,8 @@ export class PollingTokenAccountSubscriber implements TokenAccountSubscriber {
 		this.assertIsSubscribed();
 		return this.tokenAccount;
 	}
+
+	didSubscriptionSucceed(): boolean {
+		return !!this.tokenAccount;
+	}
 }

sdk/src/accounts/pollingUserAccountSubscriber.ts

@@ -52,9 +52,14 @@ export class PollingUserAccountSubscriber implements UserAccountSubscriber {
 		}
 
 		await this.addToAccountLoader();
-		await this.fetchIfUnloaded();
 
-		const subscriptionSucceeded = this.didSubscriptionSucceed();
+		let subscriptionSucceeded = false;
+		let retries = 0;
+		while (!subscriptionSucceeded && retries < 5) {
+			await this.fetchIfUnloaded();
+			subscriptionSucceeded = this.didSubscriptionSucceed();
+			retries++;
+		}
 
 		if (subscriptionSucceeded) {
 			this.eventEmitter.emit('update');