1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 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/test/sequenced_worker_pool_owner.h" 6 7 #include "base/location.h" 8 #include "base/message_loop/message_loop.h" 9 #include "base/single_thread_task_runner.h" 10 11 namespace base { 12 13 SequencedWorkerPoolOwner::SequencedWorkerPoolOwner( 14 size_t max_threads, 15 const std::string& thread_name_prefix) 16 : constructor_message_loop_(MessageLoop::current()), 17 pool_(new SequencedWorkerPool(max_threads, thread_name_prefix, this)), 18 has_work_call_count_(0) {} 19 20 SequencedWorkerPoolOwner::~SequencedWorkerPoolOwner() { 21 pool_->Shutdown(); 22 pool_ = NULL; 23 24 // Spin the current message loop until SWP destruction verified in OnDestruct. 25 exit_loop_.Run(); 26 } 27 28 const scoped_refptr<SequencedWorkerPool>& SequencedWorkerPoolOwner::pool() { 29 return pool_; 30 } 31 32 void SequencedWorkerPoolOwner::SetWillWaitForShutdownCallback( 33 const Closure& callback) { 34 will_wait_for_shutdown_callback_ = callback; 35 } 36 37 int SequencedWorkerPoolOwner::has_work_call_count() const { 38 AutoLock lock(has_work_lock_); 39 return has_work_call_count_; 40 } 41 42 void SequencedWorkerPoolOwner::OnHasWork() { 43 AutoLock lock(has_work_lock_); 44 ++has_work_call_count_; 45 } 46 47 void SequencedWorkerPoolOwner::WillWaitForShutdown() { 48 if (!will_wait_for_shutdown_callback_.is_null()) { 49 will_wait_for_shutdown_callback_.Run(); 50 51 // Release the reference to the callback to prevent retain cycles. 52 will_wait_for_shutdown_callback_ = Closure(); 53 } 54 } 55 56 void SequencedWorkerPoolOwner::OnDestruct() { 57 constructor_message_loop_->task_runner()->PostTask(FROM_HERE, 58 exit_loop_.QuitClosure()); 59 } 60 61 } // namespace base 62
