OLD | NEW |
---|---|
(Empty) | |
1 // Copyright 2017 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 #include <chrome/browser/android/contextualsearch/contextual_search_ranker_logge r_impl.h> | |
6 | |
7 #include "base/android/jni_string.h" | |
8 #include "base/android/scoped_java_ref.h" | |
9 #include "base/metrics/metrics_hashes.h" | |
10 #include "chrome/browser/browser_process.h" | |
11 #include "components/keyed_service/core/keyed_service.h" | |
12 #include "components/ukm/ukm_entry_builder.h" | |
13 #include "components/ukm/ukm_service.h" | |
14 #include "jni/ContextualSearchRankerLoggerImpl_jni.h" | |
15 | |
16 ContextualSearchRankerLoggerImpl::ContextualSearchRankerLoggerImpl( | |
17 JNIEnv* env, | |
18 jobject obj) { | |
19 java_object_.Reset(env, obj); | |
20 } | |
21 | |
22 ContextualSearchRankerLoggerImpl::~ContextualSearchRankerLoggerImpl() { | |
23 java_object_ = nullptr; | |
24 } | |
25 | |
26 void ContextualSearchRankerLoggerImpl::SetupLoggingAndRanker( | |
27 JNIEnv* env, | |
28 jobject obj, | |
29 const base::android::JavaParamRef<jstring>& j_base_page_url) { | |
30 GURL page_url = | |
31 GURL(base::android::ConvertJavaStringToUTF8(env, j_base_page_url)); | |
32 ukm::UkmService* ukm_service = g_browser_process->ukm_service(); | |
33 SetUkmService(ukm_service, page_url); | |
34 // TODO(donnd): set up the model once inference is available. | |
35 } | |
36 | |
37 void ContextualSearchRankerLoggerImpl::SetUkmService( | |
38 ukm::UkmService* ukm_service, | |
39 const GURL& page_url) { | |
40 ukm_service_ = ukm_service; | |
41 source_id_ = ukm_service_->GetNewSourceID(); | |
42 ukm_service_->UpdateSourceURL(source_id_, page_url); | |
43 builder_ = ukm_service_->GetEntryBuilder(source_id_, "ContextualSearch"); | |
rkaplow
2017/05/18 18:33:13
can you register the UKM entry name and features i
Donn Denman
2017/05/18 21:46:36
Sorry about totally missing this! Done.
| |
44 } | |
45 | |
46 void ContextualSearchRankerLoggerImpl::LogLong( | |
47 JNIEnv* env, | |
48 jobject obj, | |
49 const base::android::JavaParamRef<jstring>& j_feature, | |
50 jlong j_long) { | |
51 std::string feature = base::android::ConvertJavaStringToUTF8(env, j_feature); | |
52 builder_->AddMetric(feature.c_str(), j_long); | |
53 } | |
54 | |
55 void ContextualSearchRankerLoggerImpl::WriteLogAndReset(JNIEnv* env, | |
56 jobject obj) { | |
57 // Set up another builder for the next record (in case it's needed). | |
58 builder_ = ukm_service_->GetEntryBuilder(source_id_, "ContextualSearch"); | |
59 } | |
60 | |
61 // Java wrapper boilerplate | |
62 | |
63 void ContextualSearchRankerLoggerImpl::Destroy( | |
64 JNIEnv* env, | |
65 const base::android::JavaParamRef<jobject>& obj) { | |
66 delete this; | |
67 } | |
68 | |
69 bool RegisterContextualSearchRankerLoggerImpl(JNIEnv* env) { | |
70 return RegisterNativesImpl(env); | |
71 } | |
72 | |
73 jlong Init(JNIEnv* env, const base::android::JavaParamRef<jobject>& obj) { | |
74 ContextualSearchRankerLoggerImpl* ranker_logger_impl = | |
75 new ContextualSearchRankerLoggerImpl(env, obj); | |
76 return reinterpret_cast<intptr_t>(ranker_logger_impl); | |
77 } | |
OLD | NEW |