mirror of
https://github.com/hanzoai/agency.git
synced 2026-11-02 19:13:28 +00:00
86 lines
2.0 KiB
Plaintext
86 lines
2.0 KiB
Plaintext
// src/notifyManager.ts
|
|
import { systemSetTimeoutZero } from "./timeoutManager.js";
|
|
var defaultScheduler = systemSetTimeoutZero;
|
|
function createNotifyManager() {
|
|
let queue = [];
|
|
let transactions = 0;
|
|
let notifyFn = (callback) => {
|
|
callback();
|
|
};
|
|
let batchNotifyFn = (callback) => {
|
|
callback();
|
|
};
|
|
let scheduleFn = defaultScheduler;
|
|
const schedule = (callback) => {
|
|
if (transactions) {
|
|
queue.push(callback);
|
|
} else {
|
|
scheduleFn(() => {
|
|
notifyFn(callback);
|
|
});
|
|
}
|
|
};
|
|
const flush = () => {
|
|
const originalQueue = queue;
|
|
queue = [];
|
|
if (originalQueue.length) {
|
|
scheduleFn(() => {
|
|
batchNotifyFn(() => {
|
|
originalQueue.forEach((callback) => {
|
|
notifyFn(callback);
|
|
});
|
|
});
|
|
});
|
|
}
|
|
};
|
|
return {
|
|
batch: (callback) => {
|
|
let result;
|
|
transactions++;
|
|
try {
|
|
result = callback();
|
|
} finally {
|
|
transactions--;
|
|
if (!transactions) {
|
|
flush();
|
|
}
|
|
}
|
|
return result;
|
|
},
|
|
/**
|
|
* All calls to the wrapped function will be batched.
|
|
*/
|
|
batchCalls: (callback) => {
|
|
return (...args) => {
|
|
schedule(() => {
|
|
callback(...args);
|
|
});
|
|
};
|
|
},
|
|
schedule,
|
|
/**
|
|
* Use this method to set a custom notify function.
|
|
* This can be used to for example wrap notifications with `React.act` while running tests.
|
|
*/
|
|
setNotifyFunction: (fn) => {
|
|
notifyFn = fn;
|
|
},
|
|
/**
|
|
* Use this method to set a custom function to batch notifications together into a single tick.
|
|
* By default React Query will use the batch function provided by ReactDOM or React Native.
|
|
*/
|
|
setBatchNotifyFunction: (fn) => {
|
|
batchNotifyFn = fn;
|
|
},
|
|
setScheduler: (fn) => {
|
|
scheduleFn = fn;
|
|
}
|
|
};
|
|
}
|
|
var notifyManager = createNotifyManager();
|
|
export {
|
|
createNotifyManager,
|
|
defaultScheduler,
|
|
notifyManager
|
|
};
|
|
//# sourceMappingURL=notifyManager.js.map |