From e13f44aeb9909fb821560f6f178c1b0c4d204a55 Mon Sep 17 00:00:00 2001
From: Chenjie Yu <cjyu@google.com>
Date: Thu, 7 Feb 2019 17:51:38 -0800
Subject: [PATCH] train info pull atom

Bug: 122807604
Test: will add gts
Change-Id: I2d6bdee1c5f17d5b2dc341a110605b6aaa6f4947
---
 cmds/statsd/src/atoms.proto | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/cmds/statsd/src/atoms.proto b/cmds/statsd/src/atoms.proto
index 4f17d69f23f25..cec19b8841d76 100644
--- a/cmds/statsd/src/atoms.proto
+++ b/cmds/statsd/src/atoms.proto
@@ -296,6 +296,7 @@ message Atom {
         NumBiometricsEnrolled num_faces_enrolled = 10048;
         RoleHolder role_holder = 10049;
         DangerousPermissionState dangerous_permission_state = 10050;
+        TrainInfo train_info = 10051;
     }
 
     // DO NOT USE field numbers above 100,000 in AOSP.
@@ -5489,3 +5490,20 @@ message DeviceIdentifierAccessDenied {
     // True if the package is privileged.
     optional bool is_priv_app = 4;
 }
+
+/**
+ * Potential experiment ids that goes with a train install.
+ */
+message TrainExperimentIds {
+    repeated int64 experiment_id = 1;
+}
+
+/**
+ * Pulls the ongoing mainline install train version code.
+ * Pulled from StatsCompanionService
+ */
+message TrainInfo {
+    optional int64 train_version_code = 1;
+
+    optional TrainExperimentIds train_experiment_id = 2;
+}