parse_defs.h

/* Copyright (c) 2021 OceanBase and/or its affiliates. All rights reserved.
miniob is licensed under Mulan PSL v2.
You can use this software according to the terms and conditions of the Mulan PSL v2.
You may obtain a copy of Mulan PSL v2 at:
         http://license.coscl.org.cn/MulanPSL2
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 PSL v2 for more details. */
 
//
// Created by Meiyi
//
 
#pragma once
 
#include "common/lang/string.h"
#include "common/lang/vector.h"
#include "common/lang/memory.h"
#include "common/value.h"
#include "common/lang/utility.h"
 
class Expression;
 
struct RelAttrSqlNode
{
  string relation_name;   
  string attribute_name;  
};
 
enum CompOp
{
  EQUAL_TO,     
  LESS_EQUAL,   
  NOT_EQUAL,    
  LESS_THAN,    
  GREAT_EQUAL,  
  GREAT_THAN,   
  NO_OP
};
 
struct ConditionSqlNode
{
  int left_is_attr;              
  Value          left_value;     
  RelAttrSqlNode left_attr;      
  CompOp         comp;           
  int            right_is_attr;  
  RelAttrSqlNode right_attr;     
  Value          right_value;    
};
 
struct SelectSqlNode
{
  vector<unique_ptr<Expression>> expressions;  
  vector<string>                 relations;    
  vector<ConditionSqlNode>       conditions;   
  vector<unique_ptr<Expression>> group_by;     
};
 
struct CalcSqlNode
{
  vector<unique_ptr<Expression>> expressions;  
};
 
struct InsertSqlNode
{
  string        relation_name;  
  vector<Value> values;         
};
 
struct DeleteSqlNode
{
  string                   relation_name;  
  vector<ConditionSqlNode> conditions;
};
 
struct UpdateSqlNode
{
  string                   relation_name;   
  string                   attribute_name;  
  Value                    value;           
  vector<ConditionSqlNode> conditions;
};
 
struct AttrInfoSqlNode
{
  AttrType type;    
  string   name;    
  size_t   length;  
};
 
struct CreateTableSqlNode
{
  string                  relation_name;  
  vector<AttrInfoSqlNode> attr_infos;     
  vector<string>          primary_keys;   
  // TODO: integrate to CreateTableOptions
  string storage_format;  
  string storage_engine;  
};
 
struct DropTableSqlNode
{
  string relation_name;  
};
 
struct AnalyzeTableSqlNode
{
  string relation_name;  
};
 
struct CreateIndexSqlNode
{
  string index_name;      
  string relation_name;   
  string attribute_name;  
};
 
struct DropIndexSqlNode
{
  string index_name;     
  string relation_name;  
};
 
struct DescTableSqlNode
{
  string relation_name;
};
 
struct LoadDataSqlNode
{
  string relation_name;
  string file_name;
};
 
struct SetVariableSqlNode
{
  string name;
  Value  value;
};
 
class ParsedSqlNode;
 
struct ExplainSqlNode
{
  unique_ptr<ParsedSqlNode> sql_node;
};
 
struct ErrorSqlNode
{
  string error_msg;
  int    line;
  int    column;
};
 
enum SqlCommandFlag
{
  SCF_ERROR = 0,
  SCF_CALC,
  SCF_SELECT,
  SCF_INSERT,
  SCF_UPDATE,
  SCF_DELETE,
  SCF_CREATE_TABLE,
  SCF_DROP_TABLE,
  SCF_ANALYZE_TABLE,
  SCF_CREATE_INDEX,
  SCF_DROP_INDEX,
  SCF_SYNC,
  SCF_SHOW_TABLES,
  SCF_DESC_TABLE,
  SCF_BEGIN,  
  SCF_COMMIT,
  SCF_CLOG_SYNC,
  SCF_ROLLBACK,
  SCF_LOAD_DATA,
  SCF_HELP,
  SCF_EXIT,
  SCF_EXPLAIN,
  SCF_SET_VARIABLE,  
};
class ParsedSqlNode
{
public:
  enum SqlCommandFlag flag;
  ErrorSqlNode        error;
  CalcSqlNode         calc;
  SelectSqlNode       selection;
  InsertSqlNode       insertion;
  DeleteSqlNode       deletion;
  UpdateSqlNode       update;
  CreateTableSqlNode  create_table;
  DropTableSqlNode    drop_table;
  AnalyzeTableSqlNode analyze_table;
  CreateIndexSqlNode  create_index;
  DropIndexSqlNode    drop_index;
  DescTableSqlNode    desc_table;
  LoadDataSqlNode     load_data;
  ExplainSqlNode      explain;
  SetVariableSqlNode  set_variable;
 
public:
  ParsedSqlNode();
  explicit ParsedSqlNode(SqlCommandFlag flag);
};
 
class ParsedSqlResult
{
public:
  void add_sql_node(unique_ptr<ParsedSqlNode> sql_node);
 
  vector<unique_ptr<ParsedSqlNode>> &sql_nodes() { return sql_nodes_; }
 
private:
  vector<unique_ptr<ParsedSqlNode>> sql_nodes_;  
};