| 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 | 
|---|