Make Blockchain Subscribe methods return Unsubscribe callback

2025-04-24 07:54:51 +00:00
parent 1d5e3131c8
commit 26a61291a0
4 changed files with 19 additions and 3 deletions
@@ -173,6 +173,9 @@ export class BlockchainInMemory extends BaseBlockchain {

    // Add the address notification callback.
    this.addressSubscriptions.get(lockscriptHex)!.add(callback);
+
+    // Return an unsubscribe callback.
+    return async () => await this.unsubscribeAddress(address, callback);
  }

  async unsubscribeAddress(address: string, callback: BlockchainCallback) {
@@ -206,6 +209,10 @@ export class BlockchainInMemory extends BaseBlockchain {
      this.transactionSubscriptions.set(transactionHash, new Set());
    }
    this.transactionSubscriptions.get(transactionHash)!.add(callback);
+
+    // Return an unsubscribe callback.
+    return async () =>
+      await this.unsubscribeTransaction(transactionHash, callback);
  }

  async unsubscribeTransaction(