oceanbase/src/share/restore/ob_ls_restore_status.cpp

/**
 * 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/restore/ob_ls_restore_status.h"

using namespace oceanbase;
using namespace share;


ObLSRestoreStatus::ObLSRestoreStatus(const Status &status)
  : status_(status)
{
}

ObLSRestoreStatus &ObLSRestoreStatus::operator=(const ObLSRestoreStatus &restore_status)
{
  if (this != &restore_status) {
    status_ = restore_status.status_;
  }
  return *this;
}

#define LS_RESTORE_STATUS_CASE_TO_TYPE \
    LRS_CASE_TO_TYPE(NONE); \
    LRS_CASE_TO_TYPE(RESTORE_START); \
    LRS_CASE_TO_TYPE(RESTORE_SYS_TABLETS); \
    LRS_CASE_TO_TYPE(WAIT_RESTORE_SYS_TABLETS); \
    LRS_CASE_TO_TYPE(RESTORE_TABLETS_META); \
    LRS_CASE_TO_TYPE(WAIT_RESTORE_TABLETS_META); \
    LRS_CASE_TO_TYPE(RESTORE_TO_CONSISTENT_SCN); \
    LRS_CASE_TO_TYPE(WAIT_RESTORE_TO_CONSISTENT_SCN); \
    LRS_CASE_TO_TYPE(QUICK_RESTORE); \
    LRS_CASE_TO_TYPE(WAIT_QUICK_RESTORE); \
    LRS_CASE_TO_TYPE(QUICK_RESTORE_FINISH); \
    LRS_CASE_TO_TYPE(RESTORE_MAJOR_DATA); \
    LRS_CASE_TO_TYPE(WAIT_RESTORE_MAJOR_DATA); \
    LRS_CASE_TO_TYPE(RESTORE_FAILED); \
    LRS_CASE_TO_TYPE(CLONE_START); \
    LRS_CASE_TO_TYPE(CLONE_COPY_ALL_TABLET_META); \
    LRS_CASE_TO_TYPE(CLONE_COPY_LS_META); \
    LRS_CASE_TO_TYPE(CLONE_CLOG_REPLAY); \
    LRS_CASE_TO_TYPE(CLONE_FAILED);

const char *ObLSRestoreStatus::get_restore_status_str(const ObLSRestoreStatus &status)
{
#define LRS_CASE_TO_TYPE(state)        \
  case state: {                       \
    str = #state;                     \
    break;                            \
  }

  const char* str = "INVALID";
  switch (status) {
    LS_RESTORE_STATUS_CASE_TO_TYPE
    default:
      break;
  }
#undef LRS_CASE_TO_TYPE
  return str;
}

bool ObLSRestoreStatus::is_valid_(int32_t status) const
{
#define LRS_CASE_TO_TYPE(state)       \
  case state: {                       \
    bool_ret = true;                  \
    break;                            \
  }

  bool bool_ret = false;
  switch (status) {
    LS_RESTORE_STATUS_CASE_TO_TYPE
    default:
      break;
  }
#undef LRS_CASE_TO_TYPE
  return bool_ret;
}

#undef LS_RESTORE_STATUS_CASE_TO_TYPE
bool ObLSRestoreStatus::need_online() const
{
  return ((status_ >= WAIT_RESTORE_SYS_TABLETS
           && status_ <= WAIT_RESTORE_MAJOR_DATA)
          || status_ == NONE
          || status_ == CLONE_CLOG_REPLAY);
}

int ObLSRestoreStatus::set_status(int32_t status)
{
  int ret = OB_SUCCESS;
  if (!is_valid_(status)) {
    ret = OB_INVALID_ARGUMENT;
    LOG_WARN("invalid status", K(ret), K(status));
  } else {
    status_ = static_cast<Status>(status);
  }
  return ret;
}

ObLSRestoreStatus &ObLSRestoreStatus::operator=(const Status &status)
{
  status_ = status;
  return *this;
}

int ObLSRestoreStatus::serialize(char *buf, const int64_t len, int64_t &pos) const
{
  int ret = OB_SUCCESS;
  int64_t new_pos = pos;
  if (OB_ISNULL(buf)
      || OB_UNLIKELY(len <= 0)
      || OB_UNLIKELY(pos < 0)) {
    ret = OB_INVALID_ARGUMENT;
    LOG_WARN("invalid args", K(ret), K(buf), K(len), K(pos));
  } else if (OB_FAIL(serialization::encode_i8(buf, len, new_pos, static_cast<int8_t>(status_)))) {
    LOG_WARN("failed to serialize log stream restore status", K(ret), K(len), K_(status));
  } else {
    pos = new_pos;
  }
  return ret;
}

int ObLSRestoreStatus::deserialize(const char *buf, const int64_t len, int64_t &pos)
{
  int ret = OB_SUCCESS;
  int64_t new_pos = pos;
  if (OB_ISNULL(buf)
      || OB_UNLIKELY(len <= 0)
      || OB_UNLIKELY(pos < 0)) {
    ret = OB_INVALID_ARGUMENT;
    LOG_WARN("invalid args", K(ret), K(buf), K(len), K(pos));
  } else if (OB_FAIL(serialization::decode_i8(buf, len, new_pos, (int8_t*)(&status_)))) {
    LOG_WARN("failed to decode log stream restore status", K(ret), K(len), K(new_pos));
  } else {
    pos = new_pos;
  }
  return ret;
}

int64_t ObLSRestoreStatus::get_serialize_size() const
{
  return serialization::encoded_length_i8(static_cast<int8_t>(status_));
}