[WIP] Support PerformanceResourceTiming for Service Worker Navigation Preload

third_party/WebKit/Source/modules/serviceworkers/FetchEvent.cpp

 // 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 "modules/serviceworkers/FetchEvent.h"
 
 #include "bindings/core/v8/ScriptState.h"
 #include "bindings/core/v8/ToV8ForCore.h"
 #include "bindings/core/v8/V8PrivateProperty.h"
 #include "core/dom/ExecutionContext.h"
+#include "core/timing/WorkerGlobalScopePerformance.h"
 #include "modules/fetch/BytesConsumerForDataConsumerHandle.h"
 #include "modules/fetch/Request.h"
 #include "modules/fetch/Response.h"
 #include "modules/serviceworkers/FetchRespondWithObserver.h"
 #include "modules/serviceworkers/ServiceWorkerError.h"
 #include "modules/serviceworkers/ServiceWorkerGlobalScope.h"
+#include "platform/loader/fetch/ResourceTimingInfo.h"
 #include "platform/network/NetworkUtils.h"
 #include "platform/wtf/PtrUtil.h"
 #include "platform/wtf/RefPtr.h"
 #include "public/platform/WebURLResponse.h"
 #include "public/platform/modules/serviceworker/WebServiceWorkerError.h"
 
 namespace blink {
 
 FetchEvent* FetchEvent::Create(ScriptState* script_state,
                                const AtomicString& type,
@@ skipping 80 matching lines @@
   }
 }
 
 void FetchEvent::OnNavigationPreloadResponse(
     ScriptState* script_state,
     std::unique_ptr<WebURLResponse> response,
     std::unique_ptr<WebDataConsumerHandle> data_consume_handle) {
   if (!script_state->ContextIsValid())
     return;
   DCHECK(preload_response_property_);
+  DCHECK(!preload_response_);
   ScriptState::Scope scope(script_state);
+  preload_response_ = std::move(response);
   FetchResponseData* response_data =
       data_consume_handle
           ? FetchResponseData::CreateWithBuffer(new BodyStreamBuffer(
                 script_state, new BytesConsumerForDataConsumerHandle(
                                   ExecutionContext::From(script_state),
                                   std::move(data_consume_handle))))
           : FetchResponseData::Create();
   Vector<KURL> url_list(1);
-  url_list[0] = response->Url();
+  url_list[0] = preload_response_->Url();
   response_data->SetURLList(url_list);
-  response_data->SetStatus(response->HttpStatusCode());
-  response_data->SetStatusMessage(response->HttpStatusText());
-  response_data->SetResponseTime(response->ToResourceResponse().ResponseTime());
+  response_data->SetStatus(preload_response_->HttpStatusCode());
+  response_data->SetStatusMessage(preload_response_->HttpStatusText());
+  response_data->SetResponseTime(
+      preload_response_->ToResourceResponse().ResponseTime());
   const HTTPHeaderMap& headers(
-      response->ToResourceResponse().HttpHeaderFields());
+      preload_response_->ToResourceResponse().HttpHeaderFields());
   for (const auto& header : headers) {
     response_data->HeaderList()->Append(header.key, header.value);
   }
   FetchResponseData* tainted_response =
-      NetworkUtils::IsRedirectResponseCode(response->HttpStatusCode())
+      NetworkUtils::IsRedirectResponseCode(preload_response_->HttpStatusCode())
           ? response_data->CreateOpaqueRedirectFilteredResponse()
           : response_data->CreateBasicFilteredResponse();
   preload_response_property_->Resolve(
       Response::Create(ExecutionContext::From(script_state), tainted_response));
 }
 
 void FetchEvent::OnNavigationPreloadError(
     ScriptState* script_state,
     std::unique_ptr<WebServiceWorkerError> error) {
   if (!script_state->ContextIsValid())
     return;
   DCHECK(preload_response_property_);
   preload_response_property_->Reject(
       ServiceWorkerError::Take(nullptr, *error.get()));
 }
 
+void FetchEvent::OnNavigationPreloadComplete(
+    WorkerGlobalScope* worker_global_scope,
+    double completion_time,
+    int64_t encoded_data_length,
+    int64_t encoded_body_length,
+    int64_t decoded_body_length) {
+  DCHECK(preload_response_);
+  std::unique_ptr<WebURLResponse> response = std::move(preload_response_);
+  ResourceResponse resource_response = response->ToResourceResponse();
+  resource_response.SetEncodedDataLength(encoded_data_length);
+  resource_response.SetEncodedBodyLength(encoded_body_length);
+  resource_response.SetDecodedBodyLength(decoded_body_length);
+  // According to the current spec of Resource Timing, the initiator type of
+  // navigation preload request must be "other". But it may change when the spec
+  // discussion is settled. https://github.com/w3c/resource-timing/issues/110
+  RefPtr<ResourceTimingInfo> info = ResourceTimingInfo::Create(
+      "other", resource_response.GetResourceLoadTiming()->RequestTime(),
+      false /* is_main_resource */);
+  info->SetNegativeAllowed(true);
+  info->SetLoadFinishTime(completion_time);
+  info->SetInitialURL(request_->url());
+  info->SetFinalResponse(resource_response);
+  info->AddFinalTransferSize(encoded_data_length);
+  WorkerGlobalScopePerformance::performance(*worker_global_scope)
+      ->AddResourceTiming(*info);
+}
+
 DEFINE_TRACE(FetchEvent) {
   visitor->Trace(observer_);
   visitor->Trace(request_);
   visitor->Trace(preload_response_property_);
   ExtendableEvent::Trace(visitor);
 }
 
 }  // namespace blink