Server.cpp

#include <Test/TestMain.cpp>

#include <startup.h>
#include "Parsetree/ExecuteLogicalQueryPlan.h"
#include "/usr/include/assert.h"
#include "Test/set_up_environment.h"
#include <getopt.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){
	handle_parameters(argc,argv);
	Config::getInstance();
	handle_parameters(argc,argv);
	Config::getInstance()->print_configure();
#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(true)
			sleep(1);
	}
	else {
		sleep(1);	//wait for server init...
		Environment::getInstance(master);
		while(true)
			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
	maina(argc, argv);
	return 0;
#endif
}