mirror of
https://github.com/coredns/coredns.git
synced 2025-10-28 00:34:24 -04:00
31 lines
574 B
Go
31 lines
574 B
Go
|
package proxy
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync/atomic"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/coredns/coredns/plugin/pkg/healthcheck"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Default CheckDown functions for use in the proxy plugin.
|
||
|
|
var checkDownFunc = func(upstream *staticUpstream) healthcheck.UpstreamHostDownFunc {
|
||
|
|
return func(uh *healthcheck.UpstreamHost) bool {
|
||
|
|
|
||
|
|
down := false
|
||
|
|
|
||
|
|
uh.Lock()
|
||
|
|
until := uh.OkUntil
|
||
|
|
uh.Unlock()
|
||
|
|
|
||
|
|
if !until.IsZero() && time.Now().After(until) {
|
||
|
|
down = true
|
||
|
|
}
|
||
|
|
|
||
|
|
fails := atomic.LoadInt32(&uh.Fails)
|
||
|
|
if fails >= upstream.MaxFails && upstream.MaxFails != 0 {
|
||
|
|
down = true
|
||
|
|
}
|
||
|
|
return down
|
||
|
|
}
|
||
|
|
}
|