asio_addon.cpp

Go to the documentation of this file.

/******************************************************************************
 *   Copyright (C) 2013-2014 by Alexander Rykovanov                        *
 *   rykovanov.as@gmail.com                                                   *
 *                                                                            *
 *   This library is free software; you can redistribute it and/or modify     *
 *   it under the terms of the GNU Lesser General Public License as           *
 *   published by the Free Software Foundation; version 3 of the License.     *
 *                                                                            *
 *   This library is distributed in the hope that it will be useful,          *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of           *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the            *
 *   GNU Lesser General Public License for more details.                      *
 *                                                                            *
 *   You should have received a copy of the GNU Lesser General Public License *
 *   along with this library; if not, write to the                            *
 *   Free Software Foundation, Inc.,                                          *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.                *
 ******************************************************************************/

#include <opc/ua/server/addons/asio_addon.h>

#include <boost/asio.hpp>
#include <iostream>
#include <thread>

namespace
{
  class AsioAddonImpl : public OpcUa::Server::AsioAddon
  {
  public:
    AsioAddonImpl()
      : Work(IoService)
    {
    }


    void Initialize(Common::AddonsManager&, const Common::AddonParameters& params) override
    {
      const unsigned threadsNumber = GetThreadsNumber(params);
      //std::cout << "asio| Starting " << threadsNumber << "threads." << std::endl;
      for (unsigned i = 0; i < threadsNumber; ++i)
      {
        Threads.emplace_back([this, i](){
          //std::cout << "asio| Starting thread " << i << "." << std::endl;
          IoService.run();
          //std::cout << "asio| Thread " << i << "exited." << std::endl;
        });
      }
    }

    void Stop() override
    {
      //std::cout << "asio| stopping io service." << std::endl;
      IoService.stop();
      //std::cout << "asio| joining threads." << std::endl;
      std::for_each(Threads.begin(), Threads.end(), [](std::thread& thread){
        thread.join();
      });
    }

    virtual boost::asio::io_service& GetIoService() override
    {
      return IoService;
    }

  unsigned GetThreadsNumber(const Common::AddonParameters& params) const
  {
    unsigned num = 1;
    for (auto paramIt : params.Parameters)
    {
      if (paramIt.Name == "threads")
      {
        num = std::stoi(paramIt.Value);
        break;
      }
    }
    return num;
  }

  private:
    boost::asio::io_service IoService;
    boost::asio::io_service::work Work;
    std::vector<std::thread> Threads;
  };
}


namespace OpcUa
{
  namespace Server
  {

    Common::Addon::UniquePtr AsioAddonFactory::CreateAddon()
    {
      return Common::Addon::UniquePtr(new AsioAddonImpl);
    }

  }
}