mariadb-columnstore-engine/tools/dmldriver/dmldriver.cpp

/* Copyright (C) 2014 InfiniDB, Inc.

   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: dmldriver.cpp 2101 2013-01-21 14:12:52Z rdempsey $

//#define NDEBUG
#include <cassert>
#include <iostream>
#include <string>
#include <unistd.h>
#include <cctype>
#include <sstream>
#include <fstream>
#include <limits>
using namespace std;

#include <boost/scoped_array.hpp>
#include <boost/tokenizer.hpp>
using namespace boost;

#include "tpchrf2.h"
#include "dmlif.h"

namespace
{
bool vflg;
bool dflg;

void usage()
{
  cout << "usage: dmldriver [-vhd] [-c intvl] [-f file] [-s sid] [-t flgs] [-r file] [-p cnt] [sql_text]"
       << endl;
  cout << "   -c intvl \tcommit every intvl statements" << endl;
  cout << "   -f file  \tread statements from file (max 15KB/stmt)" << endl;
  cout << "   -s sid   \tset sid as session id" << endl;
  cout << "   -t flgs  \tset trace flags" << endl;
  cout << "   -v       \tdisplay affected row count(s)" << endl;
  cout << "   -d       \tdisplay debug info" << endl;
  cout << "   -r file  \tread orderkeys from file for TPC-H RF2" << endl;
  cout << "   -p cnt   \tpack cnt orderkeys into each delete stmt (only w/ -r)" << endl;
  cout << "   -e schema\tset the schema name (only w/ -r)" << endl;
  cout << "   -h       \tdisplay this help text" << endl;
}
}  // namespace

int main(int argc, char** argv)
{
  int c;

  opterr = 0;

  vflg = false;
  dflg = false;

  bool fflg = false;
  string infilename;

  int cIntvl = numeric_limits<int>::max();

  uint32_t sessionID = time(0) & 0x7fffffff;

  bool rflg = false;

  int packCnt = 1;

  uint32_t tflg = 0;

  string schema;

  while ((c = getopt(argc, argv, "e:t:s:c:f:r:p:vhd")) != EOF)
    switch (c)
    {
      case 'v': vflg = true; break;

      case 'd': dflg = true; break;

      case 'f':
        fflg = true;
        infilename = optarg;
        break;

      case 'c': cIntvl = static_cast<int>(strtol(optarg, 0, 0)); break;

      case 's': sessionID = static_cast<uint32_t>(strtoul(optarg, 0, 0)); break;

      case 't': tflg = static_cast<uint32_t>(strtoul(optarg, 0, 0)); break;

      case 'r':
        rflg = true;
        infilename = optarg;
        break;

      case 'p': packCnt = static_cast<int>(strtol(optarg, 0, 0)); break;

      case 'e': schema = optarg; break;

      case 'h':
      case '?':
      default:
        usage();
        return (c == 'h' ? 0 : 1);
        break;
    }

  if (!fflg && !rflg && ((argc - optind) < 1))
  {
    usage();
    return 1;
  }

  if (fflg && rflg)
  {
    cout << "-f and -r are mutually exclusive!" << endl << endl;
    usage();
    return 1;
  }

  if (!schema.empty() && !rflg)
  {
    cout << "-e requires -r!" << endl << endl;
    usage();
    return 1;
  }

  string stmtStr;

  if (!fflg && !rflg)
    stmtStr = argv[optind++];

  int rc = 0;

  dmlif::DMLIF dmlif(sessionID, tflg, dflg, vflg);

  if (fflg)
  {
    ifstream ifs(infilename.c_str());

    if (!ifs.good())
    {
      cerr << "Error accessing file " << infilename << endl;
      return 1;
    }

    const streamsize ilinelen = 15 * 1024;
    scoped_array<char> iline(new char[ilinelen]);
    int cnt = 0;

    for (;;)
    {
      ifs.getline(iline.get(), ilinelen);

      if (ifs.eof())
        break;

      rc = dmlif.sendOne(iline.get());

      if (rc != 0)
        break;

      cnt++;

      if ((cnt % cIntvl) == 0)
        dmlif.sendOne("COMMIT;");
    }
  }
  else if (rflg)
  {
    ifstream ifs(infilename.c_str());

    if (!ifs.good())
    {
      cerr << "Error accessing file " << infilename << endl;
      return 1;
    }

    if (schema.empty())
      schema = "tpch";

    tpch::RF2 rf2(schema, sessionID, tflg, cIntvl, packCnt, dflg, vflg);
    rc = rf2.run(ifs);
  }
  else
  {
    rc = dmlif.sendOne(stmtStr);
  }

  if (rc == 0)
    dmlif.sendOne("COMMIT;");

  return 0;
}