diff options
author | Devtools Arcadia <arcadia-devtools@yandex-team.ru> | 2022-02-07 18:08:42 +0300 |
---|---|---|
committer | Devtools Arcadia <arcadia-devtools@mous.vla.yp-c.yandex.net> | 2022-02-07 18:08:42 +0300 |
commit | 1110808a9d39d4b808aef724c861a2e1a38d2a69 (patch) | |
tree | e26c9fed0de5d9873cce7e00bc214573dc2195b7 /library/cpp/retry/retry.cpp | |
download | ydb-1110808a9d39d4b808aef724c861a2e1a38d2a69.tar.gz |
intermediate changes
ref:cde9a383711a11544ce7e107a78147fb96cc4029
Diffstat (limited to 'library/cpp/retry/retry.cpp')
-rw-r--r-- | library/cpp/retry/retry.cpp | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/library/cpp/retry/retry.cpp b/library/cpp/retry/retry.cpp new file mode 100644 index 00000000000..92466cdeca5 --- /dev/null +++ b/library/cpp/retry/retry.cpp @@ -0,0 +1,26 @@ +#include "retry.h" + +#include <util/stream/output.h> + +void DoWithRetry(std::function<void()> func, TRetryOptions retryOptions) { + DoWithRetry(func, retryOptions, true); +} + +bool DoWithRetryOnRetCode(std::function<bool()> func, TRetryOptions retryOptions) { + for (ui32 attempt = 0; attempt <= retryOptions.RetryCount; ++attempt) { + if (func()) { + return true; + } + auto sleep = retryOptions.SleepFunction; + sleep(retryOptions.GetTimeToSleep(attempt)); + } + return false; +} + +TRetryOptions MakeRetryOptions(const NRetry::TRetryOptionsPB& retryOptions) { + return TRetryOptions(retryOptions.GetMaxTries(), + TDuration::MilliSeconds(retryOptions.GetInitialSleepMs()), + TDuration::MilliSeconds(retryOptions.GetRandomDeltaMs()), + TDuration::MilliSeconds(retryOptions.GetSleepIncrementMs()), + TDuration::MilliSeconds(retryOptions.GetExponentalMultiplierMs())); +} |