Skip to content

Commit

Permalink
Refactor kill() again (#280)
Browse files Browse the repository at this point in the history
  • Loading branch information
ehmicky authored and sindresorhus committed Jun 12, 2019
1 parent 55337f7 commit 1b38dc9
Show file tree
Hide file tree
Showing 2 changed files with 32 additions and 10 deletions.
30 changes: 20 additions & 10 deletions index.js
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@ const onExit = require('signal-exit');
const stdio = require('./lib/stdio');

const TEN_MEGABYTES = 1000 * 1000 * 10;
const DEFAULT_FORCE_KILL_TIMEOUT = 1000 * 5;

const SPACES_REGEXP = / +/g;

Expand Down Expand Up @@ -224,18 +225,27 @@ function setKillTimeout(kill, signal, options, killResult) {
return;
}

const forceKillAfter = Number.isInteger(options.forceKillAfter) ?
options.forceKillAfter :
5000;
setTimeout(() => kill('SIGKILL'), forceKillAfter).unref();
const timeout = getForceKillAfterTimeout(options);
setTimeout(() => {
kill('SIGKILL');
}, timeout).unref();
}

function shouldForceKill(signal, options, killResult) {
return ((typeof signal === 'string' &&
signal.toUpperCase() === 'SIGTERM') ||
signal === os.constants.signals.SIGTERM) &&
options.forceKill !== false &&
killResult;
function shouldForceKill(signal, {forceKill}, killResult) {
return isSigterm(signal) && forceKill !== false && killResult;
}

function isSigterm(signal) {
return signal === os.constants.signals.SIGTERM ||
(typeof signal === 'string' && signal.toUpperCase() === 'SIGTERM');
}

function getForceKillAfterTimeout({forceKillAfter = DEFAULT_FORCE_KILL_TIMEOUT}) {
if (!Number.isInteger(forceKillAfter) || forceKillAfter < 0) {
throw new TypeError(`Expected the \`forceKillAfter\` option to be a non-negative integer, got \`${forceKillAfter}\` (${typeof forceKillAfter})`);
}

return forceKillAfter;
}

const execa = (file, args, options) => {
Expand Down
12 changes: 12 additions & 0 deletions test.js
Original file line number Diff line number Diff line change
Expand Up @@ -175,6 +175,18 @@ if (process.platform !== 'win32') {
const {signal} = await t.throwsAsync(subprocess);
t.is(signal, 'SIGKILL');
});

test('.kill() `forceKillAfter` should not be a float', t => {
t.throws(() => {
execa('noop').kill('SIGTERM', {forceKillAfter: 0.5});
}, {instanceOf: TypeError, message: /non-negative integer/});
});

test('.kill() `forceKillAfter` should not be negative', t => {
t.throws(() => {
execa('noop').kill('SIGTERM', {forceKillAfter: -1});
}, {instanceOf: TypeError, message: /non-negative integer/});
});
}

test('stripFinalNewline: true', async t => {
Expand Down

0 comments on commit 1b38dc9

Please sign in to comment.