ed/Odin
1
0
Fork 0
mirror of https://github.com/Ed94/Odin.git synced 2026-08-05 15:18:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move mutex use around in thread pool

Browse Source
This commit is contained in:
gingerBill
2022-12-20 14:45:01 +00:00
parent 8fc9566a83
commit eb0775ad53
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/thread_pool.cpp
+3 -4
View File
@@ -83,10 +83,8 @@ gb_internal void thread_pool_queue_push(ThreadPool *pool, WorkerTask *task) {
gb_internal bool thread_pool_add_task(ThreadPool *pool, WorkerTaskProc *proc, void *data) { gb_internal bool thread_pool_add_task(ThreadPool *pool, WorkerTaskProc *proc, void *data) {
GB_ASSERT(proc != nullptr); GB_ASSERT(proc != nullptr);
mutex_lock(&pool->mutex);
WorkerTask *task = gb_alloc_item(permanent_allocator(), WorkerTask); WorkerTask *task = gb_alloc_item(permanent_allocator(), WorkerTask);
if (task == nullptr) { if (task == nullptr) {
mutex_unlock(&pool->mutex);
GB_PANIC("Out of memory"); GB_PANIC("Out of memory");
return false; return false;
} }
@@ -94,9 +92,10 @@ gb_internal bool thread_pool_add_task(ThreadPool *pool, WorkerTaskProc *proc, vo
task->do_work = proc; task->do_work = proc;
task->data = data; task->data = data;
mutex_lock(&pool->mutex);
thread_pool_queue_push(pool, task); thread_pool_queue_push(pool, task);
GB_ASSERT(pool->ready >= 0); GB_ASSERT(pool->ready >= 0);
pool->ready++; pool->ready.fetch_add(1);
condition_broadcast(&pool->task_cond); condition_broadcast(&pool->task_cond);
mutex_unlock(&pool->mutex); mutex_unlock(&pool->mutex);
return true; return true;
@@ -111,7 +110,7 @@ gb_internal void thread_pool_wait(ThreadPool *pool) {
if (pool->threads.count == 0) { if (pool->threads.count == 0) {
while (!thread_pool_queue_empty(pool)) { while (!thread_pool_queue_empty(pool)) {
thread_pool_do_task(thread_pool_queue_pop(pool)); thread_pool_do_task(thread_pool_queue_pop(pool));
--pool->ready; pool->ready.fetch_sub(1);
} }
GB_ASSERT(pool->ready == 0); GB_ASSERT(pool->ready == 0);
return; return;