void
NotifyTaskUnixSignal(
PLW_TASK pTask,
siginfo_t* pInfo
)
{
LOCK_THREAD(pTask->pThread);
if (pTask->EventSignal != TASK_COMPLETE_MASK)
{
while (pTask->pUnixSignal->si_signo)
{
pthread_cond_wait(&pTask->pThread->Event, &pTask->pThread->Lock);
if (pTask->EventSignal == TASK_COMPLETE_MASK)
{
goto cleanup;
}
}
*pTask->pUnixSignal = *pInfo;
pTask->EventSignal |= LW_TASK_EVENT_UNIX_SIGNAL;
RingRemove(&pTask->SignalRing);
RingEnqueue(&pTask->pThread->Tasks, &pTask->SignalRing);
SignalThread(pTask->pThread);
}
cleanup:
UNLOCK_THREAD(pTask->pThread);
}
VOID
LwRtlCancelTaskGroup(
PLW_TASK_GROUP group
)
{
PRING ring = NULL;
PLW_TASK pTask = NULL;
ULONG i = 0;
LOCK_GROUP(group);
group->bCancelled = TRUE;
for (i = 0; i < group->pPool->ulEventThreadCount; i++)
{
LOCK_THREAD(&group->pPool->pEventThreads[i]);
}
for (ring = group->Tasks.pNext; ring != &group->Tasks; ring = ring->pNext)
{
pTask = LW_STRUCT_FROM_FIELD(ring, SELECT_TASK, GroupRing);
pTask->TriggerSet |= LW_TASK_EVENT_EXPLICIT | LW_TASK_EVENT_CANCEL;
}
for (i = 0; i < group->pPool->ulEventThreadCount; i++)
{
SignalThread(&group->pPool->pEventThreads[i]);
UNLOCK_THREAD(&group->pPool->pEventThreads[i]);
}
UNLOCK_GROUP(group);
}
VOID
LwRtlCancelTaskGroup(
PLW_TASK_GROUP pGroup
)
{
PRING ring = NULL;
PLW_TASK pTask = NULL;
LOCK_GROUP(pGroup);
pGroup->bCancelled = TRUE;
LockAllThreads(pGroup->pPool);
for (ring = pGroup->Tasks.pNext; ring != &pGroup->Tasks; ring = ring->pNext)
{
pTask = LW_STRUCT_FROM_FIELD(ring, EPOLL_TASK, GroupRing);
if (pTask->EventSignal != TASK_COMPLETE_MASK)
{
pTask->EventSignal |= LW_TASK_EVENT_EXPLICIT | LW_TASK_EVENT_CANCEL;
RingRemove(&pTask->SignalRing);
RingEnqueue(&pTask->pThread->Tasks, &pTask->SignalRing);
SignalThread(pTask->pThread);
}
}
UnlockAllThreads(pGroup->pPool);
UNLOCK_GROUP(pGroup);
}
void
NotifyTaskUnixSignal(
PLW_TASK pTask,
siginfo_t* pInfo
)
{
LOCK_THREAD(pTask->pThread);
if (pTask->TriggerSet != TASK_COMPLETE_MASK)
{
while (pTask->pUnixSignal->si_signo)
{
pthread_cond_wait(&pTask->pThread->Event, &pTask->pThread->Lock);
if (pTask->TriggerSet == TASK_COMPLETE_MASK)
{
goto cleanup;
}
}
*pTask->pUnixSignal = *pInfo;
pTask->TriggerSet |= LW_TASK_EVENT_UNIX_SIGNAL;
SignalThread(pTask->pThread);
}
cleanup:
UNLOCK_THREAD(pTask->pThread);
}
VOID
LwRtlWakeTask(
PLW_TASK pTask
)
{
LOCK_THREAD(pTask->pThread);
pTask->TriggerSet |= LW_TASK_EVENT_EXPLICIT;
SignalThread(pTask->pThread);
UNLOCK_THREAD(pTask->pThread);
}
VOID
LwRtlCancelTask(
PLW_TASK pTask
)
{
LOCK_THREAD(pTask->pThread);
pTask->TriggerSet |= LW_TASK_EVENT_EXPLICIT | LW_TASK_EVENT_CANCEL;
SignalThread(pTask->pThread);
UNLOCK_THREAD(pTask->pThread);
}
VOID
LwRtlFreeThreadPool(
PLW_THREAD_POOL* ppPool
)
{
PLW_THREAD_POOL pPool = *ppPool;
PEPOLL_THREAD pThread = NULL;
int i = 0;
if (pPool)
{
LOCK_POOL(pPool);
pPool->bShutdown = TRUE;
pthread_cond_broadcast(&pPool->Event);
UNLOCK_POOL(pPool);
if (pPool->pEventThreads)
{
for (i = 0; i < pPool->ulEventThreadCount; i++)
{
pThread = &pPool->pEventThreads[i];
LOCK_THREAD(pThread);
pThread->bShutdown = TRUE;
SignalThread(pThread);
UNLOCK_THREAD(pThread);
pthread_join(pThread->Thread, NULL);
DestroyEventThread(pThread);
}
RtlMemoryFree(pPool->pEventThreads);
}
if (pPool->pDelegate)
{
ReleaseDelegatePool(&pPool->pDelegate);
}
pthread_cond_destroy(&pPool->Event);
pthread_mutex_destroy(&pPool->Lock);
DestroyWorkThreads(&pPool->WorkThreads);
RtlMemoryFree(pPool);
*ppPool = NULL;
}
}
VOID
LwRtlWakeTask(
PLW_TASK pTask
)
{
LOCK_THREAD(pTask->pThread);
if (pTask->EventSignal != TASK_COMPLETE_MASK)
{
pTask->EventSignal |= LW_TASK_EVENT_EXPLICIT;
RingRemove(&pTask->SignalRing);
RingEnqueue(&pTask->pThread->Tasks, &pTask->SignalRing);
SignalThread(pTask->pThread);
}
UNLOCK_THREAD(pTask->pThread);
}
WELS_THREAD_ERROR_CODE CWelsThreadPool::OnTaskStop (CWelsTaskThread* pThread, IWelsTask* pTask) {
//fprintf(stdout, "CWelsThreadPool::OnTaskStop 0: Task %x at Thread %x Finished\n", pTask, pThread);
RemoveThreadFromBusyList (pThread);
AddThreadToIdleQueue (pThread);
//fprintf(stdout, "CWelsThreadPool::OnTaskStop 1: Task %x at Thread %x Finished, m_pSink=%x\n", pTask, pThread, m_pSink);
if (pTask->GetSink()) {
pTask->GetSink()->OnTaskExecuted();
}
//if (m_pSink) {
// m_pSink->OnTaskExecuted (pTask);
//}
//fprintf(stdout, "CWelsThreadPool::OnTaskStop 2: Task %x at Thread %x Finished\n", pTask, pThread);
SignalThread();
//fprintf(stdout, "ThreadPool: Task %x at Thread %x Finished\n", pTask, pThread);
return WELS_THREAD_ERROR_OK;
}
WELS_THREAD_ERROR_CODE CWelsThreadPool::QueueTask (IWelsTask* pTask) {
CWelsAutoLock cLock (m_cLockPool);
//fprintf(stdout, "CWelsThreadPool::QueueTask: %d, pTask=%x\n", m_iRefCount, pTask);
if (GetWaitedTaskNum() == 0) {
CWelsTaskThread* pThread = GetIdleThread();
if (pThread != NULL) {
//fprintf(stdout, "ThreadPool: ExecuteTask = %x at thread %x\n", pTask, pThread);
pThread->SetTask (pTask);
return WELS_THREAD_ERROR_OK;
}
}
//fprintf(stdout, "ThreadPool: AddTaskToWaitedList: %x\n", pTask);
if (false == AddTaskToWaitedList (pTask)){
return WELS_THREAD_ERROR_GENERAL;
}
//fprintf(stdout, "ThreadPool: SignalThread: %x\n", pTask);
SignalThread();
return WELS_THREAD_ERROR_OK;
}
