YugabyteDB (2.13.0.0-b42, bfc6a6643e7399ac8a0e81d06a3ee6d6571b33ab)

//--------------------------------------------------------------------------------------------------

// Copyright (c) YugaByte, Inc.

//

// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except

// in compliance with the License.  You may obtain a copy of the License at

//

// http://www.apache.org/licenses/LICENSE-2.0

//

// Unless required by applicable law or agreed to in writing, software distributed under the License

// is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express

// or implied.  See the License for the specific language governing permissions and limitations

// under the License.

//

//

// Tree node definitions for INSERT statement.

//--------------------------------------------------------------------------------------------------

#ifndef YB_YQL_CQL_QL_PTREE_PT_INSERT_H_

#define YB_YQL_CQL_QL_PTREE_PT_INSERT_H_

#include "yb/yql/cql/ql/ptree/list_node.h"

#include "yb/yql/cql/ql/ptree/pt_dml.h"

#include "yb/yql/cql/ql/ptree/pt_insert_values_clause.h"

#include "yb/yql/cql/ql/ptree/pt_name.h"

#include "yb/yql/cql/ql/ptree/tree_node.h"

namespace yb {

namespace ql {

//--------------------------------------------------------------------------------------------------

class PTInsertStmt : public PTDmlStmt {

 public:

  //------------------------------------------------------------------------------------------------

  // Public types.

  typedef MCSharedPtr<PTInsertStmt> SharedPtr;

  typedef MCSharedPtr<const PTInsertStmt> SharedPtrConst;

  //------------------------------------------------------------------------------------------------

  // Constructor and destructor.

  PTInsertStmt(MemoryContext *memctx,

               YBLocationPtr loc,

               PTQualifiedName::SharedPtr relation,

               PTQualifiedNameListNode::SharedPtr columns,

               const PTCollection::SharedPtr& inserting_value,

               PTExprPtr if_clause = nullptr,

               bool else_error = false,

               PTDmlUsingClausePtr using_clause = nullptr,

               const bool returns_status = false);

  virtual ~PTInsertStmt();

  template<typename... TypeArgs>

  inline static PTInsertStmt::SharedPtr MakeShared(MemoryContext *memctx,

                                                   TypeArgs&&... args) {

    return MCMakeShared<PTInsertStmt>(memctx, std::forward<TypeArgs>(args)...);

  }

_ZN2yb2ql12PTInsertStmt10MakeSharedIJNSt3__110shared_ptrINS0_8LocationEEERNS4_INS0_15PTQualifiedNameEEERNS4_INS0_12TreeListNodeIS7_EEEERNS4_INS0_20PTInsertValuesClauseEEEDnbRNS4_INS0_16PTDmlUsingClauseEEERbEEENS4_IS1_EEPNS_8internal9ArenaBaseINSM_11ArenaTraitsEEEDpOT_

Line

Count

Source

54

47.3k

                                                   TypeArgs&&... args) {

55

47.3k

    return MCMakeShared<PTInsertStmt>(memctx, std::forward<TypeArgs>(args)...);

56

47.3k

  }

_ZN2yb2ql12PTInsertStmt10MakeSharedIJNSt3__110shared_ptrINS0_8LocationEEERNS4_INS0_15PTQualifiedNameEEERNS4_INS0_12TreeListNodeIS7_EEEERNS4_INS0_20PTInsertValuesClauseEEERNS4_INS0_6PTExprEEERbRNS4_INS0_16PTDmlUsingClauseEEESK_EEENS4_IS1_EEPNS_8internal9ArenaBaseINSP_11ArenaTraitsEEEDpOT_

Line

Count

Source

54

78

                                                   TypeArgs&&... args) {

55

78

    return MCMakeShared<PTInsertStmt>(memctx, std::forward<TypeArgs>(args)...);

56

78

  }

_ZN2yb2ql12PTInsertStmt10MakeSharedIJNSt3__110shared_ptrINS0_8LocationEEERNS4_INS0_15PTQualifiedNameEEEDnRNS4_INS0_18PTInsertJsonClauseEEEDnbRNS4_INS0_16PTDmlUsingClauseEEERbEEENS4_IS1_EEPNS_8internal9ArenaBaseINSI_11ArenaTraitsEEEDpOT_

Line

Count

Source

54

86

                                                   TypeArgs&&... args) {

55

86

    return MCMakeShared<PTInsertStmt>(memctx, std::forward<TypeArgs>(args)...);

56

86

  }

  // Node semantics analysis.

  virtual CHECKED_STATUS Analyze(SemContext *sem_context) override;

  void PrintSemanticAnalysisResult(SemContext *sem_context);

  ExplainPlanPB AnalysisResultToPB() override;

  // Node type.

  virtual TreeNodeOpcode opcode() const override {

    return TreeNodeOpcode::kPTInsertStmt;

  }

  // Table name.

  client::YBTableName table_name() const override {

    return relation_->ToTableName();

  }

  // Returns location of table name.

  const YBLocation& table_loc() const override {

    return relation_->loc();

  }

  const PTCollection::SharedPtr& InsertingValue() const {

    return inserting_value_;

  }

  bool IsWriteOp() const override {

    return true;

  }

 private:

  //

  // Analyze helper functions

  //

  CHECKED_STATUS AnalyzeInsertingValue(PTCollection* inserting_value,

                                       SemContext* sem_context);

  CHECKED_STATUS AnanlyzeValuesClause(PTInsertValuesClause* values_clause,

                                      SemContext* sem_context);

  CHECKED_STATUS AnanlyzeJsonClause(PTInsertJsonClause* json_clause,

                                    SemContext* sem_context);

  CHECKED_STATUS ProcessColumn(const MCSharedPtr<MCString>& mc_col_name,

                               const ColumnDesc* col_desc,

                               const PTExprPtr& value_expr,

                               SemContext* sem_context);

  // Initialize all non-initialized columns according to their configured defaults

  CHECKED_STATUS InitRemainingColumns(bool is_json_clause,

                                      SemContext* sem_context);

  // --- The parser will decorate this node with the following information --

  PTQualifiedName::SharedPtr relation_;

  PTQualifiedNameListNode::SharedPtr columns_;

  PTCollection::SharedPtr inserting_value_;

  // -- The semantic analyzer will decorate this node with the following information --

};

}  // namespace ql

}  // namespace yb

#endif  // YB_YQL_CQL_QL_PTREE_PT_INSERT_H_