stasis-bLSM/newserver.cpp

#include <stasis/transactional.h>
#include <stasis/logger/safeWrites.h>
#undef end
#undef try
#undef begin

#include <signal.h>


#include "merger.h"
#include "logstore.h"
#include "simpleServer.h"

/*
 * newserver.cpp
 *
 *  Created on: Aug 11, 2010
 *      Author: sears
 */

int main(int argc, char *argv[])
{
    signal(SIGPIPE, SIG_IGN);
    int64_t c0_size = 1024 * 1024 * 512 * 1;
    int log_mode = 0; // do not log by default.
    stasis_buffer_manager_size = 1 * 1024 * 1024 * 1024 / PAGE_SIZE;  // 1.5GB total

    for(int i = 1; i < argc; i++) {
    	if(!strcmp(argv[i], "--test")) {
    		stasis_buffer_manager_size = 3 * 1024 * 1024 * 128 / PAGE_SIZE;  // 228MB total
    		c0_size = 1024 * 1024 * 100;
    		printf("warning: running w/ tiny c0 for testing\n"); // XXX build a separate test server and deployment server?
    	} else if(!strcmp(argv[i], "--benchmark")) {
    	      stasis_buffer_manager_size = (1024L * 1024L * 1024L * 2L) / PAGE_SIZE;  // 4GB total
    	      c0_size =                     1024L * 1024L * 1024L * 2L;
    	      printf("note: running w/ 2GB c0 for benchmarking\n"); // XXX build a separate test server and deployment server?
    	} else if(!strcmp(argv[i], "--log-mode")) {
    		i++;
    		log_mode = atoi(argv[i]);
    	} else {
    		fprintf(stderr, "Usage: %s [--test|--benchmark] [--log-mode <int>]", argv[0]);
    		abort();
    	}
    }

    logtable<datatuple>::init_stasis();

      int xid = Tbegin();


      recordid table_root = ROOT_RECORD;
    {
		logtable<datatuple> ltable(log_mode, c0_size);

		if(TrecordType(xid, ROOT_RECORD) == INVALID_SLOT) {
			printf("Creating empty logstore\n");
			table_root = ltable.allocTable(xid);
			assert(table_root.page == ROOT_RECORD.page &&
				   table_root.slot == ROOT_RECORD.slot);
		} else {
			printf("Opened existing logstore\n");
			table_root.size = TrecordSize(xid, ROOT_RECORD);
			ltable.openTable(xid, table_root);
		}

		Tcommit(xid);
		merge_scheduler * mscheduler = new merge_scheduler(&ltable);
		mscheduler->start();
		ltable.replayLog();

		simpleServer *lserver = new simpleServer(&ltable);

		lserver->acceptLoop();

		printf ("Stopping server...\n");
		delete lserver;

		printf("Stopping merge threads...\n");
		mscheduler->shutdown();
		delete mscheduler;

		printf("Deinitializing stasis...\n");
		fflush(stdout);
    }
    logtable<datatuple>::deinit_stasis();

    printf("Shutdown complete\n");
}