chore: make create client dynamic (#2368)

midwayjs · Oct 10, 2022 · 3b3c05b · 3b3c05b
1 parent 88b720b
commit 3b3c05b
Showing 1 changed file with 34 additions and 29 deletions.
diff --git a/packages/grpc/src/comsumer/clients.ts b/packages/grpc/src/comsumer/clients.ts
@@ -36,36 +36,41 @@ export class GRPCClients extends Map {
       return;
     }
     for (const cfg of this.grpcConfig['services']) {
-      const packageDefinition = await loadProto({
-        loaderOptions: cfg.loaderOptions,
-        protoPath: cfg.protoPath,
-      });
-      const allProto = loadPackageDefinition(packageDefinition);
-      const packageProto: any = finePackageProto(allProto, cfg.package);
-      for (const definition in packageDefinition) {
-        if (!packageDefinition[definition]['format']) {
-          const serviceName = definition.replace(`${cfg.package}.`, '');
-          const connectionService = new packageProto[serviceName](
-            cfg.url,
-            credentials.createInsecure(),
-            cfg.clientOptions
-          );
-          for (const methodName of Object.keys(packageDefinition[definition])) {
-            const originMethod = connectionService[methodName];
-            connectionService[methodName] = (
-              clientOptions: IClientOptions = {}
-            ) => {
-              return this.getClientRequestImpl(
-                connectionService,
-                originMethod,
-                clientOptions
-              );
-            };
-            connectionService[Utils.camelCase(methodName)] =
-              connectionService[methodName];
-          }
-          this.set(definition, connectionService);
+      await this.createClient(cfg);
+    }
+  }
+
+  async createClient<T>(options: IGRPCClientServiceOptions): Promise<T> {
+    const packageDefinition = await loadProto({
+      loaderOptions: options.loaderOptions,
+      protoPath: options.protoPath,
+    });
+    const allProto = loadPackageDefinition(packageDefinition);
+    const packageProto: any = finePackageProto(allProto, options.package);
+    for (const definition in packageDefinition) {
+      if (!packageDefinition[definition]['format']) {
+        const serviceName = definition.replace(`${options.package}.`, '');
+        const connectionService = new packageProto[serviceName](
+          options.url,
+          credentials.createInsecure(),
+          options.clientOptions
+        );
+        for (const methodName of Object.keys(packageDefinition[definition])) {
+          const originMethod = connectionService[methodName];
+          connectionService[methodName] = (
+            clientOptions: IClientOptions = {}
+          ) => {
+            return this.getClientRequestImpl(
+              connectionService,
+              originMethod,
+              clientOptions
+            );
+          };
+          connectionService[Utils.camelCase(methodName)] =
+            connectionService[methodName];
         }
+        this.set(definition, connectionService);
+        return connectionService;
       }
     }
   }