@sha1n/about-time 0.0.5 → 0.0.6

package/README.md

@@ -12,6 +12,7 @@ A collection of essential time related utilities.
 - [About-Time](#about-time)
   - [Install](#install)
   - [Delay](#delay)
+  - [WithTimeout](#withtimeout)
   - [Sleep](#sleep)
   - [Stopwatch](#stopwatch)
   - [Until / Eventually](#until--eventually)
@@ -35,6 +36,12 @@ npm i @sha1n/about-time
 await delay(action, 10, TimeUnit.Milliseconds);
 ```
 
+## WithTimeout
+```ts
+// Execute a function and guards it with a specified timeout
+await withTimeout(action, 10, TimeUnit.Milliseconds);
+```
+
 ## Sleep
 ```ts
 // Pause execution for a specified amount of time

package/dist/lib/utilities.js

@@ -1,7 +1,12 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.eventually = exports.until = exports.stopwatch = exports.delay = exports.sleep = void 0;
+exports.eventually = exports.until = exports.stopwatch = exports.delay = exports.sleep = exports.withTimeout = void 0;
 const timeunit_1 = require("./timeunit");
+class TimeoutError extends Error {
+    constructor(message) {
+        super(message || 'Timeout');
+    }
+}
 /**
  * Zzzz...
  *
@@ -26,7 +31,10 @@ exports.sleep = sleep;
 function delay(action, time, units) {
     const delayMs = (0, timeunit_1.toMilliseconds)(time, units);
     return new Promise((resolve, reject) => {
-        setTimeout(() => Promise.resolve(action()).then(resolve, reject), delayMs);
+        const timer = setTimeout(() => Promise.resolve(action()).then(resolve, reject), delayMs);
+        process.on('beforeExit', () => {
+            clearTimeout(timer);
+        });
     });
 }
 exports.delay = delay;
@@ -56,7 +64,7 @@ async function until(condition, opts) {
         const handle = setInterval(() => {
             if (Date.now() > deadline) {
                 clearInterval(handle);
-                reject(new Error('Timeout'));
+                reject(new TimeoutError());
             }
             try {
                 if (condition()) {
@@ -77,3 +85,35 @@ exports.until = until;
  */
 const eventually = until;
 exports.eventually = eventually;
+/**
+ * Executes an action with a specified timeout. If the action times out, rejects with TimeoutError.
+ *
+ * @param action an action to execute with timeout
+ * @param timeout the timeout to set for the action
+ * @param units the time units
+ * @returns the action result
+ */
+async function withTimeout(action, timeout, units) {
+    const promisedAction = new Promise((resolve, reject) => {
+        try {
+            resolve(action());
+        }
+        catch (e) {
+            reject(e);
+        }
+    });
+    const race = new Promise((resolve, reject) => {
+        const timer = setTimeout(() => {
+            reject(new TimeoutError());
+        }, (0, timeunit_1.toMilliseconds)(timeout, units));
+        return Promise.resolve(promisedAction).then(r => {
+            clearTimeout(timer);
+            resolve(r);
+        }, err => {
+            clearTimeout(timer);
+            reject(err);
+        });
+    });
+    return race;
+}
+exports.withTimeout = withTimeout;

package/dist/types/lib/utilities.d.ts

@@ -37,4 +37,13 @@ declare function until(condition: () => boolean, opts?: {
  * Alias to `until`
  */
 declare const eventually: typeof until;
-export { sleep, delay, stopwatch, until, eventually };
+/**
+ * Executes an action with a specified timeout. If the action times out, rejects with TimeoutError.
+ *
+ * @param action an action to execute with timeout
+ * @param timeout the timeout to set for the action
+ * @param units the time units
+ * @returns the action result
+ */
+declare function withTimeout<T>(action: () => T | Promise<T>, timeout: number, units?: TimeUnit): Promise<T>;
+export { withTimeout, sleep, delay, stopwatch, until, eventually };

package/lib/utilities.ts

@@ -1,5 +1,11 @@
 import { TimeUnit, toMilliseconds } from './timeunit';
 
+class TimeoutError extends Error {
+  constructor(message?: string) {
+    super(message || 'Timeout');
+  }
+}
+
 /**
  * Zzzz...
  *
@@ -24,7 +30,10 @@ function sleep(time: number, units?: TimeUnit): Promise<void> {
 function delay<T>(action: () => T | Promise<T>, time: number, units?: TimeUnit): Promise<T> {
   const delayMs = toMilliseconds(time, units);
   return new Promise((resolve, reject) => {
-    setTimeout(() => Promise.resolve(action()).then(resolve, reject), delayMs);
+    const timer = setTimeout(() => Promise.resolve(action()).then(resolve, reject), delayMs);
+    process.on('beforeExit', () => {
+      clearTimeout(timer);
+    });
   });
 }
 
@@ -59,7 +68,7 @@ async function until(
     const handle = setInterval(() => {
       if (Date.now() > deadline) {
         clearInterval(handle);
-        reject(new Error('Timeout'));
+        reject(new TimeoutError());
       }
 
       try {
@@ -80,4 +89,41 @@ async function until(
  */
 const eventually = until;
 
-export { sleep, delay, stopwatch, until, eventually };
+/**
+ * Executes an action with a specified timeout. If the action times out, rejects with TimeoutError.
+ *
+ * @param action an action to execute with timeout
+ * @param timeout the timeout to set for the action
+ * @param units the time units
+ * @returns the action result
+ */
+async function withTimeout<T>(action: () => T | Promise<T>, timeout: number, units?: TimeUnit): Promise<T> {
+  const promisedAction = new Promise<T>((resolve, reject) => {
+    try {
+      resolve(action());
+    } catch (e) {
+      reject(e);
+    }
+  });
+
+  const race = new Promise<T>((resolve, reject) => {
+    const timer = setTimeout(() => {
+      reject(new TimeoutError());
+    }, toMilliseconds(timeout, units));
+
+    return Promise.resolve(promisedAction).then(
+      r => {
+        clearTimeout(timer);
+        resolve(r);
+      },
+      err => {
+        clearTimeout(timer);
+        reject(err);
+      }
+    );
+  });
+
+  return race;
+}
+
+export { withTimeout, sleep, delay, stopwatch, until, eventually };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sha1n/about-time",
-  "version": "0.0.5",
+  "version": "0.0.6",
   "description": "A set of essential time related utilities",
   "repository": "https://github.com/sha1n/about-time",
   "author": "Shai Nagar",