优化

‘liusuyi’

2023-10-21 ae92b6823499cfab77201b4033922fa373fb37a3

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
'use strict';
const mimicFn = require('mimic-fn');
 
const calledFunctions = new WeakMap();
 
const onetime = (function_, options = {}) => {
    if (typeof function_ !== 'function') {
        throw new TypeError('Expected a function');
    }
 
    let returnValue;
    let callCount = 0;
    const functionName = function_.displayName || function_.name || '<anonymous>';
 
    const onetime = function (...arguments_) {
        calledFunctions.set(onetime, ++callCount);
 
        if (callCount === 1) {
            returnValue = function_.apply(this, arguments_);
            function_ = null;
        } else if (options.throw === true) {
            throw new Error(`Function \`${functionName}\` can only be called once`);
        }
 
        return returnValue;
    };
 
    mimicFn(onetime, function_);
    calledFunctions.set(onetime, callCount);
 
    return onetime;
};
 
module.exports = onetime;
// TODO: Remove this for the next major release
module.exports.default = onetime;
 
module.exports.callCount = function_ => {
    if (!calledFunctions.has(function_)) {
        throw new Error(`The given function \`${function_.name}\` is not wrapped by the \`onetime\` package`);
    }
 
    return calledFunctions.get(function_);
};