/**
 * 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 SQL_ENG

#include "ob_px_dtl_msg.h"
#include "sql/engine/ob_physical_plan_ctx.h"

using namespace oceanbase::common;
using namespace oceanbase::sql;
using namespace oceanbase::sql::dtl;

OB_SERIALIZE_MEMBER(ObPxTabletInfo,
                    tablet_id_,
                    logical_row_count_,
                    physical_row_count_);
OB_SERIALIZE_MEMBER(ObPxTaskMonitorInfo, sched_exec_time_start_, sched_exec_time_end_, exec_time_start_, exec_time_end_, metrics_);
OB_SERIALIZE_MEMBER((ObPxTaskChSet, dtl::ObDtlChSet), sqc_id_, task_id_);
OB_SERIALIZE_MEMBER(ObPxPartChMapItem, first_, second_, third_);
OB_SERIALIZE_MEMBER(ObPxReceiveDataChannelMsg, child_dfo_id_, ch_sets_, ch_total_info_, has_filled_channel_);
OB_SERIALIZE_MEMBER(ObPxTransmitDataChannelMsg, ch_sets_, part_affinity_map_, ch_total_info_, has_filled_channel_);
OB_SERIALIZE_MEMBER(ObPxInitSqcResultMsg, dfo_id_, sqc_id_, rc_, task_count_, err_msg_);
OB_SERIALIZE_MEMBER(ObPxFinishSqcResultMsg, dfo_id_, sqc_id_, rc_, trans_result_,
                    task_monitor_info_array_, sqc_affected_rows_, dml_row_info_,
                    temp_table_id_, interm_result_ids_, fb_info_, err_msg_, das_retry_rc_);
OB_SERIALIZE_MEMBER(ObPxFinishTaskResultMsg, dfo_id_, sqc_id_, task_id_, rc_);
OB_SERIALIZE_MEMBER((ObPxBloomFilterChInfo, dtl::ObDtlChTotalInfo), filter_id_);
OB_SERIALIZE_MEMBER((ObPxBloomFilterChSet, dtl::ObDtlChSet), filter_id_, sqc_id_);
OB_SERIALIZE_MEMBER(ObPxCreateBloomFilterChannelMsg, sqc_count_, sqc_id_, ch_set_info_);
OB_SERIALIZE_MEMBER(ObPxBloomFilterData, filter_, tenant_id_, filter_id_,
                    server_id_, px_sequence_id_, bloom_filter_count_);
OB_SERIALIZE_MEMBER(ObPxDmlRowInfo, row_match_count_, row_duplicated_count_, row_deleted_count_);
OB_SERIALIZE_MEMBER(ObPxTabletRange, tablet_id_, range_cut_, range_weights_);

int ObPxTaskChSet::assign(const ObPxTaskChSet &other)
{
  int ret = OB_SUCCESS;
  sqc_id_ = other.sqc_id_;
  task_id_ = other.task_id_;
  sm_group_id_ = other.sm_group_id_;
  if (OB_FAIL(dtl::ObDtlChSet::assign(other))) {
    LOG_WARN("fail assign ObPxTaskChSet", K(other), K(ret));
  }
  return ret;
}

int ObPxBloomFilterChSet::assign(const ObPxBloomFilterChSet &other)
{
  int ret = OB_SUCCESS;
  filter_id_ = other.filter_id_;
  sqc_id_ = other.sqc_id_;
  if (OB_FAIL(dtl::ObDtlChSet::assign(other))) {
    LOG_WARN("fail assign ObPxBloomFilterChSet", K(other), K(ret));
  }
  return ret;
}

void ObPxDmlRowInfo::set_px_dml_row_info(const ObPhysicalPlanCtx &plan_ctx)
{
  row_match_count_ = plan_ctx.get_row_matched_count();
  row_duplicated_count_ = plan_ctx.get_row_duplicated_count();
  row_deleted_count_ = plan_ctx.get_row_deleted_count();
}

ObPxTabletRange::ObPxTabletRange()
  : tablet_id_(OB_INVALID_ID), range_weights_(0), range_cut_()
{
  range_weights_ = 1;
}

void ObPxTabletRange::reset()
{
  tablet_id_ = OB_INVALID_ID;
  range_cut_.reset();
  range_weights_ = 1;
}

bool ObPxTabletRange::is_valid() const
{
  return tablet_id_ >= 0 && range_cut_.count() >= 0;
}

int ObPxTabletRange::assign(const ObPxTabletRange &other)
{
  int ret = OB_SUCCESS;
  if (OB_FAIL(range_cut_.assign(other.range_cut_))) {
    LOG_WARN("assign range cut failed", K(ret), K(other));
  } else {
    tablet_id_ = other.tablet_id_;
    range_weights_ = other.range_weights_;
  }
  return ret;
}