From 953fc900629dd2b04c509b4f953e75dd351e09d9 Mon Sep 17 00:00:00 2001
From: Paul Cristian Sarbu <paul.cristian.sarbu@huawei.com>
Date: Tue, 16 Jul 2024 15:51:24 +0200
Subject: Pass RetryConfig instance to WithRetry

---
 src/buildtool/common/remote/retry.hpp | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'src/buildtool/common/remote/retry.hpp')

diff --git a/src/buildtool/common/remote/retry.hpp b/src/buildtool/common/remote/retry.hpp
index c0ce318b..6b8278a4 100644
--- a/src/buildtool/common/remote/retry.hpp
+++ b/src/buildtool/common/remote/retry.hpp
@@ -23,6 +23,7 @@
 #include <utility>
 
 #include "grpcpp/grpcpp.h"
+#include "src/buildtool/common/remote/retry_config.hpp"
 #include "src/buildtool/logging/logger.hpp"
 
 // Utility class to help detecting when exit the retry loop. This class can be
@@ -51,6 +52,7 @@ using CallableReturningRetryResponse = std::function<RetryResponse(void)>;
 /// Retry loop interrupts when one of the two members of the function's returned
 /// RetryResponse object is set to true.
 [[nodiscard]] auto WithRetry(CallableReturningRetryResponse const& f,
+                             RetryConfig const& retry_config,
                              Logger const& logger) noexcept -> bool;
 
 using CallableReturningGrpcStatus = std::function<grpc::Status(void)>;
@@ -59,6 +61,7 @@ using CallableReturningGrpcStatus = std::function<grpc::Status(void)>;
 /// Retry loop interrupts when function returns an error code different from
 /// UNAVAILABLE.
 [[nodiscard]] auto WithRetry(CallableReturningGrpcStatus const& f,
+                             RetryConfig const& retry_config,
                              Logger const& logger) noexcept
     -> std::pair<bool, grpc::Status>;
 
-- 
cgit v1.2.3