source: src/Fragmentation/Automation/poolworker.cpp@ 6ee809

/*
 * Project: MoleCuilder
 * Description: creates and alters molecular systems
 * Copyright (C)  2012 University of Bonn. All rights reserved.
 * Please see the LICENSE file or "Copyright notice" in builder.cpp for details.
 */

/*
 * \file poolworker.cpp
 *
 * This file strongly follows the Serialization example from the boost::asio
 * library (see client.cpp)
 *
 *  Created on: Feb 28, 2012
 *      Author: heber
 */


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

// boost asio needs specific operator new
#include <boost/asio.hpp>

#include "CodePatterns/MemDebug.hpp"

#include <iostream>
#include <boost/lexical_cast.hpp>
#include <signal.h>

#include "atexit.hpp"
#include "CodePatterns/Info.hpp"
#include "CodePatterns/Log.hpp"
#include "Pool/PoolWorker.hpp"

boost::function<void (int)> shutdownfunction;

void signalhandler(int sig)
{
  if (shutdownfunction)
    shutdownfunction(sig);
  signal(SIGINT, NULL);
}

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

  setVerbosity(3);

  size_t exitflag = 0;
  try
  {
    // Check command line arguments.
    if (argc != 5)
    {
      std::cerr << "Usage: " << argv[0] << " <host> <port> <listenhost> <listenport>" << std::endl;
      return 1;
    }

    boost::asio::io_service io_service;
    PoolWorker client(io_service, argv[1], argv[2], argv[3], argv[4]);

    // catch ctrl-c and shutdown worker properly
    shutdownfunction = boost::bind(&PoolWorker::shutdown, boost::ref(client), _1);
//    signal(SIGABRT, &signalhandler);
//    signal(SIGTERM, &signalhandler);
    signal(SIGINT, &signalhandler);

    // process io requests
    {
      Info info("io_service");
      io_service.run();
    }
    exitflag = client.getFlag();
  }
  catch (std::exception& e)
  {
    std::cerr << e.what() << std::endl;
  }


  return exitflag;
}