Skip to content

Commit

Permalink
fix rebuilds not queued when there are failed builds with rebuild pri…
Browse files Browse the repository at this point in the history
…o in the queue
  • Loading branch information
syphar committed Oct 25, 2024
1 parent 474c467 commit 385741e
Showing 1 changed file with 48 additions and 7 deletions.
55 changes: 48 additions & 7 deletions src/build_queue.rs
Original file line number Diff line number Diff line change
Expand Up @@ -676,17 +676,17 @@ pub async fn queue_rebuilds(
config: &Config,
build_queue: &AsyncBuildQueue,
) -> Result<()> {
let already_queued_rebuilds = sqlx::query_scalar!(
r#"SELECT COUNT(*) as "count!" FROM queue WHERE priority >= $1"#,
REBUILD_PRIORITY
)
.fetch_one(&mut *conn)
.await?;
let already_queued_rebuilds: usize = build_queue
.pending_count_by_priority()
.await?
.iter()
.filter_map(|(priority, count)| (*priority >= REBUILD_PRIORITY).then_some(count))
.sum();

let rebuilds_to_queue = config
.max_queued_rebuilds
.expect("config.max_queued_rebuilds not set") as i64
- already_queued_rebuilds;
- already_queued_rebuilds as i64;

if rebuilds_to_queue <= 0 {
info!("not queueing rebuilds; queue limit reached");
Expand Down Expand Up @@ -806,6 +806,47 @@ mod tests {
})
}

#[test]
fn test_still_rebuild_when_full_with_failed() {
crate::test::async_wrapper(|env| async move {
env.override_config(|config| {
config.max_queued_rebuilds = Some(1);
config.rebuild_up_to_date = Some(NaiveDate::from_ymd_opt(2024, 1, 1).unwrap());
});

let build_queue = env.async_build_queue().await;
build_queue
.add_crate("foo1", "0.1.0", REBUILD_PRIORITY, None)
.await?;
build_queue
.add_crate("foo2", "0.1.0", REBUILD_PRIORITY, None)
.await?;

let mut conn = env.async_db().await.async_conn().await;
sqlx::query!("UPDATE queue SET attempt = 99")
.execute(&mut *conn)
.await?;

assert_eq!(build_queue.queued_crates().await?.len(), 0);

env.async_fake_release()
.await
.name("foo")
.version("0.1.0")
.builds(vec![FakeBuild::default()
.rustc_version("rustc 1.84.0-nightly (e7c0d2750 2020-10-15)")])
.create_async()
.await?;

let build_queue = env.async_build_queue().await;
queue_rebuilds(&mut conn, &env.config(), &build_queue).await?;

assert_eq!(build_queue.queued_crates().await?.len(), 1);

Ok(())
})
}

#[test]
fn test_dont_rebuild_when_full() {
crate::test::async_wrapper(|env| async move {
Expand Down

0 comments on commit 385741e

Please sign in to comment.