/**
 * 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 private public
#include "ob_simple_log_server.h"
#undef private

#include "lib/file/file_directory_utils.h"
#include "logservice/palf/log_define.h"
#include <iostream>
#define USING_LOG_PREFIX RPC_TEST
#include<dirent.h>
#include <memory>
#include <net/if.h>
#include <sys/ioctl.h>
#include <sys/socket.h>
#include "util/easy_mod_stat.h"
#include "lib/oblog/ob_log.h"
#include "lib/oblog/ob_log_module.h"
#include "lib/random/ob_random.h"
#include "common/log/ob_log_constants.h"
#include "share/ob_io_device_helper.h"
#include "share/ob_thread_mgr.h"
#include "logservice/palf/palf_options.h"

namespace oceanbase
{

namespace unittest
{
using namespace oceanbase;
using namespace oceanbase::rpc;
using namespace oceanbase::rpc::frame;
using namespace oceanbase::obrpc;
using namespace oceanbase::common;
using namespace oceanbase::share;
using namespace palf;
using namespace palf::election;
using namespace logservice;

int MockNetKeepAliveAdapter::init(unittest::ObLogDeliver *log_deliver)
{
  int ret = OB_SUCCESS;
  if (OB_ISNULL(log_deliver)) {
    ret = OB_INVALID_ARGUMENT;
  } else {
    log_deliver_ = log_deliver;
  }
  return ret;
}

bool MockNetKeepAliveAdapter::in_black_or_stopped(const common::ObAddr &server)
{
  return log_deliver_->need_filter_packet_by_blacklist(server);
}

bool MockNetKeepAliveAdapter::is_server_stopped(const common::ObAddr &server)
{
  UNUSED(server);
  return false;
}

bool MockNetKeepAliveAdapter::in_black(const common::ObAddr &server)
{
  return log_deliver_->need_filter_packet_by_blacklist(server);
}

int MockNetKeepAliveAdapter::get_last_resp_ts(const common::ObAddr &server,
                                              int64_t &last_resp_ts)
{
  if (log_deliver_->need_filter_packet_by_blacklist(server)) {
    last_resp_ts = 1;
  } else {
    last_resp_ts = common::ObTimeUtility::current_time();
  }
  return OB_SUCCESS;
}

uint32_t get_local_addr(const char *dev_name)
{
  int fd, intrface;
  struct ifreq buf[16];
  struct ifconf ifc;

  if ((fd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
    return 0;
  }

  ifc.ifc_len = sizeof(buf);
  ifc.ifc_buf = (caddr_t)buf;
  if (ioctl(fd, SIOCGIFCONF, (char *)&ifc) != 0) {
    close(fd);
    return 0;
  }

  intrface = static_cast<int>(ifc.ifc_len / sizeof(struct ifreq));
  while (intrface-- > 0) {
    if (ioctl(fd, SIOCGIFFLAGS, (char *)&buf[intrface]) != 0) {
      continue;
    }
    if ((buf[intrface].ifr_flags & IFF_LOOPBACK) != 0)
      continue;
    if (!(buf[intrface].ifr_flags & IFF_UP))
      continue;
    if (dev_name != NULL && strcmp(dev_name, buf[intrface].ifr_name))
      continue;
    if (!(ioctl(fd, SIOCGIFADDR, (char *)&buf[intrface]))) {
      close(fd);
      return ((struct sockaddr_in *)(&buf[intrface].ifr_addr))->sin_addr.s_addr;
    }
  }
  close(fd);
  return 0;
}

std::string get_local_ip()
{
  uint32_t ip = get_local_addr("bond0");
 if (ip == 0) {
    ip = get_local_addr("eth0");
  }
  if (ip == 0) {
    return "";
  }
  return inet_ntoa(*(struct in_addr *)(&ip));
}

int ObSimpleLogServer::simple_init(
    const std::string &cluster_name,
    const common::ObAddr &addr,
    const int64_t node_id,
    const bool is_bootstrap = false)
{
  int ret = OB_SUCCESS;
  ObTimeGuard guard("simple_init", 0);
  SERVER_LOG(INFO, "simple_log_server simple_init start", K(node_id), K(addr_));
  auto malloc = ObMallocAllocator::get_instance();
  if (NULL == malloc->get_tenant_ctx_allocator(node_id, 0)) {
    malloc->create_and_add_tenant_allocator(node_id);
  }
  if (is_bootstrap) {
    tenant_base_ = OB_NEW(ObTenantBase, "TestBase", node_id);
    tenant_base_->init();
    tenant_base_->set(&log_service_);
    tenant_base_->set(&detector_);
  }
  ObTenantEnv::set_tenant(tenant_base_);
  assert(&log_service_ == MTL(logservice::ObLogService*));
  guard.click("init tenant_base");
  node_id_ = node_id;

  if (is_bootstrap && OB_FAIL(init_memory_dump_timer_())) {
    SERVER_LOG(ERROR, "init_memory_dump_timer_ failed", K(ret), K_(node_id));
  } else if (FALSE_IT(guard.click("init_memory_dump_timer_"))
      || OB_FAIL(init_network_(addr, is_bootstrap))) {
    SERVER_LOG(WARN, "init_network failed", K(ret), K(addr));
  } else if (FALSE_IT(guard.click("init_network_")) || OB_FAIL(init_io_(cluster_name))) {
    SERVER_LOG(WARN, "init_io failed", K(ret), K(addr));
  } else if (FALSE_IT(guard.click("init_io_")) || OB_FAIL(init_log_service_())) {
    SERVER_LOG(WARN, "init_log_service failed", K(ret), K(addr));
  } else if (FALSE_IT(guard.click("init_log_service_")) || OB_FAIL(looper_.init(this))) {
    SERVER_LOG(WARN, "init ObLooper failed", K(ret), K(addr));
  } else {
    guard.click("init_log_service_");
    SERVER_LOG(INFO, "simple_log_server init success", KPC(this), K(guard));
  }
  return ret;
}

int ObSimpleLogServer::update_tenant_log_disk_size_(const uint64_t tenant_id,
                                                    const int64_t old_log_disk_size,
                                                    const int64_t new_log_disk_size,
                                                    int64_t &allowed_new_log_disk_size)
{
  int ret = OB_SUCCESS;
  MAKE_TENANT_SWITCH_SCOPE_GUARD(guard);
  if (OB_SUCC(guard.switch_to(tenant_id))) {
    ObLogService *log_service = MTL(ObLogService *);
    if (OB_ISNULL(log_service)) {
      ret = OB_ERR_UNEXPECTED;
    } else if (OB_FAIL(log_service->update_log_disk_usage_limit_size(new_log_disk_size))) {
      LOG_WARN("failed to update_log_disk_usage_limit_size", K(ret), K(tenant_id), K(new_log_disk_size),
               K(old_log_disk_size), K(allowed_new_log_disk_size));
    } else if (OB_FAIL(log_block_pool_.update_tenant(old_log_disk_size, new_log_disk_size, allowed_new_log_disk_size, log_service))) {
      LOG_WARN("failed to update teannt int ObServerLogBlockMGR", K(ret), K(tenant_id), K(new_log_disk_size),
               K(old_log_disk_size), K(allowed_new_log_disk_size));
    } else {
      disk_opts_.log_disk_usage_limit_size_ = allowed_new_log_disk_size;
      LOG_INFO("update_log_disk_usage_limit_size success", K(ret), K(tenant_id), K(new_log_disk_size),
               K(old_log_disk_size), K(allowed_new_log_disk_size), K(disk_opts_));
    }
  }
  return ret;
}

int ObSimpleLogServer::update_disk_opts_no_lock_(const PalfDiskOptions &opts)
{
  int ret = OB_SUCCESS;
  CLOG_LOG(INFO, "begin update_disk_opts_no_lock_", K(opts), K(disk_opts_), K(inner_table_disk_opts_));
  int64_t old_log_disk_size = disk_opts_.log_disk_usage_limit_size_;
  int64_t new_log_disk_size = opts.log_disk_usage_limit_size_;
  int64_t allowed_new_log_disk_size = 0;
  // 内部表中的disk_opts立马生效
  inner_table_disk_opts_ = opts;
  // disk_opts_表示本地持久化最新的disk_opts，log_disk_percentage_延迟生效
  disk_opts_ = opts;
  disk_opts_.log_disk_usage_limit_size_ = old_log_disk_size;
  if (!opts.is_valid()) {
    ret = OB_INVALID_ARGUMENT;
    CLOG_LOG(WARN, "invalid argument", K(opts));
  } else if (OB_FAIL(update_tenant_log_disk_size_(node_id_,
                                                  old_log_disk_size,
                                                  new_log_disk_size,
                                                  allowed_new_log_disk_size))) {
    CLOG_LOG(WARN, "update_tenant_log_disk_size_ failed", K(new_log_disk_size), K(old_log_disk_size),
             K(allowed_new_log_disk_size));
  } else {
    CLOG_LOG(INFO, "update_disk_opts success", K(opts), K(disk_opts_), K(new_log_disk_size), K(old_log_disk_size),
             K(allowed_new_log_disk_size));
  }
  return ret;
}


int ObSimpleLogServer::update_disk_opts(const PalfDiskOptions &opts)
{
  int ret = OB_SUCCESS;
  ObSpinLockGuard guard(log_disk_lock_);
  ret = update_disk_opts_no_lock_(opts);
  return ret;
}

int ObSimpleLogServer::try_resize()
{
  int ret = OB_SUCCESS;
  ObSpinLockGuard guard(log_disk_lock_);
  if (disk_opts_ != inner_table_disk_opts_) {
    if (OB_FAIL(update_disk_opts_no_lock_(inner_table_disk_opts_))) {

    }
  }
  return ret;
}

int ObSimpleLogServer::update_server_log_disk(const int64_t log_disk_size)
{
  ObSpinLockGuard guard(log_disk_lock_);
  return log_block_pool_.resize_(log_disk_size);
}

int ObSimpleLogServer::init_memory_dump_timer_()
{
  int ret = OB_SUCCESS;
  common::ObFunction<bool()> print_memory_info = [=](){
      ObMallocAllocator::get_instance()->print_tenant_memory_usage(node_id_);
      ObMallocAllocator::get_instance()->print_tenant_ctx_memory_usage(node_id_);
      ObMallocAllocator::get_instance()->print_tenant_memory_usage(OB_SERVER_TENANT_ID);
      ObMallocAllocator::get_instance()->print_tenant_ctx_memory_usage(OB_SERVER_TENANT_ID);
      return false;};
  if (OB_FAIL(timer_.init_and_start(1, 1_s, "test timer"))) {
    SERVER_LOG(ERROR, "ObOccamTimer init failed", K(ret), K_(node_id));
  } else if (OB_FAIL(timer_.schedule_task_repeat(timer_handle_, 1_s, print_memory_info))) {
    SERVER_LOG(ERROR, "schedule_task failed", K(ret), K_(node_id));
  }
  return ret;
}

int ObSimpleLogServer::init_network_(const common::ObAddr &addr, const bool is_bootstrap = false)
{
  int ret = OB_SUCCESS;
  ObNetOptions opts;
  opts.rpc_io_cnt_ = 10;
  opts.tcp_user_timeout_ = 10 * 1000 * 1000; // 10s
  addr_ = addr;
  obrpc::ObRpcNetHandler::CLUSTER_ID = 1;
  if (is_bootstrap && OB_FAIL(net_.init(opts, 0))) {
    SERVER_LOG(ERROR, "net init fail", K(ret));
  } else if (OB_FAIL(deliver_.init(addr, is_bootstrap))) {
    SERVER_LOG(ERROR, "deliver_ init failed", K(ret));
  } else if (is_bootstrap && OB_FAIL(net_.add_rpc_listen(addr_.get_port(), handler_, transport_))) {
//  } else if (is_bootstrap &&(
//             OB_FAIL(net_.set_rpc_port(addr_.get_port()))
//             || OB_FAIL(net_.rpc_net_register(handler_, transport_))
//             || OB_FAIL(net_.net_keepalive_register()))) {
    SERVER_LOG(ERROR, "net_ listen failed", K(ret));
  } else if (is_bootstrap && OB_FAIL(srv_proxy_.init(transport_))) {
    SERVER_LOG(ERROR, "init srv_proxy_ failed");
  } else {
    deliver_.node_id_ = node_id_;
    SERVER_LOG(INFO, "init_network success", K(ret), K(addr_), K(node_id_), K(opts));
  }
  return ret;
}

int ObSimpleLogServer::init_io_(const std::string &cluster_name)
{
  int ret = OB_SUCCESS;
  ObTimeGuard guard("init_io_", 0);
  const std::string logserver_dir = cluster_name + "/port_" + std::to_string(addr_.get_port());
  std::string data_dir = logserver_dir;
  std::string clog_dir = logserver_dir + "/clog";
  std::string slog_dir = logserver_dir + "/slog";
  std::string sstable_dir = logserver_dir + "/sstable";
  std::vector<std::string> dirs{data_dir, slog_dir, clog_dir, sstable_dir};

  for (auto &dir : dirs) {
    if (-1 == mkdir(dir.c_str(), 0777)) {
      if (errno == EEXIST) {
        ret = OB_SUCCESS;
        SERVER_LOG(INFO, "for restart");
      } else {
        SERVER_LOG(WARN, "mkdir failed", K(errno));
        ret = OB_IO_ERROR;
        break;
      }
    }
  }
  guard.click("mkdir");
  if (OB_SUCC(ret)) {
    io_device_ = OB_NEW(ObLocalDevice, "TestBase");

    blocksstable::ObStorageEnv storage_env;
    storage_env.data_dir_ = data_dir.c_str();
    storage_env.sstable_dir_ = sstable_dir.c_str();
    storage_env.default_block_size_ = OB_DEFAULT_MACRO_BLOCK_SIZE;
    storage_env.data_disk_size_ = 1024 * 1024 * 1024;
    storage_env.data_disk_percentage_ = 0;
    // 当disk_opts_有效时，使用disk_opts_中记录的log_disk_usage_limit_size_作为log_block_pool_的初始值，否则重启会失败
    storage_env.log_disk_size_ = disk_opts_.is_valid() ? disk_opts_.log_disk_usage_limit_size_ : 2LL * 1024 * 1024 * 1024;
    storage_env.log_disk_percentage_ = 0;

    storage_env.log_spec_.log_dir_ = slog_dir.c_str();
    storage_env.log_spec_.max_log_file_size_ = ObLogConstants::MAX_LOG_FILE_SIZE;
    storage_env.clog_dir_ = clog_dir.c_str();
    iod_opts_.opts_ = iod_opt_array_;
    iod_opt_array_[0].set("data_dir", storage_env.data_dir_);
    iod_opt_array_[1].set("sstable_dir", storage_env.sstable_dir_);
    iod_opt_array_[2].set("block_size", reinterpret_cast<const char*>(&storage_env.default_block_size_));
    iod_opt_array_[3].set("datafile_disk_percentage", storage_env.data_disk_percentage_);
    iod_opt_array_[4].set("datafile_size", storage_env.data_disk_size_);
    iod_opts_.opt_cnt_ = MAX_IOD_OPT_CNT;
    if (OB_FAIL(io_device_->init(iod_opts_))) {
      SERVER_LOG(ERROR, "init io device fail", K(ret));
    } else if (OB_FAIL(log_block_pool_.init(storage_env.clog_dir_))) {
      SERVER_LOG(ERROR, "init log pool fail", K(ret));
    } else {
      guard.click("init io_device");
      clog_dir_ = clog_dir;
      SERVER_LOG(INFO, "init_io_ successs", K(ret), K(guard));
    }
    if (OB_SUCC(ret)) {
      log_block_pool_.get_tenants_log_disk_size_func_ = [](int64_t &log_disk_size) -> int
      {
        // ObServerLogBlockMGR 率先于 ObLogService加载，此时租户使用的log_disk_size为0.
        log_disk_size = 0;
        return OB_SUCCESS;
      };
      if (OB_FAIL(log_block_pool_.start(storage_env.log_disk_size_))) {
        LOG_ERROR("log pool start failed", KR(ret));
      }
    }
  }
  return ret;
}

int ObSimpleLogServer::init_log_service_()
{
  int ret = OB_SUCCESS;
  // init deps of log_service
  palf::PalfOptions opts;
  if (disk_opts_.is_valid()) {
    opts.disk_options_ = disk_opts_;
  } else {
    opts.disk_options_.log_disk_usage_limit_size_ = 2 * 1024 * 1024 * 1024ul;
    opts.disk_options_.log_disk_utilization_threshold_ = 80;
    opts.disk_options_.log_disk_utilization_limit_threshold_ = 95;
    opts.disk_options_.log_disk_throttling_percentage_ = 100;
    opts.disk_options_.log_disk_throttling_maximum_duration_ = 2 * 3600 * 1000 * 1000L;
    opts.disk_options_.log_writer_parallelism_ = 2;
    disk_opts_ = opts.disk_options_;
    inner_table_disk_opts_ = disk_opts_;
  }
  std::string clog_dir = clog_dir_ + "/tenant_1";
  allocator_ = OB_NEW(ObTenantMutilAllocator, "TestBase", node_id_);
  ObMemAttr attr(1, "SimpleLog");
  ObMemAttr ele_attr(1, ObNewModIds::OB_ELECTION);
  net_keepalive_ = MTL_NEW(MockNetKeepAliveAdapter, "SimpleLog");

  if (OB_FAIL(net_keepalive_->init(&deliver_))) {
  } else if (OB_FAIL(log_service_.init(opts, clog_dir.c_str(), addr_, allocator_, transport_, &ls_service_,
      &location_service_, &reporter_, &log_block_pool_, &sql_proxy_, net_keepalive_))) {
    SERVER_LOG(ERROR, "init_log_service_ fail", K(ret));
  } else if (OB_FAIL(log_block_pool_.create_tenant(opts.disk_options_.log_disk_usage_limit_size_))) {
    SERVER_LOG(ERROR, "crete tenant failed", K(ret));
  } else if (OB_FAIL(mock_election_map_.init(ele_attr))) {
    SERVER_LOG(ERROR, "mock_election_map_ init fail", K(ret));
  } else {
    palf_env_ = log_service_.get_palf_env();
    palf_env_->palf_env_impl_.log_rpc_.tenant_id_ = OB_SERVER_TENANT_ID;
    SERVER_LOG(INFO, "init_log_service_ success", K(ret), K(opts), K(disk_opts_));
  }
  return ret;
}

int ObSimpleLogServer::simple_start(const bool is_bootstrap = false)
{
  int ret = OB_SUCCESS;
  ObTenantEnv::set_tenant(tenant_base_);
  if (is_bootstrap && OB_FAIL(net_.start())) {
    SERVER_LOG(ERROR, "net start fail", K(ret));
  } else if (OB_FAIL(deliver_.start())) {
    SERVER_LOG(ERROR, "deliver_ start failed", K(ret));
  } else if (OB_FAIL(log_service_.arb_service_.start())) {
    SERVER_LOG(ERROR, "arb_service start failed", K(ret));
  } else if (OB_FAIL(looper_.start())) {
    SERVER_LOG(ERROR, "ObLooper start failed", K(ret));
  }
  // do not start entire log_service_ for now, it will
  // slow down cases running
  return ret;
}

int ObSimpleLogServer::simple_close(const bool is_shutdown = false)
{
  int ret = OB_SUCCESS;
  ObTenantEnv::set_tenant(tenant_base_);
  looper_.destroy();
  ObTimeGuard guard("simple_close", 0);
  deliver_.destroy(is_shutdown);
  guard.click("destroy");
  log_service_.destroy();

  log_block_pool_.destroy();
  guard.click("destroy_palf_env");

  if (is_shutdown) {
    net_.rpc_shutdown();
    net_.stop();
    net_.wait();
    net_.destroy();

    timer_handle_.stop_and_wait();
    timer_.destroy();
  }
  SERVER_LOG(INFO, "stop LogService success", K(ret), K(is_shutdown), K(guard));
  return ret;
}

int ObSimpleLogServer::simple_restart(const std::string &cluster_name, const int64_t node_idx)
{
  int ret = OB_SUCCESS;
  ObTenantEnv::set_tenant(tenant_base_);
  ObTimeGuard guard("simple_restart", 0);
  if (OB_FAIL(simple_close())) {
    SERVER_LOG(ERROR, "simple_close failed", K(ret));
  } else if (FALSE_IT(guard.click("simple_close")) || OB_FAIL(simple_init(cluster_name, addr_, node_idx))) {
    SERVER_LOG(ERROR, "simple_init failed", K(ret));
  } else if (FALSE_IT(guard.click("simple_init")) || OB_FAIL(simple_start())) {
    SERVER_LOG(ERROR, "simple_start failed", K(ret));
  } else {
    guard.click("simple_start");
    SERVER_LOG(INFO, "simple_restart success", K(ret), K(guard));
  }
  return ret;
}

int ObMittestBlacklist::init(const common::ObAddr &self)
{
  int ret = OB_SUCCESS;
  if (OB_FAIL(blacklist_.create(1024))) {
    SERVER_LOG(WARN, "create blacklist_ failed", K(ret));
  } else if (false == blacklist_.created()) {
    SERVER_LOG(WARN, "blacklist_ created failed");
  } else if (OB_FAIL(pcode_blacklist_.create(1024))) {
    SERVER_LOG(WARN, "create pcode_blacklist_ failed", K(ret));
  } else if (false == pcode_blacklist_.created()) {
    SERVER_LOG(WARN, "pcode_blacklist_ created failed");
  } else {
    self_ = self;
  }
  return ret;
}

void ObMittestBlacklist::block_net(const ObAddr &src)
{
  blacklist_.set_refactored(src);
  SERVER_LOG(INFO, "block_net", K(src), K(blacklist_));
}

void ObMittestBlacklist::unblock_net(const ObAddr &src)
{
  blacklist_.erase_refactored(src);
  SERVER_LOG(INFO, "unblock_net", K(src), K(blacklist_));
}

void ObMittestBlacklist::block_pcode(const ObRpcPacketCode &pcode)
{
  pcode_blacklist_.set_refactored((int64_t)pcode);
  SERVER_LOG(INFO, "block_pcode", K(pcode), K(pcode_blacklist_));
}

void ObMittestBlacklist::unblock_pcode(const ObRpcPacketCode &pcode)
{
  pcode_blacklist_.erase_refactored((int64_t)pcode);
  SERVER_LOG(INFO, "unblock_pcode", K(pcode), K(pcode_blacklist_));
}

bool ObMittestBlacklist::need_filter_packet_by_blacklist(const ObAddr &addr)
{
  return OB_HASH_EXIST == blacklist_.exist_refactored(addr);
}

bool ObMittestBlacklist::need_filter_packet_by_pcode_blacklist(const ObRpcPacketCode &pcode)
{
  return OB_HASH_EXIST == pcode_blacklist_.exist_refactored((int64_t)pcode);
}

void ObMittestBlacklist::set_rpc_loss(const ObAddr &src, const int loss_rate)
{
  // not thread safe
  int old_loss_rate = -1;
  bool is_exist = false;
  for (int64_t i = 0; i < rpc_loss_config_.count(); ++i) {
    LossConfig &tmp_config = rpc_loss_config_[i];
    if (tmp_config.src_ == src) {
      is_exist = true;
      // src exists, update its loss_rate
      old_loss_rate = tmp_config.loss_rate_;
      tmp_config.loss_rate_ = loss_rate;
    }
  }
  if (!is_exist) {
    LossConfig loss_config(src, loss_rate);
    rpc_loss_config_.push_back(loss_config);
  }
  SERVER_LOG(INFO, "set_rpc_loss", K(src), K(loss_rate), K(old_loss_rate));
}

void ObMittestBlacklist::reset_rpc_loss(const ObAddr &src)
{
  // not thread safe
  int64_t idx = -1;
  for (int64_t i = 0; i < rpc_loss_config_.count(); ++i) {
    LossConfig tmp_config;
    int ret = rpc_loss_config_.at(i, tmp_config);
    if (OB_SUCCESS != ret) {
      SERVER_LOG(WARN, "at failed", K(ret), K(i), "count", rpc_loss_config_.count());
    } else if (tmp_config.src_ == src) {
      // src exists, update its loss_rate
      idx = i;
    }
  }
  if (idx >= 0) {
    rpc_loss_config_.remove(idx);
  }
  SERVER_LOG(INFO, "reset_rpc_loss", K(src), K(idx));
}

void ObMittestBlacklist::get_loss_config(const ObAddr &src, bool &exist, LossConfig &loss_config)
{
  // not thread safe
  exist = false;
  for (int64_t i = 0; i < rpc_loss_config_.count(); ++i) {
    LossConfig tmp_config;
    int ret = rpc_loss_config_.at(i, tmp_config);
    if (OB_SUCCESS != ret) {
      SERVER_LOG(WARN, "at failed", K(ret));
    } else if (tmp_config.src_ == src) {
      exist = true;
      loss_config = tmp_config;
    }
  }
  SERVER_LOG(INFO, "get_loss_config", K(src), K(loss_config));
}

bool ObMittestBlacklist::need_drop_by_loss_config(const ObAddr &addr)
{
  bool bool_ret = false;
  bool exist = false;
  LossConfig loss_config;
  get_loss_config(addr, exist, loss_config);
  if (exist && loss_config.loss_rate_ > 0) {
    const int64_t random_num = ObRandom::rand(1, 100);
    bool_ret = random_num <= loss_config.loss_rate_ ? true : false;
    if (bool_ret) {
      SERVER_LOG(INFO, "need drop req by loss config", K(addr));
    }
  }
  return bool_ret;
}

int ObLogDeliver::init(const common::ObAddr &self, const bool is_bootstrap)
{
  int ret = OB_SUCCESS;
  // init_all_propocessor_();
  if (is_bootstrap && OB_FAIL(ObMittestBlacklist::init(self))) {
    SERVER_LOG(WARN, "ObMittestBlacklist init failed", K(ret));
  } else if (OB_FAIL(TG_CREATE_TENANT(lib::TGDefIDs::TEST7, tg_id_))) {
    SERVER_LOG(WARN, "ObSimpleThreadPool::init failed", K(ret));
  } else {
    is_inited_ = true;
    SERVER_LOG(INFO, "ObLogDeliver init success", KP(&blacklist_));
  }
  return ret;
}

void ObLogDeliver::destroy(const bool is_shutdown)
{
  if (IS_NOT_INIT) {
    SERVER_LOG(INFO, "ObLogDeliver not init");
  } else {
    RWLock::WLockGuard guard(lock_);
    stop();
    wait();
    is_inited_ = false;
    TG_DESTROY(tg_id_);
    if (is_shutdown) {
      blacklist_.destroy();
      pcode_blacklist_.destroy();
    }
    tg_id_ = 0;
    SERVER_LOG(INFO, "destroy ObLogDeliver");
  }
}

int ObLogDeliver::start()
{
  int ret = OB_SUCCESS;
  if (IS_NOT_INIT) {
    ret = OB_NOT_INIT;
    SERVER_LOG(ERROR, "ObLogDeliver not init");
  } else if (OB_FAIL(TG_SET_HANDLER_AND_START(tg_id_, *this))) {
    SERVER_LOG(ERROR, "start ObLogDeliver failed");
  } else {
    is_stopped_ = false;
    SERVER_LOG(INFO, "start ObLogDeliver success");
  }
  return ret;
}

void ObLogDeliver::stop()
{
  if (IS_NOT_INIT) {
    SERVER_LOG_RET(WARN, OB_NOT_INIT, "ObLogDeliver stop failed");
  } else {
    is_stopped_ = true;
    TG_STOP(tg_id_);
    SERVER_LOG(INFO, "stop ObLogDeliver success");
  }
}

int ObLogDeliver::wait()
{
  int ret = OB_SUCCESS;
  if (IS_NOT_INIT) {
    ret = OB_NOT_INIT;
    SERVER_LOG(ERROR, "ObLogDeliver not inited!!!", K(ret));
  } else {
    TG_WAIT(tg_id_);
    SERVER_LOG(INFO, "wait ObLogDeliver success", K(tg_id_));
  }
  return ret;
}

int ObLogDeliver::deliver(rpc::ObRequest &req)
{
  int ret = OB_SUCCESS;
  const ObRpcPacket &pkt = dynamic_cast<const ObRpcPacket&>(req.get_packet());
  SERVER_LOG(INFO, "deliver request", K(pkt), K(req.ez_req_), KPC(palf_env_impl_));
  if (IS_NOT_INIT) {
    ret = OB_NOT_INIT;
  } else {
    do {
      RWLock::RLockGuard guard(lock_);
      if (false == is_stopped_) {
        ret = TG_PUSH_TASK(tg_id_, &req);
        if (OB_SUCCESS != ret && OB_EAGAIN != ret) {
          SERVER_LOG(ERROR, "deliver request failed", K(ret), K(pkt), K(req.ez_req_), KPC(palf_env_impl_));
        }
      } else {
        break;
      }
    } while (OB_EAGAIN == ret);
  }
  return ret;
}

void ObLogDeliver::handle(void *task)
{
  int ret = OB_SUCCESS;
  if (IS_NOT_INIT) {
    ret = OB_NOT_INIT;
    SERVER_LOG(WARN, "ObLogDeliver not init", K(ret));
  } else if (OB_ISNULL(task)) {
    SERVER_LOG(WARN, "invalid argument", KP(task));
  } else {
    rpc::ObRequest *req = static_cast<rpc::ObRequest *>(task);
    SERVER_LOG(INFO, "handle request", KP(req->ez_req_));
    (void) handle_req_(*req);
  }
}

int ObLogDeliver::handle_req_(rpc::ObRequest &req)
{
  int ret = OB_SUCCESS;
  // tenant_id of log servers are different for clear log record,
  // so set tenant_id of RpcPacket to OB_INVALID_TENANT_ID, this
  // will make the PALF_ENV_ID to OB_SERVER_TENANT_ID
  const ObRpcPacket &pkt = dynamic_cast<const ObRpcPacket&>(req.get_packet());
  ObRpcPacket &modify_pkt = const_cast<ObRpcPacket&>(pkt);
  modify_pkt.set_tenant_id(OB_SERVER_TENANT_ID);
  SERVER_LOG(TRACE, "handle_req_ trace", K(pkt), K(node_id_));
  const ObRpcPacketCode pcode = pkt.get_pcode();
  ObFunction<bool(const ObAddr &src)> filter = [&](const ObAddr &src) -> bool {
    if (this->need_filter_packet_by_blacklist(src)) {
      SERVER_LOG(WARN, "need_filter_packet_by_blacklist", K(ret), K(pcode), K(src), K(blacklist_), KPC(palf_env_impl_));
      return true;
    } else if (this->need_drop_by_loss_config(src)) {
      SERVER_LOG(WARN, "need_drop_by_loss_config", K(ret), K(pcode), K(src), K(rpc_loss_config_), KPC(palf_env_impl_));
      return true;
    }
    return false;
  };
  if (this->need_filter_packet_by_pcode_blacklist(pcode)) {
    SERVER_LOG(WARN, "need_filter_packet_by_pcode_blacklist", K(ret), K(pcode), K(pcode_blacklist_), KPC(palf_env_impl_));
    return OB_SUCCESS;
  }
  switch (pkt.get_pcode()) {
    #define PROCESS(processer) \
    processer p;\
    p.init();\
    p.set_ob_request(req);\
    p.set_filter(&filter);  \
    p.run();      \
    break;
    case obrpc::OB_LOG_PUSH_REQ: {
      PROCESS(LogPushReqP);
    }
    case obrpc::OB_LOG_PUSH_RESP: {
      PROCESS(LogPushRespP);
    }
    case obrpc::OB_LOG_FETCH_REQ: {
      PROCESS(LogFetchReqP);
    }
    case obrpc::OB_LOG_BATCH_FETCH_RESP: {
      PROCESS(LogBatchFetchRespP);
    }
    case obrpc::OB_LOG_PREPARE_REQ: {
      PROCESS(LogPrepareReqP);
    }
    case obrpc::OB_LOG_PREPARE_RESP: {
      PROCESS(LogPrepareRespP);
    }
    case obrpc::OB_LOG_CHANGE_CONFIG_META_REQ: {
      PROCESS(LogChangeConfigMetaReqP);
    }
    case obrpc::OB_LOG_CHANGE_CONFIG_META_RESP: {
      PROCESS(LogChangeConfigMetaRespP);
    }
    case obrpc::OB_LOG_CHANGE_MODE_META_REQ: {
      PROCESS(LogChangeModeMetaReqP);
    }
    case obrpc::OB_LOG_CHANGE_MODE_META_RESP: {
      PROCESS(LogChangeModeMetaRespP);
    }
    case obrpc::OB_LOG_NOTIFY_REBUILD_REQ: {
      PROCESS(LogNotifyRebuildReqP)
    }
    case obrpc::OB_LOG_COMMITTED_INFO: {
      PROCESS(CommittedInfoP)
    }
    case obrpc::OB_LOG_LEARNER_REQ: {
      PROCESS(LogLearnerReqP)
    }
    case obrpc::OB_LOG_REGISTER_PARENT_REQ: {
      PROCESS(LogRegisterParentReqP)
    }
    case obrpc::OB_LOG_REGISTER_PARENT_RESP: {
      PROCESS(LogRegisterParentRespP)
    }
    case obrpc::OB_LOG_ELECTION_PREPARE_REQUEST : {
      PROCESS(ElectionPrepareRequestMsgP)
    }
    case obrpc::OB_LOG_ELECTION_PREPARE_RESPONSE : {
      PROCESS(ElectionPrepareResponseMsgP)
    }
    case obrpc::OB_LOG_ELECTION_ACCEPT_REQUEST : {
      PROCESS(ElectionAcceptRequestMsgP)
    }
    case obrpc::OB_LOG_ELECTION_ACCEPT_RESPONSE : {
      PROCESS(ElectionAcceptResponseMsgP)
    }
    case obrpc::OB_LOG_ELECTION_CHANGE_LEADER_REQUEST : {
      PROCESS(ElectionChangeLeaderMsgP)
    }
    case obrpc::OB_LOG_GET_MC_ST: {
      modify_pkt.set_tenant_id(node_id_);
      PROCESS(LogGetMCStP)
    }
    case obrpc::OB_LOG_ARB_PROBE_MSG: {
      modify_pkt.set_tenant_id(node_id_);
      PROCESS(logservice::LogServerProbeP)
    }
    case obrpc::OB_LOG_CONFIG_CHANGE_CMD: {
      modify_pkt.set_tenant_id(node_id_);
      PROCESS(LogMembershipChangeP)
    }
    case obrpc::OB_LOG_GET_PALF_STAT: {
      modify_pkt.set_tenant_id(node_id_);
      PROCESS(LogGetPalfStatReqP)
    }
    case obrpc::OB_LOG_CHANGE_ACCESS_MODE_CMD: {
      modify_pkt.set_tenant_id(node_id_);
      PROCESS(LogChangeAccessModeP)
    }
    case obrpc::OB_LOG_FLASHBACK_CMD: {
      modify_pkt.set_tenant_id(node_id_);
      PROCESS(LogFlashbackMsgP)
    }
    case obrpc::OB_LOG_GET_STAT: {
      modify_pkt.set_tenant_id(node_id_);
      PROCESS(LogGetStatP)
    }
    case obrpc::OB_LOG_NOTIFY_FETCH_LOG: {
      modify_pkt.set_tenant_id(node_id_);
      PROCESS(LogNotifyFetchLogReqP)
    }
    default:
      SERVER_LOG(ERROR, "invalid req type", K(pkt.get_pcode()));
      break;
  }
  return ret;
}

ObLooper::ObLooper() : log_server_(nullptr),
                       run_interval_(0),
                       is_inited_(false) { }

ObLooper::~ObLooper()
{
  destroy();
}

int ObLooper::init(ObSimpleLogServer *log_server)
{
  int ret = OB_SUCCESS;
  if (OB_UNLIKELY(is_inited_)) {
    ret = OB_INIT_TWICE;
    PALF_LOG(WARN, "ObLooper has been inited", K(ret));
  } else if (NULL == log_server) {
    ret = OB_INVALID_ARGUMENT;
    PALF_LOG(WARN, "invalid argument", K(ret), KP(log_server));
  } else {
    log_server_ = log_server;
    share::ObThreadPool::set_run_wrapper(MTL_CTX());
    run_interval_ = ObLooper::INTERVAL_US;
    is_inited_ = true;
  }

  if ((OB_FAIL(ret)) && (OB_INIT_TWICE != ret)) {
    destroy();
  }
  PALF_LOG(INFO, "ObLooper init finished", K(ret));
  return ret;
}

void ObLooper::destroy()
{
  stop();
  wait();
  is_inited_ = false;
  log_server_ = NULL;
}

void ObLooper::run1()
{
  lib::set_thread_name("ObLooper");
  log_loop_();
}

void ObLooper::log_loop_()
{

  while (!has_set_stop()) {
    int ret = OB_SUCCESS;
    const int64_t start_ts = ObTimeUtility::current_time();

    if (OB_FAIL(log_server_->try_resize())) {
      PALF_LOG(WARN, "try_resize failed", K(ret));
    }
    const int64_t round_cost_time = ObTimeUtility::current_time() - start_ts;
    int32_t sleep_ts = run_interval_ - static_cast<const int32_t>(round_cost_time);
    if (sleep_ts < 0) {
      sleep_ts = 0;
    }
    ob_usleep(sleep_ts);

    if (REACH_TENANT_TIME_INTERVAL(5 * 1000 * 1000)) {
      PALF_LOG(INFO, "ObLooper round_cost_time(us)", K(round_cost_time));
    }
  }
}
} // unittest
} // oceanbase