stasis-bLSM/newserver.cpp

#include <stasis/transactional.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;
    stasis_buffer_manager_size = 1 * 1024 * 1024 * 1024 / PAGE_SIZE;  // 1.5GB total

    if(argc == 2 && !strcmp(argv[1], "--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?
    }

    if(argc == 2 && !strcmp(argv[1], "--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?
    }

    logtable<datatuple>::init_stasis();

      int xid = Tbegin();


      recordid table_root = ROOT_RECORD;


    logtable<datatuple> ltable(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();

    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");
}