hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[Election] support set inner priority

Browse Source
This commit is contained in:
obdev
2023-02-06 19:52:39 +08:00
committed by ob-robot
parent 0d77772dac
commit e02b126ceb
7 changed files with 110 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
unittest/logservice/test_ob_election.cpp
View File

@ -99,9 +99,9 @@ public:
};
template <typename TAKEOVER_OP>
vector<ElectionImpl *> create_election_group(const int election_num, const vector<uint64_t> &allowed_be_leader, TAKEOVER_OP &&op)
vector<ElectionImpl *> create_election_group(const int election_num, const vector<uint64_t> &priority_seed, TAKEOVER_OP &&op)
{
if (allowed_be_leader.size() != election_num) {
if (priority_seed.size() != election_num) {
abort();
}
vector<ElectionImpl *> v;
@ -134,9 +134,13 @@ vector<ElectionImpl *> create_election_group(const int election_num, const vecto
&timer,
&GlobalNetService,
ObAddr(ObAddr::VER::IPV4, "127.0.0.1", port + index),
allowed_be_leader[index],
priority_seed[index],
1,
[](const int64_t, const ObAddr &){ return OB_SUCCESS; },
[&election](const int64_t, const ObAddr &dest_addr) {
return THREAD_POOL.commit_task_ignore_ret([&election, dest_addr]() {
election->change_leader_to(dest_addr);
});
},
[op, ret](Election *election, ObRole before, ObRole after, RoleChangeReason reason) {
if (before == ObRole::FOLLOWER && after == ObRole::LEADER) {
ELECT_LOG(INFO, "i become LEADER", K(obj_to_string(reason)), KPC(election));
@ -559,6 +563,39 @@ TEST_F(TestElection, inner_priority_seed_valid_when_membership_version_equal) {
ASSERT_EQ(stop_to_be_follower_count, 1);
}
TEST_F(TestElection, set_inner_priority_seed) {
auto election_list = create_election_group(3, {(uint64_t)PRIORITY_SEED_BIT::DEFAULT_SEED, (uint64_t)PRIORITY_SEED_BIT::DEFAULT_SEED, (uint64_t)PRIORITY_SEED_BIT::DEFAULT_SEED}, [](){});
for (auto &election_1 : election_list) {
for (auto &election_2 : election_list) {
GlobalNetService.connect(election_1, election_2);
}
}
this_thread::sleep_for(chrono::seconds(7));
ObRole role;
int64_t _;
election_list[0]->get_role(role, _);
ASSERT_EQ(role, ObRole::LEADER);
ASSERT_EQ(OB_SUCCESS, election_list[0]->add_inner_priority_seed_bit(PRIORITY_SEED_BIT::SEED_IN_REBUILD_PHASE_BIT));
this_thread::sleep_for(chrono::seconds(3));
election_list[1]->get_role(role, _);
ASSERT_EQ(role, ObRole::LEADER);
for (auto iter = election_list.rbegin(); iter != election_list.rend(); ++iter)
(*iter)->stop();
this_thread::sleep_for(chrono::seconds(2));
for (auto &election_ : election_list)
delete election_;
ASSERT_EQ(leader_takeover_times, 2);
ASSERT_EQ(leader_revoke_times, 2);
ASSERT_EQ(devote_to_be_leader_count, 1);
ASSERT_EQ(lease_expired_to_be_follower_count, 0);
ASSERT_EQ(change_leader_to_be_leader_count, 1);
ASSERT_EQ(change_leader_to_be_follower_count, 1);
ASSERT_EQ(stop_to_be_follower_count, 1);
}
}
}

2
unittest/logservice/test_ob_election_message_compat.cpp
View File

@ -215,7 +215,7 @@ TEST_F(TestElectionMsgCompat, old_new_msg_serialize) {
pos = 0;
ASSERT_EQ(OB_SUCCESS, prepare_msg_new1.deserialize(buffer, buffer_size, pos));
ELECT_LOG(INFO, "test msg from old to new", K(prepare_msg_old1), K(prepare_msg_new1));
ASSERT_EQ(DEFAULT_SEED, prepare_msg_new1.get_inner_priority_seed());
ASSERT_EQ((uint64_t)PRIORITY_SEED_BIT::DEFAULT_SEED, prepare_msg_new1.get_inner_priority_seed());
pos = 0;
ASSERT_EQ(OB_SUCCESS, prepare_msg_new2.serialize(buffer, buffer_size, pos));
pos = 0;