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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
// Copyright 2022 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/segmentation_platform/internal/signals/url_signal_handler.h"
#include "base/check.h"
#include "base/functional/callback.h"
#include "components/segmentation_platform/internal/database/ukm_database.h"
namespace segmentation_platform {
bool UrlSignalHandler::HistoryDelegate::FastCheckUrl(const GURL& url) {
return false;
}
UrlSignalHandler::UrlSignalHandler(UkmDatabase* ukm_database)
: ukm_database_(ukm_database) {}
UrlSignalHandler::~UrlSignalHandler() {
DCHECK(history_delegates_.empty());
}
void UrlSignalHandler::OnUkmSourceUpdated(ukm::SourceId source_id,
const std::vector<GURL>& urls) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
const GURL& url = urls.back();
// TODO(haileywang): add profile_id in here.
auto callback = base::BindOnce(&UrlSignalHandler::OnCheckedHistory,
weak_factory_.GetWeakPtr(), source_id, url);
CheckHistoryForUrl(url, std::move(callback));
}
void UrlSignalHandler::OnHistoryVisit(const GURL& url,
const std::string& profile_id) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
ukm_database_->OnUrlValidated(url, profile_id);
}
void UrlSignalHandler::OnUrlsRemovedFromHistory(const std::vector<GURL>& urls,
bool all_urls) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
ukm_database_->RemoveUrls(urls, all_urls);
}
void UrlSignalHandler::AddHistoryDelegate(HistoryDelegate* history_delegate) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
auto it = history_delegates_.insert(history_delegate);
DCHECK(it.second);
}
void UrlSignalHandler::RemoveHistoryDelegate(
HistoryDelegate* history_delegate) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
auto count = history_delegates_.erase(history_delegate);
DCHECK(count);
}
void UrlSignalHandler::CheckHistoryForUrl(const GURL& url,
FindCallback callback) {
for (auto& history_delegate : history_delegates_) {
if (history_delegate->FastCheckUrl(url)) {
std::move(callback).Run(true, history_delegate->profile_id());
return;
}
}
if (skip_history_db_check_) {
std::move(callback).Run(false, "");
} else {
// TODO(ssid): Add metrics on what percentage of URLs will be missed if we
// do not do this check, and wait for notifications instead.
ContinueCheckingHistory(
url, std::make_unique<base::flat_set<HistoryDelegate*>>(),
std::move(callback), false, "");
}
}
void UrlSignalHandler::ContinueCheckingHistory(
const GURL& url,
std::unique_ptr<base::flat_set<HistoryDelegate*>> delegates_checked,
FindCallback callback,
bool found,
const std::string& profile_id) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
if (found) {
std::move(callback).Run(true, profile_id);
return;
}
for (auto& next_delegate : history_delegates_) {
if (!delegates_checked->count(next_delegate)) {
delegates_checked->insert(next_delegate);
next_delegate->FindUrlInHistory(
url,
base::BindOnce(&UrlSignalHandler::ContinueCheckingHistory,
weak_factory_.GetWeakPtr(), url,
std::move(delegates_checked), std::move(callback)));
return;
}
}
std::move(callback).Run(false, "");
}
void UrlSignalHandler::OnCheckedHistory(ukm::SourceId source_id,
const GURL& url,
bool in_history,
const std::string& profile_id) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
DCHECK(!in_history || !profile_id.empty());
ukm_database_->UpdateUrlForUkmSource(source_id, url, in_history, profile_id);
}
} // namespace segmentation_platform
|
