1 // Copyright (c) 2012 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 #include "base/threading/worker_pool.h"
6
7 #include "base/bind.h"
8 #include "base/callback.h"
9 #include "base/debug/trace_event.h"
10 #include "base/logging.h"
11 #include "base/pending_task.h"
12 #include "base/threading/thread_local.h"
13 #include "base/tracked_objects.h"
14
15 namespace base {
16
17 namespace {
18
19 base::LazyInstance<ThreadLocalBoolean>::Leaky
20 g_worker_pool_running_on_this_thread = LAZY_INSTANCE_INITIALIZER;
21
WorkItemCallback(void * param)22 DWORD CALLBACK WorkItemCallback(void* param) {
23 PendingTask* pending_task = static_cast<PendingTask*>(param);
24 TRACE_EVENT2("task", "WorkItemCallback::Run",
25 "src_file", pending_task->posted_from.file_name(),
26 "src_func", pending_task->posted_from.function_name());
27
28 tracked_objects::TrackedTime start_time =
29 tracked_objects::ThreadData::NowForStartOfRun(pending_task->birth_tally);
30
31 g_worker_pool_running_on_this_thread.Get().Set(true);
32 pending_task->task.Run();
33 g_worker_pool_running_on_this_thread.Get().Set(false);
34
35 tracked_objects::ThreadData::TallyRunOnWorkerThreadIfTracking(
36 pending_task->birth_tally,
37 tracked_objects::TrackedTime(pending_task->time_posted), start_time,
38 tracked_objects::ThreadData::NowForEndOfRun());
39
40 delete pending_task;
41 return 0;
42 }
43
44 // Takes ownership of |pending_task|
PostTaskInternal(PendingTask * pending_task,bool task_is_slow)45 bool PostTaskInternal(PendingTask* pending_task, bool task_is_slow) {
46 ULONG flags = 0;
47 if (task_is_slow)
48 flags |= WT_EXECUTELONGFUNCTION;
49
50 if (!QueueUserWorkItem(WorkItemCallback, pending_task, flags)) {
51 DLOG_GETLASTERROR(ERROR) << "QueueUserWorkItem failed";
52 delete pending_task;
53 return false;
54 }
55
56 return true;
57 }
58
59 } // namespace
60
61 // static
PostTask(const tracked_objects::Location & from_here,const base::Closure & task,bool task_is_slow)62 bool WorkerPool::PostTask(const tracked_objects::Location& from_here,
63 const base::Closure& task, bool task_is_slow) {
64 PendingTask* pending_task = new PendingTask(from_here, task);
65 return PostTaskInternal(pending_task, task_is_slow);
66 }
67
68 // static
RunsTasksOnCurrentThread()69 bool WorkerPool::RunsTasksOnCurrentThread() {
70 return g_worker_pool_running_on_this_thread.Get().Get();
71 }
72
73 } // namespace base
74