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
15namespace base {
16
17namespace {
18
19base::LazyInstance<ThreadLocalBoolean>::Leaky
20    g_worker_pool_running_on_this_thread = LAZY_INSTANCE_INITIALIZER;
21
22DWORD 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|
45bool 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
62bool 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
69bool WorkerPool::RunsTasksOnCurrentThread() {
70  return g_worker_pool_running_on_this_thread.Get().Get();
71}
72
73}  // namespace base
74