[TTS] Write initial Tap-features to Ranker.

chrome/android/java/src/org/chromium/chrome/browser/contextualsearch/ContextualSearchRankerLoggerImpl.java

Index: chrome/android/java/src/org/chromium/chrome/browser/contextualsearch/ContextualSearchRankerLoggerImpl.java
diff --git a/chrome/android/java/src/org/chromium/chrome/browser/contextualsearch/ContextualSearchRankerLoggerImpl.java b/chrome/android/java/src/org/chromium/chrome/browser/contextualsearch/ContextualSearchRankerLoggerImpl.java
index f56846a9e1520392f10e63580250c007eebfb28f..e1500ee702b7aab80aea1515b64ebb53c856ab0e 100644
--- a/chrome/android/java/src/org/chromium/chrome/browser/contextualsearch/ContextualSearchRankerLoggerImpl.java
+++ b/chrome/android/java/src/org/chromium/chrome/browser/contextualsearch/ContextualSearchRankerLoggerImpl.java
@@ -6,25 +6,137 @@ package org.chromium.chrome.browser.contextualsearch;
 
 import org.chromium.base.Log;
 
+import java.net.URL;
+
 /**
  * Implements the UMA logging for Ranker that's used for Contextual Search Tap Suppression.
  */
 public class ContextualSearchRankerLoggerImpl implements ContextualSearchRankerLogger {
     private static final String TAG = "ContextualSearch";
 
+    // The current version of the Ranker model that we use for Contextual Search.
+    private static final int CONTEXTUAL_SEARCH_RANKER_MODEL_VERSION = 0;
+
+    // Pointer to the native instance of this class.
+    private long mNativePointer;
+
+    // Whether logging for the current URL is currently setup.
+    private boolean mIsLoggingSetup;
+
+    // Whether the service is ready to actually record log data.
+    private boolean mCanServiceActuallyRecord;
+
+    // Whether any data has been written to the log since calling setupLoggingForPage().
+    private boolean mDidLog;
+
+    /**
+     * Constructs a Ranker Logger and associated native implementation to write Contextual Search
+     * ML data to Ranker.
+     */
+    public ContextualSearchRankerLoggerImpl() {
+        if (isEnabled()) mNativePointer = nativeInit();
+    }
+
+    /**
+     * This method should be called to clean up storage when an instance of this class is
+     * no longer in use.  The nativeDestroy will call the destructor on the native instance.
+     */
+    void destroy() {
+        if (isEnabled()) {
+            assert mNativePointer != 0;
+            writeLogAndReset();
+            nativeDestroy(mNativePointer);
+            mNativePointer = 0;
+            mIsLoggingSetup = false;
+            mCanServiceActuallyRecord = false;
+            mDidLog = false;
+        }
+    }
+
+    @Override
+    public void setupLoggingForPage(URL basePageUrl) {
+        if (isEnabled()) {
+            mIsLoggingSetup = true;
+            // The URL may be null for custom Chrome URIs like chrome://flags.
+            if (basePageUrl != null) {
+                nativeSetupLoggingAndRanker(mNativePointer, basePageUrl.toString(),
+                        getContextualSearchRankerModelSelector());
+                mCanServiceActuallyRecord = true;
+            }
+        }
+    }
+
     @Override
     public void log(Feature feature, Object value) {
-        // TODO(donnd): log to an actual persistent proto ASAP!
-        Log.v(TAG, "log %s with value %s", feature.toString(), value);
+        if (!isEnabled()) {
+            Log.v(TAG, "Ranker data %s: %s", feature, value);
+            return;
+        }
+
+        // TODO(donnd): Add some enforcement that log() calls are done before inference time.
+        assert mIsLoggingSetup;
+        if (value instanceof Boolean) {
+            logToNative(feature.toString(), ((boolean) value ? 1 : 0));
+        } else if (value instanceof Integer) {
+            Integer i = (int) value;
+            logToNative(feature.toString(), Long.valueOf(i));
+        } else if (value instanceof Long) {
+            logToNative(feature.toString(), (long) value);
+        } else if (value instanceof Character) {
+            logToNative(feature.toString(), Character.getNumericValue((char) value));
+        } else {
+            Log.w(TAG,
+                    "Could not log feature to Ranker: " + feature.toString() + " of class "
+                            + value.getClass());
+        }
     }
 
     @Override
-    public void logOutcome(Object value) {
-        log(Feature.OUTCOME_WAS_PANEL_OPENED, value);
+    public void logOutcome(Feature feature, Object value) {
+        log(feature, value);
     }
 
     @Override
     public void writeLogAndReset() {
-        Log.v(TAG, "Reset!\n");
+        if (isEnabled() && mDidLog) {
+            nativeWriteLogAndReset(mNativePointer);
+            mIsLoggingSetup = false;
+            mCanServiceActuallyRecord = false;
+            mDidLog = false;
+        }
+    }
+
+    /** Whether actually writing data is enabled.  If not, we may do nothing or print. */
+    private boolean isEnabled() {
+        return ContextualSearchFieldTrial.isRankerLoggingEnabled();
     }
+
+    /**
+     * Logs to the native instance.  All native logging must go through this bottleneck.
+     * @param feature The feature to log.
+     * @param value The value to log.
+     */
+    private void logToNative(String feature, long value) {
+        if (mCanServiceActuallyRecord) {
+            nativeLogLong(mNativePointer, feature, value);
+            mDidLog = true;
+        }
+    }
+
+    /** Gets the model selector that specifies which model to use for Ranker. */
+    private String getContextualSearchRankerModelSelector() {
+        return "org.chromium.chrome.browser.contextualsearch.v"
+                + CONTEXTUAL_SEARCH_RANKER_MODEL_VERSION;
+    }
+
+    // ============================================================================================
+    // Native methods.
+    // ============================================================================================
+    private native long nativeInit();
+    private native void nativeDestroy(long nativeContextualSearchRankerLoggerImpl);
+    private native void nativeLogLong(
+            long nativeContextualSearchRankerLoggerImpl, String featureString, long value);
+    private native void nativeSetupLoggingAndRanker(long nativeContextualSearchRankerLoggerImpl,
+            String basePageUrl, String rankerModelSelector);
+    private native void nativeWriteLogAndReset(long nativeContextualSearchRankerLoggerImpl);
 }