C++ ThreadPool::Shutdown方法代码示例

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();
}

//.........这里部分代码省略.........
	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;
}