compiler: handle cancelation from finishPrelinkQueue

3 files changed, 5 insertions, 3 deletions

diff --git a/src/Compilation.zig b/src/Compilation.zig
index 8b840b8d45..8e005992ec 100644
--- a/src/Compilation.zig
+++ b/src/Compilation.zig
@@ -5056,7 +5056,9 @@ fn dispatchPrelinkWork(comp: *Compilation, main_progress_node: std.Progress.Node
     }
 
     prelink_group.wait(io);
-    comp.link_queue.finishPrelinkQueue(comp);
+    comp.link_queue.finishPrelinkQueue(comp) catch |err| switch (err) {
+        error.Canceled => return,
+    };
 }
 
 const JobError = Allocator.Error || Io.Cancelable;
diff --git a/src/InternPool.zig b/src/InternPool.zig
index 3076aa2cf3..69a64cbc45 100644
--- a/src/InternPool.zig
+++ b/src/InternPool.zig
@@ -11,7 +11,6 @@ const assert = std.debug.assert;
 const BigIntConst = std.math.big.int.Const;
 const BigIntMutable = std.math.big.int.Mutable;
 const Cache = std.Build.Cache;
-const Io = std.Io;
 const Limb = std.math.big.Limb;
 const Hash = std.hash.Wyhash;
 const Zir = std.zig.Zir;
diff --git a/src/link/Queue.zig b/src/link/Queue.zig
index caa2cc1898..b716800bae 100644
--- a/src/link/Queue.zig
+++ b/src/link/Queue.zig
@@ -121,7 +121,7 @@ pub fn enqueueZcu(
     link.doZcuTask(comp, tid, task);
 }
 
-pub fn finishPrelinkQueue(q: *Queue, comp: *Compilation) void {
+pub fn finishPrelinkQueue(q: *Queue, comp: *Compilation) Io.Cancelable!void {
     if (q.future != null) {
         q.prelink_queue.close(comp.io);
         return;
@@ -136,6 +136,7 @@ pub fn finishPrelinkQueue(q: *Queue, comp: *Compilation) void {
         } else |err| switch (err) {
             error.OutOfMemory => comp.link_diags.setAllocFailure(),
             error.LinkFailure => {},
+            error.Canceled => |e| return e,
         }
     }
 }