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(int32_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();
   }
 
   void OnComplete(const ResourceRequestCompletionStatus& status) override {
@@ skipping 415 matching lines @@
     }
   }
   // Cancel the request if it didn't complete, and clean it up so the bridge
   // will receive no more messages.
   if (info.completion_time.is_null())
     message_sender_->Send(new ResourceHostMsg_CancelRequest(request_id));
   RemovePendingRequest(request_id);
 }
 
 void ResourceDispatcher::SetDefersLoading(int request_id, bool value) {
-  PendingRequestMap::iterator it = pending_requests_.find(request_id);
-  if (it == pending_requests_.end()) {
+  PendingRequestInfo* request_info = GetPendingRequestInfo(request_id);
+  if (!request_info) {
     DLOG(ERROR) << "unknown request";
     return;
   }
-  PendingRequestInfo* request_info = it->second.get();
   if (value) {
     request_info->is_deferred = value;
   } else if (request_info->is_deferred) {
     request_info->is_deferred = false;
 
     FollowPendingRedirect(request_id, request_info);
 
     main_thread_task_runner_->PostTask(
         FROM_HERE, base::Bind(&ResourceDispatcher::FlushDeferredMessages,
                               weak_factory_.GetWeakPtr(), request_id));
   }
 }
 
 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,
+                                               int32_t transfer_size_diff) {
+  DCHECK_GT(transfer_size_diff, 0);
+  PendingRequestInfo* request_info = GetPendingRequestInfo(request_id);
+  if (!request_info)
+    return;
+
+  // TODO(yhirano): Consider using int64_t in
+  // RequestPeer::OnTransferSizeUpdated.
+  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