actions/magic-nix-cache-action
1
0
Fork 0
mirror of https://github.com/DeterminateSystems/magic-nix-cache-action.git synced 2024-12-26 17:30:34 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #10 from DeterminateSystems/retry

Browse source 
Retry
This commit is contained in:
Graham Christensen 2023-06-27 13:08:43 -04:00 committed by GitHub
commit 8a218f9e26
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 55 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
dist/index.js generated vendored
View file

@ -6,6 +6,7 @@ import { spawn } from 'node:child_process';
import { openSync, writeSync, close, createWriteStream } from 'node:fs';
import { pipeline } from 'node:stream/promises';
import { setTimeout as setTimeout$1 } from 'timers/promises';
import { promisify as promisify$1, inspect } from 'node:util';
import require$$0 from 'os';
import require$$1 from 'fs';
import crypto from 'crypto';
@ -27,7 +28,6 @@ import crypto$1 from 'node:crypto';
import require$$0$4 from 'buffer';
import require$$0$3 from 'stream';
import require$$1$3 from 'zlib';
import { promisify as promisify$1, inspect } from 'node:util';
import net from 'node:net';
import { checkServerIdentity } from 'node:tls';
import https$4 from 'node:https';
@ -12103,6 +12103,19 @@ var got$1 = got;
// Main
const ENV_CACHE_DAEMONDIR = 'MAGIC_NIX_CACHE_DAEMONDIR';
const gotClient = got$1.extend({
retry: {
limit: 5,
methods: ['POST', 'GET', 'PUT', 'HEAD', 'DELETE', 'OPTIONS', 'TRACE'],
},
hooks: {
beforeRetry: [
(error, retryCount) => {
coreExports.info(`Retrying after error ${error.code}, retry #: ${retryCount}`);
}
],
},
});
function getCacherUrl() {
const runnerArch = process.env.RUNNER_ARCH;
const runnerOs = process.env.RUNNER_OS;
@ -12132,7 +12145,7 @@ async function fetchAutoCacher(destination) {
});
const binary_url = getCacherUrl();
coreExports.debug(`Fetching the Magic Nix Cache from ${binary_url}`);
return pipeline(got$1.stream(binary_url), stream);
return pipeline(gotClient.stream(binary_url), stream);
}
async function setUpAutoCache() {
const tmpdir = process.env['RUNNER_TEMP'] || os$2.tmpdir();
@ -12204,9 +12217,18 @@ async function notifyAutoCache() {
if (!daemonDir) {
return;
}
const res = await got$1.post(`http://${coreExports.getInput('listen')}/api/workflow-start`).json();
try {
coreExports.debug(`Indicating workflow start`);
const res = await gotClient.post(`http://${coreExports.getInput('listen')}/api/workflow-start`).json();
coreExports.debug(`back from post`);
coreExports.debug(res);
}
catch (e) {
coreExports.info(`Error marking the workflow as started:`);
coreExports.info(inspect(e));
coreExports.info(`Magic Nix Cache may not be running for this workflow.`);
}
}
async function tearDownAutoCache() {
const daemonDir = process.env[ENV_CACHE_DAEMONDIR];
if (!daemonDir) {
@ -12227,7 +12249,7 @@ async function tearDownAutoCache() {
});
try {
coreExports.debug(`about to post to localhost`);
const res = await got$1.post(`http://${coreExports.getInput('listen')}/api/workflow-finish`).json();
const res = await gotClient.post(`http://${coreExports.getInput('listen')}/api/workflow-finish`).json();
coreExports.debug(`back from post`);
coreExports.debug(res);
}

30
src/index.ts
View file

@ -7,6 +7,7 @@ import { spawn } from 'node:child_process';
import { createWriteStream, openSync, writeSync, close } from 'node:fs';
import { pipeline } from 'node:stream/promises';
import { setTimeout } from 'timers/promises';
import { inspect } from 'node:util';
import * as core from '@actions/core';
import { Tail } from 'tail';
@ -14,6 +15,21 @@ import got from "got";
const ENV_CACHE_DAEMONDIR = 'MAGIC_NIX_CACHE_DAEMONDIR';
const gotClient = got.extend({
retry: {
limit: 5,
methods: [ 'POST', 'GET', 'PUT', 'HEAD', 'DELETE', 'OPTIONS', 'TRACE' ],
},
hooks: {
beforeRetry: [
(error, retryCount) => {
core.info(`Retrying after error ${error.code}, retry #: ${retryCount}`);
}
],
},
});
function getCacherUrl() : string {
const runnerArch = process.env.RUNNER_ARCH;
const runnerOs = process.env.RUNNER_OS;
@ -53,7 +69,7 @@ async function fetchAutoCacher(destination: string) {
core.debug(`Fetching the Magic Nix Cache from ${binary_url}`);
return pipeline(
got.stream(binary_url),
gotClient.stream(binary_url),
stream
);
}
@ -141,8 +157,16 @@ async function notifyAutoCache() {
return;
}
const res: any = await got.post(`http://${core.getInput('listen')}/api/workflow-start`).json();
try {
core.debug(`Indicating workflow start`);
const res: any = await gotClient.post(`http://${core.getInput('listen')}/api/workflow-start`).json();
core.debug(`back from post`);
core.debug(res);
} catch (e) {
core.info(`Error marking the workflow as started:`);
core.info(inspect(e));
core.info(`Magic Nix Cache may not be running for this workflow.`);
}
}
async function tearDownAutoCache() {
@ -170,7 +194,7 @@ async function tearDownAutoCache() {
try {
core.debug(`about to post to localhost`);
const res: any = await got.post(`http://${core.getInput('listen')}/api/workflow-finish`).json();
const res: any = await gotClient.post(`http://${core.getInput('listen')}/api/workflow-finish`).json();
core.debug(`back from post`);
core.debug(res);
} finally {