AppCache + Quota integration

webkit/appcache/appcache_quota_client.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_QUOTA_CLIENT_H_
+#define WEBKIT_APPCACHE_APPCACHE_QUOTA_CLIENT_H_
+
+#include <deque>
+#include <map>
+#include <string>
+
+#include "base/gtest_prod_util.h"
+#include "base/memory/ref_counted.h"
+#include "base/memory/scoped_ptr.h"
+#include "net/base/completion_callback.h"
+#include "webkit/appcache/appcache_storage.h"
+#include "webkit/quota/quota_client.h"
+#include "webkit/quota/quota_task.h"
+#include "webkit/quota/quota_types.h"
+
+namespace appcache {
+
+class AppCacheService;
+class AppCacheStorageImpl;
+class AppCacheQuotaClientTest;
+
+// A QuotaClient implementation to integrate the appcache service
+// with the quota  management system. The QuotaClient interface is
+// used on the IO thread by the quota manager. This class deletes
+// itself when both the quota manager and the appcache service have
+// been destroyed.
+class AppCacheQuotaClient : public quota::QuotaClient {
+ public:
+  virtual ~AppCacheQuotaClient();
+
+  // QuotaClient method overrides
+  virtual ID id() const OVERRIDE;
+  virtual void OnQuotaManagerDestroyed();
+  virtual void GetOriginUsage(const GURL& origin,
+                              quota::StorageType type,
+                              GetUsageCallback* callback) OVERRIDE;
+  virtual void GetOriginsForType(quota::StorageType type,
+                                 GetOriginsCallback* callback) OVERRIDE;
+  virtual void GetOriginsForHost(quota::StorageType type,
+                                 const std::string& host,
+                                 GetOriginsCallback* callback) OVERRIDE;
+  virtual void DeleteOriginData(const GURL& origin,
+                                quota::StorageType type,
+                                DeletionCallback* callback) OVERRIDE;
+
+ private:
+  friend AppCacheService;  // for NotifyAppCacheIsDestroyed
+  friend AppCacheStorageImpl;  // for NotifyAppCacheIsReady
+  friend AppCacheQuotaClientTest;
+
+  struct UsageRequest {
+    GURL origin;
+    quota::StorageType type;
+    GetUsageCallback* callback;
+  };
+  struct OriginsRequest {
+    quota::StorageType type;
+    std::string opt_host;
+    GetOriginsCallback* callback;
+  };
+  struct DeleteRequest {
+    GURL origin;
+    quota::StorageType type;
+    DeletionCallback* callback;
+  };
+  typedef std::deque<UsageRequest> UsageRequestQueue;
+  typedef std::deque<OriginsRequest> OriginsRequestQueue;
+  typedef std::deque<DeleteRequest> DeleteRequestQueue;
+
+  explicit AppCacheQuotaClient(AppCacheService* service);
+
+  void DidDeleteAppCachesForOrigin(int rv);
+  void GetOriginsHelper(quota::StorageType type,
+                        const std::string& opt_host,
+                        GetOriginsCallback* callback_ptr);
+  void ProcessPendingRequests();
+  void AbortPendingRequests();
+  void DeletePendingRequests();
+  const AppCacheStorage::UsageMap* GetUsageMap();
+
+  // For use by appcache internals during initialization and shutdown.
+  void NotifyAppCacheReady();
+  void NotifyAppCacheDestroyed();
+
+  static void AbortUsageRequest(const UsageRequest& request);
+  static void AbortOriginsRequest(const OriginsRequest& request);
+  static void AbortDeleteRequest(const DeleteRequest& request);
+
+  // Prior to appcache service being ready, we have to queue
+  // up reqeusts and defer acting on them until we're ready.
+  UsageRequestQueue pending_usage_requests_;
+  OriginsRequestQueue pending_origins_requests_;
+  DeleteRequestQueue pending_delete_requests_;
+
+  // And once it's ready, we can only handle one delete request at a time,
+  // so we queue up additional requests while one is in already in progress.
+  scoped_ptr<DeletionCallback> current_delete_request_callback_;
+  scoped_refptr<net::CancelableCompletionCallback<AppCacheQuotaClient> >
+      service_delete_callback_;
+
+  AppCacheService* service_;
+  bool appcache_is_ready_;
+  bool quota_manager_is_destroyed_;
+
+  DISALLOW_COPY_AND_ASSIGN(AppCacheQuotaClient);
+};
+
+}  // namespace appcache
+
+#endif  // WEBKIT_APPCACHE_APPCACHE_QUOTA_CLIENT_H_