just-mr: support retry for temporary grpc failures

... using the same mechanism that just uses. As just-mr now also
does quite some grpc calls, e.g., fetching distribution files from
the remote-execution endpoint and negociating roots with the serve
endpoint, it may now also be affected by a bad network or a service
temporarily be unavailable.

1 files changed, 14 insertions, 0 deletions

diff --git a/src/other_tools/just_mr/launch.cpp b/src/other_tools/just_mr/launch.cpp
index 1236d9f4..693d8808 100644
--- a/src/other_tools/just_mr/launch.cpp
+++ b/src/other_tools/just_mr/launch.cpp
@@ -35,6 +35,7 @@ auto CallJust(std::optional<std::filesystem::path> const& config_file,
               MultiRepoJustSubCmdsArguments const& just_cmd_args,
               MultiRepoLogArguments const& log_args,
               MultiRepoRemoteAuthArguments const& auth_args,
+              RetryArguments const& retry_args,
               ForwardOnlyArguments const& launch_fwd,
               bool forward_build_root,
               std::string multi_repo_tool_name) -> int {
@@ -185,6 +186,19 @@ auto CallJust(std::optional<std::filesystem::path> const& config_file,
             cmd.emplace_back("--tls-client-key");
             cmd.emplace_back(auth_args.tls_client_key->string());
         }
+        if (retry_args.max_attempts) {
+            cmd.emplace_back("--max-attempts");
+            cmd.emplace_back(std::to_string(*retry_args.max_attempts));
+        }
+        if (retry_args.initial_backoff_seconds) {
+            cmd.emplace_back("--initial-backoff-seconds");
+            cmd.emplace_back(
+                std::to_string(*retry_args.initial_backoff_seconds));
+        }
+        if (retry_args.max_backoff_seconds) {
+            cmd.emplace_back("--max-backoff-seconds");
+            cmd.emplace_back(std::to_string(*retry_args.max_backoff_seconds));
+        }
     }
     if (supports_dispatch and just_cmd_args.endpoint_configuration) {
         cmd.emplace_back("--endpoint-configuration");