The C based gRPC (C++, Python, Ruby, Objective-C, PHP, C#) https://grpc.io/
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

189 lines
7.0 KiB

//
//
// 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.
//
//
8 years ago
#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"
8 years ago
namespace grpc {
namespace testing {
7 years ago
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;
7 years ago
options.set_initial_backoff(initial_backoff)
.set_multiplier(multiplier)
.set_jitter(jitter)
.set_max_backoff(max_backoff);
BackOff backoff(options);
7 years ago
grpc_core::Timestamp next_attempt_start_time = backoff.NextAttemptTime();
EXPECT_EQ(next_attempt_start_time - grpc_core::Timestamp::Now(),
initial_backoff);
8 years ago
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);
8 years ago
}
}
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;
7 years ago
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);
8 years ago
}
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;
7 years ago
options.set_initial_backoff(initial_backoff)
.set_multiplier(multiplier)
.set_jitter(jitter)
.set_max_backoff(max_backoff);
BackOff backoff(options);
8 years ago
// 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();
8 years ago
// 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));
8 years ago
}
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;
8 years ago
const double jitter = 0.1;
BackOff::Options options;
7 years ago
options.set_initial_backoff(initial_backoff)
.set_multiplier(multiplier)
.set_jitter(jitter)
.set_max_backoff(max_backoff);
BackOff backoff(options);
8 years ago
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));
8 years ago
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));
8 years ago
}
}
7 years ago
} // namespace
} // namespace testing
} // namespace grpc
8 years ago
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;
8 years ago
}