ThreadPool.h

Go to the documentation of this file.

// Copyright (c) Qualcomm Technologies, Inc. and/or its subsidiaries.
// SPDX-License-Identifier: BSD-3-Clause-Clear
 
#ifndef THREAD_POOL_H
#define THREAD_POOL_H
 
#include <thread>
#include <vector>
#include <functional>
#include <mutex>
#include <condition_variable>
#include <exception>
 
#include "Utils.h"
#include "Logger.h"
#include "MemoryPool.h"
#include "SafeOps.h"
 
class TaskNode {
public:
    std::function<void(void*)>* taskCallback;
    void* args;
    TaskNode* next;
 
    TaskNode();
    ~TaskNode();
};
 
struct ThreadNode {
    std::thread* th;
    ThreadNode* next;
};
 
class TaskQueue {
private:
    int32_t size;
    TaskNode* head;
    TaskNode* tail;
 
public:
    TaskQueue();
 
    void add(TaskNode* taskNode);
    TaskNode* poll();
    int8_t isEmpty();
    int32_t getSize();
 
    ~TaskQueue();
};
 
static const int32_t maxLoadPerThread = 3;
 
class ThreadPool {
private:
    int32_t mDesiredPoolCapacity; 
    int32_t mMaxPoolCapacity; 
 
    int32_t mCurrentThreadsCount;
    int32_t mTotalTasksCount;
    int8_t mTerminatePool;
 
    TaskQueue* mCurrentTasks;
    ThreadNode* mThreadQueueHead;
    ThreadNode* mThreadQueueTail;
 
    std::mutex mThreadPoolMutex;
    std::condition_variable mThreadPoolCond;
 
    TaskNode* createTaskNode(std::function<void(void*)> taskCallback, void* args);
    int8_t addNewThread(int8_t isCoreThread);
    int8_t threadRoutineHelper(int8_t isCoreThread);
 
public:
    ThreadPool(int32_t desiredCapacity, int32_t maxCapacity);
    ~ThreadPool();
 
    int8_t enqueueTask(std::function<void(void*)> callBack, void* arg);
};
 
#endif