YiLoadThreadManager.h

Go to the documentation of this file.

// © You i Labs Inc. 2000-2017. All rights reserved.
#ifndef _YI_LOAD_THREAD_MANAGER_INL_
#define _YI_LOAD_THREAD_MANAGER_INL_

#include "signal/YiSignal.h"
#include "thread/YiMutex.h"
#include "thread/YiAutoMutex.h"
#include "thread/YiWaitCondition.h"
#include "thread/YiThread.h"

class IYIRunOnThread
{
public:
    virtual ~IYIRunOnThread() {};
    virtual void DoInWorkerThread() = 0;
};

class CYILoadThread;
class CYIThreadManager;

class CYILoadThreadManager : public CYIThread
{
public:

    CYILoadThreadManager(int32_t nThreads);
    ~CYILoadThreadManager();

    void StopManager();

    void Load(std::unique_ptr<IYIRunOnThread> pResource);

    void Unload(const IYIRunOnThread *pResource);

private:

    struct Stats
    {
        Stats() : nLoad(0), nUnload(0), nThreadStarted(0),
                  nRemovedUnprocessed(0), nThreadStarved(0), nRegularUnload(0)
        {}
        int nLoad;
        int nUnload;
        int nThreadStarted;
        int nRemovedUnprocessed;
        int nThreadStarved;
        int nRegularUnload;
    } mStats;

    void RemoveUnprocessed();

    virtual void Run() override;

    std::list<IYIRunOnThread*> mListLoad;
    std::list<const IYIRunOnThread *> mListUnload;

    typedef std::pair<IYIRunOnThread*, CYILoadThread* > pairDataThread;
    std::list<pairDataThread> mListRunning;

    CYIMutex mQueueMutex;
    CYIWaitCondition mAvailableItem;

    bool   m_Stop;
    uint32_t m_ManagerId;
    std::unique_ptr<CYIThreadManager> m_pThreadManager;
    std::map<int32_t, CYILoadThread *> m_Threads;
};

#endif // _YI_LOAD_THREAD_MANAGER_INL_