mirror of
https://github.com/OptimalBits/bull.git
synced 2026-07-08 11:41:05 +08:00
fix(worker): better closing when disconnected
This commit is contained in:
@@ -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
@@ -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);
|
||||
|
||||
Reference in New Issue
Block a user