summaryrefslogtreecommitdiffstats
path: root/base/threading/worker_pool_win.cc
blob: b73cabdb46b4410eefbd9b290dbd99e683326b3c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
// Copyright (c) 2011 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "base/threading/worker_pool.h"

#include "base/bind.h"
#include "base/debug/trace_event.h"
#include "base/logging.h"
#include "base/task.h"
#include "base/tracked_objects.h"

namespace base {

namespace {

struct PendingTask {
  PendingTask(
      const tracked_objects::Location& posted_from,
      const base::Closure& task)
      : posted_from(posted_from),
        task(task) {
    birth_tally = tracked_objects::ThreadData::TallyABirthIfActive(posted_from);
    time_posted = tracked_objects::ThreadData::Now();
  }

  // Counter for location where the Closure was posted from.
  tracked_objects::Births* birth_tally;

  // Time the task was posted.
  tracked_objects::TrackedTime time_posted;

  // The site this PendingTask was posted from.
  tracked_objects::Location posted_from;

  // The task to run.
  base::Closure task;
};

DWORD CALLBACK WorkItemCallback(void* param) {
  PendingTask* pending_task = static_cast<PendingTask*>(param);
  UNSHIPPED_TRACE_EVENT2("task", "WorkItemCallback::Run",
                         "src_file", pending_task->posted_from.file_name(),
                         "src_func", pending_task->posted_from.function_name());

  tracked_objects::TrackedTime start_time = tracked_objects::ThreadData::Now();

  pending_task->task.Run();

  tracked_objects::ThreadData::TallyRunOnWorkerThreadIfTracking(
      pending_task->birth_tally, pending_task->time_posted,
      start_time, tracked_objects::ThreadData::Now());

  delete pending_task;
  return 0;
}

// Takes ownership of |pending_task|
bool PostTaskInternal(PendingTask* pending_task, bool task_is_slow) {
  ULONG flags = 0;
  if (task_is_slow)
    flags |= WT_EXECUTELONGFUNCTION;

  if (!QueueUserWorkItem(WorkItemCallback, pending_task, flags)) {
    DLOG(ERROR) << "QueueUserWorkItem failed: " << GetLastError();
    delete pending_task;
    return false;
  }

  return true;
}

}  // namespace

bool WorkerPool::PostTask(const tracked_objects::Location& from_here,
                          Task* task, bool task_is_slow) {
  PendingTask* pending_task =
      new PendingTask(from_here,
                      base::Bind(&subtle::TaskClosureAdapter::Run,
                                 new subtle::TaskClosureAdapter(task)));
  return PostTaskInternal(pending_task, task_is_slow);
}

bool WorkerPool::PostTask(const tracked_objects::Location& from_here,
                          const base::Closure& task, bool task_is_slow) {
  PendingTask* pending_task = new PendingTask(from_here, task);
  return PostTaskInternal(pending_task, task_is_slow);
}

}  // namespace base