OLD | NEW |
| (Empty) |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #ifndef COMPONENTS_RAPPOR_RAPPOR_SERVICE_H_ | |
6 #define COMPONENTS_RAPPOR_RAPPOR_SERVICE_H_ | |
7 | |
8 #include <stdint.h> | |
9 | |
10 #include <map> | |
11 #include <memory> | |
12 #include <string> | |
13 | |
14 #include "base/callback.h" | |
15 #include "base/macros.h" | |
16 #include "base/memory/weak_ptr.h" | |
17 #include "base/threading/thread_checker.h" | |
18 #include "base/timer/timer.h" | |
19 #include "components/metrics/daily_event.h" | |
20 #include "components/rappor/rappor_parameters.h" | |
21 #include "components/rappor/sample.h" | |
22 #include "components/rappor/sampler.h" | |
23 | |
24 class PrefRegistrySimple; | |
25 class PrefService; | |
26 | |
27 namespace net { | |
28 class URLRequestContextGetter; | |
29 } | |
30 | |
31 namespace rappor { | |
32 | |
33 class LogUploaderInterface; | |
34 class RapporMetric; | |
35 class RapporReports; | |
36 | |
37 // This class provides an interface for recording samples for rappor metrics, | |
38 // and periodically generates and uploads reports based on the collected data. | |
39 class RapporService : public base::SupportsWeakPtr<RapporService> { | |
40 public: | |
41 // Constructs a RapporService. | |
42 // Calling code is responsible for ensuring that the lifetime of | |
43 // |pref_service| is longer than the lifetime of RapporService. | |
44 // |is_incognito_callback| will be called to test if incognito mode is active. | |
45 RapporService(PrefService* pref_service, | |
46 const base::Callback<bool(void)> is_incognito_callback); | |
47 virtual ~RapporService(); | |
48 | |
49 // Add an observer for collecting daily metrics. | |
50 void AddDailyObserver( | |
51 std::unique_ptr<metrics::DailyEvent::Observer> observer); | |
52 | |
53 // Initializes the rappor service, including loading the cohort and secret | |
54 // preferences from disk. | |
55 void Initialize(net::URLRequestContextGetter* context); | |
56 | |
57 // Updates the settings for metric recording and uploading. | |
58 // The RapporService must be initialized before this method is called. | |
59 // |recording_groups| should be set of flags, e.g. | |
60 // UMA_RECORDING_GROUP | SAFEBROWSING_RECORDING_GROUP | |
61 // If it contains any enabled groups, periodic reports will be | |
62 // generated and queued for upload. | |
63 // If |may_upload| is true, reports will be uploaded from the queue. | |
64 void Update(int recording_groups, bool may_upload); | |
65 | |
66 // Constructs a Sample object for the caller to record fields in. | |
67 virtual std::unique_ptr<Sample> CreateSample(RapporType); | |
68 | |
69 // Records a Sample of rappor metric specified by |metric_name|. | |
70 // | |
71 // TODO(holte): Rename RecordSample to RecordString and then rename this | |
72 // to RecordSample. | |
73 // | |
74 // example: | |
75 // std::unique_ptr<Sample> sample = | |
76 // rappor_service->CreateSample(MY_METRIC_TYPE); | |
77 // sample->SetStringField("Field1", "some string"); | |
78 // sample->SetFlagsValue("Field2", SOME|FLAGS); | |
79 // rappor_service->RecordSample("MyMetric", std::move(sample)); | |
80 // | |
81 // This will result in a report setting two metrics "MyMetric.Field1" and | |
82 // "MyMetric.Field2", and they will both be generated from the same sample, | |
83 // to allow for correlations to be computed. | |
84 virtual void RecordSampleObj(const std::string& metric_name, | |
85 std::unique_ptr<Sample> sample); | |
86 | |
87 // Records a sample of the rappor metric specified by |metric_name|. | |
88 // Creates and initializes the metric, if it doesn't yet exist. | |
89 virtual void RecordSample(const std::string& metric_name, | |
90 RapporType type, | |
91 const std::string& sample); | |
92 | |
93 // Registers the names of all of the preferences used by RapporService in the | |
94 // provided PrefRegistry. This should be called before calling Start(). | |
95 static void RegisterPrefs(PrefRegistrySimple* registry); | |
96 | |
97 protected: | |
98 // Initializes the state of the RapporService. | |
99 void InitializeInternal(std::unique_ptr<LogUploaderInterface> uploader, | |
100 int32_t cohort, | |
101 const std::string& secret); | |
102 | |
103 // Cancels the next call to OnLogInterval. | |
104 virtual void CancelNextLogRotation(); | |
105 | |
106 // Schedules the next call to OnLogInterval. | |
107 virtual void ScheduleNextLogRotation(base::TimeDelta interval); | |
108 | |
109 // Logs all of the collected metrics to the reports proto message and clears | |
110 // the internal map. Exposed for tests. Returns true if any metrics were | |
111 // recorded. | |
112 bool ExportMetrics(RapporReports* reports); | |
113 | |
114 private: | |
115 // Records a sample of the rappor metric specified by |parameters|. | |
116 // Creates and initializes the metric, if it doesn't yet exist. | |
117 // Exposed for tests. | |
118 void RecordSampleInternal(const std::string& metric_name, | |
119 const RapporParameters& parameters, | |
120 const std::string& sample); | |
121 | |
122 // Checks if the service has been started successfully. | |
123 bool IsInitialized() const; | |
124 | |
125 // Called whenever the logging interval elapses to generate a new log of | |
126 // reports and pass it to the uploader. | |
127 void OnLogInterval(); | |
128 | |
129 // Check if recording of the metric is allowed, given it's parameters. | |
130 // This will check that we are not in incognito mode, and that the | |
131 // appropriate recording group is enabled. | |
132 bool RecordingAllowed(const RapporParameters& parameters); | |
133 | |
134 // Finds a metric in the metrics_map_, creating it if it doesn't already | |
135 // exist. | |
136 RapporMetric* LookUpMetric(const std::string& metric_name, | |
137 const RapporParameters& parameters); | |
138 | |
139 // A weak pointer to the PrefService used to read and write preferences. | |
140 PrefService* pref_service_; | |
141 | |
142 // A callback for testing if incognito mode is active; | |
143 const base::Callback<bool(void)> is_incognito_callback_; | |
144 | |
145 // Client-side secret used to generate fake bits. | |
146 std::string secret_; | |
147 | |
148 // The cohort this client is assigned to. -1 is uninitialized. | |
149 int32_t cohort_; | |
150 | |
151 // Timer which schedules calls to OnLogInterval(). | |
152 base::OneShotTimer log_rotation_timer_; | |
153 | |
154 // A daily event for collecting metrics once a day. | |
155 metrics::DailyEvent daily_event_; | |
156 | |
157 // A private LogUploader instance for sending reports to the server. | |
158 std::unique_ptr<LogUploaderInterface> uploader_; | |
159 | |
160 // The set of recording groups that metrics are being recorded, e.g. | |
161 // UMA_RECORDING_GROUP | SAFEBROWSING_RECORDING_GROUP | |
162 int recording_groups_; | |
163 | |
164 // We keep all registered metrics in a map, from name to metric. | |
165 std::map<std::string, std::unique_ptr<RapporMetric>> metrics_map_; | |
166 | |
167 internal::Sampler sampler_; | |
168 | |
169 base::ThreadChecker thread_checker_; | |
170 | |
171 DISALLOW_COPY_AND_ASSIGN(RapporService); | |
172 }; | |
173 | |
174 } // namespace rappor | |
175 | |
176 #endif // COMPONENTS_RAPPOR_RAPPOR_SERVICE_H_ | |
OLD | NEW |