OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "components/translate/core/browser/translate_ranker_impl.h" | 5 #include "components/translate/core/browser/translate_ranker_impl.h" |
6 | 6 |
7 #include <cmath> | 7 #include <cmath> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
10 #include "base/bind_helpers.h" | 10 #include "base/bind_helpers.h" |
11 #include "base/command_line.h" | 11 #include "base/command_line.h" |
12 #include "base/files/file_path.h" | 12 #include "base/files/file_path.h" |
13 #include "base/files/file_util.h" | 13 #include "base/files/file_util.h" |
14 #include "base/memory/ptr_util.h" | 14 #include "base/memory/ptr_util.h" |
15 #include "base/metrics/histogram_macros.h" | 15 #include "base/metrics/histogram_macros.h" |
16 #include "base/metrics/metrics_hashes.h" | 16 #include "base/metrics/metrics_hashes.h" |
17 #include "base/profiler/scoped_tracker.h" | 17 #include "base/profiler/scoped_tracker.h" |
18 #include "base/strings/string_number_conversions.h" | 18 #include "base/strings/string_number_conversions.h" |
19 #include "base/strings/string_util.h" | 19 #include "base/strings/string_util.h" |
20 #include "base/task_runner.h" | 20 #include "base/task_runner.h" |
21 #include "base/threading/thread_task_runner_handle.h" | 21 #include "base/threading/thread_task_runner_handle.h" |
22 #include "components/metrics/proto/translate_event.pb.h" | 22 #include "components/metrics/proto/translate_event.pb.h" |
23 #include "components/translate/core/browser/proto/ranker_model.pb.h" | 23 #include "components/translate/core/browser/proto/ranker_model.pb.h" |
24 #include "components/translate/core/browser/proto/translate_ranker_model.pb.h" | 24 #include "components/translate/core/browser/proto/translate_ranker_model.pb.h" |
25 #include "components/translate/core/browser/ranker_model.h" | 25 #include "components/translate/core/browser/ranker_model.h" |
26 #include "components/translate/core/browser/translate_download_manager.h" | 26 #include "components/translate/core/browser/translate_download_manager.h" |
27 #include "components/translate/core/browser/translate_prefs.h" | 27 #include "components/translate/core/browser/translate_prefs.h" |
28 #include "components/translate/core/browser/translate_url_fetcher.h" | 28 #include "components/translate/core/browser/translate_url_fetcher.h" |
29 #include "components/translate/core/common/translate_switches.h" | 29 #include "components/translate/core/common/translate_switches.h" |
30 #include "components/ukm/ukm_entry_builder.h" | 30 #include "components/ukm/public/ukm_entry_builder.h" |
31 #include "components/ukm/ukm_service.h" | 31 #include "components/ukm/public/ukm_recorder.h" |
32 #include "components/variations/variations_associated_data.h" | 32 #include "components/variations/variations_associated_data.h" |
33 #include "url/gurl.h" | 33 #include "url/gurl.h" |
34 | 34 |
35 namespace translate { | 35 namespace translate { |
36 | 36 |
37 namespace { | 37 namespace { |
38 | 38 |
39 using chrome_intelligence::RankerModel; | 39 using chrome_intelligence::RankerModel; |
40 using chrome_intelligence::RankerModelProto; | 40 using chrome_intelligence::RankerModelProto; |
41 using chrome_intelligence::TranslateRankerModel; | 41 using chrome_intelligence::TranslateRankerModel; |
(...skipping 87 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
129 << "denied_count=" << denied_count << ", " | 129 << "denied_count=" << denied_count << ", " |
130 << "ignored_count=" << ignored_count << ", " | 130 << "ignored_count=" << ignored_count << ", " |
131 << "total_count=" << total_count << ", " | 131 << "total_count=" << total_count << ", " |
132 << "accept_ratio=" << accepted_ratio << ", " | 132 << "accept_ratio=" << accepted_ratio << ", " |
133 << "decline_ratio=" << denied_ratio << ", " | 133 << "decline_ratio=" << denied_ratio << ", " |
134 << "ignore_ratio=" << ignored_ratio; | 134 << "ignore_ratio=" << ignored_ratio; |
135 } | 135 } |
136 | 136 |
137 TranslateRankerImpl::TranslateRankerImpl(const base::FilePath& model_path, | 137 TranslateRankerImpl::TranslateRankerImpl(const base::FilePath& model_path, |
138 const GURL& model_url, | 138 const GURL& model_url, |
139 ukm::UkmService* ukm_service) | 139 ukm::UkmRecorder* ukm_recorder) |
140 : ukm_service_(ukm_service), | 140 : ukm_recorder_(ukm_recorder), |
141 is_logging_enabled_(true), | 141 is_logging_enabled_(true), |
142 is_query_enabled_(base::FeatureList::IsEnabled(kTranslateRankerQuery)), | 142 is_query_enabled_(base::FeatureList::IsEnabled(kTranslateRankerQuery)), |
143 is_enforcement_enabled_( | 143 is_enforcement_enabled_( |
144 base::FeatureList::IsEnabled(kTranslateRankerEnforcement)), | 144 base::FeatureList::IsEnabled(kTranslateRankerEnforcement)), |
145 is_decision_override_enabled_(base::FeatureList::IsEnabled( | 145 is_decision_override_enabled_(base::FeatureList::IsEnabled( |
146 translate::kTranslateRankerDecisionOverride)), | 146 translate::kTranslateRankerDecisionOverride)), |
147 weak_ptr_factory_(this) { | 147 weak_ptr_factory_(this) { |
148 if (is_query_enabled_ || is_enforcement_enabled_ || | 148 if (is_query_enabled_ || is_enforcement_enabled_ || |
149 is_decision_override_enabled_) { | 149 is_decision_override_enabled_) { |
150 model_loader_ = base::MakeUnique<RankerModelLoader>( | 150 model_loader_ = base::MakeUnique<RankerModelLoader>( |
(...skipping 135 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
286 std::vector<metrics::TranslateEventProto>* events) { | 286 std::vector<metrics::TranslateEventProto>* events) { |
287 DCHECK(sequence_checker_.CalledOnValidSequence()); | 287 DCHECK(sequence_checker_.CalledOnValidSequence()); |
288 DVLOG(3) << "Flushing translate ranker events."; | 288 DVLOG(3) << "Flushing translate ranker events."; |
289 events->swap(event_cache_); | 289 events->swap(event_cache_); |
290 event_cache_.clear(); | 290 event_cache_.clear(); |
291 } | 291 } |
292 | 292 |
293 void TranslateRankerImpl::SendEventToUKM( | 293 void TranslateRankerImpl::SendEventToUKM( |
294 const metrics::TranslateEventProto& event, | 294 const metrics::TranslateEventProto& event, |
295 const GURL& url) { | 295 const GURL& url) { |
296 if (!ukm_service_) { | 296 if (!ukm_recorder_) { |
297 DVLOG(3) << "No UKM service."; | 297 DVLOG(3) << "No UKM service."; |
298 return; | 298 return; |
299 } | 299 } |
300 DVLOG(3) << "Sending event for url: " << url.spec(); | 300 DVLOG(3) << "Sending event for url: " << url.spec(); |
301 int32_t source_id = ukm_service_->GetNewSourceID(); | 301 ukm::SourceId source_id = ukm_recorder_->GetNewSourceID(); |
302 ukm_service_->UpdateSourceURL(source_id, url); | 302 ukm_recorder_->UpdateSourceURL(source_id, url); |
303 std::unique_ptr<ukm::UkmEntryBuilder> builder = | 303 std::unique_ptr<ukm::UkmEntryBuilder> builder = |
304 ukm_service_->GetEntryBuilder(source_id, "Translate"); | 304 ukm_recorder_->GetEntryBuilder(source_id, "Translate"); |
305 // The metrics added here should be kept in sync with the documented | 305 // The metrics added here should be kept in sync with the documented |
306 // metrics in tools/metrics/ukm/ukm.xml. | 306 // metrics in tools/metrics/ukm/ukm.xml. |
307 // TODO(hamelphi): Remove hashing functions once UKM accepts strings metrics. | 307 // TODO(hamelphi): Remove hashing functions once UKM accepts strings metrics. |
308 builder->AddMetric("SourceLanguage", | 308 builder->AddMetric("SourceLanguage", |
309 base::HashMetricName(event.source_language())); | 309 base::HashMetricName(event.source_language())); |
310 builder->AddMetric("TargetLanguage", | 310 builder->AddMetric("TargetLanguage", |
311 base::HashMetricName(event.target_language())); | 311 base::HashMetricName(event.target_language())); |
312 builder->AddMetric("Country", base::HashMetricName(event.country())); | 312 builder->AddMetric("Country", base::HashMetricName(event.country())); |
313 builder->AddMetric("AcceptCount", event.accept_count()); | 313 builder->AddMetric("AcceptCount", event.accept_count()); |
314 builder->AddMetric("DeclineCount", event.decline_count()); | 314 builder->AddMetric("DeclineCount", event.decline_count()); |
(...skipping 53 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
368 } | 368 } |
369 } | 369 } |
370 | 370 |
371 } // namespace translate | 371 } // namespace translate |
372 | 372 |
373 std::ostream& operator<<(std::ostream& stream, | 373 std::ostream& operator<<(std::ostream& stream, |
374 const translate::TranslateRankerFeatures& features) { | 374 const translate::TranslateRankerFeatures& features) { |
375 features.WriteTo(stream); | 375 features.WriteTo(stream); |
376 return stream; | 376 return stream; |
377 } | 377 } |
OLD | NEW |