forked from ChunelFeng/CGraph
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathUThreadBase.h
More file actions
245 lines (202 loc) · 6.76 KB
/
UThreadBase.h
File metadata and controls
245 lines (202 loc) · 6.76 KB
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
/***************************
@Author: Chunel
@Contact: chunel@foxmail.com
@File: UThreadBase.h
@Time: 2021/7/2 11:24 下午
@Desc:
***************************/
#ifndef CGRAPH_UTHREADBASE_H
#define CGRAPH_UTHREADBASE_H
#include <thread>
#include "../UThreadObject.h"
#include "../Queue/UQueueInclude.h"
#include "../Task/UTaskInclude.h"
#include "../Metrics/UMetrics.h"
CGRAPH_NAMESPACE_BEGIN
class UThreadBase : public UThreadObject {
protected:
explicit UThreadBase() {
done_ = true;
is_init_ = false;
is_running_ = false;
pool_task_queue_ = nullptr;
pool_priority_task_queue_ = nullptr;
config_ = nullptr;
total_task_num_ = 0;
}
~UThreadBase() override {
reset();
}
CStatus destroy() override {
CGRAPH_FUNCTION_BEGIN
CGRAPH_ASSERT_INIT(true)
reset();
CGRAPH_FUNCTION_END
}
/**
* 从线程池的队列中，获取任务
* @param task
* @return
*/
virtual CBool popPoolTask(UTaskRef task) {
CBool result = pool_task_queue_->tryPop(task);
if (!result && CGRAPH_THREAD_TYPE_SECONDARY == type_) {
// 如果辅助线程没有获取到的话，还需要再尝试从长时间任务队列中，获取一次
result = pool_priority_task_queue_->tryPop(task);
}
metrics_.calcPool(result, 1);
return result;
}
/**
* 从线程池的队列中中，获取批量任务
* @param tasks
* @return
*/
virtual CBool popPoolTask(UTaskArrRef tasks) {
CBool result = pool_task_queue_->tryPop(tasks, config_->max_pool_batch_size_);
if (!result && CGRAPH_THREAD_TYPE_SECONDARY == type_) {
result = pool_priority_task_queue_->tryPop(tasks, 1); // 从优先队列里，最多pop出来一个
}
metrics_.calcPool(result, tasks.size());
return result;
}
/**
* 执行单个任务
* @param task
*/
CVoid runTask(UTask& task) {
is_running_ = true;
task();
total_task_num_++;
is_running_ = false;
}
/**
* 批量执行任务
* @param tasks
*/
CVoid runTasks(UTaskArr& tasks) {
is_running_ = true;
for (auto& task : tasks) {
task();
}
total_task_num_ += tasks.size();
is_running_ = false;
}
/**
* 清空所有任务内容
*/
CVoid reset() {
done_ = false;
cv_.notify_one(); // 防止主线程 wait时间过长，导致的结束缓慢问题
if (thread_.joinable()) {
thread_.join(); // 等待线程结束
}
is_init_ = false;
is_running_ = false;
total_task_num_ = 0;
}
/**
* 执行单个消息
* @return
*/
virtual CVoid processTask() = 0;
/**
* 获取批量执行task信息
*/
virtual CVoid processTasks() = 0;
/**
* 循环处理任务
* @return
*/
CVoid loopProcess() {
CGRAPH_ASSERT_NOT_NULL_THROW_ERROR(config_)
if (config_->batch_task_enable_) {
while (done_) {
processTasks(); // 批量任务获取执行接口
}
} else {
while (done_) {
processTask(); // 单个任务获取执行接口
}
}
}
/**
* 设置线程优先级，仅针对非windows平台使用
*/
CVoid setSchedParam() {
#ifndef _WIN32
int priority = CGRAPH_THREAD_SCHED_OTHER;
int policy = CGRAPH_THREAD_MIN_PRIORITY;
if (type_ == CGRAPH_THREAD_TYPE_PRIMARY) {
priority = config_->primary_thread_priority_;
policy = config_->primary_thread_policy_;
} else if (type_ == CGRAPH_THREAD_TYPE_SECONDARY) {
priority = config_->secondary_thread_priority_;
policy = config_->secondary_thread_policy_;
}
auto handle = thread_.native_handle();
sched_param param = { calcPriority(priority) };
int ret = pthread_setschedparam(handle, calcPolicy(policy), ¶m);
if (0 != ret) {
CGRAPH_ECHO("warning : set thread sched param failed, system error code is [%d]", ret);
}
#endif
}
/**
* 设置线程亲和性，仅针对linux系统
*/
CVoid setAffinity(int index) {
#if defined(__linux__) && !defined(__ANDROID__)
if (!config_->bind_cpu_enable_ || CGRAPH_CPU_NUM == 0 || index < 0) {
return;
}
cpu_set_t mask;
CPU_ZERO(&mask);
CPU_SET(index % CGRAPH_CPU_NUM, &mask);
auto handle = thread_.native_handle();
int ret = pthread_setaffinity_np(handle, sizeof(cpu_set_t), &mask);
if (0 != ret) {
CGRAPH_ECHO("warning : set thread affinity failed, system error code is [%d]", ret);
}
#endif
}
private:
/**
* 设定计算线程调度策略信息，
* 非OTHER/RR/FIFO对应数值，统一返回OTHER类型
* @param policy
* @return
*/
static CInt calcPolicy(int policy) {
return (CGRAPH_THREAD_SCHED_OTHER == policy
|| CGRAPH_THREAD_SCHED_RR == policy
|| CGRAPH_THREAD_SCHED_FIFO == policy)
? policy : CGRAPH_THREAD_SCHED_OTHER;
}
/**
* 设定线程优先级信息
* 超过[min,max]范围，统一设置为min值
* @param priority
* @return
*/
static CInt calcPriority(int priority) {
return (priority >= CGRAPH_THREAD_MIN_PRIORITY
&& priority <= CGRAPH_THREAD_MAX_PRIORITY)
? priority : CGRAPH_THREAD_MIN_PRIORITY;
}
protected:
CBool done_; // 线程状态标记
CBool is_init_; // 标记初始化状态
CBool is_running_; // 是否正在执行
CInt type_ = 0; // 用于区分线程类型（主线程、辅助线程）
CULong total_task_num_ = 0; // 处理的任务的数字
UAtomicQueue<UTask>* pool_task_queue_; // 用于存放线程池中的普通任务
UAtomicPriorityQueue<UTask>* pool_priority_task_queue_; // 用于存放线程池中的包含优先级任务的队列，仅辅助线程可以执行
UThreadPoolConfigPtr config_ = nullptr; // 配置参数信息
UMetrics metrics_; // 线程中的指标信息
std::thread thread_; // 线程类
std::mutex mutex_;
std::condition_variable cv_;
};
CGRAPH_NAMESPACE_END
#endif //CGRAPH_UTHREADBASE_H