Add call options to client method stubs. (#86)

dart-archive · Jul 7, 2017 · 8134b33 · 8134b33
1 parent 021498c
commit 8134b33
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 11 deletions.
diff --git a/lib/grpc_generator.dart b/lib/grpc_generator.dart
@@ -218,9 +218,11 @@ class _GrpcMethod {
 
   void generateClientStub(IndentingWriter out) {
     out.println();
-    out.addBlock('$_clientReturnType $_dartName($_argumentType request) {', '}',
-        () {
-      out.println('final call = new ClientCall(_channel, _\$$_dartName);');
+    out.addBlock(
+        '$_clientReturnType $_dartName($_argumentType request, {CallOptions options}) {',
+        '}', () {
+      out.println(
+          'final call = new ClientCall(_channel, _\$$_dartName, options: options);');
       if (_clientStreaming) {
         out.println('request.pipe(call.request);');
       } else {

diff --git a/test/file_generator_test.dart b/test/file_generator_test.dart
@@ -545,30 +545,32 @@ class TestClient {
 
   TestClient(this._channel);
 
-  ResponseFuture<Output> unary(Input request) {
-    final call = new ClientCall(_channel, _$unary);
+  ResponseFuture<Output> unary(Input request, {CallOptions options}) {
+    final call = new ClientCall(_channel, _$unary, options: options);
     call.request
       ..add(request)
       ..close();
     return new ResponseFuture(call);
   }
 
-  ResponseFuture<Output> clientStreaming(Stream<Input> request) {
-    final call = new ClientCall(_channel, _$clientStreaming);
+  ResponseFuture<Output> clientStreaming(Stream<Input> request,
+      {CallOptions options}) {
+    final call = new ClientCall(_channel, _$clientStreaming, options: options);
     request.pipe(call.request);
     return new ResponseFuture(call);
   }
 
-  ResponseStream<Output> serverStreaming(Input request) {
-    final call = new ClientCall(_channel, _$serverStreaming);
+  ResponseStream<Output> serverStreaming(Input request, {CallOptions options}) {
+    final call = new ClientCall(_channel, _$serverStreaming, options: options);
     call.request
       ..add(request)
       ..close();
     return new ResponseStream(call);
   }
 
-  ResponseStream<Output> bidirectional(Stream<Input> request) {
-    final call = new ClientCall(_channel, _$bidirectional);
+  ResponseStream<Output> bidirectional(Stream<Input> request,
+      {CallOptions options}) {
+    final call = new ClientCall(_channel, _$bidirectional, options: options);
     request.pipe(call.request);
     return new ResponseStream(call);
   }