Extract Job::enqueue_sync_to_index() fn

Author: Turbo87

src/admin/delete_crate.rs

@@ -56,10 +56,7 @@ fn delete(opts: Opts, conn: &mut PgConnection) {
     }
 
     if dotenv::var("FEATURE_INDEX_SYNC").is_ok() {
-        Job::sync_to_git_index(&krate.name).enqueue(conn).unwrap();
-        Job::sync_to_sparse_index(&krate.name)
-            .enqueue(conn)
-            .unwrap();
+        Job::enqueue_sync_to_index(&krate.name, conn).unwrap();
     } else {
         uploader.delete_index(&client, &krate.name).unwrap();
     }

src/admin/delete_version.rs

@@ -60,9 +60,6 @@ fn delete(opts: Opts, conn: &mut PgConnection) {
     }
 
     if dotenv::var("FEATURE_INDEX_SYNC").is_ok() {
-        Job::sync_to_git_index(&krate.name).enqueue(conn).unwrap();
-        Job::sync_to_sparse_index(&krate.name)
-            .enqueue(conn)
-            .unwrap();
+        Job::enqueue_sync_to_index(&krate.name, conn).unwrap();
     }
 }

src/admin/yank_version.rs

@@ -66,10 +66,7 @@ fn yank(opts: Opts, conn: &mut PgConnection) {
         .unwrap();
 
     if dotenv::var("FEATURE_INDEX_SYNC").is_ok() {
-        Job::sync_to_git_index(&krate.name).enqueue(conn).unwrap();
-        Job::sync_to_sparse_index(&krate.name)
-            .enqueue(conn)
-            .unwrap();
+        Job::enqueue_sync_to_index(&krate.name, conn).unwrap();
     } else {
         crate::worker::sync_yanked(krate.name, v.num)
             .enqueue(conn)

src/background_jobs.rs

@@ -53,6 +53,31 @@ impl Job {
     const SYNC_TO_SPARSE_INDEX: &str = "sync_to_sparse_index";
     const UPDATE_DOWNLOADS: &str = "update_downloads";
 
+    pub fn enqueue_sync_to_index<T: ToString>(
+        krate: T,
+        conn: &mut PgConnection,
+    ) -> Result<(), EnqueueError> {
+        use crate::schema::background_jobs::dsl::*;
+
+        let to_git = Self::sync_to_git_index(krate.to_string());
+        let to_git = (
+            job_type.eq(to_git.as_type_str()),
+            data.eq(to_git.to_value()?),
+        );
+
+        let to_sparse = Self::sync_to_sparse_index(krate.to_string());
+        let to_sparse = (
+            job_type.eq(to_sparse.as_type_str()),
+            data.eq(to_sparse.to_value()?),
+        );
+
+        diesel::insert_into(background_jobs)
+            .values(vec![to_git, to_sparse])
+            .execute(conn)?;
+
+        Ok(())
+    }
+
     pub fn sync_to_git_index<T: ToString>(krate: T) -> Job {
         Job::SyncToGitIndex(SyncToIndexJob {
             krate: krate.to_string(),

src/controllers/krate/publish.rs

@@ -274,8 +274,7 @@ pub async fn publish(app: AppState, req: BytesRequest) -> AppResult<Json<GoodCra
             };
 
             if app.config.feature_index_sync {
-                Job::sync_to_git_index(&git_crate.name).enqueue(conn)?;
-                Job::sync_to_sparse_index(&git_crate.name).enqueue(conn)?;
+                Job::enqueue_sync_to_index(&krate.name, conn)?;
             } else {
                 worker::add_crate(git_crate).enqueue(conn)?;
             }

src/controllers/version/yank.rs

@@ -86,8 +86,7 @@ fn modify_yank(
     insert_version_owner_action(conn, version.id, user.id, api_token_id, action)?;
 
     if state.config.feature_index_sync {
-        Job::sync_to_git_index(&krate.name).enqueue(conn)?;
-        Job::sync_to_sparse_index(&krate.name).enqueue(conn)?;
+        Job::enqueue_sync_to_index(&krate.name, conn)?;
     } else {
         worker::sync_yanked(krate.name, version.num).enqueue(conn)?;
     }