task-queue/task_queue_manager.h at main · ouxianghui/task-queue · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
#pragma once

#include <memory>
#include <vector>
#include <string>
#include <unordered_map>
#include <mutex>

namespace vi {

class TaskQueue;

class TaskQueueManager {
public:
    static std::unique_ptr<TaskQueueManager>& instance();

    ~TaskQueueManager();

    void create(const std::vector<std::string>& nameList);

    TaskQueue* queue(const std::string& name);

    bool hasQueue(const std::string& name);

private:
    void clear();

    bool exist(const std::string& name);

private:
    TaskQueueManager();

    TaskQueueManager(TaskQueueManager&&) = delete;

    TaskQueueManager(const TaskQueueManager&) = delete;

    TaskQueueManager& operator=(const TaskQueueManager&) = delete;

private:
    std::mutex m_mutex;

    std::unordered_map<std::string, std::unique_ptr<TaskQueue>> m_queueMap;

};

}

#define TQMgr vi::TaskQueueManager::instance()

#define TQ(name) TQMgr->queue(name)