fix: handle key '/user/' not exist in dynamodb when tables are create…

…d first time (openobserve#1486)
taimingl · Sep 12, 2023 · 3441bff · 3441bff
1 parent 9d1f82d
commit 3441bff
Showing 1 changed file with 13 additions and 11 deletions.
diff --git a/src/service/db/user.rs b/src/service/db/user.rs
@@ -153,17 +153,19 @@ pub async fn cache() -> Result<(), anyhow::Error> {
 pub async fn root_user_exists() -> bool {
     let db = &infra_db::DEFAULT;
     let key = "/user/";
-    let mut ret = db.list_values(key).await.unwrap();
-    ret.retain(|item| {
-        let user: DBUser = json::from_slice(item).unwrap();
-        user.organizations
-            .first()
-            .as_ref()
-            .unwrap()
-            .role
-            .eq(&crate::common::meta::user::UserRole::Root)
-    });
-    !ret.is_empty()
+    if let Ok(mut ret) = db.list_values(key).await {
+        ret.retain(|item| {
+            let user: DBUser = json::from_slice(item).unwrap();
+            user.organizations
+                .first()
+                .as_ref()
+                .unwrap()
+                .role
+                .eq(&crate::common::meta::user::UserRole::Root)
+        });
+        return !ret.is_empty();
+    }
+    false
 }
 
 pub async fn reset() -> Result<(), anyhow::Error> {