fix(worker): better closing when disconnected

This commit is contained in:
Manuel Astudillo
2022-02-16 18:06:43 +08:00
committed by GitHub
parent ccd4537b9d
commit 41b940457b
2 changed files with 86 additions and 3 deletions
+84
View File
@@ -0,0 +1,84 @@
// Extracted from p-timeout https://github.com/sindresorhus/p-timeout
// as it is not commonjs compatible. This is vesion 5.0.2
'use strict';
class TimeoutError extends Error {
constructor(message) {
super(message);
this.name = 'TimeoutError';
}
}
module.exports.TimeoutError = TimeoutError;
module.exports.pTimeout = function pTimeout(
promise,
milliseconds,
fallback,
options
) {
let timer;
const cancelablePromise = new Promise((resolve, reject) => {
if (typeof milliseconds !== 'number' || Math.sign(milliseconds) !== 1) {
throw new TypeError(
`Expected \`milliseconds\` to be a positive number, got \`${milliseconds}\``
);
}
if (milliseconds === Number.POSITIVE_INFINITY) {
resolve(promise);
return;
}
options = {
customTimers: { setTimeout, clearTimeout },
...options
};
timer = options.customTimers.setTimeout.call(
undefined,
() => {
if (typeof fallback === 'function') {
try {
resolve(fallback());
} catch (error) {
reject(error);
}
return;
}
const message =
typeof fallback === 'string'
? fallback
: `Promise timed out after ${milliseconds} milliseconds`;
const timeoutError =
fallback instanceof Error ? fallback : new TimeoutError(message);
if (typeof promise.cancel === 'function') {
promise.cancel();
}
reject(timeoutError);
},
milliseconds
);
(async () => {
try {
resolve(await promise);
} catch (error) {
reject(error);
} finally {
options.customTimers.clearTimeout.call(undefined, timer);
}
})();
});
cancelablePromise['clear'] = () => {
clearTimeout(timer);
timer = undefined;
};
return cancelablePromise;
};
+2 -3
View File
@@ -16,7 +16,7 @@ const utils = require('./utils');
const TimerManager = require('./timer-manager');
const { promisify } = require('util');
const pTimeout = require('p-timeout');
const { pTimeout } = require('./p-timeout');
const semver = require('semver');
const debuglog = require('debuglog')('bull');
const uuid = require('uuid');
@@ -538,7 +538,7 @@ async function redisClientDisconnect(client) {
}
}),
500
).catch(() => {
).finally(() => {
client.once('error', _reject);
client.disconnect();
if (['connecting', 'reconnecting'].includes(client.status)) {
@@ -847,7 +847,6 @@ Queue.prototype.pause = function(isLocal, doNotWaitActive) {
this.bclient.connect()
);
}
return this.whenCurrentJobsFinished();
} else {
return scripts.pause(this, true);