Count site data size for important sites

chrome/browser/engagement/important_sites_usage_counter.h

Index: chrome/browser/engagement/important_sites_usage_counter.h
diff --git a/chrome/browser/engagement/important_sites_usage_counter.h b/chrome/browser/engagement/important_sites_usage_counter.h
new file mode 100644
index 0000000000000000000000000000000000000000..03c82c9bd538ba63b5a04cadbc3a12486c2c9327
--- /dev/null
+++ b/chrome/browser/engagement/important_sites_usage_counter.h
@@ -0,0 +1,79 @@
+// Copyright 2017 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 CHROME_BROWSER_ENGAGEMENT_IMPORTANT_SITES_USAGE_COUNTER_H_
+#define CHROME_BROWSER_ENGAGEMENT_IMPORTANT_SITES_USAGE_COUNTER_H_
+
+#include <string>
+#include <vector>
+
+#include "base/callback.h"
+#include "base/sequenced_task_runner_helpers.h"
+#include "chrome/browser/engagement/important_sites_util.h"
+#include "content/public/browser/local_storage_usage_info.h"
+#include "storage/browser/quota/quota_manager.h"
+
+namespace content {
+class DOMStorageContext;
+}
+
+namespace storage {
+class QuotaManager;
+}
+
+// A helper class for important storage that retrieves the localstorage and
+// quota usage for each domain in |ImportantDomainInfo|, populates
+// |ImportantDomainInfo::usage| in the |sites| entries and return the result via
+// |callback|.
+class ImportantSitesUsageCounter {
+ public:
+  using UsageCallback = base::Callback<void(
+      std::vector<ImportantSitesUtil::ImportantDomainInfo> sites)>;
+
+  // Populates the ImportantDomainInfo::usage field of each site with the amount
+  // of localstorage and quota bytes used. |callback| is synchronously invoked

[Bernhard Bauer, 2017/05/04 14:16:44]

The callback is called asynchronously, no?

[dullweber, 2017/05/05 08:35:27]

Oh, yes. Thanks!

+  // on the UI thread with the |sites| vector (populated with usage) as an
+  // argument.
+  static void GetUsage(
+      std::vector<ImportantSitesUtil::ImportantDomainInfo> sites,
+      storage::QuotaManager* quota_manager,
+      content::DOMStorageContext* dom_storage_context,
+      UsageCallback callback);

[Bernhard Bauer, 2017/05/04 14:16:44]

Nit: Callbacks are usually passed by const reference. Alternatively, you could
make the Callback a OnceCallback, which makes it move-only.

[dullweber, 2017/05/05 08:35:28]

changed to OnceCallback.

+
+ private:
+  friend class base::DeleteHelper<ImportantSitesUsageCounter>;
+
+  ImportantSitesUsageCounter(
+      std::vector<ImportantSitesUtil::ImportantDomainInfo> sites,
+      storage::QuotaManager* quota_manager,
+      content::DOMStorageContext* dom_storage_context,
+      UsageCallback callback);
+  ~ImportantSitesUsageCounter();
+
+  void RunAndDestroySelf();
+
+  void GetQuotaUsageOnIOThread();
+
+  void ReceiveQuotaUsageOnIOThread(
+      const std::vector<storage::UsageInfo>& usage_infos);
+
+  void ReceiveQuotaUsage(const std::vector<storage::UsageInfo>& usage_infos);
+
+  void ReceiveLocalStorageUsage(
+      const std::vector<content::LocalStorageUsageInfo>& storage_infos);
+
+  // Look up the corresponding ImportantDomainInfo for |url| and increase its
+  // usage by |size|.
+  void IncrementUsage(const std::string& domain, int64_t size);
+
+  void Done();
+
+  UsageCallback callback_;
+  std::vector<ImportantSitesUtil::ImportantDomainInfo> sites_;
+  storage::QuotaManager* quota_manager_;
+  content::DOMStorageContext* dom_storage_context_;
+  int tasks_;

[Bernhard Bauer, 2017/05/04 14:16:44]

DISALLOW_IMPLICIT_CONSTRUCTORS()

[dullweber, 2017/05/05 08:35:27]

Done.

+};
+
+#endif  // CHROME_BROWSER_ENGAGEMENT_IMPORTANT_SITES_USAGE_COUNTER_H_