Scale Stack 0.3 Developer Documentation

scalestack/event/libevent/service.cc

Go to the documentation of this file.

/*
 * Scale Stack
 *
 * Copyright 2010 Eric Day
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "config.h"

#include <algorithm>
#include <exception>
#include <functional>

#include <scalestack/event/libevent/handler_provider.h>
#include <scalestack/event/libevent/service.h>
#include <scalestack/event/libevent/thread.h>
#include <scalestack/kernel/delete_pointer.h>
#include <scalestack/kernel/module.h>
#include <scalestack/kernel/option.h>

using namespace std;

namespace scalestack
{
namespace event
{
namespace libevent
{

/*
 * Public methods.
 */

service::service(kernel::module& module):
  event::service(module),
  _current_thread(),
  _threads()
{
  size_t thread_count = _module.get_option("threads").get_size_value();
  if (thread_count == 0)
    _module.log_fatal("The 'threads' option must be greater than 0");

  for (size_t thread_index = 0; thread_index < thread_count; thread_index++)
  {
    thread* thread = NULL;
    try
    {
      thread = new libevent::thread(_module, thread_index);
      _threads.push_back(thread);
    }
    catch (exception&)
    {
      if (thread != NULL)
      {
        thread->shutdown();
        delete thread;
      }

      _remove_threads();
      throw;
    }
  }
}

service::~service()
{
  _remove_threads();
}

/*
 * Private methods.
 */

event::handler_provider* service::_add_handler(handler* handler)
{
  /* Choose assigned thread by round-robin. */
  _current_thread++;
  return new handler_provider(_module, handler,
                              *_threads[_current_thread % _threads.size()]);
}

void service::_remove_threads(void)
{
  for_each(_threads.begin(), _threads.end(), mem_fun(&thread::shutdown));
  for_each(_threads.begin(), _threads.end(), kernel::delete_pointer());
}

} /* namespace libevent */
} /* namespace event */
} /* namespace scalestack */

---