Define a generic interface to allow flexibility (#160)

* Define a generic interface to allow flexibility * Add intercepting call * Remove export * Use the grpc interface inside the factory * closer match to design doc
GoogleCloudPlatform · Jan 23, 2023 · 7e12a66 · 7e12a66
1 parent db52df9
commit 7e12a66
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/grpc_interface.ts b/src/grpc_interface.ts
@@ -0,0 +1,6 @@
+export interface GrpcInterface {
+  Channel: any;
+  connectivityState: any;
+  status: any;
+  InterceptingCall: any;
+};
diff --git a/src/index.ts b/src/index.ts
@@ -16,6 +16,7 @@
  *
  */
 import * as grpcType from '@grpc/grpc-js';
+import {GrpcInterface} from './grpc_interface';
 import * as util from 'util';
 
 import {ChannelRef} from './channel_ref';
@@ -30,7 +31,8 @@ import AffinityConfig = protoRoot.grpc.gcp.AffinityConfig;
 
 type GrpcModule = typeof grpcType;
 
-export = (grpc: GrpcModule) => {
+export = (grpc: GrpcInterface) => setup(grpc as GrpcModule);
+const setup = (grpc: GrpcModule) => {
   const GcpChannelFactory = getGcpChannelFactoryClass(grpc);
   /**
    * Create ApiConfig proto message from config object.