ThreadPool: allow task cancellation in EnqueueAsyncWork

TheMostDiligent · TheMostDiligent · commit 65ecd0195b7f · 2024-10-14T12:35:50.000-07:00
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);
         }
 

Original file line number	Diff line number	Diff line change
`@@ -185,7 +185,9 @@ RefCntAutoPtr<IAsyncTask> EnqueueAsyncWork(IThreadPool* pThreadPool,`
`185`	`185`
`186`	`186`	`virtual void DILIGENT_CALL_TYPE Run(Uint32 ThreadId) override final`
`187`	`187`	`{`
`188`		`- ASYNC_TASK_STATUS TaskStatus = m_Handler(ThreadId);`
	`188`	`+ ASYNC_TASK_STATUS TaskStatus = !m_bSafelyCancel.load() ?`
	`189`	`+ m_Handler(ThreadId) :`
	`190`	`+ ASYNC_TASK_STATUS_CANCELLED;`
`189`	`191`	`SetStatus(TaskStatus);`
`190`	`192`	`}`
`191`	`193`