// workqueue-internal.h -- internal work queue header for gold -*- C++ -*- // Copyright 2006, 2007, 2008 Free Software Foundation, Inc. // Written by Ian Lance Taylor <iant@google.com>. // This file is part of gold. // This program is free software; you can redistribute it and/or modify // it under the terms of the GNU General Public License as published by // the Free Software Foundation; either version 3 of the License, or // (at your option) any later version. // This program is distributed in the hope that it will be useful, // but WITHOUT ANY WARRANTY; without even the implied warranty of // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the // GNU General Public License for more details. // You should have received a copy of the GNU General Public License // along with this program; if not, write to the Free Software // Foundation, Inc., 51 Franklin Street - Fifth Floor, Boston, // MA 02110-1301, USA. #ifndef GOLD_WORKQUEUE_INTERNAL_H #define GOLD_WORKQUEUE_INTERNAL_H #include <queue> #include <csignal> #include "gold-threads.h" #include "workqueue.h" // This is an internal header file for different gold workqueue // implementations. namespace gold { class Workqueue_thread; // The Workqueue_threader abstract class. This is the interface used // by the general workqueue code to manage threads. class Workqueue_threader { public: Workqueue_threader(Workqueue* workqueue) : workqueue_(workqueue) { } virtual ~Workqueue_threader() { } // Set the number of threads to use. This is ignored when not using // threads. virtual void set_thread_count(int) = 0; // Return whether to cancel the current thread. virtual bool should_cancel_thread() = 0; protected: // Get the Workqueue. Workqueue* get_workqueue() { return this->workqueue_; } private: // The Workqueue. Workqueue* workqueue_; }; // The threaded instantiation of Workqueue_threader. class Workqueue_threader_threadpool : public Workqueue_threader { public: Workqueue_threader_threadpool(Workqueue*); ~Workqueue_threader_threadpool(); // Set the thread count. void set_thread_count(int); // Return whether to cancel a thread. bool should_cancel_thread(); // Process all tasks. This keeps running until told to cancel. void process(int thread_number) { this->get_workqueue()->process(thread_number); } private: // This is set if we need to check the thread count. volatile sig_atomic_t check_thread_count_; // Lock for the remaining members. Lock lock_; // The number of threads we want to create. This is set to zero // when all threads should exit. int desired_thread_count_; // The number of threads currently running. int threads_; }; } // End namespace gold. #endif // !defined(GOLD_WORKQUEUE_INTERNAL_H)