From b33b1ef3ddb71acb41071a4300e0015e40dece7e Mon Sep 17 00:00:00 2001
From: Irevoire <tamo@meilisearch.com>
Date: Wed, 13 Oct 2021 16:36:15 +0200
Subject: [PATCH] update the way of getting and saving the user-id to the file
 system

---
 meilisearch-http/src/analytics.rs | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/meilisearch-http/src/analytics.rs b/meilisearch-http/src/analytics.rs
index bd985c9ea..c0e699344 100644
--- a/meilisearch-http/src/analytics.rs
+++ b/meilisearch-http/src/analytics.rs
@@ -1,5 +1,6 @@
 use serde_json::Value;
 use std::fmt::Display;
+use std::fs::read_to_string;
 
 use crate::Opt;
 
@@ -14,6 +15,7 @@ mod segment {
     use segment::{AutoBatcher, Batcher, HttpClient};
     use serde_json::{json, Value};
     use std::fmt::Display;
+    use std::fs;
     use std::time::{Duration, Instant};
     use sysinfo::{DiskExt, System, SystemExt};
     use tokio::sync::Mutex;
@@ -73,12 +75,17 @@ mod segment {
         }
 
         pub async fn new(opt: &Opt, meilisearch: &MeiliSearch) -> &'static Self {
-            // see if there is already a user-id
-            let user_id = std::fs::read_to_string(opt.db_path.join("user-id"));
+            // see if there is already a user-id in the `data.ms`
+            let user_id = fs::read_to_string(opt.db_path.join("user-id"))
+                .or_else(|_| fs::read_to_string("/tmp/meilisearch-user-id"));
             let first_time_run = user_id.is_err();
             // if not, generate a new user-id and save it to the fs
             let user_id = user_id.unwrap_or_else(|_| Uuid::new_v4().to_string());
-            let _ = std::fs::write(opt.db_path.join("user-id"), user_id.as_bytes());
+            let _ = fs::write(opt.db_path.join("user-id"), user_id.as_bytes());
+            let _ = fs::write(
+                opt.db_path.join("/tmp/meilisearch-user-id"),
+                user_id.as_bytes(),
+            );
 
             let client = HttpClient::default();
             let user = User::UserId {
@@ -103,7 +110,7 @@ mod segment {
                 .await
                 .push(Identify {
                     user: segment.user.clone(),
-                    // TODO: TAMO: what should we do when meilisearch is broken at start
+                    // If meilisearch is corrupted at the start we can panic
                     traits: Self::compute_traits(
                         &segment.opt,
                         meilisearch.get_all_stats().await.unwrap(),
@@ -191,7 +198,8 @@ pub struct MockAnalytics {
 
 impl MockAnalytics {
     pub fn new(opt: &Opt) -> &'static Self {
-        let user = std::fs::read_to_string(opt.db_path.join("user-id"))
+        let user = read_to_string(opt.db_path.join("user-id"))
+            .or_else(|_| read_to_string("/tmp/meilisearch-user-id"))
             .unwrap_or_else(|_| "No user-id".to_string());
         let analytics = Box::new(Self { user });
         Box::leak(analytics)