YiThreadPools.h

Go to the documentation of this file.

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

#include "YiTask.h"
#include "YiThreadPool.h"

class CYIThreadPools
{
public:
    CYIThreadPools(bool bInstanciateDefaultPool = true, uint32_t uDefaultPoolThreadsCount = 0);
    ~CYIThreadPools();
    
    static CYIThreadPool *GetDefaultThreadPool();
    
    static CYIThreadPool *GetManagedThreadPool(const CYIString &name, uint32_t uMaxSize = 4, uint32_t uMaxSleepingSize = 4, uint32_t uInitialSize = 2, uint32_t uExpiryTimeMs = 30000, YI_PRIORITY ePriority = YI_PRIORITY_DEFAULT, uint32_t uStackSize = CYIThread::DEFAULT_STACK_SIZE);

    static bool DestroyManagedThreadPool(const CYIString &name);
    
    static uint32_t GetDefaultPoolThreadsCount();

    static uint32_t GetLogicalProcessorsCount(bool bUsableProcessorsOnly); 

public: //static helper functions
    static bool RunOnNewThread(std::unique_ptr<CYITaskBase> pTask);

    static bool RunOnUIThread(std::unique_ptr<CYITaskBase> pTask);

    static bool RunAsync(std::unique_ptr<CYITaskBase> pTask);

    CYIThreadPool *InstanceGetDefaultThreadPool(); 
    CYIThreadPool *InstanceGetManagedThreadPool(const CYIString &name, uint32_t uMaxSize = 4, uint32_t uMaxSleepingSize = 4, uint32_t uInitialSize = 2, uint32_t uExpiryTimeMs = 30000, YI_PRIORITY ePriority = YI_PRIORITY_DEFAULT, uint32_t uStackSize = CYIThread::DEFAULT_STACK_SIZE); 
    bool InstanceDestroyManagedThreadPool(const CYIString &name); 
    bool InstanceRunOnNewThread(std::unique_ptr<CYITaskBase> pTask); 
    bool InstanceRunAsync(std::unique_ptr<CYITaskBase> pTask); 

private:
    class OneOffWorkerThread : public CYIThread
    {
    public:
        OneOffWorkerThread(CYIThreadPools *pOwnerPools, std::unique_ptr<CYITaskBase> pTask);
        virtual ~OneOffWorkerThread();
        virtual void Run() override;
        void SetDeletable();
        bool IsDeletable();
        
    private:
        CYIThreadPools *m_pOwnerPools;
        std::unique_ptr<CYITaskBase> m_pTask;
        bool      m_bDeletable;
        CYIMutex     m_mutex;
    };
    
    class ThreadDeletingTask : public CYITask<bool>
    {
    public:
        ThreadDeletingTask(CYIThreadPools *pOwnerPools, OneOffWorkerThread *pThread);
        virtual ~ThreadDeletingTask();
        virtual bool Run() override;
        
    private:
        CYIThreadPools *m_pOwnerPools;
        OneOffWorkerThread *m_pThread;
    };

    static CYIThreadPools *GetInstance(); 

    CYIMutex m_mutex;
    bool m_bShutDown;
    CYIThreadPool *m_pDefaultThreadPool;
    std::map<CYIString, CYIThreadPool*> m_threadPools;
    std::atomic<int32_t> m_nOneOffWorkerID;
    std::atomic<int32_t> m_nNonDeletedOneOffWorkers;

    YI_DISALLOW_COPY_AND_ASSIGN(CYIThreadPools)
};

#endif // _YI_THREAD_POOLS_H_