M-Labs/nac3
M-Labs
/
nac3
8
0
Fork 5
Code Issues 89 Pull Requests 7 Releases Wiki Activity

core: name codegen worker threads

This commit is contained in:
lyken 2024-06-25 12:21:22 +08:00
parent 06e9d90d57
commit fb331c05b4
1 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
nac3core/src/codegen/mod.rs
View File

@ -273,11 +273,21 @@ impl WorkerRegistry {
let registry = registry.clone();
let registry2 = registry.clone();
let f = f.clone();
let handle = thread::spawn(move || {
registry.worker_thread(generator.as_mut(), &f);
});
let handle = thread::spawn(move || {
if let Err(e) = handle.join() {
// Create a thread name for the worker,
// so when it panics, the error message would be
// prettier than `thread '<unnamed>' panicked at <location>`
let worker_thread_name =
format!("codegen-worker-{worker_id}", worker_id = generator.get_name());
let worker_handle = thread::Builder::new()
.name(worker_thread_name)
.spawn(move || {
registry.worker_thread(generator.as_mut(), &f);
})
.unwrap();
let collector_handle = thread::spawn(move || {
if let Err(e) = worker_handle.join() {
if let Some(e) = e.downcast_ref::<&'static str>() {
eprintln!("Got an error: {e}");
} else {
@ -287,7 +297,7 @@ impl WorkerRegistry {
registry2.wait_condvar.notify_all();
}
});
handles.push(handle);
handles.push(collector_handle);
}
(registry, handles)
}