From adaad62fbd8701a864a134054e11a9777b3cbf38 Mon Sep 17 00:00:00 2001
From: Quentin McGaw <quentin.mcgaw@gmail.com>
Date: Fri, 12 Feb 2021 21:27:26 +0000
Subject: [PATCH] Feature: updater: no sleep for last DNS resolution

---
 internal/updater/resolver.go | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/internal/updater/resolver.go b/internal/updater/resolver.go
index 09990f1d..5c664315 100644
--- a/internal/updater/resolver.go
+++ b/internal/updater/resolver.go
@@ -86,7 +86,8 @@ func resolveRepeat(ctx context.Context, lookupIP lookupIPFunc, host string,
 	repetition int, timeBetween time.Duration) (ips []net.IP, err error) {
 	uniqueIPs := make(map[string]struct{})
 
-	for i := 0; i < repetition; i++ {
+	i := 0
+	for {
 		newIPs, err := lookupIP(ctx, host)
 		if err != nil {
 			return nil, err
@@ -95,6 +96,12 @@ func resolveRepeat(ctx context.Context, lookupIP lookupIPFunc, host string,
 			key := ip.String()
 			uniqueIPs[key] = struct{}{}
 		}
+
+		i++
+		if i == repetition {
+			break
+		}
+
 		timer := time.NewTimer(timeBetween)
 		select {
 		case <-timer.C: