Add database function to purge an already recorded artifact

Author: Kobzol

database/src/pool.rs

@@ -179,6 +179,9 @@ pub trait Connection: Send + Sync {
         profile: &str,
         cache: &str,
     ) -> Vec<(ArtifactIdNumber, i32)>;
+
+    /// Removes all data associated with the given artifact.
+    async fn purge_artifact(&self, aid: &ArtifactId);
 }
 
 #[async_trait::async_trait]

database/src/pool/postgres.rs

@@ -1380,4 +1380,14 @@ where
             _ => panic!("unknown artifact type: {:?}", ty),
         }
     }
+
+    async fn purge_artifact(&self, aid: &ArtifactId) {
+        // Once we delete the artifact, all data associated with it should also be deleted
+        // thanks to ON DELETE CASCADE.
+        let info = aid.info();
+        self.conn()
+            .execute("delete from artifact where name = $1", &[&info.name])
+            .await
+            .unwrap();
+    }
 }

database/src/pool/sqlite.rs

@@ -1231,4 +1231,13 @@ impl Connection for SqliteConnection {
             .collect::<Result<_, _>>()
             .unwrap()
     }
+
+    async fn purge_artifact(&self, aid: &ArtifactId) {
+        // Once we delete the artifact, all data associated with it should also be deleted
+        // thanks to ON DELETE CASCADE.
+        let info = aid.info();
+        self.raw_ref()
+            .execute("delete from artifact where name = ?1", [info.name])
+            .unwrap();
+    }
 }