Index: content/renderer/service_worker/embedded_worker_instance_client_impl.cc |
diff --git a/content/renderer/service_worker/embedded_worker_instance_client_impl.cc b/content/renderer/service_worker/embedded_worker_instance_client_impl.cc |
index 38014c8cc62640fcf7258ef9d588d784c2c0b1c3..4e6da670dc294a021d40ea90a76c46dc9e8cca1c 100644 |
--- a/content/renderer/service_worker/embedded_worker_instance_client_impl.cc |
+++ b/content/renderer/service_worker/embedded_worker_instance_client_impl.cc |
@@ -27,13 +27,16 @@ void EmbeddedWorkerInstanceClientImpl::Create( |
void EmbeddedWorkerInstanceClientImpl::StopWorkerCompleted() { |
DCHECK(embedded_worker_id_); |
- DCHECK(stop_callback_); |
+ DCHECK(!stop_callbacks_.empty()); |
dispatcher_->UnregisterWorker(embedded_worker_id_.value()); |
embedded_worker_id_.reset(); |
- stop_callback_.Run(); |
+ std::vector<StopWorkerCallback> callbacks; |
+ stop_callbacks_.swap(callbacks); |
+ for (auto& callback : callbacks) |
+ callback.Run(); |
+ DCHECK(stop_callbacks_.empty()); |
TRACE_EVENT0("ServiceWorker", |
"EmbeddedWorkerInstanceClientImpl::StopWorkerCompleted"); |
- stop_callback_.Reset(); |
wrapper_ = nullptr; |
} |
@@ -42,6 +45,7 @@ void EmbeddedWorkerInstanceClientImpl::StartWorker( |
mojom::ServiceWorkerEventDispatcherRequest dispatcher_request) { |
DCHECK(ChildThreadImpl::current()); |
DCHECK(!wrapper_); |
+ DCHECK(!embedded_worker_id_); |
TRACE_EVENT0("ServiceWorker", |
"EmbeddedWorkerInstanceClientImpl::StartWorker"); |
embedded_worker_id_ = params.embedded_worker_id; |
@@ -61,12 +65,19 @@ void EmbeddedWorkerInstanceClientImpl::StartWorker( |
void EmbeddedWorkerInstanceClientImpl::StopWorker( |
const StopWorkerCallback& callback) { |
DCHECK(ChildThreadImpl::current()); |
+ // StopWorker is possible to be called before StartWorker(). |
+ if (!wrapper_) { |
+ callback.Run(); |
+ return; |
+ } |
DCHECK(embedded_worker_id_); |
nhiroki
2016/12/19 09:21:55
nit: For readability, it'd be better to place this
shimazu
2016/12/19 09:47:51
Done.
|
- // StopWorker is possible to be called twice or before StartWorker(). |
- if (stop_callback_ || !wrapper_) |
+ // StopWorker is possible to be called twice. |
+ if (!stop_callbacks_.empty()) { |
+ stop_callbacks_.push_back(std::move(callback)); |
return; |
+ } |
TRACE_EVENT0("ServiceWorker", "EmbeddedWorkerInstanceClientImpl::StopWorker"); |
- stop_callback_ = std::move(callback); |
+ stop_callbacks_.push_back(std::move(callback)); |
dispatcher_->RecordStopWorkerTimer(embedded_worker_id_.value()); |
wrapper_->worker()->terminateWorkerContext(); |
} |