Event subscription (TS sdk)

explorer/client/src/utils/api/DefaultRpcClient.ts

@@ -7,5 +7,12 @@ import { getEndpoint, Network } from './rpcSetting';
 
 export { Network, getEndpoint };
 
-export const DefaultRpcClient = (network: Network | string) =>
-    new JsonRpcProvider(getEndpoint(network));
+const defaultRpcMap: Map<Network | string, JsonRpcProvider> = new Map();
+export const DefaultRpcClient = (network: Network | string) => {
+    const existingClient = defaultRpcMap.get(network);
+    if (existingClient) return existingClient;
+
+    const provider = new JsonRpcProvider(getEndpoint(network));
+    defaultRpcMap.set(network, provider);
+    return provider;
+};

sdk/typescript/package.json

@@ -86,6 +86,7 @@
     "buffer": "^6.0.3",
     "cross-fetch": "^3.1.5",
     "jayson": "^3.6.6",
+    "rpc-websockets": "^7.5.0",
     "js-sha3": "^0.8.0",
     "lossless-json": "^1.0.5",
     "tweetnacl": "^1.0.3"

sdk/typescript/src/providers/json-rpc-provider.ts

@@ -30,16 +30,22 @@ import {
   SuiObjectRef,
   getObjectReference,
   Coin,
+  SuiEventFilter,
+  SuiEventEnvelope,
+  SubscriptionId,
   ExecuteTransactionRequestType,
   SuiExecuteTransactionResponse,
 } from '../types';
 import { SignatureScheme } from '../cryptography/publickey';
+import { DEFAULT_CLIENT_OPTIONS, WebsocketClient, WebsocketClientOptions } from '../rpc/websocket-client';
 
 const isNumber = (val: any): val is number => typeof val === 'number';
 const isAny = (_val: any): _val is any => true;
 
+
 export class JsonRpcProvider extends Provider {
   protected client: JsonRpcClient;
+  protected wsClient: WebsocketClient;
   /**
    * Establish a connection to a Sui RPC endpoint
    *
@@ -54,10 +60,13 @@ export class JsonRpcProvider extends Provider {
    */
   constructor(
     public endpoint: string,
-    public skipDataValidation: boolean = false
+    public skipDataValidation: boolean = false,
+    public socketOptions: WebsocketClientOptions = DEFAULT_CLIENT_OPTIONS
   ) {
     super();
+
     this.client = new JsonRpcClient(endpoint);
+    this.wsClient = new WebsocketClient(endpoint, skipDataValidation, socketOptions);
   }
 
   // Move info
@@ -421,4 +430,15 @@ export class JsonRpcProvider extends Provider {
       );
     }
   }
+
+  async subscribeEvent(
+    filter: SuiEventFilter,
+    onMessage: (event: SuiEventEnvelope) => void
+  ): Promise<SubscriptionId> {
+    return this.wsClient.subscribeEvent(filter, onMessage);
+  }
+
+  async unsubscribeEvent(id: SubscriptionId): Promise<boolean> {
+    return this.wsClient.unsubscribeEvent(id);
+  }
 }

sdk/typescript/src/providers/provider.ts

@@ -14,6 +14,9 @@ import {
   SuiMoveNormalizedStruct,
   SuiMoveNormalizedModule,
   SuiMoveNormalizedModules,
+  SuiEventFilter,
+  SuiEventEnvelope,
+  SubscriptionId,
   ExecuteTransactionRequestType,
   SuiExecuteTransactionResponse,
 } from '../types';
@@ -136,5 +139,21 @@ export abstract class Provider {
   ): Promise<SuiMoveNormalizedStruct>;
 
   abstract syncAccountState(address: string): Promise<any>;
+
+  /**
+   * Subscribe to get notifications whenever an event matching the filter occurs
+   * @param filter - filter describing the subset of events to follow
+   * @param onMessage - function to run when we receive a notification of a new event matching the filter
+   */
+  abstract subscribeEvent(
+    filter: SuiEventFilter,
+    onMessage: (event: SuiEventEnvelope) => void
+  ): Promise<SubscriptionId>;
+
+  /**
+   * Unsubscribe from an event subscription
+   * @param id - subscription id to unsubscribe from (previously received from subscribeEvent)
+   */
+  abstract unsubscribeEvent(id: SubscriptionId): Promise<boolean>;
   // TODO: add more interface methods
 }

sdk/typescript/src/providers/void-provider.ts

@@ -16,6 +16,9 @@ import {
   SuiMoveNormalizedStruct,
   SuiMoveNormalizedModule,
   SuiMoveNormalizedModules,
+  SuiEventFilter,
+  SuiEventEnvelope,
+  SubscriptionId,
   ExecuteTransactionRequestType,
   SuiExecuteTransactionResponse,
 } from '../types';
@@ -123,6 +126,17 @@ export class VoidProvider extends Provider {
     throw this.newError('syncAccountState');
   }
 
+  async subscribeEvent(
+    _filter: SuiEventFilter,
+    _onMessage: (event: SuiEventEnvelope) => void
+  ): Promise<SubscriptionId> {
+      throw this.newError('subscribeEvent');
+  }
+
+  async unsubscribeEvent(_id: SubscriptionId): Promise<boolean> {
+    throw this.newError('unsubscribeEvent');
+  }
+
   private newError(operation: string): Error {
     return new Error(`Please use a valid provider for ${operation}`);
   }