Fix: Leak of grpc resources on terminate.

Author: MarkDuckworth

packages/firestore/src/core/component_provider.ts

@@ -483,7 +483,11 @@ export class OnlineComponentProvider {
     );
   }
 
-  terminate(): Promise<void> {
-    return remoteStoreShutdown(this.remoteStore);
+  async terminate(): Promise<void> {
+    await remoteStoreShutdown(this.remoteStore);
+
+    if (this.datastore) {
+      await this.datastore.terminate();
+    }
   }
 }

packages/firestore/src/platform/node/grpc_connection.ts

@@ -325,4 +325,11 @@ export class GrpcConnection implements Connection {
 
     return stream;
   }
+
+  terminate(): void {
+    if (this.cachedStub) {
+      this.cachedStub.close();
+      this.cachedStub = undefined;
+    }
+  }
 }

packages/firestore/src/remote/connection.ts

@@ -92,6 +92,8 @@ export interface Connection {
    */
   readonly shouldResourcePathBeIncludedInRequest: boolean;
 
+  terminate(): void;
+
   // TODO(mcg): subscribe to connection state changes.
 }
 

packages/firestore/src/remote/datastore.ts

@@ -161,6 +161,7 @@ class DatastoreImpl extends Datastore {
 
   terminate(): void {
     this.terminated = true;
+    this.connection.terminate();
   }
 }
 

packages/firestore/src/remote/rest_connection.ts

@@ -189,4 +189,8 @@ export abstract class RestConnection implements Connection {
     );
     return `${this.baseUrl}/${RPC_URL_VERSION}/${path}:${urlRpcName}`;
   }
+
+  terminate(): void {
+    // No-op
+  }
 }

packages/firestore/test/unit/remote/datastore.test.ts

@@ -41,6 +41,12 @@ use(chaiAsPromised);
 // `invokeRPC()` and `invokeStreamingRPC()`.
 describe('Datastore', () => {
   class MockConnection implements Connection {
+    terminateInvoked = false;
+
+    terminate(): void {
+      this.terminateInvoked = true;
+    }
+
     invokeRPC<Req, Resp>(
       rpcName: string,
       path: string,
@@ -131,6 +137,18 @@ describe('Datastore', () => {
       });
   });
 
+  it('Connection.terminate() invoked if Datastore terminated', async () => {
+    const connection = new MockConnection();
+    const datastore = newDatastore(
+      new EmptyAuthCredentialsProvider(),
+      new EmptyAppCheckTokenProvider(),
+      connection,
+      serializer
+    );
+    datastore.terminate();
+    await expect(connection.terminateInvoked).to.equal(true);
+  });
+
   it('DatastoreImpl.invokeRPC() rethrows a FirestoreError', async () => {
     const connection = new MockConnection();
     connection.invokeRPC = () =>