Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/task_scheduler/post_task.h" | 5 #include "base/task_scheduler/post_task.h" |
| 6 | 6 |
| 7 #include <utility> | 7 #include <utility> |
| 8 | 8 |
| 9 #include "base/logging.h" | 9 #include "base/logging.h" |
| 10 #include "base/task_scheduler/scoped_set_task_priority_for_current_thread.h" | |
| 10 #include "base/task_scheduler/task_scheduler.h" | 11 #include "base/task_scheduler/task_scheduler.h" |
| 11 #include "base/threading/post_task_and_reply_impl.h" | 12 #include "base/threading/post_task_and_reply_impl.h" |
| 12 | 13 |
| 13 namespace base { | 14 namespace base { |
| 14 | 15 |
| 15 namespace { | 16 namespace { |
| 16 | 17 |
| 17 class PostTaskAndReplyTaskRunner : public internal::PostTaskAndReplyImpl { | 18 class PostTaskAndReplyTaskRunner : public internal::PostTaskAndReplyImpl { |
| 18 public: | 19 public: |
| 19 explicit PostTaskAndReplyTaskRunner(const TaskTraits& traits) | 20 explicit PostTaskAndReplyTaskRunner(const TaskTraits& traits) |
| 20 : traits_(traits) {} | 21 : traits_(traits) {} |
| 21 | 22 |
| 22 private: | 23 private: |
| 23 bool PostTask(const tracked_objects::Location& from_here, | 24 bool PostTask(const tracked_objects::Location& from_here, |
| 24 OnceClosure task) override { | 25 OnceClosure task) override { |
| 25 PostTaskWithTraits(from_here, traits_, std::move(task)); | 26 PostTaskWithTraits(from_here, traits_, std::move(task)); |
| 26 return true; | 27 return true; |
| 27 } | 28 } |
| 28 | 29 |
| 29 const TaskTraits traits_; | 30 const TaskTraits traits_; |
| 30 }; | 31 }; |
| 31 | 32 |
| 33 // Returns TaskTraits based on |traits|. If TaskPriority hasn't been set | |
| 34 // explicitly in |traits|, the returned TaskTraits have the current | |
| 35 // TaskPriority. | |
| 36 TaskTraits CreateTaskTraitsWithExplicitPriority(const TaskTraits& traits) { | |
|
gab
2017/04/25 17:16:41
I prefer GetTaskTraitsWithExplicitPriority() since
fdoray
2017/04/25 17:57:48
Done.
| |
| 37 return traits.priority_set_explicitly() | |
| 38 ? traits | |
| 39 : TaskTraits(traits).WithPriority( | |
| 40 internal::GetTaskPriorityForCurrentThread()); | |
| 41 } | |
| 32 | 42 |
| 33 } // namespace | 43 } // namespace |
| 34 | 44 |
| 35 void PostTask(const tracked_objects::Location& from_here, OnceClosure task) { | 45 void PostTask(const tracked_objects::Location& from_here, OnceClosure task) { |
| 36 PostDelayedTask(from_here, std::move(task), TimeDelta()); | 46 PostDelayedTask(from_here, std::move(task), TimeDelta()); |
| 37 } | 47 } |
| 38 | 48 |
| 39 void PostDelayedTask(const tracked_objects::Location& from_here, | 49 void PostDelayedTask(const tracked_objects::Location& from_here, |
| 40 OnceClosure task, | 50 OnceClosure task, |
| 41 TimeDelta delay) { | 51 TimeDelta delay) { |
| (...skipping 13 matching lines...) Expand all Loading... | |
| 55 PostDelayedTaskWithTraits(from_here, traits, std::move(task), TimeDelta()); | 65 PostDelayedTaskWithTraits(from_here, traits, std::move(task), TimeDelta()); |
| 56 } | 66 } |
| 57 | 67 |
| 58 void PostDelayedTaskWithTraits(const tracked_objects::Location& from_here, | 68 void PostDelayedTaskWithTraits(const tracked_objects::Location& from_here, |
| 59 const TaskTraits& traits, | 69 const TaskTraits& traits, |
| 60 OnceClosure task, | 70 OnceClosure task, |
| 61 TimeDelta delay) { | 71 TimeDelta delay) { |
| 62 DCHECK(TaskScheduler::GetInstance()) | 72 DCHECK(TaskScheduler::GetInstance()) |
| 63 << "Ref. Prerequisite section of post_task.h"; | 73 << "Ref. Prerequisite section of post_task.h"; |
| 64 TaskScheduler::GetInstance()->PostDelayedTaskWithTraits( | 74 TaskScheduler::GetInstance()->PostDelayedTaskWithTraits( |
| 65 from_here, traits, std::move(task), std::move(delay)); | 75 from_here, CreateTaskTraitsWithExplicitPriority(traits), std::move(task), |
| 76 std::move(delay)); | |
| 66 } | 77 } |
| 67 | 78 |
| 68 void PostTaskWithTraitsAndReply(const tracked_objects::Location& from_here, | 79 void PostTaskWithTraitsAndReply(const tracked_objects::Location& from_here, |
| 69 const TaskTraits& traits, | 80 const TaskTraits& traits, |
| 70 OnceClosure task, | 81 OnceClosure task, |
| 71 OnceClosure reply) { | 82 OnceClosure reply) { |
| 72 PostTaskAndReplyTaskRunner(traits).PostTaskAndReply( | 83 PostTaskAndReplyTaskRunner(traits).PostTaskAndReply( |
| 73 from_here, std::move(task), std::move(reply)); | 84 from_here, std::move(task), std::move(reply)); |
| 74 } | 85 } |
| 75 | 86 |
| 76 scoped_refptr<TaskRunner> CreateTaskRunnerWithTraits(const TaskTraits& traits) { | 87 scoped_refptr<TaskRunner> CreateTaskRunnerWithTraits(const TaskTraits& traits) { |
| 77 DCHECK(TaskScheduler::GetInstance()) | 88 DCHECK(TaskScheduler::GetInstance()) |
| 78 << "Ref. Prerequisite section of post_task.h"; | 89 << "Ref. Prerequisite section of post_task.h"; |
| 79 return TaskScheduler::GetInstance()->CreateTaskRunnerWithTraits(traits); | 90 return TaskScheduler::GetInstance()->CreateTaskRunnerWithTraits( |
| 91 CreateTaskTraitsWithExplicitPriority(traits)); | |
| 80 } | 92 } |
| 81 | 93 |
| 82 scoped_refptr<SequencedTaskRunner> CreateSequencedTaskRunnerWithTraits( | 94 scoped_refptr<SequencedTaskRunner> CreateSequencedTaskRunnerWithTraits( |
| 83 const TaskTraits& traits) { | 95 const TaskTraits& traits) { |
| 84 DCHECK(TaskScheduler::GetInstance()) | 96 DCHECK(TaskScheduler::GetInstance()) |
| 85 << "Ref. Prerequisite section of post_task.h"; | 97 << "Ref. Prerequisite section of post_task.h"; |
| 86 return TaskScheduler::GetInstance()->CreateSequencedTaskRunnerWithTraits( | 98 return TaskScheduler::GetInstance()->CreateSequencedTaskRunnerWithTraits( |
| 87 traits); | 99 CreateTaskTraitsWithExplicitPriority(traits)); |
| 88 } | 100 } |
| 89 | 101 |
| 90 scoped_refptr<SingleThreadTaskRunner> CreateSingleThreadTaskRunnerWithTraits( | 102 scoped_refptr<SingleThreadTaskRunner> CreateSingleThreadTaskRunnerWithTraits( |
| 91 const TaskTraits& traits) { | 103 const TaskTraits& traits) { |
| 92 DCHECK(TaskScheduler::GetInstance()) | 104 DCHECK(TaskScheduler::GetInstance()) |
| 93 << "Ref. Prerequisite section of post_task.h"; | 105 << "Ref. Prerequisite section of post_task.h"; |
| 94 return TaskScheduler::GetInstance()->CreateSingleThreadTaskRunnerWithTraits( | 106 return TaskScheduler::GetInstance()->CreateSingleThreadTaskRunnerWithTraits( |
| 95 traits); | 107 CreateTaskTraitsWithExplicitPriority(traits)); |
| 96 } | 108 } |
| 97 | 109 |
| 98 #if defined(OS_WIN) | 110 #if defined(OS_WIN) |
| 99 scoped_refptr<SingleThreadTaskRunner> CreateCOMSTATaskRunnerWithTraits( | 111 scoped_refptr<SingleThreadTaskRunner> CreateCOMSTATaskRunnerWithTraits( |
| 100 const TaskTraits& traits) { | 112 const TaskTraits& traits) { |
| 101 DCHECK(TaskScheduler::GetInstance()) | 113 DCHECK(TaskScheduler::GetInstance()) |
| 102 << "Ref. Prerequisite section of post_task.h"; | 114 << "Ref. Prerequisite section of post_task.h"; |
| 103 return TaskScheduler::GetInstance()->CreateCOMSTATaskRunnerWithTraits(traits); | 115 return TaskScheduler::GetInstance()->CreateCOMSTATaskRunnerWithTraits( |
| 116 CreateTaskTraitsWithExplicitPriority(traits)); | |
| 104 } | 117 } |
| 105 #endif // defined(OS_WIN) | 118 #endif // defined(OS_WIN) |
| 106 | 119 |
| 107 } // namespace base | 120 } // namespace base |
| OLD | NEW |