15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/test/sequenced_worker_pool_owner.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/location.h" 8ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch#include "base/message_loop/message_loop.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace base { 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)SequencedWorkerPoolOwner::SequencedWorkerPoolOwner( 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t max_threads, 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& thread_name_prefix) 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : constructor_message_loop_(MessageLoop::current()), 16c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) pool_(new SequencedWorkerPool(max_threads, thread_name_prefix, this)), 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) has_work_call_count_(0) {} 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)SequencedWorkerPoolOwner::~SequencedWorkerPoolOwner() { 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pool_ = NULL; 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MessageLoop::current()->Run(); 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const scoped_refptr<SequencedWorkerPool>& SequencedWorkerPoolOwner::pool() { 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return pool_; 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void SequencedWorkerPoolOwner::SetWillWaitForShutdownCallback( 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const Closure& callback) { 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) will_wait_for_shutdown_callback_ = callback; 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int SequencedWorkerPoolOwner::has_work_call_count() const { 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AutoLock lock(has_work_lock_); 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return has_work_call_count_; 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void SequencedWorkerPoolOwner::OnHasWork() { 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AutoLock lock(has_work_lock_); 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ++has_work_call_count_; 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void SequencedWorkerPoolOwner::WillWaitForShutdown() { 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!will_wait_for_shutdown_callback_.is_null()) { 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) will_wait_for_shutdown_callback_.Run(); 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void SequencedWorkerPoolOwner::OnDestruct() { 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) constructor_message_loop_->PostTask( 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FROM_HERE, 522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) constructor_message_loop_->QuitWhenIdleClosure()); 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace base 56