[Mojo-Loading] Implement URLLoader::SetPriority

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 69 matching lines @@
            blink::WebReferrerPolicyNoReferrerWhenDowngrade) &&
       request.referrer.SchemeIsCryptographic() &&
       !request.url.SchemeIsCryptographic()) {
     LOG(FATAL) << "Trying to send secure referrer for insecure request "
                << "without an appropriate referrer policy.\n"
                << "URL = " << request.url << "\n"
                << "Referrer = " << request.referrer;
   }
 }
 
+mojom::RequestPriority MojoRequestPriorityForNetRequestPriority(
+    net::RequestPriority priority) {
+  switch (priority) {
+    case net::THROTTLED:
+      return mojom::RequestPriority::kThrottled;
+    case net::IDLE:
+      return mojom::RequestPriority::kIdle;
+    case net::LOWEST:
+      return mojom::RequestPriority::kLowest;
+    case net::LOW:
+      return mojom::RequestPriority::kLow;
+    case net::MEDIUM:
+      return mojom::RequestPriority::kMedium;
+    case net::HIGHEST:
+      return mojom::RequestPriority::kHighest;
+  }
+
+  NOTREACHED();
+  return static_cast<mojom::RequestPriority>(priority);
+}
+
 }  // namespace
 
 ResourceDispatcher::ResourceDispatcher(
     IPC::Sender* sender,
     scoped_refptr<base::SingleThreadTaskRunner> main_thread_task_runner)
     : message_sender_(sender),
       delegate_(NULL),
       io_timestamp_(base::TimeTicks()),
       main_thread_task_runner_(main_thread_task_runner),
       weak_factory_(this) {
@@ skipping 397 matching lines @@
 
     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));
+  PendingRequestInfo* request_info = GetPendingRequestInfo(request_id);
+  DCHECK(request_info);
+  if (request_info->url_loader) {
+    request_info->url_loader->SetPriority(
+        MojoRequestPriorityForNetRequestPriority(new_priority),
+        intra_priority_value);
+  } else {
+    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
@@ skipping 293 matching lines @@
     delete message;
   }
 }
 
 void ResourceDispatcher::SetResourceSchedulingFilter(
     scoped_refptr<ResourceSchedulingFilter> resource_scheduling_filter) {
   resource_scheduling_filter_ = resource_scheduling_filter;
 }
 
 }  // namespace content