@certik/skynet 0.20.0 → 0.20.2

package/CHANGELOG.md

@@ -1,5 +1,15 @@
 # Changelog
 
+## 0.20.2
+
+- Make `options` in `withRetry` function optional
+
+## 0.20.1
+
+- Update dynamodb `scanWholeTable` types
+- Add `logger` in log library
+- Add `withRetry` in availability library
+
 ## 0.20.0
 
 - Fix databricks with bun (extra step needed for projects using the lib):

package/availability.ts

@@ -46,3 +46,29 @@ export async function exponentialRetry<T>(
 
   return result;
 }
+
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+export function withRetry<F extends (...args: any[]) => any>(
+  func: F,
+  options?: { maxRetry?: number; initialDuration?: number; growFactor?: number },
+) {
+  let retries = options?.maxRetry || 3;
+  let duration = options?.initialDuration || 500;
+  const growFactorFinal = options?.growFactor || 2;
+
+  return async (...args: Parameters<F>): Promise<ReturnType<F>> => {
+    do {
+      try {
+        return await func(...args);
+      } catch (error) {
+        retries = retries - 1;
+        if (retries === 0) {
+          throw error;
+        }
+        await wait(duration);
+        duration = duration * growFactorFinal;
+      }
+    } while (retries >= 0);
+    throw new Error("unreachable");
+  };
+}

package/dynamodb.ts

@@ -56,7 +56,7 @@ function getDocClient(forceNew = false) {
   return _docClient;
 }
 
-async function scanWholeTable(options: ScanCommandInput) {
+async function scanWholeTable<T>(options: ScanCommandInput) {
   const dynamodb = getDocClient();
 
   let items: ScanCommandOutput["Items"] = [];
@@ -82,7 +82,7 @@ async function scanWholeTable(options: ScanCommandInput) {
   scannedCount += data.ScannedCount || 0;
 
   return {
-    Items: items,
+    Items: items as T[],
     Count: count,
     ScannedCount: scannedCount,
   };

package/log.ts

@@ -42,4 +42,13 @@ const inline = {
   },
 };
 
-export { print, getLine, inline };
+const logger = {
+  log: function (...args: unknown[]) {
+    console.log(`[${timestamp()}]`, ...args);
+  },
+  error: function (...args: unknown[]) {
+    console.error(`[${timestamp()}]`, ...args);
+  },
+};
+
+export { print, getLine, inline, logger };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@certik/skynet",
-  "version": "0.20.0",
+  "version": "0.20.2",
   "description": "Skynet Shared JS library",
   "type": "module",
   "main": "index.ts",