Skip to content
This repository was archived by the owner on May 28, 2025. It is now read-only.
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit d308586

Browse files
committedMar 10, 2025·
Auto merge of rust-lang#138307 - Kobzol:citool-alias, r=<try>
Allow specifying glob patterns for try jobs This PR modifies the `try-job` lookup logic to allow glob patterns. So you can e.g. request all MSVC-related jobs with `try-job: *msvc*`. Best reviewed commit by commit. r? `@marcoieni` try-job: *msvc*
2 parents 2b285cd + d32fddf commit d308586

File tree

6 files changed

+344
-236
lines changed

6 files changed

+344
-236
lines changed
 

‎src/ci/citool/Cargo.lock

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,7 @@ dependencies = [
107107
"build_helper",
108108
"clap",
109109
"csv",
110+
"glob-match",
110111
"insta",
111112
"serde",
112113
"serde_json",
@@ -308,6 +309,12 @@ dependencies = [
308309
"wasi",
309310
]
310311

312+
[[package]]
313+
name = "glob-match"
314+
version = "0.2.1"
315+
source = "registry+https://github.com/rust-lang/crates.io-index"
316+
checksum = "9985c9503b412198aa4197559e9a318524ebc4519c229bfa05a535828c950b9d"
317+
311318
[[package]]
312319
name = "hashbrown"
313320
version = "0.15.2"

‎src/ci/citool/Cargo.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ edition = "2021"
77
anyhow = "1"
88
clap = { version = "4.5", features = ["derive"] }
99
csv = "1"
10+
glob-match = "0.2"
1011
serde = { version = "1", features = ["derive"] }
1112
serde_yaml = "0.9"
1213
serde_json = "1"

‎src/ci/citool/src/jobs.rs

Lines changed: 309 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,309 @@
1+
use std::collections::BTreeMap;
2+
3+
use serde_yaml::Value;
4+
5+
use crate::GitHubContext;
6+
7+
/// Representation of a job loaded from the `src/ci/github-actions/jobs.yml` file.
8+
#[derive(serde::Deserialize, Debug, Clone)]
9+
pub struct Job {
10+
/// Name of the job, e.g. mingw-check
11+
pub name: String,
12+
/// GitHub runner on which the job should be executed
13+
pub os: String,
14+
pub env: BTreeMap<String, Value>,
15+
/// Should the job be only executed on a specific channel?
16+
#[serde(default)]
17+
pub only_on_channel: Option<String>,
18+
/// Do not cancel the whole workflow if this job fails.
19+
#[serde(default)]
20+
pub continue_on_error: Option<bool>,
21+
/// Free additional disk space in the job, by removing unused packages.
22+
#[serde(default)]
23+
pub free_disk: Option<bool>,
24+
}
25+
26+
impl Job {
27+
/// By default, the Docker image of a job is based on its name.
28+
/// However, it can be overridden by its IMAGE environment variable.
29+
pub fn image(&self) -> String {
30+
self.env
31+
.get("IMAGE")
32+
.map(|v| v.as_str().expect("IMAGE value should be a string").to_string())
33+
.unwrap_or_else(|| self.name.clone())
34+
}
35+
36+
fn is_linux(&self) -> bool {
37+
self.os.contains("ubuntu")
38+
}
39+
}
40+
41+
#[derive(serde::Deserialize, Debug)]
42+
struct JobEnvironments {
43+
#[serde(rename = "pr")]
44+
pr_env: BTreeMap<String, Value>,
45+
#[serde(rename = "try")]
46+
try_env: BTreeMap<String, Value>,
47+
#[serde(rename = "auto")]
48+
auto_env: BTreeMap<String, Value>,
49+
}
50+
51+
#[derive(serde::Deserialize, Debug)]
52+
pub struct JobDatabase {
53+
#[serde(rename = "pr")]
54+
pub pr_jobs: Vec<Job>,
55+
#[serde(rename = "try")]
56+
pub try_jobs: Vec<Job>,
57+
#[serde(rename = "auto")]
58+
pub auto_jobs: Vec<Job>,
59+
60+
/// Shared environments for the individual run types.
61+
envs: JobEnvironments,
62+
}
63+
64+
impl JobDatabase {
65+
/// Find `auto` jobs that correspond to the passed `pattern`.
66+
/// Patterns are matched using the glob syntax.
67+
/// For example `dist-*` matches all jobs starting with `dist-`.
68+
fn find_auto_jobs_by_patter(&self, pattern: &str) -> Vec<Job> {
69+
self.auto_jobs
70+
.iter()
71+
.filter(|j| glob_match::glob_match(pattern, &j.name))
72+
.cloned()
73+
.collect()
74+
}
75+
}
76+
77+
pub fn load_job_db(db: &str) -> anyhow::Result<JobDatabase> {
78+
let mut db: Value = serde_yaml::from_str(&db)?;
79+
80+
// We need to expand merge keys (<<), because serde_yaml can't deal with them
81+
// `apply_merge` only applies the merge once, so do it a few times to unwrap nested merges.
82+
db.apply_merge()?;
83+
db.apply_merge()?;
84+
85+
let db: JobDatabase = serde_yaml::from_value(db)?;
86+
Ok(db)
87+
}
88+
89+
/// Representation of a job outputted to a GitHub Actions workflow.
90+
#[derive(serde::Serialize, Debug)]
91+
struct GithubActionsJob {
92+
/// The main identifier of the job, used by CI scripts to determine what should be executed.
93+
name: String,
94+
/// Helper label displayed in GitHub Actions interface, containing the job name and a run type
95+
/// prefix (PR/try/auto).
96+
full_name: String,
97+
os: String,
98+
env: BTreeMap<String, serde_json::Value>,
99+
#[serde(skip_serializing_if = "Option::is_none")]
100+
continue_on_error: Option<bool>,
101+
#[serde(skip_serializing_if = "Option::is_none")]
102+
free_disk: Option<bool>,
103+
}
104+
105+
/// Skip CI jobs that are not supposed to be executed on the given `channel`.
106+
fn skip_jobs(jobs: Vec<Job>, channel: &str) -> Vec<Job> {
107+
jobs.into_iter()
108+
.filter(|job| {
109+
job.only_on_channel.is_none() || job.only_on_channel.as_deref() == Some(channel)
110+
})
111+
.collect()
112+
}
113+
114+
/// Type of workflow that is being executed on CI
115+
#[derive(Debug)]
116+
pub enum RunType {
117+
/// Workflows that run after a push to a PR branch
118+
PullRequest,
119+
/// Try run started with @bors try
120+
TryJob { job_patterns: Option<Vec<String>> },
121+
/// Merge attempt workflow
122+
AutoJob,
123+
}
124+
125+
/// Maximum number of custom try jobs that can be requested in a single
126+
/// `@bors try` request.
127+
const MAX_TRY_JOBS_COUNT: usize = 20;
128+
129+
fn calculate_jobs(
130+
run_type: &RunType,
131+
db: &JobDatabase,
132+
channel: &str,
133+
) -> anyhow::Result<Vec<GithubActionsJob>> {
134+
let (jobs, prefix, base_env) = match run_type {
135+
RunType::PullRequest => (db.pr_jobs.clone(), "PR", &db.envs.pr_env),
136+
RunType::TryJob { job_patterns } => {
137+
let jobs = if let Some(patterns) = job_patterns {
138+
let mut jobs = vec![];
139+
let mut unknown_patterns = vec![];
140+
for pattern in patterns {
141+
let matched_jobs = db.find_auto_jobs_by_patter(pattern);
142+
if matched_jobs.is_empty() {
143+
unknown_patterns.push(pattern.clone());
144+
} else {
145+
jobs.extend(matched_jobs);
146+
}
147+
}
148+
if !unknown_patterns.is_empty() {
149+
return Err(anyhow::anyhow!(
150+
"Patterns `{}` did not match any auto jobs",
151+
unknown_patterns.join(", ")
152+
));
153+
}
154+
if jobs.len() > MAX_TRY_JOBS_COUNT {
155+
return Err(anyhow::anyhow!(
156+
"It is only possible to schedule up to {MAX_TRY_JOBS_COUNT} custom jobs, received {} custom jobs expanded from {} pattern(s)",
157+
jobs.len(),
158+
patterns.len()
159+
));
160+
}
161+
jobs
162+
} else {
163+
db.try_jobs.clone()
164+
};
165+
(jobs, "try", &db.envs.try_env)
166+
}
167+
RunType::AutoJob => (db.auto_jobs.clone(), "auto", &db.envs.auto_env),
168+
};
169+
let jobs = skip_jobs(jobs, channel);
170+
let jobs = jobs
171+
.into_iter()
172+
.map(|job| {
173+
let mut env: BTreeMap<String, serde_json::Value> = crate::yaml_map_to_json(base_env);
174+
env.extend(crate::yaml_map_to_json(&job.env));
175+
let full_name = format!("{prefix} - {}", job.name);
176+
177+
GithubActionsJob {
178+
name: job.name,
179+
full_name,
180+
os: job.os,
181+
env,
182+
continue_on_error: job.continue_on_error,
183+
free_disk: job.free_disk,
184+
}
185+
})
186+
.collect();
187+
188+
Ok(jobs)
189+
}
190+
191+
pub fn calculate_job_matrix(
192+
db: JobDatabase,
193+
gh_ctx: GitHubContext,
194+
channel: &str,
195+
) -> anyhow::Result<()> {
196+
let run_type = gh_ctx.get_run_type().ok_or_else(|| {
197+
anyhow::anyhow!("Cannot determine the type of workflow that is being executed")
198+
})?;
199+
eprintln!("Run type: {run_type:?}");
200+
201+
let jobs = calculate_jobs(&run_type, &db, channel)?;
202+
if jobs.is_empty() {
203+
return Err(anyhow::anyhow!("Computed job list is empty"));
204+
}
205+
206+
let run_type = match run_type {
207+
RunType::PullRequest => "pr",
208+
RunType::TryJob { .. } => "try",
209+
RunType::AutoJob => "auto",
210+
};
211+
212+
eprintln!("Output");
213+
eprintln!("jobs={jobs:?}");
214+
eprintln!("run_type={run_type}");
215+
println!("jobs={}", serde_json::to_string(&jobs)?);
216+
println!("run_type={run_type}");
217+
218+
Ok(())
219+
}
220+
221+
pub fn find_linux_job<'a>(jobs: &'a [Job], name: &str) -> anyhow::Result<&'a Job> {
222+
let Some(job) = jobs.iter().find(|j| j.name == name) else {
223+
let available_jobs: Vec<&Job> = jobs.iter().filter(|j| j.is_linux()).collect();
224+
let mut available_jobs =
225+
available_jobs.iter().map(|j| j.name.to_string()).collect::<Vec<_>>();
226+
available_jobs.sort();
227+
return Err(anyhow::anyhow!(
228+
"Job {name} not found. The following jobs are available:\n{}",
229+
available_jobs.join(", ")
230+
));
231+
};
232+
if !job.is_linux() {
233+
return Err(anyhow::anyhow!("Only Linux jobs can be executed locally"));
234+
}
235+
236+
Ok(job)
237+
}
238+
239+
#[cfg(test)]
240+
mod tests {
241+
use crate::jobs::{JobDatabase, load_job_db};
242+
243+
#[test]
244+
fn lookup_job_pattern() {
245+
let db = load_job_db(
246+
r#"
247+
envs:
248+
pr:
249+
try:
250+
auto:
251+
252+
pr:
253+
try:
254+
auto:
255+
- name: dist-a
256+
os: ubuntu
257+
env: {}
258+
- name: dist-a-alt
259+
os: ubuntu
260+
env: {}
261+
- name: dist-b
262+
os: ubuntu
263+
env: {}
264+
- name: dist-b-alt
265+
os: ubuntu
266+
env: {}
267+
- name: test-a
268+
os: ubuntu
269+
env: {}
270+
- name: test-a-alt
271+
os: ubuntu
272+
env: {}
273+
- name: test-i686
274+
os: ubuntu
275+
env: {}
276+
- name: dist-i686
277+
os: ubuntu
278+
env: {}
279+
- name: test-msvc-i686-1
280+
os: ubuntu
281+
env: {}
282+
- name: test-msvc-i686-2
283+
os: ubuntu
284+
env: {}
285+
"#,
286+
)
287+
.unwrap();
288+
check_pattern(
289+
&db,
290+
"dist-*",
291+
&["dist-a", "dist-a-alt", "dist-b", "dist-b-alt", "dist-i686"],
292+
);
293+
check_pattern(&db, "*-alt", &["dist-a-alt", "dist-b-alt", "test-a-alt"]);
294+
check_pattern(&db, "dist*-alt", &["dist-a-alt", "dist-b-alt"]);
295+
check_pattern(
296+
&db,
297+
"*i686*",
298+
&["test-i686", "dist-i686", "test-msvc-i686-1", "test-msvc-i686-2"],
299+
);
300+
}
301+
302+
#[track_caller]
303+
fn check_pattern(db: &JobDatabase, pattern: &str, expected: &[&str]) {
304+
let jobs =
305+
db.find_auto_jobs_by_patter(pattern).into_iter().map(|j| j.name).collect::<Vec<_>>();
306+
307+
assert_eq!(jobs, expected);
308+
}
309+
}

‎src/ci/citool/src/main.rs

Lines changed: 18 additions & 230 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
mod cpu_usage;
22
mod datadog;
3+
mod jobs;
34
mod merge_report;
45
mod metrics;
56
mod utils;
@@ -10,10 +11,12 @@ use std::process::Command;
1011

1112
use anyhow::Context;
1213
use clap::Parser;
14+
use jobs::JobDatabase;
1315
use serde_yaml::Value;
1416

1517
use crate::cpu_usage::load_cpu_usage;
1618
use crate::datadog::upload_datadog_metric;
19+
use crate::jobs::RunType;
1720
use crate::merge_report::post_merge_report;
1821
use crate::metrics::postprocess_metrics;
1922
use crate::utils::load_env_var;
@@ -22,104 +25,6 @@ const CI_DIRECTORY: &str = concat!(env!("CARGO_MANIFEST_DIR"), "/..");
2225
const DOCKER_DIRECTORY: &str = concat!(env!("CARGO_MANIFEST_DIR"), "/../docker");
2326
const JOBS_YML_PATH: &str = concat!(env!("CARGO_MANIFEST_DIR"), "/../github-actions/jobs.yml");
2427

25-
/// Representation of a job loaded from the `src/ci/github-actions/jobs.yml` file.
26-
#[derive(serde::Deserialize, Debug, Clone)]
27-
struct Job {
28-
/// Name of the job, e.g. mingw-check
29-
name: String,
30-
/// GitHub runner on which the job should be executed
31-
os: String,
32-
env: BTreeMap<String, Value>,
33-
/// Should the job be only executed on a specific channel?
34-
#[serde(default)]
35-
only_on_channel: Option<String>,
36-
/// Rest of attributes that will be passed through to GitHub actions
37-
#[serde(flatten)]
38-
extra_keys: BTreeMap<String, Value>,
39-
}
40-
41-
impl Job {
42-
fn is_linux(&self) -> bool {
43-
self.os.contains("ubuntu")
44-
}
45-
46-
/// By default, the Docker image of a job is based on its name.
47-
/// However, it can be overridden by its IMAGE environment variable.
48-
fn image(&self) -> String {
49-
self.env
50-
.get("IMAGE")
51-
.map(|v| v.as_str().expect("IMAGE value should be a string").to_string())
52-
.unwrap_or_else(|| self.name.clone())
53-
}
54-
}
55-
56-
#[derive(serde::Deserialize, Debug)]
57-
struct JobEnvironments {
58-
#[serde(rename = "pr")]
59-
pr_env: BTreeMap<String, Value>,
60-
#[serde(rename = "try")]
61-
try_env: BTreeMap<String, Value>,
62-
#[serde(rename = "auto")]
63-
auto_env: BTreeMap<String, Value>,
64-
}
65-
66-
#[derive(serde::Deserialize, Debug)]
67-
struct JobDatabase {
68-
#[serde(rename = "pr")]
69-
pr_jobs: Vec<Job>,
70-
#[serde(rename = "try")]
71-
try_jobs: Vec<Job>,
72-
#[serde(rename = "auto")]
73-
auto_jobs: Vec<Job>,
74-
75-
/// Shared environments for the individual run types.
76-
envs: JobEnvironments,
77-
}
78-
79-
impl JobDatabase {
80-
fn find_auto_job_by_name(&self, name: &str) -> Option<Job> {
81-
self.auto_jobs.iter().find(|j| j.name == name).cloned()
82-
}
83-
}
84-
85-
fn load_job_db(path: &Path) -> anyhow::Result<JobDatabase> {
86-
let db = utils::read_to_string(path)?;
87-
let mut db: Value = serde_yaml::from_str(&db)?;
88-
89-
// We need to expand merge keys (<<), because serde_yaml can't deal with them
90-
// `apply_merge` only applies the merge once, so do it a few times to unwrap nested merges.
91-
db.apply_merge()?;
92-
db.apply_merge()?;
93-
94-
let db: JobDatabase = serde_yaml::from_value(db)?;
95-
Ok(db)
96-
}
97-
98-
/// Representation of a job outputted to a GitHub Actions workflow.
99-
#[derive(serde::Serialize, Debug)]
100-
struct GithubActionsJob {
101-
/// The main identifier of the job, used by CI scripts to determine what should be executed.
102-
name: String,
103-
/// Helper label displayed in GitHub Actions interface, containing the job name and a run type
104-
/// prefix (PR/try/auto).
105-
full_name: String,
106-
os: String,
107-
env: BTreeMap<String, serde_json::Value>,
108-
#[serde(flatten)]
109-
extra_keys: BTreeMap<String, serde_json::Value>,
110-
}
111-
112-
/// Type of workflow that is being executed on CI
113-
#[derive(Debug)]
114-
enum RunType {
115-
/// Workflows that run after a push to a PR branch
116-
PullRequest,
117-
/// Try run started with @bors try
118-
TryJob { custom_jobs: Option<Vec<String>> },
119-
/// Merge attempt workflow
120-
AutoJob,
121-
}
122-
12328
struct GitHubContext {
12429
event_name: String,
12530
branch_ref: String,
@@ -130,21 +35,21 @@ impl GitHubContext {
13035
fn get_run_type(&self) -> Option<RunType> {
13136
match (self.event_name.as_str(), self.branch_ref.as_str()) {
13237
("pull_request", _) => Some(RunType::PullRequest),
133-
("push", "refs/heads/try-perf") => Some(RunType::TryJob { custom_jobs: None }),
38+
("push", "refs/heads/try-perf") => Some(RunType::TryJob { job_patterns: None }),
13439
("push", "refs/heads/try" | "refs/heads/automation/bors/try") => {
135-
let custom_jobs = self.get_custom_jobs();
136-
let custom_jobs = if !custom_jobs.is_empty() { Some(custom_jobs) } else { None };
137-
Some(RunType::TryJob { custom_jobs })
40+
let patterns = self.get_try_job_patterns();
41+
let patterns = if !patterns.is_empty() { Some(patterns) } else { None };
42+
Some(RunType::TryJob { job_patterns: patterns })
13843
}
13944
("push", "refs/heads/auto") => Some(RunType::AutoJob),
14045
_ => None,
14146
}
14247
}
14348

144-
/// Tries to parse names of specific CI jobs that should be executed in the form of
145-
/// try-job: <job-name>
49+
/// Tries to parse patterns of CI jobs that should be executed in the form of
50+
/// try-job: <job-pattern>
14651
/// from the commit message of the passed GitHub context.
147-
fn get_custom_jobs(&self) -> Vec<String> {
52+
fn get_try_job_patterns(&self) -> Vec<String> {
14853
if let Some(ref msg) = self.commit_message {
14954
msg.lines()
15055
.filter_map(|line| line.trim().strip_prefix("try-job: "))
@@ -164,15 +69,6 @@ fn load_github_ctx() -> anyhow::Result<GitHubContext> {
16469
Ok(GitHubContext { event_name, branch_ref: load_env_var("GITHUB_REF")?, commit_message })
16570
}
16671

167-
/// Skip CI jobs that are not supposed to be executed on the given `channel`.
168-
fn skip_jobs(jobs: Vec<Job>, channel: &str) -> Vec<Job> {
169-
jobs.into_iter()
170-
.filter(|job| {
171-
job.only_on_channel.is_none() || job.only_on_channel.as_deref() == Some(channel)
172-
})
173-
.collect()
174-
}
175-
17672
fn yaml_map_to_json(map: &BTreeMap<String, Value>) -> BTreeMap<String, serde_json::Value> {
17773
map.into_iter()
17874
.map(|(key, value)| {
@@ -184,124 +80,13 @@ fn yaml_map_to_json(map: &BTreeMap<String, Value>) -> BTreeMap<String, serde_jso
18480
.collect()
18581
}
18682

187-
/// Maximum number of custom try jobs that can be requested in a single
188-
/// `@bors try` request.
189-
const MAX_TRY_JOBS_COUNT: usize = 20;
190-
191-
fn calculate_jobs(
192-
run_type: &RunType,
193-
db: &JobDatabase,
194-
channel: &str,
195-
) -> anyhow::Result<Vec<GithubActionsJob>> {
196-
let (jobs, prefix, base_env) = match run_type {
197-
RunType::PullRequest => (db.pr_jobs.clone(), "PR", &db.envs.pr_env),
198-
RunType::TryJob { custom_jobs } => {
199-
let jobs = if let Some(custom_jobs) = custom_jobs {
200-
if custom_jobs.len() > MAX_TRY_JOBS_COUNT {
201-
return Err(anyhow::anyhow!(
202-
"It is only possible to schedule up to {MAX_TRY_JOBS_COUNT} custom jobs, received {} custom jobs",
203-
custom_jobs.len()
204-
));
205-
}
206-
207-
let mut jobs = vec![];
208-
let mut unknown_jobs = vec![];
209-
for custom_job in custom_jobs {
210-
if let Some(job) = db.find_auto_job_by_name(custom_job) {
211-
jobs.push(job);
212-
} else {
213-
unknown_jobs.push(custom_job.clone());
214-
}
215-
}
216-
if !unknown_jobs.is_empty() {
217-
return Err(anyhow::anyhow!(
218-
"Custom job(s) `{}` not found in auto jobs",
219-
unknown_jobs.join(", ")
220-
));
221-
}
222-
jobs
223-
} else {
224-
db.try_jobs.clone()
225-
};
226-
(jobs, "try", &db.envs.try_env)
227-
}
228-
RunType::AutoJob => (db.auto_jobs.clone(), "auto", &db.envs.auto_env),
229-
};
230-
let jobs = skip_jobs(jobs, channel);
231-
let jobs = jobs
232-
.into_iter()
233-
.map(|job| {
234-
let mut env: BTreeMap<String, serde_json::Value> = yaml_map_to_json(base_env);
235-
env.extend(yaml_map_to_json(&job.env));
236-
let full_name = format!("{prefix} - {}", job.name);
237-
238-
GithubActionsJob {
239-
name: job.name,
240-
full_name,
241-
os: job.os,
242-
env,
243-
extra_keys: yaml_map_to_json(&job.extra_keys),
244-
}
245-
})
246-
.collect();
247-
248-
Ok(jobs)
249-
}
250-
251-
fn calculate_job_matrix(
252-
db: JobDatabase,
253-
gh_ctx: GitHubContext,
254-
channel: &str,
255-
) -> anyhow::Result<()> {
256-
let run_type = gh_ctx.get_run_type().ok_or_else(|| {
257-
anyhow::anyhow!("Cannot determine the type of workflow that is being executed")
258-
})?;
259-
eprintln!("Run type: {run_type:?}");
260-
261-
let jobs = calculate_jobs(&run_type, &db, channel)?;
262-
if jobs.is_empty() {
263-
return Err(anyhow::anyhow!("Computed job list is empty"));
264-
}
265-
266-
let run_type = match run_type {
267-
RunType::PullRequest => "pr",
268-
RunType::TryJob { .. } => "try",
269-
RunType::AutoJob => "auto",
270-
};
271-
272-
eprintln!("Output");
273-
eprintln!("jobs={jobs:?}");
274-
eprintln!("run_type={run_type}");
275-
println!("jobs={}", serde_json::to_string(&jobs)?);
276-
println!("run_type={run_type}");
277-
278-
Ok(())
279-
}
280-
281-
fn find_linux_job<'a>(jobs: &'a [Job], name: &str) -> anyhow::Result<&'a Job> {
282-
let Some(job) = jobs.iter().find(|j| j.name == name) else {
283-
let available_jobs: Vec<&Job> = jobs.iter().filter(|j| j.is_linux()).collect();
284-
let mut available_jobs =
285-
available_jobs.iter().map(|j| j.name.to_string()).collect::<Vec<_>>();
286-
available_jobs.sort();
287-
return Err(anyhow::anyhow!(
288-
"Job {name} not found. The following jobs are available:\n{}",
289-
available_jobs.join(", ")
290-
));
291-
};
292-
if !job.is_linux() {
293-
return Err(anyhow::anyhow!("Only Linux jobs can be executed locally"));
294-
}
295-
296-
Ok(job)
297-
}
298-
29983
fn run_workflow_locally(db: JobDatabase, job_type: JobType, name: String) -> anyhow::Result<()> {
30084
let jobs = match job_type {
30185
JobType::Auto => &db.auto_jobs,
30286
JobType::PR => &db.pr_jobs,
30387
};
304-
let job = find_linux_job(jobs, &name).with_context(|| format!("Cannot find job {name}"))?;
88+
let job =
89+
jobs::find_linux_job(jobs, &name).with_context(|| format!("Cannot find job {name}"))?;
30590

30691
let mut custom_env: BTreeMap<String, String> = BTreeMap::new();
30792
// Replicate src/ci/scripts/setup-environment.sh
@@ -385,7 +170,7 @@ enum Args {
385170
}
386171

387172
#[derive(clap::ValueEnum, Clone)]
388-
enum JobType {
173+
pub enum JobType {
389174
/// Merge attempt ("auto") job
390175
Auto,
391176
/// Pull request job
@@ -395,7 +180,10 @@ enum JobType {
395180
fn main() -> anyhow::Result<()> {
396181
let args = Args::parse();
397182
let default_jobs_file = Path::new(JOBS_YML_PATH);
398-
let load_db = |jobs_path| load_job_db(jobs_path).context("Cannot load jobs.yml");
183+
let load_db = |jobs_path| {
184+
let db = utils::read_to_string(jobs_path)?;
185+
Ok::<_, anyhow::Error>(jobs::load_job_db(&db).context("Cannot load jobs.yml")?)
186+
};
399187

400188
match args {
401189
Args::CalculateJobMatrix { jobs_file } => {
@@ -407,7 +195,7 @@ fn main() -> anyhow::Result<()> {
407195
.trim()
408196
.to_string();
409197

410-
calculate_job_matrix(load_db(jobs_path)?, gh_ctx, &channel)
198+
jobs::calculate_job_matrix(load_db(jobs_path)?, gh_ctx, &channel)
411199
.context("Failed to calculate job matrix")?;
412200
}
413201
Args::RunJobLocally { job_type, name } => {

‎src/ci/citool/src/merge_report.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ use std::collections::HashMap;
44
use anyhow::Context;
55
use build_helper::metrics::{JsonRoot, TestOutcome};
66

7-
use crate::JobDatabase;
7+
use crate::jobs::JobDatabase;
88
use crate::metrics::get_test_suites;
99

1010
type Sha = String;

‎src/doc/rustc-dev-guide/src/tests/ci.md

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -134,15 +134,17 @@ There are several use-cases for try builds:
134134
the [dist-x86_64-linux] CI job.
135135
- Run a specific CI job (e.g. Windows tests) on a PR, to quickly test if it
136136
passes the test suite executed by that job. You can select which CI jobs will
137-
be executed in the try build by adding up to 10 lines containing `try-job:
138-
<name of job>` to the PR description. All such specified jobs will be executed
137+
be executed in the try build by adding lines containing `try-job:
138+
<job patter>` to the PR description. All such specified jobs will be executed
139139
in the try build once the `@bors try` command is used on the PR. If no try
140140
jobs are specified in this way, the jobs defined in the `try` section of
141-
[`jobs.yml`] will be executed by default.
141+
[`jobs.yml`] will be executed by default. Each pattern can either be an exact
142+
name of a job or a glob pattern that matches multiple jobs, for example
143+
`*msvc*` or `*-alt`. You can start at most 20 jobs in a single try build.
142144

143145
> **Using `try-job` PR description directives**
144146
>
145-
> 1. Identify which set of try-jobs (max 10) you would like to exercise. You can
147+
> 1. Identify which set of try-jobs you would like to exercise. You can
146148
> find the name of the CI jobs in [`jobs.yml`].
147149
>
148150
> 2. Amend PR description to include (usually at the end of the PR description)
@@ -153,9 +155,10 @@ There are several use-cases for try builds:
153155
>
154156
> try-job: x86_64-msvc
155157
> try-job: test-various
158+
> try-job: *-alt
156159
> ```
157160
>
158-
> Each `try-job` directive must be on its own line.
161+
> Each `try-job` pattern must be on its own line.
159162
>
160163
> 3. Run the prescribed try jobs with `@bors try`. As aforementioned, this
161164
> requires the user to either (1) have `try` permissions or (2) be delegated

0 commit comments

Comments
 (0)
This repository has been archived.