|
|
|
//
|
|
|
|
//
|
|
|
|
// Copyright 2016 gRPC authors.
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
// you may not use this file except in compliance with the License.
|
|
|
|
// You may obtain a copy of the License at
|
|
|
|
//
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
//
|
|
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
// See the License for the specific language governing permissions and
|
|
|
|
// limitations under the License.
|
|
|
|
//
|
|
|
|
//
|
|
|
|
|
|
|
|
#include "src/core/lib/backoff/backoff.h"
|
|
|
|
|
|
|
|
#include <algorithm>
|
|
|
|
#include <memory>
|
|
|
|
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
|
|
|
|
#include <grpc/grpc.h>
|
|
|
|
|
|
|
|
#include "src/core/lib/gprpp/time.h"
|
|
|
|
#include "src/core/lib/iomgr/exec_ctx.h"
|
|
|
|
#include "test/core/test_util/test_config.h"
|
|
|
|
|
|
|
|
namespace grpc {
|
|
|
|
namespace testing {
|
|
|
|
namespace {
|
|
|
|
|
|
|
|
using grpc_core::BackOff;
|
|
|
|
|
|
|
|
TEST(BackOffTest, ConstantBackOff) {
|
|
|
|
const auto initial_backoff = grpc_core::Duration::Milliseconds(200);
|
|
|
|
const double multiplier = 1.0;
|
|
|
|
const double jitter = 0.0;
|
|
|
|
const auto max_backoff = grpc_core::Duration::Seconds(1);
|
|
|
|
grpc_core::ExecCtx exec_ctx;
|
|
|
|
BackOff::Options options;
|
|
|
|
options.set_initial_backoff(initial_backoff)
|
|
|
|
.set_multiplier(multiplier)
|
|
|
|
.set_jitter(jitter)
|
|
|
|
.set_max_backoff(max_backoff);
|
|
|
|
BackOff backoff(options);
|
|
|
|
|
|
|
|
grpc_core::Timestamp next_attempt_start_time = backoff.NextAttemptTime();
|
|
|
|
EXPECT_EQ(next_attempt_start_time - grpc_core::Timestamp::Now(),
|
|
|
|
initial_backoff);
|
|
|
|
for (int i = 0; i < 10000; i++) {
|
|
|
|
next_attempt_start_time = backoff.NextAttemptTime();
|
|
|
|
EXPECT_EQ(next_attempt_start_time - grpc_core::Timestamp::Now(),
|
|
|
|
initial_backoff);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(BackOffTest, MinConnect) {
|
|
|
|
const auto initial_backoff = grpc_core::Duration::Milliseconds(100);
|
|
|
|
const double multiplier = 1.0;
|
|
|
|
const double jitter = 0.0;
|
|
|
|
const auto max_backoff = grpc_core::Duration::Seconds(1);
|
|
|
|
grpc_core::ExecCtx exec_ctx;
|
|
|
|
BackOff::Options options;
|
|
|
|
options.set_initial_backoff(initial_backoff)
|
|
|
|
.set_multiplier(multiplier)
|
|
|
|
.set_jitter(jitter)
|
|
|
|
.set_max_backoff(max_backoff);
|
|
|
|
BackOff backoff(options);
|
|
|
|
grpc_core::Timestamp next = backoff.NextAttemptTime();
|
|
|
|
EXPECT_EQ(next - grpc_core::Timestamp::Now(), initial_backoff);
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(BackOffTest, NoJitterBackOff) {
|
|
|
|
const auto initial_backoff = grpc_core::Duration::Milliseconds(2);
|
|
|
|
const double multiplier = 2.0;
|
|
|
|
const double jitter = 0.0;
|
|
|
|
const auto max_backoff = grpc_core::Duration::Milliseconds(513);
|
|
|
|
BackOff::Options options;
|
|
|
|
options.set_initial_backoff(initial_backoff)
|
|
|
|
.set_multiplier(multiplier)
|
|
|
|
.set_jitter(jitter)
|
|
|
|
.set_max_backoff(max_backoff);
|
|
|
|
BackOff backoff(options);
|
|
|
|
// x_1 = 2
|
|
|
|
// x_n = 2**i + x_{i-1} ( = 2**(n+1) - 2 )
|
|
|
|
grpc_core::ExecCtx exec_ctx;
|
|
|
|
grpc_core::ExecCtx::Get()->TestOnlySetNow(
|
|
|
|
grpc_core::Timestamp::FromMillisecondsAfterProcessEpoch(0));
|
|
|
|
grpc_core::Timestamp next = backoff.NextAttemptTime();
|
|
|
|
EXPECT_EQ(next, grpc_core::Timestamp::FromMillisecondsAfterProcessEpoch(2));
|
|
|
|
grpc_core::ExecCtx::Get()->TestOnlySetNow(next);
|
|
|
|
next = backoff.NextAttemptTime();
|
|
|
|
EXPECT_EQ(next, grpc_core::Timestamp::FromMillisecondsAfterProcessEpoch(6));
|
|
|
|
grpc_core::ExecCtx::Get()->TestOnlySetNow(next);
|
|
|
|
next = backoff.NextAttemptTime();
|
|
|
|
EXPECT_EQ(next, grpc_core::Timestamp::FromMillisecondsAfterProcessEpoch(14));
|
|
|
|
grpc_core::ExecCtx::Get()->TestOnlySetNow(next);
|
|
|
|
next = backoff.NextAttemptTime();
|
|
|
|
EXPECT_EQ(next, grpc_core::Timestamp::FromMillisecondsAfterProcessEpoch(30));
|
|
|
|
grpc_core::ExecCtx::Get()->TestOnlySetNow(next);
|
|
|
|
next = backoff.NextAttemptTime();
|
|
|
|
EXPECT_EQ(next, grpc_core::Timestamp::FromMillisecondsAfterProcessEpoch(62));
|
|
|
|
grpc_core::ExecCtx::Get()->TestOnlySetNow(next);
|
|
|
|
next = backoff.NextAttemptTime();
|
|
|
|
EXPECT_EQ(next, grpc_core::Timestamp::FromMillisecondsAfterProcessEpoch(126));
|
|
|
|
grpc_core::ExecCtx::Get()->TestOnlySetNow(next);
|
|
|
|
next = backoff.NextAttemptTime();
|
|
|
|
EXPECT_EQ(next, grpc_core::Timestamp::FromMillisecondsAfterProcessEpoch(254));
|
|
|
|
grpc_core::ExecCtx::Get()->TestOnlySetNow(next);
|
|
|
|
next = backoff.NextAttemptTime();
|
|
|
|
EXPECT_EQ(next, grpc_core::Timestamp::FromMillisecondsAfterProcessEpoch(510));
|
|
|
|
grpc_core::ExecCtx::Get()->TestOnlySetNow(next);
|
|
|
|
next = backoff.NextAttemptTime();
|
|
|
|
EXPECT_EQ(next,
|
|
|
|
grpc_core::Timestamp::FromMillisecondsAfterProcessEpoch(1022));
|
|
|
|
grpc_core::ExecCtx::Get()->TestOnlySetNow(next);
|
|
|
|
next = backoff.NextAttemptTime();
|
|
|
|
// Hit the maximum timeout. From this point onwards, retries will increase
|
|
|
|
// only by max timeout.
|
|
|
|
EXPECT_EQ(next,
|
|
|
|
grpc_core::Timestamp::FromMillisecondsAfterProcessEpoch(1535));
|
|
|
|
grpc_core::ExecCtx::Get()->TestOnlySetNow(next);
|
|
|
|
next = backoff.NextAttemptTime();
|
|
|
|
EXPECT_EQ(next,
|
|
|
|
grpc_core::Timestamp::FromMillisecondsAfterProcessEpoch(2048));
|
|
|
|
grpc_core::ExecCtx::Get()->TestOnlySetNow(next);
|
|
|
|
next = backoff.NextAttemptTime();
|
|
|
|
EXPECT_EQ(next,
|
|
|
|
grpc_core::Timestamp::FromMillisecondsAfterProcessEpoch(2561));
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(BackOffTest, JitterBackOff) {
|
|
|
|
const auto initial_backoff = grpc_core::Duration::Milliseconds(500);
|
|
|
|
auto current_backoff = initial_backoff;
|
|
|
|
const auto max_backoff = grpc_core::Duration::Seconds(1);
|
|
|
|
const double multiplier = 1.0;
|
|
|
|
const double jitter = 0.1;
|
|
|
|
BackOff::Options options;
|
|
|
|
options.set_initial_backoff(initial_backoff)
|
|
|
|
.set_multiplier(multiplier)
|
|
|
|
.set_jitter(jitter)
|
|
|
|
.set_max_backoff(max_backoff);
|
|
|
|
BackOff backoff(options);
|
|
|
|
|
|
|
|
grpc_core::ExecCtx exec_ctx;
|
|
|
|
grpc_core::Timestamp next = backoff.NextAttemptTime();
|
|
|
|
EXPECT_EQ(next - grpc_core::Timestamp::Now(), initial_backoff);
|
|
|
|
|
|
|
|
auto expected_next_lower_bound = grpc_core::Duration::Milliseconds(
|
|
|
|
static_cast<double>(current_backoff.millis()) * (1 - jitter));
|
|
|
|
auto expected_next_upper_bound = grpc_core::Duration::Milliseconds(
|
|
|
|
static_cast<double>(current_backoff.millis()) * (1 + jitter));
|
|
|
|
|
|
|
|
for (int i = 0; i < 10000; i++) {
|
|
|
|
next = backoff.NextAttemptTime();
|
|
|
|
// next-now must be within (jitter*100)% of the current backoff (which
|
|
|
|
// increases by * multiplier up to max_backoff).
|
|
|
|
const grpc_core::Duration timeout_millis =
|
|
|
|
next - grpc_core::Timestamp::Now();
|
|
|
|
EXPECT_GE(timeout_millis, expected_next_lower_bound);
|
|
|
|
EXPECT_LE(timeout_millis, expected_next_upper_bound);
|
|
|
|
current_backoff = std::min(
|
|
|
|
grpc_core::Duration::Milliseconds(
|
|
|
|
static_cast<double>(current_backoff.millis()) * multiplier),
|
|
|
|
max_backoff);
|
|
|
|
expected_next_lower_bound = grpc_core::Duration::Milliseconds(
|
|
|
|
static_cast<double>(current_backoff.millis()) * (1 - jitter));
|
|
|
|
expected_next_upper_bound = grpc_core::Duration::Milliseconds(
|
|
|
|
static_cast<double>(current_backoff.millis()) * (1 + jitter));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
} // namespace
|
|
|
|
} // namespace testing
|
|
|
|
} // namespace grpc
|
|
|
|
|
|
|
|
int main(int argc, char** argv) {
|
|
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
|
|
grpc::testing::TestEnvironment env(&argc, argv);
|
|
|
|
grpc_init();
|
|
|
|
int ret = RUN_ALL_TESTS();
|
|
|
|
grpc_shutdown();
|
|
|
|
return ret;
|
|
|
|
}
|