| // Copyright 2018 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. |
| |
| #ifndef CONTENT_BROWSER_SCHEDULER_BROWSER_TASK_EXECUTOR_H_ |
| #define CONTENT_BROWSER_SCHEDULER_BROWSER_TASK_EXECUTOR_H_ |
| |
| #include <memory> |
| |
| #include "base/gtest_prod_util.h" |
| #include "base/memory/scoped_refptr.h" |
| #include "base/task/task_executor.h" |
| #include "build/build_config.h" |
| #include "content/browser/scheduler/browser_io_task_environment.h" |
| #include "content/browser/scheduler/browser_ui_thread_scheduler.h" |
| #include "content/common/content_export.h" |
| #include "content/public/browser/browser_task_traits.h" |
| #include "content/public/browser/browser_thread.h" |
| |
| namespace content { |
| |
| class BrowserTaskExecutorTest; |
| class BrowserProcessSubThread; |
| |
| // This class's job is to map base::TaskTraits to actual task queues for the |
| // browser process. |
| class CONTENT_EXPORT BrowserTaskExecutor : public base::TaskExecutor { |
| public: |
| // Creates and registers a BrowserTaskExecutor on the current thread which |
| // owns a BrowserUIThreadScheduler. This facilitates posting tasks to a |
| // BrowserThread via //base/task/post_task.h. |
| // All BrowserThread::UI task queues except best effort ones are also enabled. |
| // TODO(carlscab): These queues should be enabled in |
| // BrowserMainLoop::InitializeMainThread() but some Android tests fail if we |
| // do so. |
| static void Create(); |
| |
| // Creates the IO thread using the scheduling infrastructure set up in the |
| // Create() method. That means that clients have access to TaskRunners |
| // associated with the IO thread before that thread is even created. In order |
| // to do so this class will own the TaskEnvironment for the IO thread until |
| // the thread is created, at which point ownership will be transferred and the |
| // |BrowserTaskExecutor| will only communicate with it via TaskRunner |
| // instances. |
| // |
| // Browser task queues will initially be disabled, that is tasks posted to |
| // them will not run. But the default task runner of the thread (the one you |
| // get via ThreadTaskRunnerHandle::Get()) will be active. This is the same |
| // task runner you get by calling BrowserProcessSubThread::task_runner(). The |
| // queues can be initialized by calling InitializeIOThread which is done |
| // during Chromium starup in BrowserMainLoop::CreateThreads. |
| // |
| // Early on during Chromium startup we initialize the ServiceManager and it |
| // needs to run tasks immediately. The ServiceManager itself does not know |
| // about the IO thread (it does not use the browser task traits), it only uses |
| // the task runner provided to it during initialization and possibly |
| // ThreadTaskRunnerHandle::Get() from tasks it posts. But we currently run it |
| // on the IO thread so we need the default task runner to be active for its |
| // tasks to run. Note that since tasks posted via the browser task traits will |
| // not run they won't be able to access the default task runner either, so for |
| // those tasks the default task queue is also "disabled". |
| // |
| // Attention: This method can only be called once (as there must be only one |
| // IO thread). |
| static std::unique_ptr<BrowserProcessSubThread> CreateIOThread(); |
| |
| // Enables non best effort queues on the IO thread. Usually called from |
| // BrowserMainLoop::CreateThreads. |
| static void InitializeIOThread(); |
| |
| // Enables all queues on all threads. |
| // Can be called multiple times. |
| static void EnableAllQueues(); |
| |
| // As Create but with the user provided objects. |
| static void CreateForTesting( |
| std::unique_ptr<BrowserUIThreadScheduler> browser_ui_thread_scheduler, |
| std::unique_ptr<BrowserIOTaskEnvironment> browser_io_task_environment); |
| |
| // This must be called after the FeatureList has been initialized in order |
| // for scheduling experiments to function. |
| static void PostFeatureListSetup(); |
| |
| // Winds down the BrowserTaskExecutor, after this no tasks can be executed |
| // and the base::TaskExecutor APIs are non-functional but won't crash if |
| // called. |
| static void Shutdown(); |
| |
| // Unregister and delete the TaskExecutor after a test. |
| static void ResetForTesting(); |
| |
| // Runs all pending tasks for the given thread. Tasks posted after this method |
| // is called (in particular any task posted from within any of the pending |
| // tasks) will be queued but not run. Conceptually this call will disable all |
| // queues, run any pending tasks, and re-enable all the queues. |
| // |
| // If any of the pending tasks posted a task, these could be run by calling |
| // this method again or running a regular RunLoop. But if that were the case |
| // you should probably rewrite you tests to wait for a specific event instead. |
| static void RunAllPendingTasksOnThreadForTesting( |
| BrowserThread::ID identifier); |
| |
| struct ThreadIdAndQueueType { |
| BrowserThread::ID thread_id; |
| BrowserTaskQueues::QueueType queue_type; |
| }; |
| |
| static ThreadIdAndQueueType GetThreadIdAndQueueType( |
| const base::TaskTraits& traits); |
| |
| // base::TaskExecutor implementation. |
| bool PostDelayedTask(const base::Location& from_here, |
| const base::TaskTraits& traits, |
| base::OnceClosure task, |
| base::TimeDelta delay) override; |
| |
| scoped_refptr<base::TaskRunner> CreateTaskRunner( |
| const base::TaskTraits& traits) override; |
| |
| scoped_refptr<base::SequencedTaskRunner> CreateSequencedTaskRunner( |
| const base::TaskTraits& traits) override; |
| |
| scoped_refptr<base::SingleThreadTaskRunner> CreateSingleThreadTaskRunner( |
| const base::TaskTraits& traits, |
| base::SingleThreadTaskRunnerThreadMode thread_mode) override; |
| |
| #if defined(OS_WIN) |
| scoped_refptr<base::SingleThreadTaskRunner> CreateCOMSTATaskRunner( |
| const base::TaskTraits& traits, |
| base::SingleThreadTaskRunnerThreadMode thread_mode) override; |
| #endif // defined(OS_WIN) |
| |
| private: |
| friend class BrowserTaskExecutorTest; |
| |
| static void CreateInternal( |
| std::unique_ptr<BrowserUIThreadScheduler> browser_ui_thread_scheduler, |
| std::unique_ptr<BrowserIOTaskEnvironment> browser_io_task_environment); |
| |
| // For GetProxyTaskRunnerForThread(). |
| FRIEND_TEST_ALL_PREFIXES(BrowserTaskExecutorTest, |
| EnsureUIThreadTraitPointsToExpectedQueue); |
| FRIEND_TEST_ALL_PREFIXES(BrowserTaskExecutorTest, |
| EnsureIOThreadTraitPointsToExpectedQueue); |
| FRIEND_TEST_ALL_PREFIXES(BrowserTaskExecutorTest, |
| BestEffortTasksRunAfterStartup); |
| |
| explicit BrowserTaskExecutor( |
| std::unique_ptr<BrowserUIThreadScheduler> browser_ui_thread_scheduler, |
| std::unique_ptr<BrowserIOTaskEnvironment> browser_io_task_environment); |
| ~BrowserTaskExecutor() override; |
| |
| scoped_refptr<base::SingleThreadTaskRunner> GetTaskRunner( |
| const base::TaskTraits& traits) const; |
| |
| std::unique_ptr<BrowserUIThreadScheduler> browser_ui_thread_scheduler_; |
| BrowserUIThreadScheduler::Handle browser_ui_thread_handle_; |
| |
| std::unique_ptr<BrowserIOTaskEnvironment> browser_io_task_environment_; |
| BrowserIOTaskEnvironment::Handle browser_io_thread_handle_; |
| |
| DISALLOW_COPY_AND_ASSIGN(BrowserTaskExecutor); |
| }; |
| |
| } // namespace content |
| |
| #endif // CONTENT_BROWSER_SCHEDULER_BROWSER_TASK_EXECUTOR_H_ |