209 lines
7.7 KiB
C++
209 lines
7.7 KiB
C++
/**
|
|
* 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.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "ob_simple_log_cluster_testbase.h"
|
|
#include "logservice/arbserver/ob_arb_srv_network_frame.h"
|
|
#include "logservice/arbserver/palf_env_lite_mgr.h"
|
|
#include "logservice/arbserver/ob_arb_server_timer.h"
|
|
#include "logservice/arbserver/palf_handle_lite.h"
|
|
|
|
namespace oceanbase
|
|
{
|
|
namespace unittest
|
|
{
|
|
struct PalfEnvLiteGuard
|
|
{
|
|
PalfEnvLiteGuard() : palf_env_lite_(NULL), palf_env_mgr_(NULL) {}
|
|
~PalfEnvLiteGuard()
|
|
{
|
|
if (NULL != palf_env_lite_ && NULL != palf_env_mgr_) {
|
|
palf_env_mgr_->revert_palf_env_lite(palf_env_lite_);
|
|
}
|
|
palf_env_lite_ = NULL;
|
|
palf_env_mgr_ = NULL;
|
|
}
|
|
palflite::PalfEnvLite *palf_env_lite_;
|
|
palflite::PalfEnvLiteMgr *palf_env_mgr_;
|
|
};
|
|
|
|
class DummyBlockPool : public palf::ILogBlockPool {
|
|
public:
|
|
virtual int create_block_at(const palf::FileDesc &dir_fd,
|
|
const char *block_path,
|
|
const int64_t block_size)
|
|
{
|
|
int fd = -1;
|
|
if (-1 == (fd = ::openat(dir_fd, block_path, palf::LOG_WRITE_FLAG | O_CREAT, 0664))) {
|
|
CLOG_LOG_RET(WARN, common::OB_ERR_SYS, "openat failed", K(block_path), K(errno));
|
|
return OB_IO_ERROR;
|
|
} else if (-1 == ::fallocate(fd, 0, 0, block_size)) {
|
|
CLOG_LOG_RET(WARN, common::OB_ERR_SYS, "fallocate failed", K(block_path), K(errno));
|
|
return OB_IO_ERROR;
|
|
} else {
|
|
CLOG_LOG(INFO, "create_block_at success", K(block_path));
|
|
}
|
|
if (-1 != fd) {
|
|
::close(fd);
|
|
}
|
|
return OB_SUCCESS;
|
|
}
|
|
virtual int remove_block_at(const palf::FileDesc &dir_fd,
|
|
const char *block_path)
|
|
{
|
|
if (-1 == ::unlinkat(dir_fd, block_path, 0)) {
|
|
return OB_IO_ERROR;
|
|
}
|
|
return OB_SUCCESS;
|
|
}
|
|
};
|
|
|
|
class ObSimpleArbServer : public ObISimpleLogServer
|
|
{
|
|
public:
|
|
static const int64_t cluster_id_ = 1;
|
|
ObSimpleArbServer();
|
|
~ObSimpleArbServer() override;
|
|
ILogBlockPool *get_block_pool() override final
|
|
{return &dummy_block_pool_;};
|
|
ObILogAllocator *get_allocator() override final
|
|
{return &allocator_;}
|
|
virtual int update_disk_opts(const PalfDiskOptions &opts) override final
|
|
{return OB_NOT_SUPPORTED;};
|
|
virtual int get_disk_opts(PalfDiskOptions &opts) override final
|
|
{return OB_NOT_SUPPORTED;};
|
|
virtual int get_palf_env(PalfEnv *&palf_env)
|
|
{return OB_NOT_SUPPORTED;};
|
|
bool is_valid() const override final
|
|
{return true == is_inited_;}
|
|
IPalfEnvImpl *get_palf_env() override final
|
|
{
|
|
palflite::PalfEnvLite *palf_env_lite = NULL;
|
|
palflite::PalfEnvKey palf_env_key(cluster_id_, OB_SERVER_TENANT_ID);
|
|
palf_env_mgr_.get_palf_env_lite(palf_env_key, palf_env_lite);
|
|
PALF_LOG(INFO, "yunlong trace get_palf_env", KP(palf_env_lite));
|
|
return palf_env_lite;
|
|
}
|
|
void revert_palf_env(IPalfEnvImpl *palf_env) override final
|
|
{
|
|
palflite::PalfEnvLite *palf_env_lite = dynamic_cast<palflite::PalfEnvLite*>(palf_env);
|
|
palf_env_mgr_.revert_palf_env_lite(palf_env_lite);
|
|
}
|
|
const std::string& get_clog_dir() const override final
|
|
{return base_dir_;}
|
|
common::ObAddr get_addr() const override final
|
|
{return self_;}
|
|
ObTenantBase *get_tenant_base() const override final
|
|
{return tenant_base_;}
|
|
logservice::ObLogFlashbackService *get_flashback_service() override final
|
|
{return NULL;}
|
|
void set_need_drop_packet(const bool need_drop_packet) override final
|
|
{blacklist_.set_need_drop_packet(need_drop_packet);}
|
|
void block_net(const ObAddr &src) override final
|
|
{blacklist_.block_net(src);}
|
|
void unblock_net(const ObAddr &src) override final
|
|
{blacklist_.unblock_net(src);}
|
|
void block_pcode(const ObRpcPacketCode &pcode) override final
|
|
{blacklist_.block_pcode(pcode);}
|
|
void unblock_pcode(const ObRpcPacketCode &pcode) override final
|
|
{blacklist_.unblock_pcode(pcode);}
|
|
void set_rpc_loss(const ObAddr &src, const int loss_rate) override final
|
|
{blacklist_.set_rpc_loss(src, loss_rate);}
|
|
void reset_rpc_loss(const ObAddr &src) override final
|
|
{blacklist_.reset_rpc_loss(src);}
|
|
bool is_arb_server() const override final {return true;}
|
|
int64_t get_node_id() {return node_id_;}
|
|
int create_mock_election(const int64_t palf_id, MockElection *&mock_election) override final
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
mock_election = NULL;
|
|
void *buf = NULL;
|
|
ObMemAttr attr(1, ObNewModIds::OB_ELECTION);
|
|
if (OB_ISNULL(buf = ob_malloc(sizeof(MockElection), attr))) {
|
|
ret = OB_ERR_UNEXPECTED;
|
|
SERVER_LOG(ERROR, "ob_malloc failed", K(palf_id));
|
|
} else if (FALSE_IT(mock_election = new (buf) MockElection)) {
|
|
} else if (OB_FAIL(mock_election->init(palf_id, self_))) {
|
|
SERVER_LOG(WARN, "mock_election->init failed", K(palf_id), K_(self));
|
|
} else if (OB_FAIL(mock_election_map_.insert_and_get(palf::LSKey(palf_id), mock_election))) {
|
|
SERVER_LOG(WARN, "create_mock_election failed", K(palf_id));
|
|
} else {
|
|
SERVER_LOG(INFO, "create_mock_election success", K(palf_id), K_(self), KP(mock_election));
|
|
}
|
|
if (OB_FAIL(ret) && NULL != mock_election) {
|
|
ob_free(mock_election);
|
|
mock_election = NULL;
|
|
}
|
|
return ret;
|
|
}
|
|
int remove_mock_election(const int64_t palf_id) override final
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
if (OB_FAIL(mock_election_map_.del(palf::LSKey(palf_id))) && OB_ENTRY_NOT_EXIST != ret) {
|
|
SERVER_LOG(WARN, "del failed", K(palf_id));
|
|
} else {
|
|
ret = OB_SUCCESS;
|
|
SERVER_LOG(INFO, "remove_mock_election success", K(palf_id), K_(self));
|
|
}
|
|
return ret;
|
|
}
|
|
int set_leader(const int64_t palf_id, const common::ObAddr &leader, const int64_t new_epoch = 0)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
MockElection *mock_election= NULL;
|
|
if (OB_FAIL(mock_election_map_.get(palf::LSKey(palf_id), mock_election))) {
|
|
SERVER_LOG(WARN, "get failed", K(palf_id));
|
|
} else if (OB_FAIL(mock_election->set_leader(leader, new_epoch))) {
|
|
SERVER_LOG(WARN, "set_leader failed", K(palf_id), KP(mock_election), K(leader), K(new_epoch));
|
|
}
|
|
if (OB_NOT_NULL(mock_election)) {
|
|
mock_election_map_.revert(mock_election);
|
|
}
|
|
return ret;
|
|
}
|
|
int update_server_log_disk(const int64_t log_disk_size)
|
|
{return OB_SUCCESS;}
|
|
MockObLocalityManager *get_locality_manager() { return NULL; }
|
|
public:
|
|
int simple_init(const std::string &cluster_name,
|
|
const common::ObAddr &addr,
|
|
const int64_t node_id,
|
|
LogMemberRegionMap *region_map,
|
|
bool is_bootstrap) override final;
|
|
int simple_start(const bool is_bootstrap) override final;
|
|
int simple_close(const bool is_shutdown) override final;
|
|
int simple_restart(const std::string &cluster_name,
|
|
const int64_t node_idx) override final;
|
|
void destroy();
|
|
int create_palf_env_lite(const int64_t tenant_id);
|
|
int remove_palf_env_lite(const int64_t tenant_id);
|
|
int get_palf_env_lite(const int64_t tenant_id, PalfEnvLiteGuard &guard);
|
|
TO_STRING_KV(K_(palf_env_mgr), K_(self), K_(node_id));
|
|
private:
|
|
palflite::PalfEnvLiteMgr palf_env_mgr_;
|
|
arbserver::ObArbSrvNetworkFrame srv_network_frame_;
|
|
arbserver::ObArbServerTimer timer_;
|
|
DummyBlockPool dummy_block_pool_;
|
|
ObTenantMutilAllocator allocator_;
|
|
std::string base_dir_;
|
|
common::ObAddr self_;
|
|
ObTenantBase *tenant_base_;
|
|
int64_t node_id_;
|
|
ObMittestBlacklist blacklist_;
|
|
ObFunction<bool(const ObAddr &src)> filter_;
|
|
MockElectionMap mock_election_map_;
|
|
bool is_inited_;
|
|
};
|
|
}
|
|
}
|