blob: db9b1aa38a333c505debb80375fa32c95189fef0 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
// workqueue-internal.h -- internal work queue header for gold -*- C++ -*-
// Copyright (C) 2006-2019 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(int thread_number) = 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(int thread_number);
// 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)
|
