feat: ✨ Add retry logic and error handeling

service is persistent and does not error out

src/index.ts

@@ -1,7 +1,7 @@
 import type { Record } from './schemas';
 import { updatedDNSRecordSchema, createdDNSRecord } from './schemas';
 import { recordSchema } from './schemas';
-import { log, vercelAxios } from './utils';
+import { log, retry, vercelAxios } from './utils';
 import env from './env';
 import { z } from 'zod';
 
@@ -60,14 +60,22 @@ async function createDNSRecord(name: string, value: string) {
 }
 
 async function main() {
-	const publicIp = await getPublicIp();
-	const records = await getDNSRecords();
+	try {
+		const publicIp = await retry(getPublicIp);
+		const records = await retry(getDNSRecords);
 
-	env.subdomains.forEach(async (subdomain) => {
-		const record = records?.find((record) => record.name === subdomain);
-		if (!record) await createDNSRecord(subdomain, publicIp);
-		if (record && record.value !== publicIp) await updateDNSRecord(record, publicIp);
-	});
+		env.subdomains.forEach(async (subdomain) => {
+			const record = records?.find((record) => record.name === subdomain);
+			if (!record) await retry(() => createDNSRecord(subdomain, publicIp));
+			if (record && record.value !== publicIp) await retry(() => updateDNSRecord(record, publicIp));
+		});
+	} catch (error) {
+		log({
+			status: 'ERROR',
+			function: 'main',
+			error: 'Failed after 3 attempts, will try again in 30 min',
+		});
+	}
 }
 
 main();

src/utils.test.ts

@@ -1,9 +1,9 @@
 /* eslint-disable no-console */
-import { retry } from 'utils';
+import { retry } from './utils';
 
 function getValue() {
 	return new Promise<string>((resolve, reject) =>
-		Math.random() < 0.1 ? resolve('SUCCESS') : reject('ERROR')
+		Math.random() < 0 ? resolve('SUCCESS') : reject('ERROR')
 	);
 }
 

src/utils.ts

@@ -18,17 +18,15 @@ export function log(props: ConsoleError) {
 	}
 }
 
-export function retry<T>(func: () => Promise<T>, attempt = 0): Promise<T> {
-	return new Promise(
-		(resolve) =>
-			setTimeout(async () => {
-				try {
-					const value = await func();
-					resolve(value);
-				} catch (error) {
-					console.error('ERROR:', func.name, 'attempt', attempt, '->', error);
-					resolve(retry(func, attempt + 1));
-				}
-			}, 1000 * 10) //10 seconds
-	);
+export function retry<T>(func: () => Promise<T>, maxAttempt = 3, currentAttempt = 1): Promise<T> {
+	return new Promise(async (resolve, reject) => {
+		try {
+			const value = await func();
+			resolve(value);
+		} catch (error) {
+			console.error('ERROR:', func.name, 'attempt', currentAttempt, '->', error);
+			if (currentAttempt >= maxAttempt) return reject(error);
+			resolve(retry(func, maxAttempt, currentAttempt + 1));
+		}
+	});
 }