index-watcher: handle deletion requests when the crate is only queued

Author: syphar

src/build_queue.rs

@@ -245,6 +245,39 @@ impl AsyncBuildQueue {
         .await?
         .is_some())
     }
+
+    async fn remove_crate_from_queue(&self, name: &str) -> Result<()> {
+        let mut conn = self.db.get_async().await?;
+        sqlx::query!(
+            "DELETE
+             FROM queue
+             WHERE name = $1
+             ",
+            name
+        )
+        .execute(&mut *conn)
+        .await?;
+
+        Ok(())
+    }
+
+    async fn remove_version_from_queue(&self, name: &str, version: &Version) -> Result<()> {
+        let mut conn = self.db.get_async().await?;
+        sqlx::query!(
+            "DELETE
+             FROM queue
+             WHERE
+                name = $1 AND
+                version = $2
+             ",
+            name,
+            version as _,
+        )
+        .execute(&mut *conn)
+        .await?;
+
+        Ok(())
+    }
 }
 
 /// Locking functions.
@@ -331,19 +364,22 @@ impl AsyncBuildQueue {
                 }
 
                 self.queue_crate_invalidation(&mut conn, krate).await;
+                self.remove_crate_from_queue(krate).await?;
                 continue;
             }
 
             if let Some(release) = change.version_deleted() {
+                let version: Version = release
+                    .version
+                    .parse()
+                    .context("couldn't parse release version as semver")?;
+
                 match delete_version(
                     &mut conn,
                     &self.storage,
                     &self.config,
                     &release.name,
-                    &release
-                        .version
-                        .parse()
-                        .context("couldn't parse release version as semver")?,
+                    &version,
                 )
                 .await
                 .with_context(|| {
@@ -361,6 +397,8 @@ impl AsyncBuildQueue {
 
                 self.queue_crate_invalidation(&mut conn, &release.name)
                     .await;
+                self.remove_version_from_queue(&release.name, &version)
+                    .await?;
                 continue;
             }
 

src/db/delete.rs

@@ -15,20 +15,17 @@ use super::{CrateId, update_latest_version_id};
 static LIBRARY_STORAGE_PATHS_TO_DELETE: &[&str] = &["rustdoc", "rustdoc-json", "sources"];
 static OTHER_STORAGE_PATHS_TO_DELETE: &[&str] = &["sources"];
 
-#[derive(Debug, thiserror::Error)]
-enum CrateDeletionError {
-    #[error("crate is missing: {0}")]
-    MissingCrate(String),
-}
-
 #[context("error trying to delete crate {name} from database")]
 pub async fn delete_crate(
     conn: &mut sqlx::PgConnection,
     storage: &AsyncStorage,
     config: &Config,
     name: &str,
 ) -> Result<()> {
-    let crate_id = get_id(conn, name).await?;
+    let Some(crate_id) = get_id(conn, name).await? else {
+        return Ok(());
+    };
+
     let is_library = delete_crate_from_database(conn, name, crate_id).await?;
     // #899
     let paths = if is_library {
@@ -68,7 +65,11 @@ pub async fn delete_version(
     name: &str,
     version: &Version,
 ) -> Result<()> {
-    let is_library = delete_version_from_database(conn, name, version).await?;
+    let Some(crate_id) = get_id(conn, name).await? else {
+        return Ok(());
+    };
+
+    let is_library = delete_version_from_database(conn, crate_id, version).await?;
     let paths = if is_library {
         LIBRARY_STORAGE_PATHS_TO_DELETE
     } else {
@@ -105,7 +106,7 @@ pub async fn delete_version(
     Ok(())
 }
 
-async fn get_id(conn: &mut sqlx::PgConnection, name: &str) -> Result<CrateId> {
+async fn get_id(conn: &mut sqlx::PgConnection, name: &str) -> Result<Option<CrateId>> {
     Ok(sqlx::query_scalar!(
         r#"
         SELECT id as "id: CrateId"
@@ -115,8 +116,7 @@ async fn get_id(conn: &mut sqlx::PgConnection, name: &str) -> Result<CrateId> {
         name
     )
     .fetch_optional(&mut *conn)
-    .await?
-    .ok_or_else(|| CrateDeletionError::MissingCrate(name.into()))?)
+    .await?)
 }
 
 // metaprogramming!
@@ -131,10 +131,9 @@ const METADATA: &[(&str, &str)] = &[
 /// Returns whether this release was a library
 async fn delete_version_from_database(
     conn: &mut sqlx::PgConnection,
-    name: &str,
+    crate_id: CrateId,
     version: &Version,
 ) -> Result<bool> {
-    let crate_id = get_id(conn, name).await?;
     let mut transaction = conn.begin().await?;
     for &(table, column) in METADATA {
         sqlx::query(
@@ -152,20 +151,6 @@ async fn delete_version_from_database(
 
     update_latest_version_id(&mut transaction, crate_id).await?;
 
-    let paths = if is_library {
-        LIBRARY_STORAGE_PATHS_TO_DELETE
-    } else {
-        OTHER_STORAGE_PATHS_TO_DELETE
-    };
-    for prefix in paths {
-        sqlx::query!(
-            "DELETE FROM files WHERE path LIKE $1;",
-            format!("{prefix}/{name}/{version}/%"),
-        )
-        .execute(&mut *transaction)
-        .await?;
-    }
-
     transaction.commit().await?;
     Ok(is_library)
 }