2016-08-19 17:14:17 -07:00
|
|
|
package proxy
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"github.com/miekg/coredns/core/dnsserver"
|
2016-09-10 09:16:25 +01:00
|
|
|
"github.com/miekg/coredns/middleware"
|
2016-08-19 17:14:17 -07:00
|
|
|
|
|
|
|
|
"github.com/mholt/caddy"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
func init() {
|
|
|
|
|
caddy.RegisterPlugin("proxy", caddy.Plugin{
|
|
|
|
|
ServerType: "dns",
|
|
|
|
|
Action: setup,
|
|
|
|
|
})
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func setup(c *caddy.Controller) error {
|
2016-09-16 23:49:35 -07:00
|
|
|
upstreams, err := NewStaticUpstreams(&c.Dispenser)
|
2016-08-19 17:14:17 -07:00
|
|
|
if err != nil {
|
2016-09-10 09:16:25 +01:00
|
|
|
return middleware.Error("proxy", err)
|
2016-08-19 17:14:17 -07:00
|
|
|
}
|
|
|
|
|
dnsserver.GetConfig(c).AddMiddleware(func(next dnsserver.Handler) dnsserver.Handler {
|
|
|
|
|
return Proxy{Next: next, Client: Clients(), Upstreams: upstreams}
|
|
|
|
|
})
|
|
|
|
|
|
|
|
|
|
return nil
|
|
|
|
|
}
|