Move the load_workqueue function to pr_tracking

Author: Kobzol

src/handlers/pr_tracking.rs

@@ -130,6 +130,22 @@ pub(super) async fn handle_input<'a>(
     Ok(())
 }
 
+/// Loads the workqueue (mapping of open PRs assigned to users) from GitHub
+pub async fn load_workqueue(client: &Octocrab) -> anyhow::Result<ReviewerWorkqueue> {
+    tracing::debug!("Loading workqueue for rust-lang/rust");
+    let prs = retrieve_pull_request_assignments("rust-lang", "rust", &client).await?;
+
+    // Aggregate PRs by user
+    let aggregated: HashMap<UserId, HashSet<PullRequestNumber>> =
+        prs.into_iter().fold(HashMap::new(), |mut acc, (user, pr)| {
+            let prs = acc.entry(user.id).or_default();
+            prs.insert(pr);
+            acc
+        });
+    tracing::debug!("PR assignments\n{aggregated:?}");
+    Ok(ReviewerWorkqueue::new(aggregated))
+}
+
 /// Retrieve tuples of (user, PR number) where
 /// the given user is assigned as a reviewer for that PR.
 /// Only non-draft, non-rollup and open PRs are taken into account.

src/handlers/pull_requests_assignment_update.rs

@@ -1,10 +1,6 @@
-use crate::github::PullRequestNumber;
-use crate::github::UserId;
-use crate::handlers::pr_tracking::{retrieve_pull_request_assignments, ReviewerWorkqueue};
+use crate::handlers::pr_tracking::load_workqueue;
 use crate::jobs::Job;
 use async_trait::async_trait;
-use octocrab::Octocrab;
-use std::collections::{HashMap, HashSet};
 
 pub struct PullRequestAssignmentUpdate;
 
@@ -23,19 +19,3 @@ impl Job for PullRequestAssignmentUpdate {
         Ok(())
     }
 }
-
-/// Loads the workqueue (mapping of open PRs assigned to users) from GitHub
-pub async fn load_workqueue(client: &Octocrab) -> anyhow::Result<ReviewerWorkqueue> {
-    tracing::debug!("Loading workqueue for rust-lang/rust");
-    let prs = retrieve_pull_request_assignments("rust-lang", "rust", &client).await?;
-
-    // Aggregate PRs by user
-    let aggregated: HashMap<UserId, HashSet<PullRequestNumber>> =
-        prs.into_iter().fold(HashMap::new(), |mut acc, (user, pr)| {
-            let prs = acc.entry(user.id).or_default();
-            prs.insert(pr);
-            acc
-        });
-    tracing::debug!("PR assignments\n{aggregated:?}");
-    Ok(ReviewerWorkqueue::new(aggregated))
-}

src/main.rs

@@ -12,8 +12,8 @@ use tokio::{task, time};
 use tower::{Service, ServiceExt};
 use tracing as log;
 use tracing::Instrument;
+use triagebot::handlers::pr_tracking::load_workqueue;
 use triagebot::handlers::pr_tracking::ReviewerWorkqueue;
-use triagebot::handlers::pull_requests_assignment_update::load_workqueue;
 use triagebot::jobs::{
     default_jobs, JOB_PROCESSING_CADENCE_IN_SECS, JOB_SCHEDULING_CADENCE_IN_SECS,
 };