Scale Stack 0.3 Developer Documentation

scalestack/event/test/test.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 <cassert>
#include <memory>

#include <scalestack/event/test/handler.h>
#include <scalestack/event/test/test.h>
#include <scalestack/kernel/core.h>
#include <scalestack/kernel/module.h>

using namespace std;
using namespace scalestack;

namespace scalestack
{
namespace event
{
namespace test
{

/*
 * Public methods.
 */

test::test(kernel::module& module, size_t handler_count):
  kernel::service(module),
  _handler_count(handler_count),
  _mutex(module),
  _handler_service(module)
{
}

test::~test()
{
  _handler_service.stop();
  assert(_handler_count == 0);
}

void test::remove_handler(void)
{
  _mutex.lock();
  _handler_count--;
  if (_handler_count == 0)
    _module.get_core().shutdown();
  _mutex.unlock();
}

} /* namespace test */
} /* namespace event */
} /* namespace scalestack */

static void start(kernel::module& module)
{
  auto_ptr<event::test::test> test(new event::test::test(module, 6));
  module.get_core().add_service("event::test", test.get());
  test.release();
}

static void stop(kernel::module& module)
{
  delete module.get_core().remove_service("event::test");
}

static void run(kernel::module& module)
{
  event::test::test& test = static_cast<event::test::test&>(
    module.get_core().get_service("event::test"));

  new event::test::handler(test, event::test::handler::STATE_DONE);
  new event::test::handler(test, event::test::handler::STATE_START);
  new event::test::handler(test, event::test::handler::STATE_RUN);
  new event::test::handler(test, event::test::handler::STATE_TIMER);
  new event::test::handler(test, event::test::handler::STATE_READ);
  new event::test::handler(test, event::test::handler::STATE_WRITE);
}

SCALESTACK_KERNEL_MODULE(NULL, start, stop, run, NULL);

---