Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(394)

Side by Side Diff: runtime/vm/thread_pool.h

Issue 1279733003: Clean VM thread shutdown: (Closed) Base URL: git@github.com:dart-lang/sdk.git@master
Patch Set: Merge Created 5 years, 4 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « runtime/vm/thread.cc ('k') | runtime/vm/thread_pool.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 // Copyright (c) 2012, the Dart project authors. Please see the AUTHORS file 1 // Copyright (c) 2012, the Dart project authors. Please see the AUTHORS file
2 // for details. All rights reserved. Use of this source code is governed by a 2 // for details. All rights reserved. Use of this source code is governed by a
3 // BSD-style license that can be found in the LICENSE file. 3 // BSD-style license that can be found in the LICENSE file.
4 4
5 #ifndef VM_THREAD_POOL_H_ 5 #ifndef VM_THREAD_POOL_H_
6 #define VM_THREAD_POOL_H_ 6 #define VM_THREAD_POOL_H_
7 7
8 #include "vm/allocation.h"
8 #include "vm/globals.h" 9 #include "vm/globals.h"
9 #include "vm/os_thread.h" 10 #include "vm/os_thread.h"
10 11
11 namespace dart { 12 namespace dart {
12 13
13 class ThreadPool { 14 class ThreadPool {
14 public: 15 public:
15 // Subclasses of Task are able to run on a ThreadPool. 16 // Subclasses of Task are able to run on a ThreadPool.
16 class Task { 17 class Task {
17 protected: 18 protected:
18 Task(); 19 Task();
19 20
20 public: 21 public:
21 virtual ~Task(); 22 virtual ~Task();
22 23
23 // Override this to provide task-specific behavior. 24 // Override this to provide task-specific behavior.
24 virtual void Run() = 0; 25 virtual void Run() = 0;
25 26
26 private: 27 private:
27 DISALLOW_COPY_AND_ASSIGN(Task); 28 DISALLOW_COPY_AND_ASSIGN(Task);
28 }; 29 };
29 30
30 ThreadPool(); 31 ThreadPool();
31 32
32 // Shuts down this thread pool. Causes workers to terminate 33 // Shuts down this thread pool. Causes workers to terminate
33 // themselves when they are active again. 34 // themselves when they are active again.
34 ~ThreadPool(); 35 ~ThreadPool();
35 36
36 // Runs a task on the thread pool. 37 // Runs a task on the thread pool.
37 void Run(Task* task); 38 void Run(Task* task);
38 39
39 // Some simple stats. 40 // Some simple stats.
40 uint64_t workers_running() const { return count_running_; } 41 uint64_t workers_running() const { return count_running_; }
41 uint64_t workers_idle() const { return count_idle_; } 42 uint64_t workers_idle() const { return count_idle_; }
42 uint64_t workers_started() const { return count_started_; } 43 uint64_t workers_started() const { return count_started_; }
43 uint64_t workers_stopped() const { return count_stopped_; } 44 uint64_t workers_stopped() const { return count_stopped_; }
44 45
45 private: 46 private:
46 friend class ThreadPoolTestPeer;
47
48 class Worker { 47 class Worker {
49 public: 48 public:
50 explicit Worker(ThreadPool* pool); 49 explicit Worker(ThreadPool* pool);
51 50
52 // Sets a task on the worker. 51 // Sets a task on the worker.
53 void SetTask(Task* task); 52 void SetTask(Task* task);
54 53
55 // Starts the thread for the worker. This should only be called 54 // Starts the thread for the worker. This should only be called
56 // after a task has been set by the initial call to SetTask(). 55 // after a task has been set by the initial call to SetTask().
57 void StartThread(); 56 void StartThread();
58 57
59 // Main loop for a worker. 58 // Main loop for a worker. Returns true if worker is removed from thread
60 void Loop(); 59 // lists, false otherwise.
60 bool Loop();
61 61
62 // Causes worker to terminate eventually. 62 // Causes worker to terminate eventually.
63 void Shutdown(); 63 void Shutdown();
64 64
65 // Get the Worker's thread id.
66 ThreadId id() { return id_; }
67
65 private: 68 private:
66 friend class ThreadPool; 69 friend class ThreadPool;
67 70
68 // The main entry point for new worker threads. 71 // The main entry point for new worker threads.
69 static void Main(uword args); 72 static void Main(uword args);
70 73
71 bool IsDone() const { return pool_ == NULL; } 74 bool IsDone() const { return done_; }
72 75
73 // Fields owned by Worker. 76 // Fields owned by Worker.
74 Monitor monitor_; 77 Monitor monitor_;
75 ThreadPool* pool_; 78 ThreadPool* pool_;
79 bool done_;
76 Task* task_; 80 Task* task_;
81 ThreadId id_;
82 bool started_;
77 83
78 // Fields owned by ThreadPool. Workers should not look at these 84 // Fields owned by ThreadPool. Workers should not look at these
79 // directly. It's like looking at the sun. 85 // directly. It's like looking at the sun.
80 bool owned_; // Protected by ThreadPool::mutex_ 86 bool owned_; // Protected by ThreadPool::mutex_
81 Worker* all_next_; // Protected by ThreadPool::mutex_ 87 Worker* all_next_; // Protected by ThreadPool::mutex_
82 Worker* idle_next_; // Protected by ThreadPool::mutex_ 88 Worker* idle_next_; // Protected by ThreadPool::mutex_
83 89
90 Worker* shutdown_next_; // Protected by ThreadPool::exit_monitor
91
84 DISALLOW_COPY_AND_ASSIGN(Worker); 92 DISALLOW_COPY_AND_ASSIGN(Worker);
85 }; 93 };
86 94
87 void Shutdown(); 95 void Shutdown();
88 96
89 // Expensive. Use only in assertions. 97 // Expensive. Use only in assertions.
90 bool IsIdle(Worker* worker); 98 bool IsIdle(Worker* worker);
91 99
92 bool RemoveWorkerFromIdleList(Worker* worker); 100 bool RemoveWorkerFromIdleList(Worker* worker);
93 bool RemoveWorkerFromAllList(Worker* worker); 101 bool RemoveWorkerFromAllList(Worker* worker);
94 102
103 void AddWorkerToShutdownList(Worker* worker);
104 bool RemoveWorkerFromShutdownList(Worker* worker);
105
95 // Worker operations. 106 // Worker operations.
96 void SetIdle(Worker* worker); 107 void SetIdle(Worker* worker);
97 bool ReleaseIdleWorker(Worker* worker); 108 bool ReleaseIdleWorker(Worker* worker);
98 109
99 Mutex mutex_; 110 Mutex mutex_;
100 bool shutting_down_; 111 bool shutting_down_;
101 Worker* all_workers_; 112 Worker* all_workers_;
102 Worker* idle_workers_; 113 Worker* idle_workers_;
103 uint64_t count_started_; 114 uint64_t count_started_;
104 uint64_t count_stopped_; 115 uint64_t count_stopped_;
105 uint64_t count_running_; 116 uint64_t count_running_;
106 uint64_t count_idle_; 117 uint64_t count_idle_;
107 118
108 static Monitor* exit_monitor_; // Used only in testing. 119 Monitor exit_monitor_;
109 static int* exit_count_; // Used only in testing. 120 Worker* shutting_down_workers_;
110 121
111 DISALLOW_COPY_AND_ASSIGN(ThreadPool); 122 DISALLOW_COPY_AND_ASSIGN(ThreadPool);
112 }; 123 };
113 124
114 } // namespace dart 125 } // namespace dart
115 126
116 #endif // VM_THREAD_POOL_H_ 127 #endif // VM_THREAD_POOL_H_
OLDNEW
« no previous file with comments | « runtime/vm/thread.cc ('k') | runtime/vm/thread_pool.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698