Relintai/thread_pool
1
0
Fork 0
mirror of https://github.com/Relintai/thread_pool.git synced 2024-11-14 10:27:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

Added a has_job getter to the ThreadPool singleton.

Browse Source
This commit is contained in:
Relintai 2021-10-31 12:06:00 +01:00
parent 1ed776e266
commit 3bf2cae157
2 changed files with 32 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
thread_pool.cpp
View File

@ -180,6 +180,36 @@ Ref<ThreadPoolJob> ThreadPool::get_queued_job(const Variant &object, const Strin
return Ref<ThreadPoolJob>(); return Ref<ThreadPoolJob>();
} }
bool ThreadPool::has_job(const Ref<ThreadPoolJob> &job) {
_THREAD_SAFE_LOCK_
if (_use_threads) {
for (int i = 0; i < _threads.size(); ++i) {
ThreadPoolContext *context = _threads.get(i);
if (context->job == job) {
_THREAD_SAFE_UNLOCK_
return true;
}
}
}
for (int i = _current_queue_head; i < _current_queue_tail; ++i) {
if (_queue.write[i] == job) {
_THREAD_SAFE_UNLOCK_
return true;
}
}
_THREAD_SAFE_UNLOCK_
return false;
}
void ThreadPool::add_job(const Ref<ThreadPoolJob> &job) { void ThreadPool::add_job(const Ref<ThreadPoolJob> &job) {
_THREAD_SAFE_LOCK_ _THREAD_SAFE_LOCK_
@ -470,6 +500,7 @@ void ThreadPool::_bind_methods() {
ClassDB::bind_method(D_METHOD("get_running_job", "object", "method"), &ThreadPool::get_running_job); ClassDB::bind_method(D_METHOD("get_running_job", "object", "method"), &ThreadPool::get_running_job);
ClassDB::bind_method(D_METHOD("get_queued_job", "object", "method"), &ThreadPool::get_queued_job); ClassDB::bind_method(D_METHOD("get_queued_job", "object", "method"), &ThreadPool::get_queued_job);
ClassDB::bind_method(D_METHOD("has_job", "job"), &ThreadPool::has_job);
ClassDB::bind_method(D_METHOD("add_job", "job"), &ThreadPool::add_job); ClassDB::bind_method(D_METHOD("add_job", "job"), &ThreadPool::add_job);
ClassDB::bind_method(D_METHOD("register_update"), &ThreadPool::register_update); ClassDB::bind_method(D_METHOD("register_update"), &ThreadPool::register_update);

1
thread_pool.h
View File

@ -85,6 +85,7 @@ public:
Ref<ThreadPoolJob> get_running_job(const Variant &object, const StringName &method); Ref<ThreadPoolJob> get_running_job(const Variant &object, const StringName &method);
Ref<ThreadPoolJob> get_queued_job(const Variant &object, const StringName &method); Ref<ThreadPoolJob> get_queued_job(const Variant &object, const StringName &method);
bool has_job(const Ref<ThreadPoolJob> &job);
void add_job(const Ref<ThreadPoolJob> &job); void add_job(const Ref<ThreadPoolJob> &job);
Ref<ThreadPoolExecuteJob> create_execute_job_simple(const Variant &object, const StringName &method); Ref<ThreadPoolExecuteJob> create_execute_job_simple(const Variant &object, const StringName &method);