52 lines
1.3 KiB
TypeScript
52 lines
1.3 KiB
TypeScript
import promiseRetry from '../retry';
|
|
|
|
/**
|
|
* 异步函数重试 Interface
|
|
* @param {number} [retries = 5] 重试次数
|
|
* @param {number} [timeout = 2000] 重试间隔时间
|
|
* @param {Function=} onError 重试错误回调
|
|
* @param {Function=} onRetrying 重试回调
|
|
* @param {Function=} onRetryFailed 重试失败回调
|
|
*/
|
|
interface IPromiseRetryDecoratorSettings {
|
|
retries?: number;
|
|
timeout?: number;
|
|
onError?: Function;
|
|
onRetrying?: Function;
|
|
onRetryFailed?: Function;
|
|
};
|
|
|
|
/**
|
|
* 装饰器函数：给异步函数增加重试
|
|
* @param {Object} settings 入参
|
|
* @returns {Function}
|
|
* @example
|
|
* class LocalStream {
|
|
* @promiseRetryDecorator({
|
|
* retries: 10,
|
|
* timeout: 3000,
|
|
* onRetryFailed: function(error) {
|
|
* }
|
|
* })
|
|
* async recoverCapture(options) {}
|
|
* }
|
|
*/
|
|
export default function promiseRetryDecorator(settings: IPromiseRetryDecoratorSettings) {
|
|
return function(target, name, descriptor) {
|
|
const { retries = 5, timeout = 2000, onError, onRetrying, onRetryFailed } = settings;
|
|
const oldFn = promiseRetry({
|
|
retryFunction: descriptor.value,
|
|
settings: { retries, timeout },
|
|
onError,
|
|
onRetrying,
|
|
onRetryFailed,
|
|
context: null,
|
|
});
|
|
|
|
descriptor.value = function(...args) {
|
|
return oldFn.apply(this, args);
|
|
};
|
|
return descriptor;
|
|
};
|
|
}
|