| // 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_DATABASE_DATABASE_QUOTA_CLIENT_H_ |
| #define WEBKIT_DATABASE_DATABASE_QUOTA_CLIENT_H_ |
| |
| #include <set> |
| #include <string> |
| |
| #include "base/memory/ref_counted.h" |
| #include "base/message_loop_proxy.h" |
| #include "webkit/quota/quota_client.h" |
| #include "webkit/quota/quota_task.h" |
| #include "webkit/quota/quota_types.h" |
| |
| namespace webkit_database { |
| |
| class DatabaseTracker; |
| |
| // A QuotaClient implementation to integrate WebSQLDatabases |
| // with the quota management system. This interface is used |
| // on the IO thread by the quota manager. |
| class DatabaseQuotaClient : public quota::QuotaClient, |
| public quota::QuotaTaskObserver { |
| public: |
| DatabaseQuotaClient( |
| base::MessageLoopProxy* tracker_thread, |
| DatabaseTracker* tracker); |
| virtual ~DatabaseQuotaClient(); |
| |
| // QuotaClient method overrides |
| virtual ID id() const OVERRIDE; |
| virtual void OnQuotaManagerDestroyed() OVERRIDE; |
| virtual void GetOriginUsage(const GURL& origin_url, |
| quota::StorageType type, |
| const GetUsageCallback& callback) OVERRIDE; |
| virtual void GetOriginsForType(quota::StorageType type, |
| const GetOriginsCallback& callback) OVERRIDE; |
| virtual void GetOriginsForHost(quota::StorageType type, |
| const std::string& host, |
| const GetOriginsCallback& callback) OVERRIDE; |
| virtual void DeleteOriginData(const GURL& origin, |
| quota::StorageType type, |
| const DeletionCallback& callback) OVERRIDE; |
| private: |
| class HelperTask; |
| class GetOriginUsageTask; |
| class GetOriginsTaskBase; |
| class GetAllOriginsTask; |
| class GetOriginsForHostTask; |
| class DeleteOriginTask; |
| |
| typedef quota::CallbackQueueMap1 |
| <GetUsageCallback, |
| GURL, // origin |
| int64 |
| > UsageForOriginCallbackMap; |
| typedef quota::CallbackQueue2 |
| <GetOriginsCallback, |
| const std::set<GURL>&, |
| quota::StorageType |
| > OriginsForTypeCallbackQueue; |
| typedef quota::CallbackQueueMap2 |
| <GetOriginsCallback, |
| std::string, // host |
| const std::set<GURL>&, |
| quota::StorageType |
| > OriginsForHostCallbackMap; |
| |
| void DidGetOriginUsage(const GURL& origin_url, int64 usage); |
| void DidGetAllOrigins(const std::set<GURL>& origins, quota::StorageType type); |
| void DidGetOriginsForHost(const std::string& host, |
| const std::set<GURL>& origins, |
| quota::StorageType type); |
| |
| scoped_refptr<base::MessageLoopProxy> db_tracker_thread_; |
| scoped_refptr<DatabaseTracker> db_tracker_; // only used on its thread |
| UsageForOriginCallbackMap usage_for_origin_callbacks_; |
| OriginsForTypeCallbackQueue origins_for_type_callbacks_; |
| OriginsForHostCallbackMap origins_for_host_callbacks_; |
| |
| DISALLOW_COPY_AND_ASSIGN(DatabaseQuotaClient); |
| }; |
| |
| } // namespace webkit_database |
| |
| #endif // WEBKIT_DATABASE_DATABASE_QUOTA_CLIENT_H_ |