polardbxengine/sql/consensus/consensus_common.cc

216 lines
7.2 KiB
C++

/* Copyright (c) 2018, 2021, Alibaba and/or its affiliates. All rights reserved.
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License, version 2.0,
as published by the Free Software Foundation.
This program is also distributed with certain software (including
but not limited to OpenSSL) that is licensed under separate terms,
as designated in a particular file or component or in included license
documentation. The authors of MySQL/PolarDB-X Engine hereby grant you an
additional permission to link the program and your derivative works with the
separately licensed software that they have included with
MySQL/PolarDB-X Engine.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License, version 2.0, for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA */
#include <sstream>
#include "my_alloc.h"
#include "sql/consensus/consensus_common.h"
#include "sql/sys_vars_consensus.h"
#include "sql/consensus_log_manager.h"
namespace im {
void collect_show_global_results(MEM_ROOT *mem_root, std::vector<Consensus_show_global_result *> &results) {
if (opt_consensus_force_recovery)
return;
alisql::Paxos::MemberInfoType mi;
std::vector<alisql::Paxos::ClusterInfoType> cis;
consensus_ptr->getMemberInfo(&mi);
/* if node is not LEADER, global information is empty */
if (mi.role != alisql::Paxos::StateType::LEADER)
return;
consensus_ptr->getClusterInfo(cis);
for (auto &ci : cis) {
Consensus_show_global_result *result = new (mem_root) Consensus_show_global_result();
result->id = ci.serverId;
result->ip_port.str = strmake_root(mem_root, ci.ipPort.c_str(), ci.ipPort.length());
result->ip_port.length = ci.ipPort.length();
result->match_index = ci.matchIndex;
result->next_index = ci.nextIndex;
const char* res = NULL;
switch (ci.role) {
case alisql::Paxos::StateType::FOLLOWER:
res = "Follower";
break;
case alisql::Paxos::StateType::CANDIDATE:
res = "Candidate";
break;
case alisql::Paxos::StateType::LEADER:
res = "Leader";
break;
case alisql::Paxos::StateType::LEARNER:
res = "Learner";
break;
default:
res = "No Role";
break;
}
result->role.str = res;
result->role.length = strlen(res);
if (ci.forceSync)
res = "Yes";
else
res = "No";
result->force_sync.str = res;
result->force_sync.length = strlen(res);
result->election_weight = ci.electionWeight;
result->applied_index = ci.appliedIndex;
result->learner_source = ci.learnerSource;
if (ci.pipelining)
res = "Yes";
else
res = "No";
result->pipelining.str = res;
result->pipelining.length = strlen(res);
if (ci.useApplied)
res = "Yes";
else
res = "No";
result->send_applied.str = res;
result->send_applied.length = strlen(res);
results.push_back(result);
}
return;
}
void collect_show_local_results(MEM_ROOT *mem_root, Consensus_show_local_result *result) {
if (opt_consensus_force_recovery)
return;
const char* res = NULL;
alisql::Paxos::MemberInfoType mi;
consensus_ptr->getMemberInfo(&mi);
result->id = mi.serverId;
result->current_term = mi.currentTerm;
result->current_leader.str = strmake_root(mem_root, mi.currentLeaderAddr.c_str(), mi.currentLeaderAddr.length());
result->current_leader.length = mi.currentLeaderAddr.length();
result->commit_index = mi.commitIndex;
result->last_log_term = mi.lastLogTerm;
result->last_log_index = mi.lastLogIndex;
switch (mi.role) {
case alisql::Paxos::StateType::FOLLOWER:
res = "Follower";
break;
case alisql::Paxos::StateType::CANDIDATE:
res = "Candidate";
break;
case alisql::Paxos::StateType::LEADER:
res = "Leader";
break;
case alisql::Paxos::StateType::LEARNER:
res = "Learner";
break;
default:
res = "No Role";
break;
}
result->role.str = res;
result->role.length = strlen(res);
result->vote_for = mi.votedFor;
result->applied_index = mi.lastAppliedIndex;
Consensus_Log_System_Status rw_status = consensus_log_manager.get_status();
if (opt_cluster_log_type_instance) {
res = "No";
} else {
if (rw_status == BINLOG_WORKING)
res = "Yes";
else
res = "No";
}
result->server_ready_for_rw.str = res;
result->server_ready_for_rw.length = strlen(res);
if (opt_cluster_log_type_instance)
res = "Log";
else
res = "Normal";
result->instance_type.str = res;
result->instance_type.length = strlen(res);
return;
}
void collect_show_logs_results(MEM_ROOT *mem_root, std::vector<Consensus_show_logs_result *> &results) {
IO_CACHE *index_file;
LOG_INFO cur;
File file;
char fname[FN_REFLEN];
List<Item> field_list;
size_t length;
size_t cur_dir_len;
consensus_log_manager.lock_consensus(TRUE);
MYSQL_BIN_LOG *log = consensus_log_manager.get_status() == BINLOG_WORKING ? &mysql_bin_log : &consensus_log_manager.get_relay_log_info()->relay_log;
if (!log->is_open()) {
consensus_log_manager.unlock_consensus();
my_error(ER_NO_BINARY_LOGGING, MYF(0));
return;
}
mysql_mutex_lock(log->get_log_lock());
log->lock_index();
index_file= log->get_index_file();
log->raw_get_current_log(&cur); // dont take mutex
mysql_mutex_unlock(log->get_log_lock()); // lockdep, OK
cur_dir_len= dirname_length(cur.log_file_name);
reinit_io_cache(index_file, READ_CACHE, (my_off_t) 0, 0, 0);
/* The file ends with EOF or empty line */
while ((length=my_b_gets(index_file, fname, sizeof(fname))) > 1) {
size_t dir_len;
ulonglong file_length= 0; // Length if open fails
fname[--length] = '\0'; // remove the newline
Consensus_show_logs_result *result = new (mem_root) Consensus_show_logs_result();
dir_len= dirname_length(fname);
length-= dir_len;
result->log_name.str = strmake_root(mem_root, fname + dir_len, length);
result->log_name.length = length;
if (!(strncmp(fname+dir_len, cur.log_file_name+cur_dir_len, length))) {
file_length= cur.pos; /* The active log, use the active position */
} else {
/* this is an old log, open it and find the size */
if ((file= mysql_file_open(key_file_binlog,
fname, O_RDONLY | O_SHARE | O_BINARY,
MYF(0))) >= 0) {
file_length= (ulonglong) mysql_file_seek(file, 0L, MY_SEEK_END, MYF(0));
mysql_file_close(file, MYF(0));
}
}
result->file_size = file_length;
uint64 start_index = consensus_log_manager.get_log_file_index()->get_start_index_of_file(std::string(fname));
result->start_log_index = start_index;
results.push_back(result);
}
if(index_file->error == -1)
goto err;
log->unlock_index();
consensus_log_manager.unlock_consensus();
return;
err:
log->unlock_index();
consensus_log_manager.unlock_consensus();
return;
}
} /* namespace im */