Chromium Code Reviews| 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(); |
| } |