Fix error handling when query fails (coredns#2255)

DataDog · Mar 4, 2019 · 555ddea · 555ddea
1 parent 107a194
commit 555ddea
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/plugin/loop/loop.go b/plugin/loop/loop.go
@@ -77,6 +77,12 @@ func (l *Loop) inc() {
 	l.i++
 }
 
+func (l *Loop) reset() {
+	l.Lock()
+	defer l.Unlock()
+	l.i = 0
+}
+
 func (l *Loop) setDisabled() {
 	l.Lock()
 	defer l.Unlock()

diff --git a/plugin/loop/setup.go b/plugin/loop/setup.go
@@ -42,6 +42,7 @@ func setup(c *caddy.Controller) error {
 
 			for time.Now().Before(deadline) {
 				if _, err := l.exchange(addr); err != nil {
+					l.reset()
 					time.Sleep(1 * time.Second)
 					continue
 				}