mariadb-columnstore-engine/dbcon/ddlpackage/sqlparser.cpp

/* Copyright (C) 2014 InfiniDB, Inc.
   Copyright (C) 2016 MariaDB Corporation

   This program is free software; you can redistribute it and/or
   modify it under the terms of the GNU General Public License
   as published by the Free Software Foundation; version 2 of
   the License.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
   MA 02110-1301, USA. */

/***********************************************************************
 *   $Id: sqlparser.cpp 9210 2013-01-21 14:10:42Z rdempsey $
 *
 *
 ***********************************************************************/

#include <fstream>
#include <errno.h>

#define DDLPKGSQLPARSER_DLLEXPORT
#include "sqlparser.h"
#undef DDLPKGSQLPARSER_DLLEXPORT

#include "ddl-gram.h"

void scanner_finish(void* yyscanner);
void scanner_init(const char* str, void* yyscanner);
int ddllex_init_extra(void* user_defined, void** yyscanner);
int ddllex_destroy(void* yyscanner);
int ddlparse(ddlpackage::pass_to_bison* x);
void set_schema(std::string schema);
namespace ddlpackage
{
using namespace std;

SqlParser::SqlParser() : fStatus(-1), fDebug(false), x(&fParseTree)
{
}

void SqlParser::SetDebug(bool debug)
{
  fDebug = debug;
}

void SqlParser::setDefaultSchema(std::string schema)
{
  x.fDBSchema = schema;
}

void SqlParser::setDefaultCharset(const CHARSET_INFO* default_charset)
{
  x.default_table_charset = default_charset;
}

int SqlParser::Parse(const char* sqltext)
{
  ddllex_init_extra(&scanData, &x.scanner);
  scanner_init(sqltext, x.scanner);
  fStatus = ddlparse(&x);
  return fStatus;
}

const ParseTree& SqlParser::GetParseTree(void)
{
  if (!Good())
  {
    throw logic_error("The ParseTree is invalid");
  }

  return fParseTree;
}

bool SqlParser::Good()
{
  return fStatus == 0;
}

SqlParser::~SqlParser()
{
  scanner_finish(x.scanner);  // free scanner allocated memory
  ddllex_destroy(x.scanner);
}

SqlFileParser::SqlFileParser() : SqlParser()
{
}

int SqlFileParser::Parse(const string& sqlfile)
{
  fStatus = -1;

  ifstream ifsql;
  ifsql.open(sqlfile.c_str());

  if (!ifsql.is_open())
  {
    perror(sqlfile.c_str());
    return fStatus;
  }

  char sqlbuf[1024 * 1024];
  unsigned length;
  ifsql.seekg(0, ios::end);
  length = ifsql.tellg();
  ifsql.seekg(0, ios::beg);

  if (length > sizeof(sqlbuf) - 1)
  {
    throw length_error("SqlFileParser has file size hard limit of 16K.");
  }

  std::streamsize rcount;
  rcount = ifsql.readsome(sqlbuf, sizeof(sqlbuf) - 1);

  if (rcount < 0)
    return fStatus;

  sqlbuf[rcount] = 0;

  // cout << endl << sqlfile << "(" << rcount << ")" << endl;
  // cout << "----------------------" << endl;
  // cout << sqlbuf << endl;

  return SqlParser::Parse(sqlbuf);
}
}  // namespace ddlpackage