1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
5 #ifndef CONTENT_BROWSER_INDEXED_DB_INDEXED_DB_CONTEXT_IMPL_H_
6 #define CONTENT_BROWSER_INDEXED_DB_INDEXED_DB_CONTEXT_IMPL_H_
12 #include "base/compiler_specific.h"
13 #include "base/files/file_path.h"
14 #include "base/gtest_prod_util.h"
15 #include "base/memory/scoped_ptr.h"
16 #include "content/browser/browser_main_loop.h"
17 #include "content/browser/indexed_db/indexed_db_factory.h"
18 #include "content/public/browser/indexed_db_context.h"
20 #include "webkit/common/quota/quota_types.h"
27 class SequencedTaskRunner;
31 class QuotaManagerProxy;
32 class SpecialStoragePolicy;
37 class IndexedDBConnection;
39 class CONTENT_EXPORT IndexedDBContextImpl
40 : NON_EXPORTED_BASE(public IndexedDBContext) {
42 // If |data_path| is empty, nothing will be saved to disk.
43 IndexedDBContextImpl(const base::FilePath& data_path,
44 quota::SpecialStoragePolicy* special_storage_policy,
45 quota::QuotaManagerProxy* quota_manager_proxy,
46 base::SequencedTaskRunner* task_runner);
48 IndexedDBFactory* GetIDBFactory();
50 // The indexed db directory.
51 static const base::FilePath::CharType kIndexedDBDirectory[];
53 // Disables the exit-time deletion of session-only data.
54 void SetForceKeepSessionState() { force_keep_session_state_ = true; }
56 // IndexedDBContext implementation:
57 virtual base::TaskRunner* TaskRunner() const OVERRIDE;
58 virtual std::vector<GURL> GetAllOrigins() OVERRIDE;
59 virtual std::vector<IndexedDBInfo> GetAllOriginsInfo() OVERRIDE;
60 virtual int64 GetOriginDiskUsage(const GURL& origin_url) OVERRIDE;
61 virtual base::Time GetOriginLastModified(const GURL& origin_url) OVERRIDE;
62 virtual void DeleteForOrigin(const GURL& origin_url) OVERRIDE;
63 virtual base::FilePath GetFilePathForTesting(
64 const std::string& origin_id) const OVERRIDE;
65 virtual void SetTaskRunnerForTesting(base::SequencedTaskRunner* task_runner)
68 // Methods called by IndexedDBDispatcherHost for quota support.
69 void ConnectionOpened(const GURL& origin_url, IndexedDBConnection* db);
70 void ConnectionClosed(const GURL& origin_url, IndexedDBConnection* db);
71 void TransactionComplete(const GURL& origin_url);
72 bool WouldBeOverQuota(const GURL& origin_url, int64 additional_bytes);
73 bool IsOverQuota(const GURL& origin_url);
75 quota::QuotaManagerProxy* quota_manager_proxy();
77 base::ListValue* GetAllOriginsDetails();
78 // ForceClose takes a value rather than a reference since it may release the
80 void ForceClose(const GURL origin_url);
81 base::FilePath GetFilePath(const GURL& origin_url) const;
82 base::FilePath data_path() const { return data_path_; }
83 bool IsInOriginSet(const GURL& origin_url) {
84 std::set<GURL>* set = GetOriginSet();
85 return set->find(origin_url) != set->end();
87 size_t GetConnectionCount(const GURL& origin_url);
89 // For unit tests allow to override the |data_path_|.
90 void set_data_path_for_testing(const base::FilePath& data_path) {
91 data_path_ = data_path;
95 virtual ~IndexedDBContextImpl();
98 FRIEND_TEST_ALL_PREFIXES(IndexedDBTest, ClearLocalState);
99 FRIEND_TEST_ALL_PREFIXES(IndexedDBTest, ClearSessionOnlyDatabases);
100 FRIEND_TEST_ALL_PREFIXES(IndexedDBTest, SetForceKeepSessionState);
101 FRIEND_TEST_ALL_PREFIXES(IndexedDBTest, ForceCloseOpenDatabasesOnDelete);
102 friend class IndexedDBQuotaClientTest;
104 typedef std::map<GURL, int64> OriginToSizeMap;
105 class IndexedDBGetUsageAndQuotaCallback;
107 base::FilePath GetIndexedDBFilePath(const std::string& origin_id) const;
108 int64 ReadUsageFromDisk(const GURL& origin_url) const;
109 void EnsureDiskUsageCacheInitialized(const GURL& origin_url);
110 void QueryDiskAndUpdateQuotaUsage(const GURL& origin_url);
111 void GotUsageAndQuota(const GURL& origin_url,
112 quota::QuotaStatusCode,
115 void GotUpdatedQuota(const GURL& origin_url, int64 usage, int64 quota);
116 void QueryAvailableQuota(const GURL& origin_url);
118 std::set<GURL>* GetOriginSet();
119 bool AddToOriginSet(const GURL& origin_url) {
120 return GetOriginSet()->insert(origin_url).second;
122 void RemoveFromOriginSet(const GURL& origin_url) {
123 GetOriginSet()->erase(origin_url);
129 scoped_refptr<IndexedDBFactory> factory_;
130 base::FilePath data_path_;
131 // If true, nothing (not even session-only data) should be deleted on exit.
132 bool force_keep_session_state_;
133 scoped_refptr<quota::SpecialStoragePolicy> special_storage_policy_;
134 scoped_refptr<quota::QuotaManagerProxy> quota_manager_proxy_;
135 base::SequencedTaskRunner* task_runner_;
136 scoped_ptr<std::set<GURL> > origin_set_;
137 OriginToSizeMap origin_size_map_;
138 OriginToSizeMap space_available_map_;
139 typedef std::set<IndexedDBConnection*> ConnectionSet;
140 std::map<GURL, ConnectionSet> connections_;
142 DISALLOW_COPY_AND_ASSIGN(IndexedDBContextImpl);
145 } // namespace content
147 #endif // CONTENT_BROWSER_INDEXED_DB_INDEXED_DB_CONTEXT_IMPL_H_