diff options
Diffstat (limited to 'middleware/autopath/setup.go')
| -rw-r--r-- | middleware/autopath/setup.go | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/middleware/autopath/setup.go b/middleware/autopath/setup.go index 1855dd440..f9b12d98e 100644 --- a/middleware/autopath/setup.go +++ b/middleware/autopath/setup.go @@ -5,6 +5,7 @@ import ( "github.com/coredns/coredns/core/dnsserver" "github.com/coredns/coredns/middleware" + "github.com/coredns/coredns/middleware/erratic" "github.com/coredns/coredns/middleware/kubernetes" "github.com/mholt/caddy" @@ -32,8 +33,11 @@ func setup(c *caddy.Controller) error { if m == nil { return nil } - if k, ok := m.(kubernetes.Kubernetes); ok { - ap.searchFunc = k.AutoPath + if x, ok := m.(kubernetes.Kubernetes); ok { + ap.searchFunc = x.AutoPath + } + if x, ok := m.(*erratic.Erratic); ok { + ap.searchFunc = x.AutoPath } return nil }) @@ -50,6 +54,7 @@ func setup(c *caddy.Controller) error { // need to register themselves with dnsserver.RegisterHandler. var allowedMiddleware = map[string]bool{ "@kubernetes": true, + "@erratic": true, } func autoPathParse(c *caddy.Controller) (*AutoPath, string, error) { |