Dispatch transfer size update notification on mojo-loading

content/child/resource_dispatcher.cc

 // Copyright (c) 2012 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.
 
 // See http://dev.chromium.org/developers/design-documents/multi-process-resource-loading
 
 #include "content/child/resource_dispatcher.h"
 
 #include <utility>
 
@@ skipping 96 matching lines @@
     DCHECK(!body_consumer_);
     resource_dispatcher_->OnMessageReceived(ResourceMsg_ReceivedRedirect(
         request_id_, redirect_info, response_head));
   }
 
   void OnDataDownloaded(int64_t data_len, int64_t encoded_data_len) override {
     resource_dispatcher_->OnMessageReceived(
         ResourceMsg_DataDownloaded(request_id_, data_len, encoded_data_len));
   }
 
+  void OnTransferSizeUpdated(int64_t transfer_size_diff) override {
+    resource_dispatcher_->OnTransferSizeUpdated(request_id_,
+                                                transfer_size_diff);
+  }
+
   void OnStartLoadingResponseBody(
       mojo::ScopedDataPipeConsumerHandle body) override {
     DCHECK(!body_consumer_);
     body_consumer_ = new URLResponseBodyConsumer(
         request_id_, resource_dispatcher_, std::move(body), task_runner_);
     if (has_received_response_)
       body_consumer_->Start(task_runner_.get());
   }
 
   void OnComplete(const ResourceRequestCompletionStatus& status) override {
@@ skipping 442 matching lines @@
 }
 
 void ResourceDispatcher::DidChangePriority(int request_id,
                                            net::RequestPriority new_priority,
                                            int intra_priority_value) {
   DCHECK(base::ContainsKey(pending_requests_, request_id));
   message_sender_->Send(new ResourceHostMsg_DidChangePriority(
       request_id, new_priority, intra_priority_value));
 }
 
+void ResourceDispatcher::OnTransferSizeUpdated(int request_id,
+                                               int transfer_size_diff) {

[dcheng, 2016/12/15 06:29:02]

Can we use int64_t or int uniformly? It seems like the current code mostly send
this as an int, but I'm not familiar enough with the loading machinery to say
what we want long-term here.

[yhirano, 2016/12/15 11:28:00]

I added a TODO comment.

+  DCHECK_GT(transfer_size_diff, 0);
+  PendingRequestInfo* request_info = GetPendingRequestInfo(request_id);

[dcheng, 2016/12/15 06:29:02]

Unrelated, but I wish this class used GetPendingRequestInfo() or
pending_requests_.find() more uniformly.

[yhirano, 2016/12/15 11:28:00]

There are four pending_request_.find() in this file and SetDefersLoading in the
only call that can be replaced with GetPendingRequestInfo(). Done.

+  if (!request_info)
+    return;
+
+  request_info->peer->OnTransferSizeUpdated(transfer_size_diff);
+}
+
 ResourceDispatcher::PendingRequestInfo::PendingRequestInfo(
     std::unique_ptr<RequestPeer> peer,
     ResourceType resource_type,
     int origin_pid,
     const url::Origin& frame_origin,
     const GURL& request_url,
     bool download_to_file)
     : peer(std::move(peer)),
       resource_type(resource_type),
       origin_pid(origin_pid),
@@ skipping 272 matching lines @@
     delete message;
   }
 }
 
 void ResourceDispatcher::SetResourceSchedulingFilter(
     scoped_refptr<ResourceSchedulingFilter> resource_scheduling_filter) {
   resource_scheduling_filter_ = resource_scheduling_filter;
 }
 
 }  // namespace content