-
Notifications
You must be signed in to change notification settings - Fork 11
/
driftless.js
87 lines (76 loc) · 2.07 KB
/
driftless.js
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
import present from 'present';
export const DEFAULT_THRESHOLD_MS = 1;
export const DEFAULT_AGGRESSION = 1.1;
export const DEFAULT_NOW_FN = (...args) => present(...args);
export const DEFAULT_SET_TIMEOUT = (...args) => setTimeout(...args);
export const DEFAULT_CLEAR_TIMEOUT = (...args) => clearTimeout(...args);
const timerHandles = {};
let nextId = 0;
function tryDriftless(id, opts) {
const {
atMs,
fn,
thresholdMs = DEFAULT_THRESHOLD_MS,
aggression = DEFAULT_AGGRESSION,
customNow = DEFAULT_NOW_FN,
customSetTimeout = DEFAULT_SET_TIMEOUT,
} = opts;
const delayMs = atMs - customNow();
const handle = delayMs > thresholdMs
? (
customSetTimeout(() => {
tryDriftless.apply(this, arguments); // eslint-disable-line prefer-rest-params
}, delayMs / aggression)
)
: (
customSetTimeout(() => {
// Call the function using setTimeout to ensure asynchrony
fn();
}, 0)
);
timerHandles[id] = handle;
}
export function setDriftless(...args) {
return setDriftless.setDriftlessSpyable(...args);
}
// Separate function for testing
setDriftless.setDriftlessSpyable = (opts) => {
const id = nextId;
nextId += 1;
tryDriftless(id, opts);
return id;
};
export function clearDriftless(id, opts = {}) {
const {
customClearTimeout = DEFAULT_CLEAR_TIMEOUT,
} = opts;
customClearTimeout(timerHandles[id]);
}
function castToFn(fn) {
return typeof fn === 'function'
? fn
: new Function(fn); // eslint-disable-line no-new-func
}
export function setDriftlessTimeout(fn, delayMs, ...params) {
const callFn = castToFn(fn);
return setDriftless({
atMs: DEFAULT_NOW_FN() + delayMs,
fn(...args) {
return callFn.call(this, ...args, ...params);
},
});
}
export function setDriftlessInterval(fn, delayMs, ...params) {
const callFn = castToFn(fn);
let id;
const opts = {
atMs: DEFAULT_NOW_FN() + delayMs,
fn(...args) {
opts.atMs += delayMs;
tryDriftless(id, opts);
return callFn.call(this, ...args, ...params);
},
};
id = setDriftless(opts);
return id;
}