ThreadPool: allow task cancellation in EnqueueAsyncWork

DiligentGraphics · Oct 14, 2024 · 65ecd01 · 65ecd01
1 parent eb9af01
commit 65ecd01
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Common/interface/ThreadPool.hpp b/Common/interface/ThreadPool.hpp
@@ -185,7 +185,9 @@ RefCntAutoPtr<IAsyncTask> EnqueueAsyncWork(IThreadPool* pThreadPool,
 
         virtual void DILIGENT_CALL_TYPE Run(Uint32 ThreadId) override final
         {
-            ASYNC_TASK_STATUS TaskStatus = m_Handler(ThreadId);
+            ASYNC_TASK_STATUS TaskStatus = !m_bSafelyCancel.load() ?
+                m_Handler(ThreadId) :
+                ASYNC_TASK_STATUS_CANCELLED;
             SetStatus(TaskStatus);
         }