int main(int argc, char **argv){
bool ret = true;
long size = sysconf(_SC_NPROCESSORS_ONLN);
cout<<"number of workers: "<<size<<endl;
ThreadPool p;
ret = p.Init(size, start_routine, arg);
if (!ret){
return -1;
}
ret = p.Start();
if (!ret){
p.Shutdown();
return -1;
}
int n = 3;
while(n-->0){
sleep(2);
}
p.Shutdown();
return 0;
}
void PelotonInit::Shutdown() {
// shut down index tuner
if (settings::SettingsManager::GetBool(settings::SettingId::index_tuner)) {
auto &index_tuner = tuning::IndexTuner::GetInstance();
index_tuner.Stop();
}
// shut down layout tuner
if (settings::SettingsManager::GetBool(settings::SettingId::layout_tuner)) {
auto &layout_tuner = tuning::LayoutTuner::GetInstance();
layout_tuner.Stop();
}
// shut down GC.
gc::GCManagerFactory::GetInstance().StopGC();
// shut down epoch.
concurrency::EpochManagerFactory::GetInstance().StopEpoch();
// stop worker pool
threadpool::MonoQueuePool::GetInstance().Shutdown();
// stop indextuner thread pool
if (settings::SettingsManager::GetBool(settings::SettingId::brain)) {
threadpool::MonoQueuePool::GetBrainInstance().Shutdown();
}
thread_pool.Shutdown();
// shutdown protocol buf library
google::protobuf::ShutdownProtobufLibrary();
// clear parameters
google::ShutDownCommandLineFlags();
}
// Load with multithreaded
void Level::LoadMultiple()
{
ThreadPool * pool = new ThreadPool();
pool->Initialize(prefs.threads, ThreadPoolWork);
stack<ThreadString *> strings;
// Preparing region object
RegionReader region = RegionReader();
int current = 0;
bool useRegion = (prefs.version == BETA);
int n = 0;
// Start swimming
while (
(!files.empty() ||
pool->Progress()) &&
!cancel)
{
// Get latest result
if (!pool->EmptyResult())
{
ThreadChunk * chnk = (ThreadChunk *)pool->PopResult();
if (chnk != 0)
{
++result;
}
else
{
--total;
}
}
// Push work
if (useRegion)
{
if (current < region.GetAmountChunks() && !pool->FullWork())
{
UINT size = 0;
BYTE * data = region.GetChunk(current++, size);
ThreadString * str = new ThreadString();
str->file.assign(data, data+size);
str->prefs = prefs;
chunks.push_back(str->chunk = new Chunk());
pool->PushWork(str);
strings.push(str);
++n;
}
if (files.empty() && region.GetAmountChunks() == current)
useRegion = false;
else if (!files.empty() && current >= region.GetAmountChunks())
{
region.Load(files.top());
files.pop();
current = 0;
// Reduce amount
total -= 1024 - region.GetAmountChunks();
}
}
else if (!files.empty())
{
if (!pool->FullWork())
{
ThreadString * str = new ThreadString();
str->file = files.top();
str->prefs = prefs;
chunks.push_back(str->chunk = new Chunk());
if (pool->PushWork(str))
{
files.pop();
strings.push(str);
++n;
}
// Evade memory spikes
else
{
delete str->chunk;
delete str;
}
}
}
// No more work
else if (pool->NoMoreWork())
{
// Wait for pool to close
if (pool->Kill())
break;
else
continue;
}
work = (total - files.size()) - result;
done = ((float)result/(float)total);
Port::Rest(0);
}
int amount = pool->GetClosedThreads();
if (cancel)
while (!pool->Kill());
#ifdef WIN32
if (pool->GetClosedThreads() < prefs.threads)
MessageBoxA(0, "Some threads were not released.", "PixelMap", MB_OK);
#endif
// Clean up after you
while (!strings.empty())
{
ThreadString * t = strings.top();
strings.pop();
if (t)
delete t;
}
pool->Shutdown();
delete pool;
}