AppCache + Quota integration

webkit/appcache/appcache_storage.h

 // Copyright (c) 2011 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.
 
 #ifndef WEBKIT_APPCACHE_APPCACHE_STORAGE_H_
 #define WEBKIT_APPCACHE_APPCACHE_STORAGE_H_
 
 #include <map>
 #include <vector>
 
@@ skipping 13 matching lines @@
 class AppCacheEntry;
 class AppCacheGroup;
 class AppCacheResponseReader;
 class AppCacheResponseWriter;
 class AppCacheService;
 struct AppCacheInfoCollection;
 struct HttpResponseInfoIOBuffer;
 
 class AppCacheStorage {
  public:
+  typedef std::map<GURL, int64> UsageMap;
 
   class Delegate {
    public:
     virtual ~Delegate() {}
 
     // If retrieval fails, 'collection' will be NULL.
     virtual void OnAllInfo(AppCacheInfoCollection* collection) {}
 
     // If a load fails the 'cache' will be NULL.
     virtual void OnCacheLoaded(AppCache* cache, int64 cache_id) {}
@@ skipping 129 matching lines @@
   int64 NewCacheId() {
     return ++last_cache_id_;
   }
   int64 NewGroupId() {
     return ++last_group_id_;
   }
 
   // The working set of object instances currently in memory.
   AppCacheWorkingSet* working_set() { return &working_set_; }
 
+  // A map of origins to usage.
+  const UsageMap* usage_map() { return &usage_map_; }
+
   // Simple ptr back to the service object that owns us.
   AppCacheService* service() { return service_; }
 
  protected:
+  friend class AppCacheQuotaClientTest;
   friend class AppCacheResponseTest;
   friend class AppCacheStorageTest;
 
   // Helper to call a collection of delegates.
   #define FOR_EACH_DELEGATE(delegates, func_and_args)                \
     do {                                                             \
       for (DelegateReferenceVector::iterator it = delegates.begin(); \
            it != delegates.end(); ++it) {                            \
         if (it->get()->delegate)                                     \
           it->get()->delegate->func_and_args;                        \
@@ skipping 76 matching lines @@
     if (iter != pending_info_loads_.end())
       return iter->second;
     return new ResponseInfoLoadTask(manifest_url, response_id, this);
   }
 
   // Should only be called when creating a new response writer.
   int64 NewResponseId() {
     return ++last_response_id_;
   }
 
+  // Helpers to query and notify the QuotaManager.
+  void UpdateUsageMapAndNotify(const GURL& origin, int64 new_usage);
+  void ClearUsageMapAndNotify();
+  void NotifyStorageAccessed(const GURL& origin);
+
   // The last storage id used for different object types.
   int64 last_cache_id_;
   int64 last_group_id_;
   int64 last_response_id_;
 
+  UsageMap usage_map_;  // maps origin to usage
   AppCacheWorkingSet working_set_;
   AppCacheService* service_;
   DelegateReferenceMap delegate_references_;
   PendingResponseInfoLoads pending_info_loads_;
 
   // The set of last ids must be retrieved from storage prior to being used.
   static const int64 kUnitializedId;
 
   FRIEND_TEST_ALL_PREFIXES(AppCacheStorageTest, DelegateReferences);
+  FRIEND_TEST_ALL_PREFIXES(AppCacheStorageTest, UsageMap);
 
   DISALLOW_COPY_AND_ASSIGN(AppCacheStorage);
 };
 
 }  // namespace appcache
 
 #endif  // WEBKIT_APPCACHE_APPCACHE_STORAGE_H_