perf(storage_mapping): Use oc_filecache.path_hash which is covered by an index

Signed-off-by: provokateurin <kate@provokateurin.de>

Author: provokateurin

Cargo.lock

@@ -33,6 +33,7 @@ nextcloud-config-parser = "0.14.1"
 url = "2.5.4"
 clap = { version = "4.5.43", features = ["derive"] }
 sd-notify = { version = "0.4.5", optional = true }
+md5 = "0.8.0"
 
 [dev-dependencies]
 mini-redis = "0.4.1"

Cargo.toml

@@ -10,6 +10,7 @@ use ahash::RandomState;
 use dashmap::mapref::one::Ref;
 use dashmap::DashMap;
 use log::debug;
+use md5;
 use rand::{thread_rng, Rng};
 use sqlx::any::AnyConnectOptions;
 use sqlx::{query_as, Any, AnyPool, FromRow};
@@ -20,8 +21,8 @@ use tokio::time::Duration;
 pub struct UserStorageAccess {
     #[sqlx(rename = "user_id")]
     user: UserId,
-    #[sqlx(rename = "path")]
-    root: String,
+    #[sqlx(rename = "path_hash")]
+    root_hash: String,
 }
 
 struct CachedAccess {
@@ -86,12 +87,25 @@ impl StorageMapping {
         storage: u32,
         path: &str,
     ) -> Result<impl Iterator<Item = UserId>, DatabaseError> {
+        let parents: DashMap<String, ()> = DashMap::new();
+        parents.insert(format!("{:x}", md5::compute("")), ());
+
+        let mut current_path = "".to_string();
+        for name in path.split("/") {
+            if current_path.is_empty() {
+                current_path = name.to_string();
+            } else {
+                current_path = format!("{}/{}", current_path.clone(), name);
+            }
+            parents.insert(format!("{:x}", md5::compute(&current_path)), ());
+        }
+
         let cached = self.get_storage_mapping(storage).await?;
         Ok(cached
             .access
             .iter()
             .filter_map(move |access| {
-                if path.starts_with(&access.root) {
+                if parents.contains_key(&access.root_hash) {
                     Some(access.user.clone())
                 } else {
                     None
@@ -108,7 +122,7 @@ impl StorageMapping {
         debug!("querying storage mapping for {storage}");
         let users = query_as::<Any, UserStorageAccess>(&format!(
             "\
-                SELECT user_id, path \
+                SELECT user_id, path_hash \
                 FROM {prefix}mounts \
                 INNER JOIN {prefix}filecache ON root_id = fileid \
                 WHERE storage_id = {storage}",

src/storage_mapping.rs

@@ -82,7 +82,7 @@ impl Services {
         .await
         .expect("Failed to connect sqlite database");
 
-        sqlx::query("CREATE TABLE oc_filecache(fileid BIGINT, path TEXT)")
+        sqlx::query("CREATE TABLE oc_filecache(fileid BIGINT, path_hash TEXT)")
             .execute(&db)
             .await
             .unwrap();
@@ -237,9 +237,9 @@ impl Services {
     }
 
     async fn add_filecache_item(&self, fileid: u32, path: &str) {
-        sqlx::query("INSERT INTO oc_filecache(fileid, path) VALUES(?, ?)")
+        sqlx::query("INSERT INTO oc_filecache(fileid, path_hash) VALUES(?, ?)")
             .bind(fileid as i64)
-            .bind(path)
+            .bind(format!("{:x}", md5::compute(path)))
             .execute(&self.db)
             .await
             .unwrap();