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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
// Copyright 2023 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/enterprise/browser/reporting/real_time_report_controller.h"
#include "base/containers/fixed_flat_map.h"
#include "components/enterprise/browser/reporting/real_time_report_generator.h"
#include "components/enterprise/browser/reporting/real_time_report_type.h"
#include "components/enterprise/browser/reporting/real_time_uploader.h"
#include "components/enterprise/browser/reporting/reporting_delegate_factory.h"
#include "components/policy/core/common/cloud/dm_token.h"
namespace enterprise_reporting {
namespace {
void OnReportEnqueued(RealTimeReportType type, bool success) {
// So far, there is nothing handle the enqueue failure as the CBCM status
// report will cover all failed requests. However, we may need a retry logic
// in the future.
LOG_IF(ERROR, !success)
<< "Real time request failed to be added to the pipeline: " << type;
}
} // namespace
RealTimeReportController::RealTimeReportController(
ReportingDelegateFactory* delegate_factory)
: real_time_report_generator_(
std::make_unique<RealTimeReportGenerator>(delegate_factory)),
delegate_(delegate_factory->GetRealTimeReportControllerDelegate()) {
if (delegate_) {
delegate_->SetTriggerCallback(
base::BindRepeating(&RealTimeReportController::GenerateAndUploadReport,
weak_ptr_factory_.GetWeakPtr()));
}
}
RealTimeReportController::~RealTimeReportController() = default;
RealTimeReportController::Delegate::Delegate() = default;
RealTimeReportController::Delegate::~Delegate() = default;
void RealTimeReportController::Delegate::SetTriggerCallback(
RealTimeReportController::TriggerCallback callback) {
DCHECK(!trigger_callback_);
DCHECK(callback);
trigger_callback_ = std::move(callback);
}
void RealTimeReportController::OnDMTokenUpdated(policy::DMToken&& dm_token) {
if (!delegate_) {
return;
}
dm_token_ = dm_token;
if (dm_token_.is_valid()) {
delegate_->StartWatchingExtensionRequestIfNeeded();
} else {
delegate_->StopWatchingExtensionRequest();
report_uploaders_.clear();
}
}
void RealTimeReportController::GenerateAndUploadReport(
RealTimeReportType type,
const RealTimeReportGenerator::Data& data) {
if (!dm_token_.is_valid()) {
return;
}
static constexpr auto kConfigs =
base::MakeFixedFlatMap<RealTimeReportType, ReportConfig>({
{RealTimeReportType::kExtensionRequest,
{RealTimeReportType::kExtensionRequest,
reporting::Destination::EXTENSIONS_WORKFLOW,
reporting::Priority::FAST_BATCH}},
{RealTimeReportType::kLegacyTech,
{RealTimeReportType::kLegacyTech,
reporting::Destination::LEGACY_TECH,
reporting::Priority::BACKGROUND_BATCH}},
});
UploadReport(data, kConfigs.at(type));
}
void RealTimeReportController::UploadReport(
const RealTimeReportGenerator::Data& data,
const ReportConfig& config) {
DCHECK(real_time_report_generator_);
VLOG(1) << "Create real time event and add it to the pipeline: "
<< config.type;
if (!dm_token_.is_valid()) {
return;
}
if (!report_uploaders_.contains(config.type)) {
report_uploaders_[config.type] = RealTimeUploader::Create(
dm_token_.value(), config.destination, config.priority);
}
auto* uploader = report_uploaders_[config.type].get();
auto reports = real_time_report_generator_->Generate(config.type, data);
for (auto& report : reports) {
uploader->Upload(std::move(report),
base::BindOnce(&OnReportEnqueued, config.type));
}
}
void RealTimeReportController::SetUploaderForTesting(
RealTimeReportType type,
std::unique_ptr<RealTimeUploader> uploader) {
report_uploaders_[type] = std::move(uploader);
}
void RealTimeReportController::SetReportGeneratorForTesting(
std::unique_ptr<RealTimeReportGenerator> generator) {
real_time_report_generator_ = std::move(generator);
}
RealTimeReportController::Delegate*
RealTimeReportController::GetDelegateForTesting() {
return delegate_.get();
}
} // namespace enterprise_reporting
|
