Improve IndexedDB's quota support

content/browser/in_process_webkit/indexed_db_context.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 CONTENT_BROWSER_IN_PROCESS_WEBKIT_INDEXED_DB_CONTEXT_H_
 #define CONTENT_BROWSER_IN_PROCESS_WEBKIT_INDEXED_DB_CONTEXT_H_
 #pragma once
 
+#include <map>
+
 #include "base/basictypes.h"
 #include "base/file_path.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
 #include "content/browser/browser_thread.h"
 
 class GURL;
 class FilePath;
 class WebKitContext;
 
@@ skipping 27 matching lines @@
   // The indexed db file extension.
   static const FilePath::CharType kIndexedDBExtension[];
 
   // Get the file name of the indexed db file for the given origin.
   FilePath GetIndexedDBFilePath(const string16& origin_id) const;
 
   void set_clear_local_state_on_exit(bool clear_local_state) {
     clear_local_state_on_exit_ = clear_local_state;
   }
 
-  // Deletes a single indexed db file.
-  void DeleteIndexedDBFile(const FilePath& file_path);
-
   // Deletes all indexed db files for the given origin.
-  void DeleteIndexedDBForOrigin(const string16& origin_id);
+  void DeleteIndexedDBForOrigin(const GURL& origin_url);
 
   // Does a particular origin get unlimited storage?
   bool IsUnlimitedStorageGranted(const GURL& origin) const;
 
+  // Methods used in response to QuotaManager requests.
   void GetAllOriginIdentifiers(std::vector<string16>* origin_ids);
+  int64 GetOriginDiskUsage(const GURL& origin_url);
+
+  // Methods called by IndexedDBDispatcherHost for quota support.
+  void ConnectionOpened(const GURL& origin_url);
+  void ConnectionClosed(const GURL& origin_url);
+  void TransactionComplete(const GURL& origin_url);
+  bool WouldBeOverQuota(const GURL& origin_url, int64 additional_bytes);
+  bool IsOverQuota(const GURL& origin_url);
 
   quota::QuotaManagerProxy* quota_manager_proxy();
 
 #ifdef UNIT_TEST
   // For unit tests allow to override the |data_path_|.
   void set_data_path(const FilePath& data_path) { data_path_ = data_path; }
 #endif
 
  private:
+  typedef std::map<GURL, int64> OriginToSizeMap;
+  class IndexedDBGetUsageAndQuotaCallback;
+
+  int64 ReadUsageFromDisk(const GURL& origin_url) const;
+  void EnsureDiskUsageCacheInitialized(const GURL& origin_url);
+  void QueryDiskAndUpdateQuotaUsage(const GURL& origin_url);
+  void GotUpdatedQuota(const GURL& origin_url, int64 usage, int64 quota);
+  void QueryAvailableQuota(const GURL& origin_url);
+  int64 ResetDiskUsageCache(const GURL& origin_url);
+
   scoped_ptr<WebKit::WebIDBFactory> idb_factory_;
 
   // Path where the indexed db data is stored
   FilePath data_path_;
 
   // True if the destructor should delete its files.
   bool clear_local_state_on_exit_;
 
   scoped_refptr<quota::SpecialStoragePolicy> special_storage_policy_;
 
   scoped_refptr<quota::QuotaManagerProxy> quota_manager_proxy_;
 
+  OriginToSizeMap origin_size_map_;
+  OriginToSizeMap space_available_map_;
+  std::map<GURL, unsigned int> connection_count_;
+
   DISALLOW_COPY_AND_ASSIGN(IndexedDBContext);
 };
 
 #endif  // CONTENT_BROWSER_IN_PROCESS_WEBKIT_INDEXED_DB_CONTEXT_H_