fix stream to be of type Observable

debugged · May 3, 2020 · dd5a808 · dd5a808
1 parent c331747
commit dd5a808
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/main.ts b/src/main.ts
@@ -990,8 +990,8 @@ function generateService(
       requestFn = requestFn.addParameter('metadata?', "Metadata@grpc");
     }
 
-    // Return observable for interface only configuration and passing returnObservable=true
-    if (options.returnObservable) {
+    // Return observable for interface only configuration, passing returnObservable=true and methodDesc.serverStreaming=true
+    if (options.returnObservable || methodDesc.serverStreaming) {
       requestFn = requestFn.returns(responseObservable(typeMap, methodDesc));
     } else {
       requestFn = requestFn.returns(responsePromise(typeMap, methodDesc));
@@ -1270,6 +1270,9 @@ function generateDataLoadersType(): InterfaceSpec {
 }
 
 function requestType(typeMap: TypeMap, methodDesc: MethodDescriptorProto): TypeName {
+  if(methodDesc.clientStreaming) {
+    return TypeNames.anyType("Observable@rxjs").param(messageToTypeName(typeMap, methodDesc.inputType));
+  }
   return messageToTypeName(typeMap, methodDesc.inputType);
 }