| // Copyright 2014 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. |
| |
| #include "content/child/service_worker/service_worker_provider_context.h" |
| |
| #include "base/bind.h" |
| #include "base/message_loop/message_loop_proxy.h" |
| #include "base/stl_util.h" |
| #include "content/child/child_thread.h" |
| #include "content/child/service_worker/service_worker_dispatcher.h" |
| #include "content/child/service_worker/service_worker_handle_reference.h" |
| #include "content/child/thread_safe_sender.h" |
| #include "content/child/worker_task_runner.h" |
| #include "content/common/service_worker/service_worker_messages.h" |
| |
| namespace content { |
| |
| ServiceWorkerProviderContext::ServiceWorkerProviderContext(int provider_id) |
| : provider_id_(provider_id), |
| main_thread_loop_proxy_(base::MessageLoopProxy::current()) { |
| if (!ChildThread::current()) |
| return; // May be null in some tests. |
| thread_safe_sender_ = ChildThread::current()->thread_safe_sender(); |
| ServiceWorkerDispatcher* dispatcher = |
| ServiceWorkerDispatcher::GetOrCreateThreadSpecificInstance( |
| thread_safe_sender_); |
| DCHECK(dispatcher); |
| dispatcher->AddProviderContext(this); |
| } |
| |
| ServiceWorkerProviderContext::~ServiceWorkerProviderContext() { |
| if (ServiceWorkerDispatcher* dispatcher = |
| ServiceWorkerDispatcher::GetThreadSpecificInstance()) { |
| dispatcher->RemoveProviderContext(this); |
| } |
| } |
| |
| ServiceWorkerHandleReference* ServiceWorkerProviderContext::waiting() { |
| DCHECK(main_thread_loop_proxy_->RunsTasksOnCurrentThread()); |
| return waiting_.get(); |
| } |
| |
| ServiceWorkerHandleReference* ServiceWorkerProviderContext::current() { |
| DCHECK(main_thread_loop_proxy_->RunsTasksOnCurrentThread()); |
| return current_.get(); |
| } |
| |
| void ServiceWorkerProviderContext::OnServiceWorkerStateChanged( |
| int handle_id, |
| blink::WebServiceWorkerState state) { |
| ServiceWorkerHandleReference* which = NULL; |
| if (handle_id == current_handle_id()) { |
| which = current_.get(); |
| } else if (handle_id == waiting_handle_id()) { |
| which = waiting_.get(); |
| } |
| |
| // We should only get messages for ServiceWorkers associated with |
| // this provider. |
| DCHECK(which); |
| |
| which->set_state(state); |
| |
| // TODO(kinuko): We can forward the message to other threads here |
| // when we support navigator.serviceWorker in dedicated workers. |
| } |
| |
| void ServiceWorkerProviderContext::OnSetWaitingServiceWorker( |
| int provider_id, |
| const ServiceWorkerObjectInfo& info) { |
| DCHECK_EQ(provider_id_, provider_id); |
| waiting_ = ServiceWorkerHandleReference::Adopt(info, thread_safe_sender_); |
| } |
| |
| void ServiceWorkerProviderContext::OnSetCurrentServiceWorker( |
| int provider_id, |
| const ServiceWorkerObjectInfo& info) { |
| DCHECK_EQ(provider_id_, provider_id); |
| |
| // This context is is the primary owner of this handle, keeps the |
| // initial reference until it goes away. |
| current_ = ServiceWorkerHandleReference::Adopt(info, thread_safe_sender_); |
| |
| // TODO(kinuko): We can forward the message to other threads here |
| // when we support navigator.serviceWorker in dedicated workers. |
| } |
| |
| int ServiceWorkerProviderContext::current_handle_id() const { |
| DCHECK(main_thread_loop_proxy_->RunsTasksOnCurrentThread()); |
| return current_ ? current_->info().handle_id : kInvalidServiceWorkerHandleId; |
| } |
| |
| int ServiceWorkerProviderContext::waiting_handle_id() const { |
| DCHECK(main_thread_loop_proxy_->RunsTasksOnCurrentThread()); |
| return waiting_ ? waiting_->info().handle_id : kInvalidServiceWorkerHandleId; |
| } |
| |
| } // namespace content |