1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/browser_ui/site_settings/android/storage_info_fetcher.h"
#include "base/functional/bind.h"
#include "content/public/browser/browser_context.h"
#include "content/public/browser/browser_task_traits.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/browser/storage_partition.h"
#include "storage/browser/quota/quota_client_type.h"
#include "storage/browser/quota/quota_manager.h"
using content::BrowserContext;
using content::BrowserThread;
namespace browser_ui {
StorageInfoFetcher::StorageInfoFetcher(content::BrowserContext* context) {
quota_manager_ = context->GetDefaultStoragePartition()->GetQuotaManager();
}
StorageInfoFetcher::~StorageInfoFetcher() = default;
void StorageInfoFetcher::FetchStorageInfo(FetchCallback fetch_callback) {
// Balanced in OnFetchCompleted.
AddRef();
fetch_callback_ = std::move(fetch_callback);
// QuotaManager must be called on IO thread, but the callback must then be
// called on the UI thread.
content::GetIOThreadTaskRunner({})->PostTask(
FROM_HERE,
base::BindOnce(
&StorageInfoFetcher::GetUsageInfo, this,
base::BindOnce(&StorageInfoFetcher::OnGetUsageInfoInternal, this)));
}
void StorageInfoFetcher::ClearStorage(const std::string& host,
ClearCallback clear_callback) {
// Balanced in OnUsageCleared.
AddRef();
clear_callback_ = std::move(clear_callback);
content::GetIOThreadTaskRunner({})->PostTask(
FROM_HERE,
base::BindOnce(
&storage::QuotaManager::DeleteHostData, quota_manager_, host,
base::BindOnce(&StorageInfoFetcher::OnUsageClearedInternal, this)));
}
void StorageInfoFetcher::GetUsageInfo(storage::GetUsageInfoCallback callback) {
DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
quota_manager_->GetUsageInfo(std::move(callback));
}
void StorageInfoFetcher::OnGetUsageInfoInternal(
storage::UsageInfoEntries entries) {
DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
entries_ = std::move(entries);
content::GetUIThreadTaskRunner({})->PostTask(
FROM_HERE, base::BindOnce(&StorageInfoFetcher::OnFetchCompleted, this));
}
void StorageInfoFetcher::OnFetchCompleted() {
DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
std::move(fetch_callback_).Run(entries_);
Release();
}
void StorageInfoFetcher::OnUsageClearedInternal(
blink::mojom::QuotaStatusCode code) {
DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
quota_manager_->ResetUsageTracker();
content::GetUIThreadTaskRunner({})->PostTask(
FROM_HERE,
base::BindOnce(&StorageInfoFetcher::OnClearCompleted, this, code));
}
void StorageInfoFetcher::OnClearCompleted(blink::mojom::QuotaStatusCode code) {
DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
std::move(clear_callback_).Run(code);
Release();
}
} // namespace browser_ui
|
