| 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 "services/service_manager/public/cpp/service_context_ref.h" | 5 #include "services/service_manager/public/cpp/service_context_ref.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/memory/ptr_util.h" | 8 #include "base/memory/ptr_util.h" |
| 9 #include "base/message_loop/message_loop.h" | 9 #include "base/message_loop/message_loop.h" |
| 10 #include "base/threading/thread_checker.h" | 10 #include "base/threading/thread_checker.h" |
| 11 #include "base/threading/thread_task_runner_handle.h" | 11 #include "base/threading/thread_task_runner_handle.h" |
| 12 | 12 |
| 13 namespace service_manager { | 13 namespace service_manager { |
| 14 | 14 |
| 15 class ServiceContextRefImpl : public ServiceContextRef { | 15 class ServiceContextRefImpl : public ServiceContextRef { |
| 16 public: | 16 public: |
| 17 ServiceContextRefImpl( | 17 ServiceContextRefImpl( |
| 18 base::WeakPtr<ServiceContextRefFactory> factory, | 18 base::WeakPtr<ServiceContextRefFactory> factory, |
| 19 scoped_refptr<base::SingleThreadTaskRunner> service_task_runner) | 19 scoped_refptr<base::SingleThreadTaskRunner> service_task_runner) |
| 20 : factory_(factory), | 20 : factory_(factory), service_task_runner_(service_task_runner) { |
| 21 service_task_runner_(service_task_runner) { | |
| 22 // This object is not thread-safe but may be used exclusively on a different | 21 // This object is not thread-safe but may be used exclusively on a different |
| 23 // thread from the one which constructed it. | 22 // thread from the one which constructed it. |
| 24 thread_checker_.DetachFromThread(); | 23 thread_checker_.DetachFromThread(); |
| 25 } | 24 } |
| 26 | 25 |
| 27 ~ServiceContextRefImpl() override { | 26 ~ServiceContextRefImpl() override { |
| 28 DCHECK(thread_checker_.CalledOnValidThread()); | 27 DCHECK(thread_checker_.CalledOnValidThread()); |
| 29 | 28 |
| 30 if (service_task_runner_->BelongsToCurrentThread() && factory_) { | 29 if (service_task_runner_->BelongsToCurrentThread() && factory_) { |
| 31 factory_->Release(); | 30 factory_->Release(); |
| 32 } else { | 31 } else { |
| 33 service_task_runner_->PostTask( | 32 service_task_runner_->PostTask( |
| 34 FROM_HERE, | 33 FROM_HERE, base::Bind(&ServiceContextRefFactory::Release, factory_)); |
| 35 base::Bind(&ServiceContextRefFactory::Release, factory_)); | |
| 36 } | 34 } |
| 37 } | 35 } |
| 38 | 36 |
| 39 private: | 37 private: |
| 40 // ServiceContextRef: | 38 // ServiceContextRef: |
| 41 std::unique_ptr<ServiceContextRef> Clone() override { | 39 std::unique_ptr<ServiceContextRef> Clone() override { |
| 42 DCHECK(thread_checker_.CalledOnValidThread()); | 40 DCHECK(thread_checker_.CalledOnValidThread()); |
| 43 | 41 |
| 44 if (service_task_runner_->BelongsToCurrentThread() && factory_) { | 42 if (service_task_runner_->BelongsToCurrentThread() && factory_) { |
| 45 factory_->AddRef(); | 43 factory_->AddRef(); |
| 46 } else { | 44 } else { |
| 47 service_task_runner_->PostTask( | 45 service_task_runner_->PostTask( |
| 48 FROM_HERE, | 46 FROM_HERE, base::Bind(&ServiceContextRefFactory::AddRef, factory_)); |
| 49 base::Bind(&ServiceContextRefFactory::AddRef, factory_)); | |
| 50 } | 47 } |
| 51 | 48 |
| 52 return base::MakeUnique<ServiceContextRefImpl>(factory_, | 49 return base::MakeUnique<ServiceContextRefImpl>(factory_, |
| 53 service_task_runner_); | 50 service_task_runner_); |
| 54 } | 51 } |
| 55 | 52 |
| 56 base::WeakPtr<ServiceContextRefFactory> factory_; | 53 base::WeakPtr<ServiceContextRefFactory> factory_; |
| 57 scoped_refptr<base::SingleThreadTaskRunner> service_task_runner_; | 54 scoped_refptr<base::SingleThreadTaskRunner> service_task_runner_; |
| 58 base::ThreadChecker thread_checker_; | 55 base::ThreadChecker thread_checker_; |
| 59 | 56 |
| (...skipping 17 matching lines...) Expand all Loading... |
| 77 void ServiceContextRefFactory::AddRef() { | 74 void ServiceContextRefFactory::AddRef() { |
| 78 ++ref_count_; | 75 ++ref_count_; |
| 79 } | 76 } |
| 80 | 77 |
| 81 void ServiceContextRefFactory::Release() { | 78 void ServiceContextRefFactory::Release() { |
| 82 if (!--ref_count_) | 79 if (!--ref_count_) |
| 83 quit_closure_.Run(); | 80 quit_closure_.Run(); |
| 84 } | 81 } |
| 85 | 82 |
| 86 } // namespace service_manager | 83 } // namespace service_manager |
| OLD | NEW |