• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4 
5 // WARNING: You should probably be using Thread (thread.h) instead.  Thread is
6 //          Chrome's message-loop based Thread abstraction, and if you are a
7 //          thread running in the browser, there will likely be assumptions
8 //          that your thread will have an associated message loop.
9 //
10 // This is a simple thread interface that backs to a native operating system
11 // thread.  You should use this only when you want a thread that does not have
12 // an associated MessageLoop.  Unittesting is the best example of this.
13 //
14 // The simplest interface to use is DelegateSimpleThread, which will create
15 // a new thread, and execute the Delegate's virtual Run() in this new thread
16 // until it has completed, exiting the thread.
17 //
18 // NOTE: You *MUST* call Join on the thread to clean up the underlying thread
19 // resources.  You are also responsible for destructing the SimpleThread object.
20 // It is invalid to destroy a SimpleThread while it is running, or without
21 // Start() having been called (and a thread never created).  The Delegate
22 // object should live as long as a DelegateSimpleThread.
23 //
24 // Thread Safety: A SimpleThread is not completely thread safe.  It is safe to
25 // access it from the creating thread or from the newly created thread.  This
26 // implies that the creator thread should be the thread that calls Join.
27 //
28 // Example:
29 //   class MyThreadRunner : public DelegateSimpleThread::Delegate { ... };
30 //   MyThreadRunner runner;
31 //   DelegateSimpleThread thread(&runner, "good_name_here");
32 //   thread.Start();
33 //   // Start will return after the Thread has been successfully started and
34 //   // initialized.  The newly created thread will invoke runner->Run(), and
35 //   // run until it returns.
36 //   thread.Join();  // Wait until the thread has exited.  You *MUST* Join!
37 //   // The SimpleThread object is still valid, however you may not call Join
38 //   // or Start again.
39 
40 #ifndef BASE_THREADING_SIMPLE_THREAD_H_
41 #define BASE_THREADING_SIMPLE_THREAD_H_
42 
43 #include <stddef.h>
44 
45 #include <queue>
46 #include <string>
47 #include <vector>
48 
49 #include "base/base_export.h"
50 #include "base/compiler_specific.h"
51 #include "base/synchronization/lock.h"
52 #include "base/synchronization/waitable_event.h"
53 #include "base/threading/platform_thread.h"
54 
55 namespace base {
56 
57 // This is the base SimpleThread.  You can derive from it and implement the
58 // virtual Run method, or you can use the DelegateSimpleThread interface.
59 class BASE_EXPORT SimpleThread : public PlatformThread::Delegate {
60  public:
61   class BASE_EXPORT Options {
62    public:
Options()63     Options() : stack_size_(0), priority_(ThreadPriority::NORMAL) {}
Options(ThreadPriority priority)64     explicit Options(ThreadPriority priority)
65         : stack_size_(0), priority_(priority) {}
~Options()66     ~Options() {}
67 
68     // We use the standard compiler-supplied copy constructor.
69 
70     // A custom stack size, or 0 for the system default.
set_stack_size(size_t size)71     void set_stack_size(size_t size) { stack_size_ = size; }
stack_size()72     size_t stack_size() const { return stack_size_; }
73 
74     // A custom thread priority.
set_priority(ThreadPriority priority)75     void set_priority(ThreadPriority priority) { priority_ = priority; }
priority()76     ThreadPriority priority() const { return priority_; }
77    private:
78     size_t stack_size_;
79     ThreadPriority priority_;
80   };
81 
82   // Create a SimpleThread.  |options| should be used to manage any specific
83   // configuration involving the thread creation and management.
84   // Every thread has a name, in the form of |name_prefix|/TID, for example
85   // "my_thread/321".  The thread will not be created until Start() is called.
86   explicit SimpleThread(const std::string& name_prefix);
87   SimpleThread(const std::string& name_prefix, const Options& options);
88 
89   ~SimpleThread() override;
90 
91   virtual void Start();
92   virtual void Join();
93 
94   // Subclasses should override the Run method.
95   virtual void Run() = 0;
96 
97   // Return the thread name prefix, or "unnamed" if none was supplied.
name_prefix()98   std::string name_prefix() { return name_prefix_; }
99 
100   // Return the completed name including TID, only valid after Start().
name()101   std::string name() { return name_; }
102 
103   // Return the thread id, only valid after Start().
tid()104   PlatformThreadId tid() { return tid_; }
105 
106   // Return True if Start() has ever been called.
107   bool HasBeenStarted();
108 
109   // Return True if Join() has evern been called.
HasBeenJoined()110   bool HasBeenJoined() { return joined_; }
111 
112   // Overridden from PlatformThread::Delegate:
113   void ThreadMain() override;
114 
115  private:
116   const std::string name_prefix_;
117   std::string name_;
118   const Options options_;
119   PlatformThreadHandle thread_;  // PlatformThread handle, invalid after Join!
120   WaitableEvent event_;          // Signaled if Start() was ever called.
121   PlatformThreadId tid_;         // The backing thread's id.
122   bool joined_;                  // True if Join has been called.
123 };
124 
125 class BASE_EXPORT DelegateSimpleThread : public SimpleThread {
126  public:
127   class BASE_EXPORT Delegate {
128    public:
Delegate()129     Delegate() { }
~Delegate()130     virtual ~Delegate() { }
131     virtual void Run() = 0;
132   };
133 
134   DelegateSimpleThread(Delegate* delegate,
135                        const std::string& name_prefix);
136   DelegateSimpleThread(Delegate* delegate,
137                        const std::string& name_prefix,
138                        const Options& options);
139 
140   ~DelegateSimpleThread() override;
141   void Run() override;
142 
143  private:
144   Delegate* delegate_;
145 };
146 
147 // DelegateSimpleThreadPool allows you to start up a fixed number of threads,
148 // and then add jobs which will be dispatched to the threads.  This is
149 // convenient when you have a lot of small work that you want done
150 // multi-threaded, but don't want to spawn a thread for each small bit of work.
151 //
152 // You just call AddWork() to add a delegate to the list of work to be done.
153 // JoinAll() will make sure that all outstanding work is processed, and wait
154 // for everything to finish.  You can reuse a pool, so you can call Start()
155 // again after you've called JoinAll().
156 class BASE_EXPORT DelegateSimpleThreadPool
157     : public DelegateSimpleThread::Delegate {
158  public:
159   typedef DelegateSimpleThread::Delegate Delegate;
160 
161   DelegateSimpleThreadPool(const std::string& name_prefix, int num_threads);
162   ~DelegateSimpleThreadPool() override;
163 
164   // Start up all of the underlying threads, and start processing work if we
165   // have any.
166   void Start();
167 
168   // Make sure all outstanding work is finished, and wait for and destroy all
169   // of the underlying threads in the pool.
170   void JoinAll();
171 
172   // It is safe to AddWork() any time, before or after Start().
173   // Delegate* should always be a valid pointer, NULL is reserved internally.
174   void AddWork(Delegate* work, int repeat_count);
AddWork(Delegate * work)175   void AddWork(Delegate* work) {
176     AddWork(work, 1);
177   }
178 
179   // We implement the Delegate interface, for running our internal threads.
180   void Run() override;
181 
182  private:
183   const std::string name_prefix_;
184   int num_threads_;
185   std::vector<DelegateSimpleThread*> threads_;
186   std::queue<Delegate*> delegates_;
187   base::Lock lock_;            // Locks delegates_
188   WaitableEvent dry_;    // Not signaled when there is no work to do.
189 };
190 
191 }  // namespace base
192 
193 #endif  // BASE_THREADING_SIMPLE_THREAD_H_
194