/home/jenkins-slave/workspace/core-os-doxygen/initdb.cpp File Reference

#include <thrift/Thrift.h>
#include <array>
#include <boost/filesystem.hpp>
#include <boost/program_options.hpp>
#include <exception>
#include <iostream>
#include <memory>
#include <string>
#include "Catalog/Catalog.h"
#include "CudaMgr/CudaMgr.h"
#include "ImportExport/Importer.h"
#include "Logger/Logger.h"
#include "OSDependent/omnisci_path.h"
#include "QueryRunner/QueryRunner.h"

Include dependency graph for initdb.cpp:

Go to the source code of this file.

Macros
#define	CALCITEPORT   3279

Functions
int	main (int argc, char *argv[])

Variables
static const std::array < std::string, 3 >	SampleGeoFileNames
static const std::array < std::string, 3 >	SampleGeoTableNames
bool	g_enable_thrift_logs {false}

Macro Definition Documentation

#define CALCITEPORT   3279

Definition at line 33 of file initdb.cpp.

Referenced by main().

Function Documentation

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 44 of file initdb.cpp.

References CALCITEPORT, cat(), CHECK, test_exceptions::data_path, g_base_path, g_enable_thrift_logs, logger::LogOptions::get_options(), omnisci::get_root_abs_path(), i, logger::init(), Catalog_Namespace::SysCatalog::instance(), OMNISCI_DEFAULT_DB, OMNISCI_ROOT_USER, run, SampleGeoFileNames, SampleGeoTableNames, and logger::LogOptions::set_base_path().

                                 {
  std::string base_path;
  bool force = false;
  bool skip_geo = false;
  namespace po = boost::program_options;

  po::options_description desc("Options");
  desc.add_options()("help,h", "Print help messages ")(
      "data",
      po::value<std::string>(&base_path)->required(),
      "Directory path to OmniSci catalogs")(
      "force,f", "Force overwriting of existing OmniSci instance")(
      "skip-geo", "Skip inserting sample geo data");

  desc.add_options()("enable-thrift-logs",
                     po::value<bool>(&g_enable_thrift_logs)
                         ->default_value(g_enable_thrift_logs)
                         ->implicit_value(true),
                     "Enable writing messages directly from thrift to stdout/stderr.");

  logger::LogOptions log_options(argv[0]);
  desc.add(log_options.get_options());

  po::positional_options_description positionalOptions;
  positionalOptions.add("data", 1);

  po::variables_map vm;

  try {
    po::store(po::command_line_parser(argc, argv)
                  .options(desc)
                  .positional(positionalOptions)
                  .run(),
              vm);
    if (vm.count("help")) {
      std::cout << desc;
      return 0;
    }
    if (vm.count("force")) {
      force = true;
    }
    if (vm.count("skip-geo")) {
      skip_geo = true;
    }
    po::notify(vm);
  } catch (boost::program_options::error& e) {
    std::cerr << "Usage Error: " << e.what() << std::endl;
    return 1;
  }

  if (!g_enable_thrift_logs) {
    apache::thrift::GlobalOutput.setOutputFunction([](const char* msg) {});
  }

  if (!boost::filesystem::exists(base_path)) {
    std::cerr << "Catalog basepath " + base_path + " does not exist.\n";
    return 1;
  }
  std::string catalogs_path = base_path + "/mapd_catalogs";
  if (boost::filesystem::exists(catalogs_path)) {
    if (force) {
      boost::filesystem::remove_all(catalogs_path);
    } else {
      std::cerr << "OmniSci catalogs already initialized at " + base_path +
                       ". Use -f to force reinitialization.\n";
      return 1;
    }
  }
  std::string data_path = base_path + "/mapd_data";
  if (boost::filesystem::exists(data_path)) {
    if (force) {
      boost::filesystem::remove_all(data_path);
    } else {
      std::cerr << "OmniSci data directory already exists at " + base_path +
                       ". Use -f to force reinitialization.\n";
      return 1;
    }
  }
  std::string export_path = base_path + "/mapd_export";
  if (boost::filesystem::exists(export_path)) {
    if (force) {
      boost::filesystem::remove_all(export_path);
    } else {
      std::cerr << "OmniSci export directory already exists at " + base_path +
                       ". Use -f to force reinitialization.\n";
      return 1;
    }
  }
  if (!boost::filesystem::create_directory(catalogs_path)) {
    std::cerr << "Cannot create mapd_catalogs subdirectory under " << base_path
              << std::endl;
  }
  if (!boost::filesystem::create_directory(export_path)) {
    std::cerr << "Cannot create mapd_export subdirectory under " << base_path
              << std::endl;
  }

  log_options.set_base_path(base_path);
  logger::init(log_options);

  try {
    SystemParameters sys_parms;
    auto dummy = std::make_shared<Data_Namespace::DataMgr>(
        data_path, sys_parms, nullptr, false, 0);
    auto calcite =
        std::make_shared<Calcite>(-1, CALCITEPORT, base_path, 1024, 5000, true, "");
    g_base_path = base_path;
    auto& sys_cat = Catalog_Namespace::SysCatalog::instance();
    sys_cat.init(base_path, dummy, {}, calcite, true, false, {});

    if (!skip_geo) {
      // Add geo samples to the system database using the root user
      Catalog_Namespace::DBMetadata cur_db;
      const std::string db_name(OMNISCI_DEFAULT_DB);
      CHECK(sys_cat.getMetadataForDB(db_name, cur_db));
      auto cat = sys_cat.getCatalog(cur_db, false);
      Catalog_Namespace::UserMetadata user;
      CHECK(sys_cat.getMetadataForUser(OMNISCI_ROOT_USER, user));

      QueryRunner::ImportDriver import_driver(cat, user);

      const size_t num_samples = SampleGeoFileNames.size();
      for (size_t i = 0; i < num_samples; i++) {
        const std::string table_name = SampleGeoTableNames[i];
        const std::string file_name = SampleGeoFileNames[i];

        const auto file_path = boost::filesystem::path(
            omnisci::get_root_abs_path() + "/ThirdParty/geo_samples/" + file_name);
        if (!boost::filesystem::exists(file_path)) {
          throw std::runtime_error(
              "Unable to populate geo sample data. File does not exist: " +
              file_path.string());
        }

        import_driver.importGeoTable(file_path.string(), table_name, true, true, false);
      }
    }

  } catch (std::exception& e) {
    std::cerr << "Exception: " << e.what() << "\n";
  }
  return 0;
}

Here is the call graph for this function:

Variable Documentation

bool g_enable_thrift_logs {false}

Definition at line 42 of file initdb.cpp.

Referenced by CommandLineOptions::fillOptions(), main(), and startMapdServer().

const std::array<std::string, 3> SampleGeoFileNames

static

Initial value:

{"us-states.json",
                                                           "us-counties.json",
                                                           "countries.json"}

Definition at line 35 of file initdb.cpp.

Referenced by main().

const std::array<std::string, 3> SampleGeoTableNames

static

Initial value:

{"omnisci_states",
                                                            "omnisci_counties",
                                                            "omnisci_countries"}

Definition at line 38 of file initdb.cpp.

Referenced by main().