1 | 'use strict';
|
---|
2 |
|
---|
3 | Object.defineProperty(exports, "__esModule", {
|
---|
4 | value: true
|
---|
5 | });
|
---|
6 | exports.default = timeout;
|
---|
7 |
|
---|
8 | var _initialParams = require('./internal/initialParams');
|
---|
9 |
|
---|
10 | var _initialParams2 = _interopRequireDefault(_initialParams);
|
---|
11 |
|
---|
12 | var _wrapAsync = require('./internal/wrapAsync');
|
---|
13 |
|
---|
14 | var _wrapAsync2 = _interopRequireDefault(_wrapAsync);
|
---|
15 |
|
---|
16 | function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
|
---|
17 |
|
---|
18 | /**
|
---|
19 | * Sets a time limit on an asynchronous function. If the function does not call
|
---|
20 | * its callback within the specified milliseconds, it will be called with a
|
---|
21 | * timeout error. The code property for the error object will be `'ETIMEDOUT'`.
|
---|
22 | *
|
---|
23 | * @name timeout
|
---|
24 | * @static
|
---|
25 | * @memberOf module:Utils
|
---|
26 | * @method
|
---|
27 | * @category Util
|
---|
28 | * @param {AsyncFunction} asyncFn - The async function to limit in time.
|
---|
29 | * @param {number} milliseconds - The specified time limit.
|
---|
30 | * @param {*} [info] - Any variable you want attached (`string`, `object`, etc)
|
---|
31 | * to timeout Error for more information..
|
---|
32 | * @returns {AsyncFunction} Returns a wrapped function that can be used with any
|
---|
33 | * of the control flow functions.
|
---|
34 | * Invoke this function with the same parameters as you would `asyncFunc`.
|
---|
35 | * @example
|
---|
36 | *
|
---|
37 | * function myFunction(foo, callback) {
|
---|
38 | * doAsyncTask(foo, function(err, data) {
|
---|
39 | * // handle errors
|
---|
40 | * if (err) return callback(err);
|
---|
41 | *
|
---|
42 | * // do some stuff ...
|
---|
43 | *
|
---|
44 | * // return processed data
|
---|
45 | * return callback(null, data);
|
---|
46 | * });
|
---|
47 | * }
|
---|
48 | *
|
---|
49 | * var wrapped = async.timeout(myFunction, 1000);
|
---|
50 | *
|
---|
51 | * // call `wrapped` as you would `myFunction`
|
---|
52 | * wrapped({ bar: 'bar' }, function(err, data) {
|
---|
53 | * // if `myFunction` takes < 1000 ms to execute, `err`
|
---|
54 | * // and `data` will have their expected values
|
---|
55 | *
|
---|
56 | * // else `err` will be an Error with the code 'ETIMEDOUT'
|
---|
57 | * });
|
---|
58 | */
|
---|
59 | function timeout(asyncFn, milliseconds, info) {
|
---|
60 | var fn = (0, _wrapAsync2.default)(asyncFn);
|
---|
61 |
|
---|
62 | return (0, _initialParams2.default)(function (args, callback) {
|
---|
63 | var timedOut = false;
|
---|
64 | var timer;
|
---|
65 |
|
---|
66 | function timeoutCallback() {
|
---|
67 | var name = asyncFn.name || 'anonymous';
|
---|
68 | var error = new Error('Callback function "' + name + '" timed out.');
|
---|
69 | error.code = 'ETIMEDOUT';
|
---|
70 | if (info) {
|
---|
71 | error.info = info;
|
---|
72 | }
|
---|
73 | timedOut = true;
|
---|
74 | callback(error);
|
---|
75 | }
|
---|
76 |
|
---|
77 | args.push(function () {
|
---|
78 | if (!timedOut) {
|
---|
79 | callback.apply(null, arguments);
|
---|
80 | clearTimeout(timer);
|
---|
81 | }
|
---|
82 | });
|
---|
83 |
|
---|
84 | // setup timer and call original function
|
---|
85 | timer = setTimeout(timeoutCallback, milliseconds);
|
---|
86 | fn.apply(null, args);
|
---|
87 | });
|
---|
88 | }
|
---|
89 | module.exports = exports['default']; |
---|