EventLoopThreadPool.h

Go to the documentation of this file.

/**
* @file EventLoopThreadPool.h.
* @brief The EventLoopThreadPool Class Definitions.
* @author Spices & Muduo.
*/
 
#pragma once
#include "Core/Thread/ThreadPoolBasic.h"
#include "EventLoop.h"
#include "Core/Thread/Semaphore.h"
 
namespace Spices {
 
namespace Net {
 
    /**
    * @brief ThreadPool of Multiple threading EventLoop.
    */
    class EventLoopThreadPool : public ThreadPool
    {
    public:
 
        using ThreadInitCallback = std::function<void(EventLoop*)>;
 
    public:
 
        /**
        * @brief Constructor Function.
        * Create Specific ThreadPool.
        * @param[in] listenAddress InetAddress.
        */
        EventLoopThreadPool(const InetAddress& listenAddress)
            : ThreadPool_Basic(listenAddress.ToIPPort())
            , m_Next(0)
            , m_ListenAddress(listenAddress)
            , m_ThreadInitCallback(nullptr)
        {}
 
        /**
        * @brief Destructor Function.
        */
        virtual ~EventLoopThreadPool() override = default;
 
        /**
        * @brief Get next thread EventLoop.
        * @return Returns Next Thread EventLoop.
        */
        EventLoop* GetNextLoop();
 
        /**
        * @brief Get all thread EventLoop.
        * @return Returns all thread EventLoop.
        */
        std::vector<EventLoop*> GetAllLoops();
 
        /**
        * @brief Get ThreadPool Name.
        * @return Returns ThreadPool Name.
        */
        const std::string& Name() const { return m_PoolName; }
 
        /**
        * @brief Thread Function.
        * @param[in] thread Thread Entity.
        */
        void ThreadFunc(Thread<>* thread);
 
        /**
        * @brief Start Run this thread pool.
        * @param[in] initThreadSize Thread Size.
        * @param[in] cb ThreadInitCallback.
        */
        void Start(int initThreadSize = 3, ThreadInitCallback cb = nullptr);
 
    private:
 
        /**
        * @brief Next EventLoop index.
        */
        int m_Next;
 
        /**
        * @brief ListenAddress.
        */
        InetAddress m_ListenAddress;
 
        /**
        * @brief All threads EventLoop collection.
        */
        std::vector<EventLoop*> m_Loops;
 
        /**
        * @brief ThreadInitCallback.
        */
        ThreadInitCallback m_ThreadInitCallback;
 
        /**
        * @brief ThreadsPrepared condition.
        */
        semaphore m_IsThreadsPrepared;
 
    };
 
}
 
}