| Avi Drissman | e4622aa | 2022-09-08 20:36:06 | [diff] [blame] | 1 | // Copyright 2019 The Chromium Authors |
| Alex Clarke | 636e705 | 2019-05-30 10:49:37 | [diff] [blame] | 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #include "base/task/single_thread_task_executor.h" |
| 6 | |
| Carlos Caballero | f5b6f91c | 2019-12-20 14:07:38 | [diff] [blame] | 7 | #include "base/message_loop/message_pump.h" |
| 8 | #include "base/message_loop/message_pump_type.h" |
| Alex Clarke | 636e705 | 2019-05-30 10:49:37 | [diff] [blame] | 9 | #include "base/task/sequence_manager/sequence_manager.h" |
| Alex Clarke | f7fb8a8 | 2019-06-06 15:41:53 | [diff] [blame] | 10 | #include "base/task/sequence_manager/sequence_manager_impl.h" |
| 11 | #include "build/build_config.h" |
| Alex Clarke | 636e705 | 2019-05-30 10:49:37 | [diff] [blame] | 12 | |
| 13 | namespace base { |
| 14 | |
| Patrick Monette | a2e2f9c | 2025-08-06 14:25:01 | [diff] [blame] | 15 | SingleThreadTaskExecutor::SingleThreadTaskExecutor(MessagePumpType type, |
| 16 | bool is_main_thread) |
| 17 | : SingleThreadTaskExecutor(type, |
| 18 | MessagePump::Create(type), |
| 19 | is_main_thread) { |
| Carlos Caballero | f5b6f91c | 2019-12-20 14:07:38 | [diff] [blame] | 20 | DCHECK_NE(type, MessagePumpType::CUSTOM); |
| 21 | } |
| 22 | |
| 23 | SingleThreadTaskExecutor::SingleThreadTaskExecutor( |
| Patrick Monette | a2e2f9c | 2025-08-06 14:25:01 | [diff] [blame] | 24 | std::unique_ptr<MessagePump> pump, |
| 25 | bool is_main_thread) |
| 26 | : SingleThreadTaskExecutor(MessagePumpType::CUSTOM, |
| 27 | std::move(pump), |
| 28 | is_main_thread) {} |
| Carlos Caballero | f5b6f91c | 2019-12-20 14:07:38 | [diff] [blame] | 29 | |
| 30 | SingleThreadTaskExecutor::SingleThreadTaskExecutor( |
| 31 | MessagePumpType type, |
| Patrick Monette | a2e2f9c | 2025-08-06 14:25:01 | [diff] [blame] | 32 | std::unique_ptr<MessagePump> pump, |
| 33 | bool is_main_thread) |
| Alex Clarke | f7fb8a8 | 2019-06-06 15:41:53 | [diff] [blame] | 34 | : sequence_manager_(sequence_manager::CreateUnboundSequenceManager( |
| 35 | sequence_manager::SequenceManager::Settings::Builder() |
| 36 | .SetMessagePumpType(type) |
| Patrick Monette | a2e2f9c | 2025-08-06 14:25:01 | [diff] [blame] | 37 | .SetIsMainThread(is_main_thread) |
| Alex Clarke | f7fb8a8 | 2019-06-06 15:41:53 | [diff] [blame] | 38 | .Build())), |
| Aman Verma | 566a5d2 | 2022-09-22 16:30:48 | [diff] [blame] | 39 | default_task_queue_( |
| 40 | sequence_manager_->CreateTaskQueue(sequence_manager::TaskQueue::Spec( |
| 41 | sequence_manager::QueueName::DEFAULT_TQ))), |
| Michael Thiessen | 7f5dfb9 | 2023-03-30 16:50:21 | [diff] [blame] | 42 | type_(type) { |
| Aiguo Ma | 88b7086 | 2022-11-02 19:07:10 | [diff] [blame] | 43 | sequence_manager_->SetDefaultTaskRunner(task_runner()); |
| Carlos Caballero | f5b6f91c | 2019-12-20 14:07:38 | [diff] [blame] | 44 | sequence_manager_->BindToMessagePump(std::move(pump)); |
| Alex Clarke | 636e705 | 2019-05-30 10:49:37 | [diff] [blame] | 45 | } |
| 46 | |
| 47 | SingleThreadTaskExecutor::~SingleThreadTaskExecutor() = default; |
| 48 | |
| Gabriel Charette | 9d059b7 | 2021-05-05 18:47:55 | [diff] [blame] | 49 | const scoped_refptr<SingleThreadTaskRunner>& |
| 50 | SingleThreadTaskExecutor::task_runner() const { |
| Alex Clarke | 636e705 | 2019-05-30 10:49:37 | [diff] [blame] | 51 | return default_task_queue_->task_runner(); |
| 52 | } |
| 53 | |
| Peter Kasting | 12bf330 | 2022-06-07 19:52:07 | [diff] [blame] | 54 | void SingleThreadTaskExecutor::SetWorkBatchSize(int work_batch_size) { |
| Etienne Pierre-doray | 9eb4f5a | 2020-01-15 16:29:24 | [diff] [blame] | 55 | sequence_manager_->SetWorkBatchSize(work_batch_size); |
| 56 | } |
| 57 | |
| Alex Clarke | 636e705 | 2019-05-30 10:49:37 | [diff] [blame] | 58 | } // namespace base |
