source: ThirdParty/JobMarket/src/JobMarket/server_main.cpp@ ab2e834

/*
 * Project: JobMarket
 * Description: asynchronous Server/Controller/Client-approach to parallel computing, based on boost::asio
 * Copyright (C)  2011-2012 Frederik Heber. All rights reserved.
 * 
 */

/*
 * \file server_main.cpp
 *
 * This file strongly follows the Serialization example from the boost::asio
 * library (see server.cpp)
 *
 *  Created on: Oct 21, 2011
 *      Author: heber
 */

// include config.h
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

// boost asio needs specific operator new
#include <boost/asio.hpp>
// program_options must have some strange static stuff, causes double free or
// corruption if included after MemDebug
#include <boost/program_options.hpp>

//#include "CodePatterns/MemDebug.hpp"

#include <cstdlib>
#include <iostream>
#include <vector>

#include "CodePatterns/Info.hpp"
#include "CodePatterns/Log.hpp"
#include "JobMarket/atexit.hpp"
#include "JobMarket/FragmentScheduler.hpp"
#include "JobMarket/Jobs/SystemCommandJob.hpp"
#include "JobMarket/ServerOptions.hpp"
#include "JobMarket/SignalHandler.hpp"

int server_main(int argc, char* argv[])
{
  // from this moment on, we need to be sure to deinitialize in the correct order
  // this is handled by the cleanup function
  atexit(cleanUp);

  ServerOptions ServerOpts;

  // Declare the supported options.
  boost::program_options::options_description desc("Allowed options");
  desc.add_options()
      ("help,h", "produce help message")
      ("verbosity,v", boost::program_options::value<size_t>(), "set verbosity level")
      ("signal", boost::program_options::value< std::vector<size_t> >(), "set signal to catch (can be given multiple times)")
      ("workeraddress", boost::program_options::value< std::string>(), "connect to all workers at this address (host:port)")
      ("workerport", boost::program_options::value< unsigned short >(), "listen on this port for connecting workers")
      ("controllerport", boost::program_options::value< unsigned short >(), "listen on this port for connecting controller")
      ("timeout", boost::program_options::value< size_t >(), "timeout in seconds for alive status of workers")
  ;

  boost::program_options::variables_map vm;
  boost::program_options::store(boost::program_options::parse_command_line(argc, argv, desc), vm);
  boost::program_options::notify(vm);

  int status = 0;
  status = ServerOpts.parseHelp(vm, desc);
  if (status) return status;
  status = ServerOpts.parseVerbosity(vm);
  if (status) return status;
  status = ServerOpts.parseWorkerPort(vm);
  if (status) return status;
  status = ServerOpts.parseControllerPort(vm);
  if (status) return status;
  status = ServerOpts.parseTimeout(vm);
  if (status) return status;
  status = ServerOpts.parseSignals(vm);
  if (status) return status;
  status = ServerOpts.parseWorkerAddress(vm);
  if (status) return status;

  size_t Exitflag = 0;
  try
  {
    boost::asio::io_service io_service;
    FragmentScheduler Server(
      io_service, 
      ServerOpts.workerport, 
      ServerOpts.controllerport, 
      WorkerAddress(ServerOpts.workeraddresshost, ServerOpts.workeraddressport), 
      ServerOpts.timeout);

    // catch ctrl-c and shutdown worker properly
    boost::function<void (int)> shutdownfunction = boost::bind(&FragmentScheduler::shutdown, boost::ref(Server), _1);
    SignalHandler signalhandler(shutdownfunction, ServerOpts.signals);

    {
      Info info("io_service");
      io_service.run();
    }
    Exitflag = Server.getExitflag();
  }
  catch (std::exception& e)
  {
    std::cerr << e.what() << std::endl;
  }

  return Exitflag;
}