diff --git a/src/dynamic_batch_scheduler.cc b/src/dynamic_batch_scheduler.cc index d8ebe5ae6..95d65119b 100644 --- a/src/dynamic_batch_scheduler.cc +++ b/src/dynamic_batch_scheduler.cc @@ -50,7 +50,7 @@ IsStaleState(Payload::State payload_state) void FinishSkippedRequests( std::vector>>&& requests, - const Status& response_status) + const Status& response_status, FailureReason reason) { for (auto& queue : requests) { for (auto& request : queue) { @@ -71,8 +71,8 @@ FinishRejectedCancelledRequests( const static Status rejected_status = Status(Status::Code::UNAVAILABLE, "Request timeout expired"); const static Status cancelled_status = Status(Status::Code::CANCELLED); - FinishSkippedRequests(std::move(rejected_requests), rejected_status); - FinishSkippedRequests(std::move(cancelled_requests), cancelled_status); + FinishSkippedRequests(std::move(rejected_requests), rejected_status, FailureReason::REJECTED); + FinishSkippedRequests(std::move(cancelled_requests), cancelled_status, FailureReason::CANCELED); } DynamicBatchScheduler::DynamicBatchScheduler(