浏览代码

Fixing trailing metadata ops.

Vizerai 7 年之前
父节点
当前提交
c0d251a10e
共有 1 个文件被更改,包括 4 次插入2 次删除
  1. 4 2
      src/cpp/ext/filters/census/client_filter.cc

+ 4 - 2
src/cpp/ext/filters/census/client_filter.cc

@@ -119,8 +119,10 @@ void CensusClientCallData::StartTransportStreamOpBatch(
   }
   }
   if (op->recv_trailing_metadata() != nullptr) {
   if (op->recv_trailing_metadata() != nullptr) {
     recv_trailing_metadata_ = op->recv_trailing_metadata()->batch();
     recv_trailing_metadata_ = op->recv_trailing_metadata()->batch();
-    initial_on_done_recv_trailing_metadata_ = op->on_complete();
-    op->set_on_complete(&on_done_recv_trailing_metadata_);
+    initial_on_done_recv_trailing_metadata_ =
+        op->op()->payload->recv_trailing_metadata.recv_trailing_metadata_ready;
+    op->op()->payload->recv_trailing_metadata.recv_trailing_metadata_ready =
+        &on_done_recv_trailing_metadata_;
   }
   }
   // Call next op.
   // Call next op.
   grpc_call_next_op(elem, op->op());
   grpc_call_next_op(elem, op->op());