Skip to content

Commit ea31442

Browse files
authored
fix(generator): do not override default sleeper in streaming-read RPCs (#12920)
1 parent 2c26eb9 commit ea31442

7 files changed

Lines changed: 7 additions & 14 deletions

File tree

generator/integration_tests/golden/v1/internal/golden_kitchen_sink_connection_impl.cc

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -173,8 +173,7 @@ GoldenKitchenSinkConnectionImpl::StreamingRead(google::test::admin::database::v1
173173
};
174174
auto resumable =
175175
internal::MakeResumableStreamingReadRpc<google::test::admin::database::v1::Response, google::test::admin::database::v1::Request>(
176-
retry_policy(*current), backoff_policy(*current),
177-
[](std::chrono::milliseconds) {}, factory,
176+
retry_policy(*current), backoff_policy(*current), factory,
178177
GoldenKitchenSinkStreamingReadStreamingUpdater, request);
179178
return internal::MakeStreamRange(internal::StreamReader<google::test::admin::database::v1::Response>(
180179
[resumable] { return resumable->Read(); }));

generator/internal/connection_impl_generator.cc

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -310,8 +310,7 @@ StreamRange<$response_type$>
310310
};
311311
auto resumable =
312312
internal::MakeResumableStreamingReadRpc<$response_type$, $request_type$>(
313-
retry_policy(*current), backoff_policy(*current),
314-
[](std::chrono::milliseconds) {}, factory,
313+
retry_policy(*current), backoff_policy(*current), factory,
315314
$service_name$$method_name$StreamingUpdater, request);
316315
return internal::MakeStreamRange(internal::StreamReader<$response_type$>(
317316
[resumable] { return resumable->Read(); }));

google/cloud/aiplatform/v1/internal/featurestore_online_serving_connection_impl.cc

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -99,8 +99,7 @@ FeaturestoreOnlineServingServiceConnectionImpl::StreamingReadFeatureValues(
9999
auto resumable = internal::MakeResumableStreamingReadRpc<
100100
google::cloud::aiplatform::v1::ReadFeatureValuesResponse,
101101
google::cloud::aiplatform::v1::StreamingReadFeatureValuesRequest>(
102-
retry_policy(*current), backoff_policy(*current),
103-
[](std::chrono::milliseconds) {}, factory,
102+
retry_policy(*current), backoff_policy(*current), factory,
104103
FeaturestoreOnlineServingServiceStreamingReadFeatureValuesStreamingUpdater,
105104
request);
106105
return internal::MakeStreamRange(

google/cloud/aiplatform/v1/internal/prediction_connection_impl.cc

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -102,8 +102,7 @@ PredictionServiceConnectionImpl::ServerStreamingPredict(
102102
auto resumable = internal::MakeResumableStreamingReadRpc<
103103
google::cloud::aiplatform::v1::StreamingPredictResponse,
104104
google::cloud::aiplatform::v1::StreamingPredictRequest>(
105-
retry_policy(*current), backoff_policy(*current),
106-
[](std::chrono::milliseconds) {}, factory,
105+
retry_policy(*current), backoff_policy(*current), factory,
107106
PredictionServiceServerStreamingPredictStreamingUpdater, request);
108107
return internal::MakeStreamRange(
109108
internal::StreamReader<

google/cloud/aiplatform/v1/internal/tensorboard_connection_impl.cc

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -668,8 +668,7 @@ TensorboardServiceConnectionImpl::ReadTensorboardBlobData(
668668
auto resumable = internal::MakeResumableStreamingReadRpc<
669669
google::cloud::aiplatform::v1::ReadTensorboardBlobDataResponse,
670670
google::cloud::aiplatform::v1::ReadTensorboardBlobDataRequest>(
671-
retry_policy(*current), backoff_policy(*current),
672-
[](std::chrono::milliseconds) {}, factory,
671+
retry_policy(*current), backoff_policy(*current), factory,
673672
TensorboardServiceReadTensorboardBlobDataStreamingUpdater, request);
674673
return internal::MakeStreamRange(
675674
internal::StreamReader<

google/cloud/bigquery/storage/v1/internal/bigquery_read_connection_impl.cc

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -89,8 +89,7 @@ BigQueryReadConnectionImpl::ReadRows(
8989
auto resumable = internal::MakeResumableStreamingReadRpc<
9090
google::cloud::bigquery::storage::v1::ReadRowsResponse,
9191
google::cloud::bigquery::storage::v1::ReadRowsRequest>(
92-
retry_policy(*current), backoff_policy(*current),
93-
[](std::chrono::milliseconds) {}, factory,
92+
retry_policy(*current), backoff_policy(*current), factory,
9493
BigQueryReadReadRowsStreamingUpdater, request);
9594
return internal::MakeStreamRange(
9695
internal::StreamReader<

google/cloud/osconfig/agentendpoint/v1/internal/agent_endpoint_connection_impl.cc

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,8 +84,7 @@ AgentEndpointServiceConnectionImpl::ReceiveTaskNotification(
8484
ReceiveTaskNotificationResponse,
8585
google::cloud::osconfig::agentendpoint::v1::
8686
ReceiveTaskNotificationRequest>(
87-
retry_policy(*current), backoff_policy(*current),
88-
[](std::chrono::milliseconds) {}, factory,
87+
retry_policy(*current), backoff_policy(*current), factory,
8988
AgentEndpointServiceReceiveTaskNotificationStreamingUpdater, request);
9089
return internal::MakeStreamRange(
9190
internal::StreamReader<google::cloud::osconfig::agentendpoint::v1::

0 commit comments

Comments
 (0)