| 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/scoped_set_task_priority_for_current_thread.h" |
| 11 #include "base/task_scheduler/task_scheduler.h" | 11 #include "base/task_scheduler/task_scheduler.h" |
| 12 #include "base/threading/post_task_and_reply_impl.h" | 12 #include "base/threading/post_task_and_reply_impl.h" |
| 13 | 13 |
| 14 namespace base { | 14 namespace base { |
| 15 | 15 |
| 16 namespace { | 16 namespace { |
| 17 | 17 |
| 18 class PostTaskAndReplyTaskRunner : public internal::PostTaskAndReplyImpl { | 18 class PostTaskWithTraitsAndReplyTaskRunner : |
| 19 public internal::PostTaskAndReplyImpl { |
| 19 public: | 20 public: |
| 20 explicit PostTaskAndReplyTaskRunner(const TaskTraits& traits) | 21 explicit PostTaskWithTraitsAndReplyTaskRunner(const TaskTraits& traits) |
| 21 : traits_(traits) {} | 22 : traits_(traits) {} |
| 22 | 23 |
| 23 private: | 24 private: |
| 24 bool PostTask(const Location& from_here, OnceClosure task) override { | 25 bool PostTask(const Location& from_here, 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 }; |
| (...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 74 "base::test::ScopedTaskEnvironment member in your fixture.\n"; | 75 "base::test::ScopedTaskEnvironment member in your fixture.\n"; |
| 75 TaskScheduler::GetInstance()->PostDelayedTaskWithTraits( | 76 TaskScheduler::GetInstance()->PostDelayedTaskWithTraits( |
| 76 from_here, GetTaskTraitsWithExplicitPriority(traits), std::move(task), | 77 from_here, GetTaskTraitsWithExplicitPriority(traits), std::move(task), |
| 77 std::move(delay)); | 78 std::move(delay)); |
| 78 } | 79 } |
| 79 | 80 |
| 80 void PostTaskWithTraitsAndReply(const Location& from_here, | 81 void PostTaskWithTraitsAndReply(const Location& from_here, |
| 81 const TaskTraits& traits, | 82 const TaskTraits& traits, |
| 82 OnceClosure task, | 83 OnceClosure task, |
| 83 OnceClosure reply) { | 84 OnceClosure reply) { |
| 84 PostTaskAndReplyTaskRunner(traits).PostTaskAndReply( | 85 PostTaskWithTraitsAndReplyTaskRunner(traits).PostTaskAndReply( |
| 85 from_here, std::move(task), std::move(reply)); | 86 from_here, std::move(task), std::move(reply)); |
| 86 } | 87 } |
| 87 | 88 |
| 88 scoped_refptr<TaskRunner> CreateTaskRunnerWithTraits(const TaskTraits& traits) { | 89 scoped_refptr<TaskRunner> CreateTaskRunnerWithTraits(const TaskTraits& traits) { |
| 89 DCHECK(TaskScheduler::GetInstance()) | 90 DCHECK(TaskScheduler::GetInstance()) |
| 90 << "Ref. Prerequisite section of post_task.h.\n\n" | 91 << "Ref. Prerequisite section of post_task.h.\n\n" |
| 91 "Hint: if this is in a unit test, you're likely merely missing a " | 92 "Hint: if this is in a unit test, you're likely merely missing a " |
| 92 "base::test::ScopedTaskEnvironment member in your fixture.\n"; | 93 "base::test::ScopedTaskEnvironment member in your fixture.\n"; |
| 93 return TaskScheduler::GetInstance()->CreateTaskRunnerWithTraits( | 94 return TaskScheduler::GetInstance()->CreateTaskRunnerWithTraits( |
| 94 GetTaskTraitsWithExplicitPriority(traits)); | 95 GetTaskTraitsWithExplicitPriority(traits)); |
| (...skipping 27 matching lines...) Expand all Loading... |
| 122 DCHECK(TaskScheduler::GetInstance()) | 123 DCHECK(TaskScheduler::GetInstance()) |
| 123 << "Ref. Prerequisite section of post_task.h.\n\n" | 124 << "Ref. Prerequisite section of post_task.h.\n\n" |
| 124 "Hint: if this is in a unit test, you're likely merely missing a " | 125 "Hint: if this is in a unit test, you're likely merely missing a " |
| 125 "base::test::ScopedTaskEnvironment member in your fixture.\n"; | 126 "base::test::ScopedTaskEnvironment member in your fixture.\n"; |
| 126 return TaskScheduler::GetInstance()->CreateCOMSTATaskRunnerWithTraits( | 127 return TaskScheduler::GetInstance()->CreateCOMSTATaskRunnerWithTraits( |
| 127 GetTaskTraitsWithExplicitPriority(traits), thread_mode); | 128 GetTaskTraitsWithExplicitPriority(traits), thread_mode); |
| 128 } | 129 } |
| 129 #endif // defined(OS_WIN) | 130 #endif // defined(OS_WIN) |
| 130 | 131 |
| 131 } // namespace base | 132 } // namespace base |
| OLD | NEW |