let nextHandle = 1;
|
// The promise needs to be created lazily otherwise it won't be patched by Zones
|
let resolved: Promise<any>;
|
const activeHandles: { [key: number]: any } = {};
|
|
/**
|
* Finds the handle in the list of active handles, and removes it.
|
* Returns `true` if found, `false` otherwise. Used both to clear
|
* Immediate scheduled tasks, and to identify if a task should be scheduled.
|
*/
|
function findAndClearHandle(handle: number): boolean {
|
if (handle in activeHandles) {
|
delete activeHandles[handle];
|
return true;
|
}
|
return false;
|
}
|
|
/**
|
* Helper functions to schedule and unschedule microtasks.
|
*/
|
export const Immediate = {
|
setImmediate(cb: () => void): number {
|
const handle = nextHandle++;
|
activeHandles[handle] = true;
|
if (!resolved) {
|
resolved = Promise.resolve();
|
}
|
resolved.then(() => findAndClearHandle(handle) && cb());
|
return handle;
|
},
|
|
clearImmediate(handle: number): void {
|
findAndClearHandle(handle);
|
},
|
};
|
|
/**
|
* Used for internal testing purposes only. Do not export from library.
|
*/
|
export const TestTools = {
|
pending() {
|
return Object.keys(activeHandles).length;
|
}
|
};
|
