Adding URLResponse Disk Cache to mojo.

shell/application_manager/network_fetcher.cc

 // Copyright 2015 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 "shell/application_manager/network_fetcher.h"
 
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/files/file.h"
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/process/process.h"
 #include "base/stl_util.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
 #include "base/trace_event/trace_event.h"
 #include "crypto/secure_hash.h"
 #include "crypto/sha2.h"
 #include "mojo/common/common_type_converters.h"
 #include "mojo/common/data_pipe_utils.h"
 #include "mojo/services/network/public/interfaces/network_service.mojom.h"
 #include "shell/application_manager/data_pipe_peek.h"
 #include "shell/switches.h"
 
 namespace shell {
 
-NetworkFetcher::NetworkFetcher(bool disable_cache,
-                               const GURL& url,
-                               mojo::NetworkService* network_service,
-                               const FetchCallback& loader_callback)
+NetworkFetcher::NetworkFetcher(
+    bool disable_cache,
+    const GURL& url,
+    mojo::NetworkService* network_service,
+    mojo::url_response_disk_cache::URLResponseDiskCache*
+        url_response_disk_cache,
+    const FetchCallback& loader_callback)
     : Fetcher(loader_callback),
       disable_cache_(false),
       url_(url),
+      url_response_disk_cache_(url_response_disk_cache),
       weak_ptr_factory_(this) {
   StartNetworkRequest(url, network_service);
 }
 
 NetworkFetcher::~NetworkFetcher() {
 }
 
 const GURL& NetworkFetcher::GetURL() const {
   return url_;
 }
@@ skipping 95 matching lines @@
   // The directory is leaked, because it can be reused at any time if the same
   // application is downloaded. Deleting it would be racy. This is only
   // happening when --predictable-app-filenames is used.
   bool result = base::CreateDirectoryAndGetError(app_dir, nullptr);
   DCHECK(result);
   std::string unique_name = base::StringPrintf("%s.mojo", app_id.c_str());
   *new_path = app_dir.Append(unique_name);
   return base::Move(old_path, *new_path);
 }
 
-void NetworkFetcher::CopyCompleted(
+void NetworkFetcher::OnFileRetrievedFromCache(
     base::Callback<void(const base::FilePath&, bool)> callback,
-    bool success) {
+    mojo::Array<uint8_t> path_as_array,
+    mojo::Array<uint8_t> cache_dir) {
+  bool success = !path_as_array.is_null();
   if (success) {
+    path_ = base::FilePath(std::string(
+        reinterpret_cast<char*>(&path_as_array.front()), path_as_array.size()));
     if (base::CommandLine::ForCurrentProcess()->HasSwitch(
             switches::kPredictableAppFilenames)) {
       // The copy completed, now move to $TMP/$APP_ID.mojo before the dlopen.
-      success = false;
       base::FilePath new_path;
       if (RenameToAppId(url_, path_, &new_path)) {
         if (base::PathExists(new_path)) {
           path_ = new_path;
           success = true;
         }
       }
     }
   }
 
   if (success)
     RecordCacheToURLMapping(path_, url_);
 
   base::MessageLoop::current()->PostTask(FROM_HERE,
                                          base::Bind(callback, path_, success));
 }
 
 void NetworkFetcher::AsPath(
     base::TaskRunner* task_runner,
     base::Callback<void(const base::FilePath&, bool)> callback) {
-  if (!path_.empty() || !response_) {
-    base::MessageLoop::current()->PostTask(
-        FROM_HERE, base::Bind(callback, path_, base::PathExists(path_)));
-    return;
-  }
+  // This should only called once, when we have a response.
+  DCHECK(response_.get());
 
-  base::CreateTemporaryFile(&path_);
-  mojo::common::CopyToFile(
-      response_->body.Pass(), path_, task_runner,
-      base::Bind(&NetworkFetcher::CopyCompleted, weak_ptr_factory_.GetWeakPtr(),
-                 callback));
+  url_response_disk_cache_->GetFile(
+      response_.Pass(), base::Bind(&NetworkFetcher::OnFileRetrievedFromCache,
+                                   weak_ptr_factory_.GetWeakPtr(), callback));
 }
 
 std::string NetworkFetcher::MimeType() {
   return response_->mime_type;
 }
 
 bool NetworkFetcher::HasMojoMagic() {
   std::string magic;
   return BlockingPeekNBytes(response_->body.get(), &magic, strlen(kMojoMagic),
                             kPeekTimeout) &&
@@ skipping 38 matching lines @@
                << "while fetching " << response->url;
     loader_callback_.Run(nullptr);
     return;
   }
 
   response_ = response.Pass();
   loader_callback_.Run(owner.Pass());
 }
 
 }  // namespace shell