base/threading/worker_pool_win.cc

   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
  22 DWORD CALLBACK WorkItemCallback(void* param) {
  23   PendingTask* pending_task = static_cast<PendingTask*>(param);
  24   TRACE_EVENT2("toplevel", "WorkItemCallback::Run",
  25                "src_file", pending_task->posted_from.file_name(),
  26                "src_func", pending_task->posted_from.function_name());
  27
  28   tracked_objects::ThreadData::PrepareForStartOfRun(pending_task->birth_tally);
  29
  30   g_worker_pool_running_on_this_thread.Get().Set(true);
  31
  32   tracked_objects::TaskStopwatch stopwatch;
  33   stopwatch.Start();
  34   pending_task->task.Run();
  35   stopwatch.Stop();
  36
  37   g_worker_pool_running_on_this_thread.Get().Set(false);
  38
  39   tracked_objects::ThreadData::TallyRunOnWorkerThreadIfTracking(
  40       pending_task->birth_tally,
  41       tracked_objects::TrackedTime(pending_task->time_posted),
  42       stopwatch);
  43
  44   delete pending_task;
  45   return 0;
  46 }
  47
  48 // Takes ownership of |pending_task|
  49 bool PostTaskInternal(PendingTask* pending_task, bool task_is_slow) {
  50   ULONG flags = 0;
  51   if (task_is_slow)
  52     flags |= WT_EXECUTELONGFUNCTION;
  53
  54   if (!QueueUserWorkItem(WorkItemCallback, pending_task, flags)) {
  55     DPLOG(ERROR) << "QueueUserWorkItem failed";
  56     delete pending_task;
  57     return false;
  58   }
  59
  60   return true;
  61 }
  62
  63 }  // namespace
  64
  65 // static
  66 bool WorkerPool::PostTask(const tracked_objects::Location& from_here,
  67                           const base::Closure& task, bool task_is_slow) {
  68   PendingTask* pending_task = new PendingTask(from_here, task);
  69   return PostTaskInternal(pending_task, task_is_slow);
  70 }
  71
  72 // static
  73 bool WorkerPool::RunsTasksOnCurrentThread() {
  74   return g_worker_pool_running_on_this_thread.Get().Get();
  75 }
  76
  77 }  // namespace base