hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
Files
5f836a3c3b586747b34df9f9423f05affec7f3a9
oceanbase/src/share/ob_zone_table_operation.cpp
obdev 6b8425d8b0 [FEAT MERGE] server manager 微服务化改造 
Co-authored-by: maosy <630014370@qq.com>
2023-04-27 16:13:23 +08:00

545 lines
19 KiB
C++
Raw Blame History

/**
* Copyright (c) 2021 OceanBase
* OceanBase CE is licensed under Mulan PubL v2.
* You can use this software according to the terms and conditions of the Mulan PubL v2.
* You may obtain a copy of Mulan PubL v2 at:
* http://license.coscl.org.cn/MulanPubL-2.0
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
* See the Mulan PubL v2 for more details.
*/
#define USING_LOG_PREFIX SHARE
#include "share/ob_zone_table_operation.h"
#include "lib/mysqlclient/ob_mysql_result.h"
#include "lib/oblog/ob_log.h"
#include "lib/string/ob_sql_string.h"
#include "share/inner_table/ob_inner_table_schema.h"
#include "share/ob_zone_info.h"
#include "common/ob_timeout_ctx.h"
#include "rootserver/ob_root_utils.h"
namespace oceanbase
{
using namespace oceanbase::common;
using namespace oceanbase::common::sqlclient;
namespace share
{
template <typename T>
int ObZoneTableOperation::set_info_item(
const char *name, const int64_t value, const char *info_str, T &info)
{
int ret = OB_SUCCESS;
// %value and %info_str can be arbitrary values
if (NULL == name) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid name", K(ret));
} else {
ObZoneInfoItem *it = info.list_.get_first();
while (OB_SUCCESS == ret && it != info.list_.get_header()) {
if (NULL == it) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("null iter", K(ret));
} else {
if (strncasecmp(it->name_, name, OB_MAX_COLUMN_NAME_LENGTH) == 0) {
it->value_ = value;
it->info_ = info_str;
break;
}
it = it->get_next();
}
}
if (OB_SUCC(ret)) {
// ignore unknown item
if (it == info.list_.get_header()) {
LOG_WARN("unknown item", K(name), K(value), "info", info_str);
}
}
}
return ret;
}
template <typename T>
int ObZoneTableOperation::load_info(
common::ObISQLClient &sql_client,
T &info,
const bool check_zone_exists)
{
int ret = OB_SUCCESS;
ObSqlString sql;
bool zone_exists = false;
SMART_VAR(ObMySQLProxy::MySQLResult, res) {
ObMySQLResult *result = NULL;
ObTimeoutCtx ctx;
if (OB_FAIL(rootserver::ObRootUtils::get_rs_default_timeout_ctx(ctx))) {
LOG_WARN("fail to get timeout ctx", K(ret), K(ctx));
} else if (OB_FAIL(sql.assign_fmt("SELECT name, value, info FROM %s WHERE zone = '%s'",
OB_ALL_ZONE_TNAME, info.zone_.ptr()))) {
LOG_WARN("append sql failed", K(ret));
} else if (OB_FAIL(sql_client.read(res, sql.ptr()))) {
LOG_WARN("execute sql failed", K(ret), K(sql));
} else if (NULL == (result = res.get_result())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("failed to get sql result", K(ret));
} else {
int64_t tmp_real_str_len = 0; // only used for output parameter
char name[OB_MAX_COLUMN_NAME_BUF_LENGTH] = "";
int64_t value = 0;
char info_str[MAX_ZONE_INFO_LENGTH + 1] = "";
while (OB_SUCCESS == ret && OB_SUCCESS == (ret = result->next())) {
zone_exists = true;
EXTRACT_STRBUF_FIELD_MYSQL(*result, "name", name,
static_cast<int64_t>(sizeof(name)), tmp_real_str_len);
EXTRACT_INT_FIELD_MYSQL(*result, "value", value, int64_t);
EXTRACT_STRBUF_FIELD_MYSQL(*result, "info", info_str,
static_cast<int64_t>(sizeof(info_str)), tmp_real_str_len);
(void) tmp_real_str_len; // make compiler happy
if (OB_SUCC(ret)) {
if (OB_FAIL(set_info_item(name, value, info_str, info))) {
LOG_WARN("set info item failed", K(ret), K(name), K(value), K(info_str));
}
}
}
if (OB_ITER_END == ret) {
ret = OB_SUCCESS;
if (check_zone_exists && !zone_exists) {
ret = OB_ZONE_INFO_NOT_EXIST;
LOG_WARN("zone not exists", KR(ret), K(sql));
}
} else {
LOG_WARN("get result failed", K(ret), K(sql));
}
}
}
return ret;
}
int ObZoneTableOperation::load_global_info(
ObISQLClient &sql_client,
ObGlobalInfo &info,
const bool check_zone_exists /* = false */)
{
return load_info(sql_client, info, check_zone_exists);
}
int ObZoneTableOperation::load_zone_info(
ObISQLClient &sql_client,
ObZoneInfo &info,
const bool check_zone_exists /* = false */)
{
return load_info(sql_client, info, check_zone_exists);
}
template <typename T>
int ObZoneTableOperation::insert_info(ObISQLClient &sql_client, T &info)
{
int ret = OB_SUCCESS;
if (!info.is_valid()) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid argument", K(ret), K(info));
} else {
DLIST_FOREACH(it, info.list_) {
const bool insert = true;
if (OB_FAIL(update_info_item(sql_client, info.zone_, *it, insert))) {
LOG_WARN("insert item failed", K(ret), "zone", info.zone_, "item", *it);
break;
}
}
}
return ret;
}
int ObZoneTableOperation::insert_global_info(ObISQLClient &sql_client, ObGlobalInfo &info)
{
int ret = OB_SUCCESS;
if (!info.is_valid()) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid argument", K(ret), K(info));
} else if (OB_FAIL(insert_info(sql_client, info))) {
LOG_WARN("insert info failed", K(ret), K(info));
}
return ret;
}
int ObZoneTableOperation::insert_zone_info(ObISQLClient &sql_client, ObZoneInfo &info)
{
int ret = OB_SUCCESS;
if (!info.is_valid()) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid argument", K(ret), K(info));
} else if (OB_FAIL(insert_info(sql_client, info))) {
LOG_WARN("insert info failed", K(ret), K(info));
}
return ret;
}
int ObZoneTableOperation::update_info_item(common::ObISQLClient &sql_client,
const common::ObZone &zone, const ObZoneInfoItem &item, bool insert /* = false */)
{
int ret = OB_SUCCESS;
ObSqlString sql;
// %zone can be empty
if (!item.is_valid()) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid argument", K(ret), K(item));
} else {
if (insert) {
if (OB_FAIL(sql.assign_fmt("INSERT INTO %s (zone, name, value, info, gmt_modified, gmt_create)"
" VALUES('%s', '%s', %ld, '%s', now(6), now(6))",
OB_ALL_ZONE_TNAME, zone.ptr(),
item.name_, item.value_, item.info_.ptr()))) {
LOG_WARN("assign sql failed", K(ret));
}
} else {
if (OB_FAIL(sql.assign_fmt(
"UPDATE %s SET value = %ld, info = '%s', gmt_modified = now(6) "
"WHERE zone = '%s' AND name = '%s'",
OB_ALL_ZONE_TNAME, item.value_, item.info_.ptr(), zone.ptr(), item.name_))) {
LOG_WARN("assign sql failed", K(ret));
}
}
}
int64_t affected_rows = 0;
if (OB_FAIL(ret)) {
} else if (OB_FAIL(sql_client.write(sql.ptr(), affected_rows))) {
LOG_WARN("execute sql failed", K(ret), K(sql));
} else if (!(is_single_row(affected_rows) || is_zero_row(affected_rows))) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("unexpected affected rows", K(ret), K(affected_rows));
} else {
LOG_TRACE("execute sql success", K(sql));
}
return ret;
}
int ObZoneTableOperation::get_zone_lease_info(ObISQLClient &sql_client,
ObZoneLeaseInfo &info)
{
int ret = OB_SUCCESS;
HEAP_VARS_3((ObGlobalInfo, global_info), (ObZoneInfo, zone_info),
(ObMySQLProxy::MySQLResult, res)) {
zone_info.zone_= info.zone_;
ObMySQLResult *result = NULL;
ObSqlString sql;
if (OB_FAIL(sql.assign_fmt(
"SELECT zone, name, value, info FROM %s WHERE zone = '' OR zone = '%s'",
OB_ALL_ZONE_TNAME, info.zone_.ptr()))) {
LOG_WARN("assign sql failed", K(ret));
} else if (OB_FAIL(sql_client.read(res, sql.ptr()))) {
LOG_WARN("execute sql failed", K(ret), K(sql));
} else if (NULL == (result = res.get_result())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("failed to get sql result", K(ret));
} else {
int64_t tmp_real_str_len = 0; // only used for output parameter
while (OB_SUCCESS == ret && OB_SUCCESS == (ret = result->next())) {
ObZone zone;
char name[OB_MAX_COLUMN_NAME_BUF_LENGTH] = "";
int64_t value = 0;
char info_str[MAX_ZONE_INFO_LENGTH + 1] = "";
EXTRACT_STRBUF_FIELD_MYSQL(*result, "zone", zone.ptr(),
MAX_ZONE_LENGTH, tmp_real_str_len);
EXTRACT_STRBUF_FIELD_MYSQL(*result, "name", name,
static_cast<int64_t>(sizeof(name)), tmp_real_str_len);
EXTRACT_INT_FIELD_MYSQL(*result, "value", value, int64_t);
EXTRACT_STRBUF_FIELD_MYSQL(*result, "info", info_str,
static_cast<int64_t>(sizeof(info_str)), tmp_real_str_len);
(void) tmp_real_str_len; // make compiler happy
if (OB_SUCC(ret)) {
if (zone.is_empty()) {
if (OB_FAIL(set_info_item(name, value, info_str, global_info))) {
LOG_WARN("set info item failed", K(ret), K(name), K(value), K(info_str));
}
} else {
if (OB_FAIL(set_info_item(name, value, info_str, zone_info))) {
LOG_WARN("set info item failed", K(ret), K(name), K(value), K(info_str));
}
}
}
}
if (OB_ITER_END == ret) {
ret = OB_SUCCESS;
} else {
LOG_WARN("get result failed", K(ret), K(sql));
}
}
if (OB_SUCC(ret)) {
ObZoneLeaseInfo t;
t.zone_ = zone_info.zone_;
t.privilege_version_ = global_info.privilege_version_;
t.config_version_ = global_info.config_version_;
t.lease_info_version_ = global_info.lease_info_version_;
t.time_zone_info_version_ = global_info.time_zone_info_version_;
if (t.privilege_version_ < 0 || t.config_version_ < 0
|| t.lease_info_version_ < 0) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("invalid zone lease info", K(ret), "lease_info", t);
} else if (t.lease_info_version_ < info.lease_info_version_) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("lease_info_version must not less than old lease_info_version",
K(ret), "new_value", t.lease_info_version_, "old_value", info.lease_info_version_);
} else {
info = t;
}
}
}
return ret;
}
int ObZoneTableOperation::get_zone_list(
ObISQLClient &sql_client, common::ObIArray<ObZone> &zone_list)
{
int ret = OB_SUCCESS;
zone_list.reset();
ObZone zone;
SMART_VAR(ObMySQLProxy::MySQLResult, res) {
ObMySQLResult *result = NULL;
char sql[OB_SHORT_SQL_LENGTH];
int n = snprintf(sql, sizeof(sql), "select distinct(zone) as zone "
"from %s where zone != ''", OB_ALL_ZONE_TNAME);
ObTimeoutCtx ctx;
if (n < 0 || n >= OB_SHORT_SQL_LENGTH) {
ret = OB_BUF_NOT_ENOUGH;
LOG_WARN("sql buf not enough", K(ret), K(n));
} else if (OB_FAIL(rootserver::ObRootUtils::get_rs_default_timeout_ctx(ctx))) {
LOG_WARN("fail to get timeout ctx", K(ret), K(ctx));
} else if (OB_FAIL(sql_client.read(res, sql))) {
LOG_WARN("failed to do read", K(sql), K(ret));
} else if (NULL == (result = res.get_result())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("failed to get result", K(sql), K(ret));
} else {
int64_t tmp_real_str_len = 0; // only used for output parameter
while (OB_SUCCESS == ret && OB_SUCCESS == (ret = result->next())) {
EXTRACT_STRBUF_FIELD_MYSQL(*result, "zone", zone.ptr(), MAX_ZONE_LENGTH, tmp_real_str_len);
(void) tmp_real_str_len; // make compiler happy
if (OB_FAIL(zone_list.push_back(zone))) {
LOG_WARN("failed to add zone list", K(ret));
}
}
if (OB_ITER_END != ret) {
LOG_WARN("failed to get zone list", K(sql), K(ret));
} else {
ret = OB_SUCCESS;
}
}
}
return ret;
}
int ObZoneTableOperation::remove_zone_info(ObISQLClient &sql_client,
const ObZone &zone)
{
int ret = OB_SUCCESS;
ObSqlString sql;
int64_t affected_rows = 0;
int64_t item_cnt = 0;
if (zone.is_empty()) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid argument", K(ret), K(zone));
} else if (OB_FAIL(sql.assign_fmt("DELETE FROM %s WHERE zone = '%s'",
OB_ALL_ZONE_TNAME, zone.ptr()))) {
LOG_WARN("sql assign_fmt failed", K(ret));
} else if (OB_FAIL(sql_client.write(sql.ptr(), affected_rows))) {
LOG_WARN("execute sql failed", K(sql), K(ret));
} else if (OB_FAIL(get_zone_item_count(item_cnt))) {
LOG_WARN("get zone item count failed", K(ret));
} else if (item_cnt != affected_rows) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("affected_rows not right", "expected affected_rows",
item_cnt, K(affected_rows), K(ret));
}
return ret;
}
int ObZoneTableOperation::get_zone_item_count(int64_t &cnt)
{
int ret = OB_SUCCESS;
ObMalloc alloc(ObModIds::OB_TEMP_VARIABLES);
ObPtrGuard<ObZoneInfo> zone_info_guard(alloc);
if (OB_FAIL(zone_info_guard.init())) {
LOG_WARN("init temporary variable failed", K(ret));
} else if (NULL == zone_info_guard.ptr()) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("null zone info ptr", K(ret));
} else {
cnt = zone_info_guard.ptr()->get_item_count();
}
return ret;
}
int ObZoneTableOperation::get_region_list(
common::ObISQLClient &sql_client, common::ObIArray<common::ObRegion> &region_list)
{
int ret = OB_SUCCESS;
region_list.reset();
ObRegion region;
SMART_VAR(ObMySQLProxy::MySQLResult, res) {
ObMySQLResult *result = NULL;
char sql[OB_SHORT_SQL_LENGTH];
int n = snprintf(sql, sizeof(sql), "select info as region"
" from %s where zone != '' and name = 'region' ", OB_ALL_ZONE_TNAME);
ObTimeoutCtx ctx;
if (n < 0 || n >= OB_SHORT_SQL_LENGTH) {
ret = OB_BUF_NOT_ENOUGH;
LOG_WARN("sql buf not enough", K(ret), K(n));
} else if (OB_FAIL(rootserver::ObRootUtils::get_rs_default_timeout_ctx(ctx))) {
LOG_WARN("fail to get timeout ctx", K(ret), K(ctx));
} else if (OB_FAIL(sql_client.read(res, sql))) {
LOG_WARN("failed to do read", K(sql), K(ret));
} else if (NULL == (result = res.get_result())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("failed to get result", K(sql), K(ret));
} else {
int64_t tmp_real_str_len = 0; // 仅用于填充出参，不起作用，需保证对应的字符串中间没有'\0'字符
while (OB_SUCCESS == ret && OB_SUCCESS == (ret = result->next())) {
EXTRACT_STRBUF_FIELD_MYSQL(*result, "region", region.ptr(), MAX_REGION_LENGTH, tmp_real_str_len);
(void) tmp_real_str_len; // make compiler happy
if (OB_FAIL(region_list.push_back(region))) {
LOG_WARN("failed to add zone list", K(ret));
}
}
if (OB_ITER_END != ret) {
LOG_WARN("failed to get zone list", K(sql), K(ret));
} else {
ret = OB_SUCCESS;
}
}
}
return ret;
}
int ObZoneTableOperation::check_encryption_zone(
common::ObISQLClient &sql_client,
const common::ObZone &zone,
bool &encryption)
{
int ret = OB_SUCCESS;
encryption = false;
HEAP_VAR(ObZoneInfo, zone_info) {
if (OB_UNLIKELY(zone.is_empty())) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("the zone is empty", KR(ret), K(zone));
} else if (OB_FAIL(get_zone_info(zone, sql_client, zone_info))) {
LOG_WARN("fail to get zone info", KR(ret), K(zone));
} else {
encryption = zone_info.is_encryption();
}
}
return ret;
}
int ObZoneTableOperation::check_zone_active(
common::ObISQLClient &sql_client,
const common::ObZone &zone,
bool &is_active)
{
int ret = OB_SUCCESS;
is_active = false;
HEAP_VAR(ObZoneInfo, zone_info) {
if (OB_UNLIKELY(zone.is_empty())) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("the zone is empty", KR(ret), K(zone));
} else if (OB_FAIL(get_zone_info(zone, sql_client, zone_info))) {
LOG_WARN("fail to get zone info", KR(ret), K(zone));
} else {
is_active = zone_info.is_active();
}
}
return ret;
}
int ObZoneTableOperation::get_inactive_zone_list(
common::ObISQLClient &sql_client,
common::ObIArray<common::ObZone> &zone_list)
{
return get_zone_list_(sql_client, zone_list, false /* is_active */);
}
int ObZoneTableOperation::get_active_zone_list(
common::ObISQLClient &sql_client,
common::ObIArray<common::ObZone> &zone_list)
{
return get_zone_list_(sql_client, zone_list, true /* is_active */);
}
int ObZoneTableOperation::get_zone_list_(
common::ObISQLClient &sql_client,
common::ObIArray<common::ObZone> &zone_list,
const bool is_active)
{
int ret = OB_SUCCESS;
ObSqlString sql;
ObTimeoutCtx ctx;
zone_list.reset();
ObZone zone;
if (OB_FAIL(rootserver::ObRootUtils::get_rs_default_timeout_ctx(ctx))) {
LOG_WARN("fail to get timeout ctx", K(ret), K(ctx));
} else if (OB_FAIL(sql.assign_fmt("SELECT zone FROM %s WHERE name = 'status' AND info = '%s'",
OB_ALL_ZONE_TNAME, is_active ? "ACTIVE" : "INACTIVE"))) {
LOG_WARN("fail to append sql", KR(ret));
} else {
SMART_VAR(ObMySQLProxy::MySQLResult, res) {
ObMySQLResult *result = NULL;
if (OB_FAIL(sql_client.read(res, OB_SYS_TENANT_ID, sql.ptr()))) {
LOG_WARN("fail to execute sql", KR(ret), K(sql));
} else if (OB_ISNULL(result = res.get_result())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("fail to get sql result", KR(ret), K(sql));
} else {
while (OB_SUCC(ret)) {
if (OB_FAIL(result->next())) {
if (OB_ITER_END != ret) {
LOG_WARN("result next failed", KR(ret));
} else {
ret = OB_SUCCESS;
break;
}
} else {
int64_t tmp_real_str_len = 0;
zone.reset();
EXTRACT_STRBUF_FIELD_MYSQL(*result, "zone", zone.ptr(), MAX_ZONE_LENGTH, tmp_real_str_len);
(void) tmp_real_str_len; // make compiler happy
if (OB_FAIL(zone_list.push_back(zone))) {
LOG_WARN("fail to push an element into zone_list", KR(ret), K(zone));
}
}
}
}
}
}
FLOG_INFO("get inactive zone_list", KR(ret), K(zone_list));
return ret;
}
int ObZoneTableOperation::get_zone_info(
const ObZone &zone,
common::ObISQLClient &sql_client,
ObZoneInfo &zone_info)
{
int ret = OB_SUCCESS;
zone_info.reset();
zone_info.zone_ = zone;
bool check_zone_exists = true;
if (OB_UNLIKELY(zone.is_empty())) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("the zone is empty", KR(ret), K(zone));
} else if (OB_FAIL(load_zone_info(sql_client, zone_info, check_zone_exists))) {
LOG_WARN("fail to load zone info", KR(ret), K(zone));
} else if (OB_UNLIKELY(!zone_info.is_valid())) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("zone_info is unexpectedly invalid",
KR(ret), K(zone), K(zone_info));
} else {}
return ret;
}
}//end namespace share
}//end namespace oceanbase
Reference in New Issue View Git Blame Copy Permalink