146 lines
2.5 KiB
C++
Executable File
146 lines
2.5 KiB
C++
Executable File
/*
|
|
* This file is a part of Winix
|
|
* and is not publicly distributed
|
|
*
|
|
* Copyright (c) 2011-2014, Tomasz Sowa
|
|
* All rights reserved.
|
|
*
|
|
*/
|
|
|
|
#include <signal.h>
|
|
#include "threadmanager.h"
|
|
#include "log.h"
|
|
|
|
|
|
namespace Winix
|
|
{
|
|
|
|
|
|
|
|
|
|
ThreadManager::ThreadManager()
|
|
{
|
|
were_started = false;
|
|
}
|
|
|
|
|
|
void ThreadManager::SetSynchro(Synchro * psynchro)
|
|
{
|
|
synchro = psynchro;
|
|
}
|
|
|
|
|
|
void ThreadManager::Init()
|
|
{
|
|
sigset_t set;
|
|
|
|
sigemptyset(&set);
|
|
sigaddset(&set, SIGTERM);
|
|
sigaddset(&set, SIGINT);
|
|
|
|
// blocking SIGTERM and SIGINT
|
|
// new threads will have the signals blocked too
|
|
pthread_sigmask(SIG_BLOCK, &set, 0);
|
|
}
|
|
|
|
|
|
|
|
|
|
void ThreadManager::Add(BaseThread * pbase, const wchar_t * thread_name)
|
|
{
|
|
ThreadItem item;
|
|
|
|
item.object = pbase;
|
|
item.name = thread_name;
|
|
thread_tab.push_back(item);
|
|
|
|
if( were_started )
|
|
Start(thread_tab.size() - 1);
|
|
else
|
|
log << log4 << "TM: added a thread to the queue, number: " << (thread_tab.size()-1)
|
|
<< ", name: " << thread_name << logend;
|
|
}
|
|
|
|
|
|
void ThreadManager::Add(BaseThread & pbase, const wchar_t * thread_name)
|
|
{
|
|
Add(&pbase, thread_name);
|
|
}
|
|
|
|
|
|
void ThreadManager::Add(BaseThread * pbase, const std::wstring & thread_name)
|
|
{
|
|
Add(pbase, thread_name.c_str());
|
|
}
|
|
|
|
void ThreadManager::Add(BaseThread & pbase, const std::wstring & thread_name)
|
|
{
|
|
Add(&pbase, thread_name.c_str());
|
|
}
|
|
|
|
|
|
|
|
void ThreadManager::StartAll()
|
|
{
|
|
synchro->Lock();
|
|
|
|
for(size_t i=0 ; i<thread_tab.size() ; ++i)
|
|
Start(i);
|
|
|
|
synchro->Unlock();
|
|
|
|
were_started = true;
|
|
}
|
|
|
|
|
|
void ThreadManager::Start(size_t i)
|
|
{
|
|
if( i < thread_tab.size() )
|
|
{
|
|
thread_tab[i].object->SetSynchro(synchro);
|
|
|
|
if( thread_tab[i].object->StartThread() )
|
|
{
|
|
log << log4 << "TM: thread " << i << " (" << thread_tab[i].object->ThreadId() << ", name: "
|
|
<< thread_tab[i].name << ") started" << logend;
|
|
}
|
|
else
|
|
{
|
|
log << log4 << "TM: cannot run a thread, thread number: " << i
|
|
<< ", name: " << thread_tab[i].name << logend;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
void ThreadManager::StopAll()
|
|
{
|
|
if( !were_started )
|
|
return;
|
|
|
|
// WakeUpThread() should be used with Lock/Unlock
|
|
synchro->Lock();
|
|
|
|
for(size_t i=0 ; i<thread_tab.size() ; ++i)
|
|
thread_tab[i].object->WakeUpThread();
|
|
|
|
synchro->Unlock();
|
|
|
|
|
|
for(size_t i=0 ; i<thread_tab.size() ; ++i)
|
|
{
|
|
log << log4 << "TM: waiting for thread " << i << " (" << thread_tab[i].object->ThreadId()
|
|
<< ", name: " << thread_tab[i].name << ")" << logend;
|
|
|
|
thread_tab[i].object->WaitForThread();
|
|
|
|
log << log4 << "TM: thread " << i << " terminated" << logend;
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
} // namespace Winix
|
|
|