[yt/cpp/mapreduce] YT-21405: Don't ignore backoff and pass actual exception in Retry()

Don't ignore backoff and pass actual exception in Retry()
b821c02fd21c9f8115cd2a4896372a9fda69e5f6

1 files changed, 21 insertions, 5 deletions

diff --git a/yt/cpp/mapreduce/client/client_reader.cpp b/yt/cpp/mapreduce/client/client_reader.cpp
index b080da26048..e1051fda964 100644
--- a/yt/cpp/mapreduce/client/client_reader.cpp
+++ b/yt/cpp/mapreduce/client/client_reader.cpp
@@ -82,15 +82,31 @@ TClientReader::TClientReader(
 
 bool TClientReader::Retry(
     const TMaybe<ui32>& rangeIndex,
-    const TMaybe<ui64>& rowIndex)
+    const TMaybe<ui64>& rowIndex,
+    const std::exception_ptr& error)
 {
     if (CurrentRequestRetryPolicy_) {
-        // TODO we should pass actual exception in Retry function
-        yexception genericError;
-        auto backoff = CurrentRequestRetryPolicy_->OnGenericError(genericError);
-        if (!backoff) {
+        TMaybe<TDuration> backoffDuration;
+        try {
+            std::rethrow_exception(error);
+        } catch (const TErrorResponse& ex) {
+            if (!IsRetriable(ex)) {
+                throw;
+            }
+            backoffDuration = CurrentRequestRetryPolicy_->OnRetriableError(ex);
+        } catch (const std::exception& ex) {
+            if (!IsRetriable(ex)) {
+                throw;
+            }
+            backoffDuration = CurrentRequestRetryPolicy_->OnGenericError(ex);
+        } catch (...) {
+        }
+
+        if (!backoffDuration) {
             return false;
         }
+
+        NDetail::TWaitProxy::Get()->Sleep(*backoffDuration);
     }
 
     try {