Show data usage on Site Details (MDSettings)

chrome/browser/storage/storage_info_fetcher.cc

+// Copyright 2016 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.
+
+#include "chrome/browser/storage/storage_info_fetcher.h"
+
+#include "content/public/browser/browser_thread.h"
+
+using content::BrowserThread;
+
+StorageInfoFetcher::StorageInfoFetcher(storage::QuotaManager* quota_manager)
+    : quota_manager_(quota_manager) {
+}
+
+StorageInfoFetcher::~StorageInfoFetcher() {
+}
+
+void StorageInfoFetcher::Run() {
+  // QuotaManager must be called on IO thread, but the callback must then be
+  // called on the UI thread.
+  BrowserThread::PostTask(
+      BrowserThread::IO, FROM_HERE,
+      base::Bind(&StorageInfoFetcher::GetUsageInfo, this,
+          base::Bind(&StorageInfoFetcher::OnGetUsageInfoInternal, this)));
+}
+
+void StorageInfoFetcher::AddObserver(Observer* observer) {
+  observers_.AddObserver(observer);
+}
+
+void StorageInfoFetcher::RemoveObserver(Observer* observer) {
+  observers_.RemoveObserver(observer);
+}
+
+void StorageInfoFetcher::GetUsageInfo(
+    const storage::GetUsageInfoCallback& callback) {
+  DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
+  quota_manager_->GetUsageInfo(callback);
+}
+
+void StorageInfoFetcher::OnGetUsageInfoInternal(
+    const storage::UsageInfoEntries& entries) {
+  DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
+
+  entries_.insert(entries_.begin(), entries.begin(), entries.end());
+  BrowserThread::PostTask(
+      BrowserThread::UI, FROM_HERE,
+      base::Bind(&StorageInfoFetcher::InvokeCallback, this));
+}
+
+void StorageInfoFetcher::InvokeCallback() {
+  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
+
+  FOR_EACH_OBSERVER(Observer, observers_, OnGetUsageInfo(entries_));
+}