Count exact cache size for time ranges

chrome/browser/browsing_data/cache_counter.cc

 // Copyright (c) 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 "chrome/browser/browsing_data/cache_counter.h"
 #include "chrome/browser/profiles/profile.h"
 #include "components/browsing_data/content/conditional_cache_counting_helper.h"
 #include "components/browsing_data/core/pref_names.h"
 #include "content/public/browser/browser_thread.h"
 #include "net/base/net_errors.h"
 
+namespace {
+base::TimeDelta kDefaultCacheCounterTimeout = base::TimeDelta::FromSeconds(3);
+}
+
 CacheCounter::CacheResult::CacheResult(const CacheCounter* source,
                                        int64_t cache_size,
                                        bool is_upper_limit)
     : FinishedResult(source, cache_size),
       cache_size_(cache_size),
       is_upper_limit_(is_upper_limit) {}
 
 CacheCounter::CacheResult::~CacheResult() {}
 
 CacheCounter::CacheCounter(Profile* profile)
     : profile_(profile),
       pending_(false),
+      timeout_(kDefaultCacheCounterTimeout),
       weak_ptr_factory_(this) {}
 
 CacheCounter::~CacheCounter() {
 }
 
 const char* CacheCounter::GetPrefName() const {
   return browsing_data::prefs::kDeleteCache;
 }
 
 void CacheCounter::Count() {
-  bool is_upper_limit = !GetPeriodStart().is_null();
   base::WeakPtr<browsing_data::ConditionalCacheCountingHelper> counter =
       browsing_data::ConditionalCacheCountingHelper::CreateForRange(
           content::BrowserContext::GetDefaultStoragePartition(profile_),
-          base::Time(), base::Time::Max())
+          GetPeriodStart(), base::Time::Max())
           ->CountAndDestroySelfWhenFinished(
               base::Bind(&CacheCounter::OnCacheSizeCalculated,
-                         weak_ptr_factory_.GetWeakPtr(), is_upper_limit));
+                         weak_ptr_factory_.GetWeakPtr(), false));
+  // Counting a subset of the cache can take a long time. If it doesn't succeed
+  // before |cache_counter_timeout| runs out, get the total cache size as an
+  // estimate instead. We don't need a timeout if GetPeriodStart() is null as
+  // this already retrieves the full cache size.
+  if (!GetPeriodStart().is_null()) {
+    content::BrowserThread::PostDelayedTask(
+        content::BrowserThread::UI, FROM_HERE,
+        base::Bind(&CacheCounter::FetchEstimate, weak_ptr_factory_.GetWeakPtr(),
+                   counter),
+        timeout_);
+  }
   pending_ = true;
 }
 
+void CacheCounter::FetchEstimate(
+    base::WeakPtr<browsing_data::ConditionalCacheCountingHelper> counter) {
+  if (!counter || counter->IsFinished())
+    return;
+  counter->CancelCounting();
+  // Try again for the whole cache.
+  browsing_data::ConditionalCacheCountingHelper::CreateForRange(
+      content::BrowserContext::GetDefaultStoragePartition(profile_),
+      base::Time(), base::Time::Max())
+      ->CountAndDestroySelfWhenFinished(
+          base::Bind(&CacheCounter::OnCacheSizeCalculated,
+                     weak_ptr_factory_.GetWeakPtr(), true));
+}
+
+void CacheCounter::SetTimeout(base::TimeDelta timeout) {
+  timeout_ = timeout;
+}
+
 void CacheCounter::OnCacheSizeCalculated(bool is_upper_limit,
                                          int64_t result_bytes) {
+  if (result_bytes == net::ERR_ABORTED) {
+    return;
+  }
   pending_ = false;
 
   // A value less than 0 means a net error code.
   if (result_bytes < 0)
     return;
   auto result =
       base::MakeUnique<CacheResult>(this, result_bytes, is_upper_limit);
   ReportResult(std::move(result));
 }
 
 bool CacheCounter::Pending() {
   return pending_;
 }