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

#ifndef OCEANBASE_SQL_RESOLVER_CMD_VARIABLE_SET_STMT_
#define OCEANBASE_SQL_RESOLVER_CMD_VARIABLE_SET_STMT_

#include "lib/string/ob_string.h"
#include "lib/container/ob_array.h"
#include "sql/resolver/ddl/ob_ddl_stmt.h"
#include "share/system_variable/ob_system_variable.h"

namespace oceanbase
{
namespace sql
{
class ObVariableSetStmt : public ObDDLStmt
{
public:
  class VariableSetNode
  {
  public:
    VariableSetNode() : variable_name_(),
                        is_system_variable_(false),
                        set_scope_(share::ObSetVar::SET_SCOPE_NEXT_TRANS),
                        value_expr_(NULL),
                        is_set_default_(false)
    {}
    virtual ~VariableSetNode() {}
    TO_STRING_KV(K_(variable_name), K_(is_system_variable),
                 K_(set_scope), K_(value_expr), K_(is_set_default));

    common::ObString variable_name_;
    bool is_system_variable_;
    share::ObSetVar::SetScopeType set_scope_;
    ObRawExpr *value_expr_;
    bool is_set_default_;
  };

  ObVariableSetStmt() : ObDDLStmt(stmt::T_VARIABLE_SET),
                        actual_tenant_id_(common::OB_INVALID_ID),
                        variable_nodes_(),
                        has_global_variable_(false)
  {}
  virtual ~ObVariableSetStmt() {}

  inline void set_actual_tenant_id(uint64_t actual_tenant_id) { actual_tenant_id_ = actual_tenant_id; }
  inline uint64_t get_actual_tenant_id() const { return actual_tenant_id_; }
  inline int add_variable_node(const VariableSetNode &node);
  inline int64_t get_variables_size() const;
  int get_variable_node(int64_t index, VariableSetNode &node) const;
  inline void set_has_global_variable(bool has_global_variable)
  { has_global_variable_ = has_global_variable; }
  inline bool has_global_variable() const { return has_global_variable_; }

  virtual int get_cmd_type() const { return get_stmt_type(); }
  virtual bool cause_implicit_commit() const {
    /* when VariableSet is SET AUTOCOMMIT=? statement,
     * and autocommit turns from 0 to 1,
     * it cause implicit commit in its executor.
     * Other set statement do not trigger implicit commit.
     * If it does, SET AUTOCOMMIT should use another new stmt
     */
    return false;
  }

  const common::ObIArray<VariableSetNode> &get_variable_nodes() const
  { return variable_nodes_; }
  virtual obrpc::ObDDLArg &get_ddl_arg() { return modify_sysvar_arg_; }
  TO_STRING_KV(K_(actual_tenant_id), K_(variable_nodes));
private:
  uint64_t actual_tenant_id_;
  common::ObArray<VariableSetNode, common::ModulePageAllocator, true> variable_nodes_;
  bool has_global_variable_;
  obrpc::ObModifySysVarArg modify_sysvar_arg_; // 用于返回exec_tenant_id_
  DISALLOW_COPY_AND_ASSIGN(ObVariableSetStmt);
};

inline int ObVariableSetStmt::add_variable_node(const VariableSetNode &node)
{
  return variable_nodes_.push_back(node);
}

inline int64_t ObVariableSetStmt::get_variables_size() const
{
  return variable_nodes_.count();
}
}//end of namespace sql
}//end of namespace oceanbase

#endif //OCEANBASE_SQL_RESOLVER_CMD_VARIABLE_SET_STMT_