actions
/
toolkit
mirror of https://github.com/actions/toolkit
1
0
Fork 0
Code

Added abort controller for timer

pull/1140/head
Sankalp Kotewar 2022-08-03 11:11:30 +00:00
parent e5e69a3171
commit 970264135a
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
packages/cache/src/internal/downloadUtils.ts vendored
View File

@ -251,6 +251,7 @@ export async function downloadCacheStorageSDK(
try {
downloadProgress.startDisplayTimer()
const controller = new AbortController();
const timerController = new AbortController();
const abortSignal = controller.signal;
while (!downloadProgress.isDone()) {
const segmentStart =
@ -262,7 +263,6 @@ export async function downloadCacheStorageSDK(
)
downloadProgress.nextSegment(segmentSize)
const result = await Promise.race([client.downloadToBuffer(
segmentStart,
segmentSize,
@ -272,11 +272,14 @@ export async function downloadCacheStorageSDK(
onProgress: downloadProgress.onProgress()
}
),
timer.setTimeout(options.abortTimeInMs, 'timeout')]);
timer.setTimeout(options.abortTimeInMs, 'timeout',{ signal: timerController.signal })]);
if(result === 'timeout') {
controller.abort();
throw new Error("Download aborted, segment download timed out.");
} else {
timerController.abort();
core.debug("Download completely successfully, cancelling timer.")
}
fs.writeFileSync(fd, result)