rpc_thread_pool.hpp

Go to the documentation of this file.

#ifndef KAGOME_CORE_API_RPC_THREAD_POOL_HPP
#define KAGOME_CORE_API_RPC_THREAD_POOL_HPP

#include <boost/asio/io_service.hpp>
#include <boost/asio/signal_set.hpp>
#include <set>
#include <thread>

#include "api/transport/rpc_io_context.hpp"
#include "log/logger.hpp"

namespace kagome::api {

  class RpcThreadPool : public std::enable_shared_from_this<RpcThreadPool> {
   public:
    using Context = RpcContext;

    struct Configuration {
      size_t min_thread_number = 1;
      size_t max_thread_number = 10;
    };

    RpcThreadPool(std::shared_ptr<Context> context,
                  const Configuration &configuration);

    ~RpcThreadPool() = default;

    void start();

    void stop();

   private:
    std::shared_ptr<Context> context_;
    const Configuration config_;

    std::vector<std::shared_ptr<std::thread>> threads_;

    log::Logger logger_ = log::createLogger("RpcThreadPool", "rpc_transport");
  };

}  // namespace kagome::api

#endif  // KAGOME_CORE_API_RPC_THREAD_POOL_HPP