Convert QuotaManager::GetLRUOrigin into a QuotaEvictionPolicy.

storage/browser/quota/quota_manager.cc

 // Copyright 2013 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 "storage/browser/quota/quota_manager.h"
 
 #include <algorithm>
 #include <functional>
 
 #include "base/bind.h"
@@ skipping 774 matching lines @@
 
  private:
   bool AppendEntry(const OriginInfoTableEntry& entry) {
     entries_.push_back(entry);
     return true;
   }
 
   OriginInfoTableEntries entries_;
 };
 
+class QuotaManager::LRUOriginEvictionPolicy : public QuotaEvictionPolicy {
+ public:
+  LRUOriginEvictionPolicy(QuotaManager* manager, StorageType type)
+      : manager_(manager), type_(type), weak_factory_(this) {}
+
+  ~LRUOriginEvictionPolicy() override {}
+
+  // Overridden from storage::QuotaEvictionPolicy:
+  void GetEvictionOrigin(const scoped_refptr<storage::SpecialStoragePolicy>&
+                             special_storage_policy,
+                         const std::set<GURL>& exceptions,
+                         const storage::GetOriginCallback& callback) override {
+    if (manager_->db_disabled_) {
+      callback.Run(GURL());
+      return;
+    }
+
+    GURL* url = new GURL;
+    manager_->PostTaskAndReplyWithResultForDBThread(
+        FROM_HERE,
+        base::Bind(&GetLRUOriginOnDBThread, type_, exceptions,
+                   manager_->special_storage_policy_, base::Unretained(url)),
+        base::Bind(&LRUOriginEvictionPolicy::DidGetLRUOrigin,
+                   weak_factory_.GetWeakPtr(), callback, base::Owned(url)));
+  }
+
+  void DidGetLRUOrigin(const storage::GetOriginCallback& callback,
+                       const GURL* origin,
+                       bool success) {
+    manager_->DidDatabaseWork(success);
+    callback.Run(*origin);
+  }
+
+ private:
+  QuotaManager* manager_;
+  StorageType type_;
+  base::WeakPtrFactory<LRUOriginEvictionPolicy> weak_factory_;
+
+  DISALLOW_COPY_AND_ASSIGN(LRUOriginEvictionPolicy);
+};
+
 // QuotaManager ---------------------------------------------------------------
 
 QuotaManager::QuotaManager(
     bool is_incognito,
     const base::FilePath& profile_path,
     const scoped_refptr<base::SingleThreadTaskRunner>& io_thread,
     const scoped_refptr<base::SequencedTaskRunner>& db_thread,
     const scoped_refptr<SpecialStoragePolicy>& special_storage_policy)
     : is_incognito_(is_incognito),
       profile_path_(profile_path),
       proxy_(new QuotaManagerProxy(this, io_thread)),
       db_disabled_(false),
       eviction_disabled_(false),
       io_thread_(io_thread),
       db_thread_(db_thread),
       temporary_quota_initialized_(false),
       temporary_quota_override_(-1),
       desired_available_space_(-1),
       special_storage_policy_(special_storage_policy),
       get_disk_space_fn_(&CallSystemGetAmountOfFreeDiskSpace),
       storage_monitor_(new StorageMonitor(this)),
       is_getting_eviction_origin_(false),
-      weak_factory_(this) {}
+      weak_factory_(this) {
+  eviction_policy_map_.resize(kStorageTypeLast + 1);
+}
 
 void QuotaManager::GetUsageInfo(const GetUsageInfoCallback& callback) {
   LazyInitialize();
   GetUsageInfoTask* get_usage_info = new GetUsageInfoTask(this, callback);
   get_usage_info->Start();
 }
 
 void QuotaManager::GetUsageAndQuotaForWebApps(
     const GURL& origin,
     StorageType type,
@@ skipping 93 matching lines @@
                                         StorageType type,
                                         bool enabled) {
   LazyInitialize();
   DCHECK(GetUsageTracker(type));
   GetUsageTracker(type)->SetUsageCacheEnabled(client_id, origin, enabled);
 }
 
 void QuotaManager::SetQuotaEvictionPolicy(
     StorageType type,
     scoped_ptr<QuotaEvictionPolicy> policy) {
-  eviction_policy_map_.set(type, policy.Pass());
+  eviction_policy_map_[type] = policy.release();
 }
 
 void QuotaManager::DeleteOriginData(
     const GURL& origin, StorageType type, int quota_client_mask,
     const StatusCallback& callback) {
   LazyInitialize();
 
   if (origin.is_empty() || clients_.empty()) {
     callback.Run(kQuotaStatusOk);
     return;
@@ skipping 345 matching lines @@
   DCHECK(GetUsageTracker(type));
   GetUsageTracker(type)->GetCachedOrigins(origins);
 }
 
 void QuotaManager::NotifyStorageAccessedInternal(
     QuotaClient::ID client_id,
     const GURL& origin, StorageType type,
     base::Time accessed_time) {
   LazyInitialize();
   if (type == kStorageTypeTemporary && is_getting_eviction_origin_) {
-    // Record the accessed origins while GetLRUOrigin task is runing
-    // to filter out them from eviction.
+    // Record the accessed origins while an eviction origin task is running
+    // to filter them from eviction.
     access_notified_origins_.insert(origin);
   }
 
   if (db_disabled_)
     return;
   PostTaskAndReplyWithResultForDBThread(
       FROM_HERE,
       base::Bind(&UpdateAccessTimeOnDBThread, origin, type, accessed_time),
       base::Bind(&QuotaManager::DidDatabaseWork,
                  weak_factory_.GetWeakPtr()));
@@ skipping 167 matching lines @@
   is_getting_eviction_origin_ = false;
 }
 
 void QuotaManager::GetEvictionOrigin(StorageType type,
                                      const GetOriginCallback& callback) {
   LazyInitialize();
   // This must not be called while there's an in-flight task.
   DCHECK(!is_getting_eviction_origin_);
   is_getting_eviction_origin_ = true;
 
-  GetOriginCallback did_get_origin_callback =
-      base::Bind(&QuotaManager::DidGetEvictionOrigin,
-                 weak_factory_.GetWeakPtr(), callback);
+  QuotaEvictionPolicy* policy = eviction_policy_map_[type];
 
-  auto eviction_policy = eviction_policy_map_.find(type);
-  if (eviction_policy != eviction_policy_map_.end()) {
-    eviction_policy->second->GetEvictionOrigin(special_storage_policy_,
-                                               GetEvictionOriginExceptions(),
-                                               did_get_origin_callback);
-    return;
+  if (!policy) {
+    policy = new LRUOriginEvictionPolicy(this, type);
+    eviction_policy_map_[type] = policy;
   }
 
-  // TODO(calamity): convert LRU origin retrieval into a QuotaEvictionPolicy.
-  GetLRUOrigin(type, did_get_origin_callback);
+  policy->GetEvictionOrigin(
+      special_storage_policy_, GetEvictionOriginExceptions(),
+      base::Bind(&QuotaManager::DidGetEvictionOrigin,
+                 weak_factory_.GetWeakPtr(), callback));
 }
 
 void QuotaManager::EvictOriginData(const GURL& origin,
                                    StorageType type,
                                    const EvictOriginDataCallback& callback) {
   DCHECK(io_thread_->BelongsToCurrentThread());
   DCHECK_EQ(type, kStorageTypeTemporary);
 
   eviction_context_.evicted_origin = origin;
   eviction_context_.evicted_type = type;
@@ skipping 11 matching lines @@
 
   UsageAndQuotaCallbackDispatcher* dispatcher =
       new UsageAndQuotaCallbackDispatcher(this);
   GetUsageTracker(kStorageTypeTemporary)
       ->GetGlobalLimitedUsage(dispatcher->GetGlobalLimitedUsageCallback());
   GetTemporaryGlobalQuota(dispatcher->GetQuotaCallback());
   GetAvailableSpace(dispatcher->GetAvailableSpaceCallback());
   dispatcher->WaitForResults(callback);
 }
 
-void QuotaManager::GetLRUOrigin(StorageType type,
-                                const GetOriginCallback& callback) {
-  LazyInitialize();
-  // This must not be called while there's an in-flight task.
-  DCHECK(lru_origin_callback_.is_null());
-  lru_origin_callback_ = callback;
-  if (db_disabled_) {
-    lru_origin_callback_.Run(GURL());
-    lru_origin_callback_.Reset();
-    return;
-  }
-
-  GURL* url = new GURL;
-  PostTaskAndReplyWithResultForDBThread(
-      FROM_HERE,
-      base::Bind(&GetLRUOriginOnDBThread, type, GetEvictionOriginExceptions(),
-                 special_storage_policy_, base::Unretained(url)),
-      base::Bind(&QuotaManager::DidGetLRUOrigin, weak_factory_.GetWeakPtr(),
-                 base::Owned(url)));
-}
-
 void QuotaManager::DidSetTemporaryGlobalOverrideQuota(
     const QuotaCallback& callback,
     const int64* new_quota,
     bool success) {
   QuotaStatusCode status = kQuotaErrorInvalidAccess;
   DidDatabaseWork(success);
   if (success) {
     temporary_quota_override_ = *new_quota;
     status = kQuotaStatusOk;
   }
@@ skipping 31 matching lines @@
                          base::TimeDelta::FromMilliseconds(
                              kReportHistogramInterval),
                          this, &QuotaManager::ReportHistogram);
 
   db_initialization_callbacks_.Run();
   GetTemporaryGlobalQuota(
       base::Bind(&QuotaManager::DidGetInitialTemporaryGlobalQuota,
                  weak_factory_.GetWeakPtr()));
 }
 
-void QuotaManager::DidGetLRUOrigin(const GURL* origin,
-                                   bool success) {
-  DidDatabaseWork(success);
-
-  lru_origin_callback_.Run(*origin);
-  lru_origin_callback_.Reset();
-}
-
 void QuotaManager::DidGetInitialTemporaryGlobalQuota(
     QuotaStatusCode status, int64 quota_unused) {
   if (eviction_disabled_)
     return;
 
   std::set<GURL>* origins = new std::set<GURL>;
   temporary_usage_tracker_->GetCachedOrigins(origins);
   // This will call the StartEviction() when initial origin registration
   // is completed.
   PostTaskAndReplyWithResultForDBThread(
@@ skipping 34 matching lines @@
   // |database_|, therefore we can be sure that database_ is alive when this
   // task runs.
   base::PostTaskAndReplyWithResult(
       db_thread_.get(),
       from_here,
       base::Bind(task, base::Unretained(database_.get())),
       reply);
 }
 
 }  // namespace storage