Server.cpp

#define GLOG_NO_ABBREVIATED_SEVERITIES
#include <startup.h>
#include <getopt.h>
#include <string>
// #include "Test/set_up_environment.h"
#include "./Test/TestMain.cpp"
#include "common/log/logging.h"
#define AUTU_MASTER
// #define FORK
//#define DEBUG_MODE

struct option long_options[] = {{"config-file", required_argument, 0, 'c'},
                                {"help", no_argument, 0, 'h'},
                                {"disable-codegen", no_argument, 0, 'd'},
                                {"catalog", required_argument, 0, 256},
                                {"elastic", no_argument, 0, 'e'},
                                {"init-dop", required_argument, 0, 257},
                                {"max-dop", required_argument, 0, 258},
                                {"datadir", required_argument, 0, 259},
                                {"scheduler_cycle", required_argument, 0, 260},
                                {nullptr, 0, 0, 0}};

std::string help_info =
    std::string(
        "-c --config-file FILE_NAME\n\t\t Specify the configure file\n") +
    "-h --help\n\t\t Print the help info\n" +
    "-d --disable-codegen\n\t\t disable code generation feature\n" +
    "-e --elastic \n\t\t enable elasticity feature\n" +
    "   --catalog CATALOG_FILE\n\t\t specified the catalog file\n" +
    "   --init-dop VALUE\n" +
    "\t\t specified the initial degree of parallelism for each segment\n" +
    "   --max-dop  VALUE\n"
    "\t\t specified the max degree of parallelism for each segment\n" +
    "   --datadir  VALUE\n"
    "\t\t specified the data directory.\n" +
    "   --scheduler_cycle VALUE\n"
    "\t\t specified the cycle(in us) for the elastic scheduler\n";

void handle_parameters(int argc, char** argv) {
  optind = 0;
  int opt;
  while ((opt = getopt_long(argc, argv, "c:h", long_options, NULL)) != -1) {
    switch (opt) {
      case 'c':
        Config::config_file = std::string(optarg);
        break;
      case 'h':
        printf("%s", help_info.c_str());
        exit(1);
        break;
      case 'd':
        Config::enable_codegen = false;
        break;
      case 'e':
        Config::enable_expander_adaptivity = true;
        break;
      case 256:
        Config::catalog_file = std::string(optarg);
        break;
      case 257:
        Config::initial_degree_of_parallelism = atoi(optarg);
        break;
      case 258:
        Config::max_degree_of_parallelism = atoi(optarg);
        break;
      case 259:
        Config::data_dir = std::string(optarg);
        break;
      case 260:
        Config::expander_adaptivity_check_frequency = atoi(optarg);
        break;
      default:
        printf("Invalid parameters! Try -h/--help\n");
        exit(1);
        break;
    }
  }
}
int main(int argc, char** argv) {
  using claims::common::Logging;
  handle_parameters(argc, argv);
  Config::getInstance();
  handle_parameters(argc, argv);
  Config::getInstance()->print_configure();
  Logging claims_logging(argv[0]);

#ifndef DEBUG_MODE
  bool master;
#ifndef AUTU_MASTER
  printf("If Master, please print 1!\n");
  int master = 0;
  scanf("%d", &master);
#else
  master = Config::master;
#endif
  std::string actor;
  if (master)
    actor = "master";
  else
    actor = "slave";

#ifndef FORK
  if (master) {
    Environment::getInstance(master);
    //  create_poc_data_four_partitions();
    //  create_poc_data_one_partitions();
    //  print_welcome();
    //  ExecuteLogicalQueryPlan();
    //    while (std::cin.get() != 'q') sleep(1);
    //  } else {
    //    Environment::getInstance(master);
    //    while (std::cin.get() != 'q') sleep(1);
    //  }
    while (1) sleep(1);
  } else {
    Environment::getInstance(master);
    while (1) sleep(1);
  }
#else
  int pid = fork();
  if (pid != 0) {
    printf("Daemon process id is %d\n", pid);
    sleep(1);
    return 0;
  }
  Config::getInstance();
  printf("Creating %s daemon...\n", actor.c_str());
  Environment::getInstance(master);
  printf("The %s daemon is successfully created.!\n", actor.c_str());
  printf("Logs are redirected to %s\n", Config::logfile.c_str());
  FILE* file;
  if ((file = freopen(Config::logfile.c_str(), "a", stdout)) == NULL) {
    printf("output redirection fails!\n");
  }
  freopen(Config::logfile.c_str(), "a", stderr);

  fflush(stdout);
  fclose(stdout);
  // redirect back to console
  freopen("/dev/tty", "a", stdout);
  printf("The Slave daemon is successfully created!\n");
  // redirect again to the log file.
  if ((file = freopen(Config::logfile.c_str(), "a", stdout)) == NULL) {
    printf("output redirection fails!\n");
  }
  freopen(Config::logfile.c_str(), "a", stderr);
  while (true) sleep(1);
#endif
#else
  Environment::getInstance(true);
  MainDebug(argc, argv);
  Environment::getInstance(1)->~Environment();
  return 0;

#endif
}