12b2ff9740
* update docs * plugins: use plugin specific logging Hooking up pkg/log also changed NewWithPlugin to just take a string instead of a plugin.Handler as that is more flexible and for instance the Root "plugin" doesn't implement it fully. Same logging from the reload plugin: .:1043 2018/04/22 08:56:37 [INFO] CoreDNS-1.1.1 2018/04/22 08:56:37 [INFO] linux/amd64, go1.10.1, CoreDNS-1.1.1 linux/amd64, go1.10.1, 2018/04/22 08:56:37 [INFO] plugin/reload: Running configuration MD5 = ec4c9c55cd19759ea1c46b8c45742b06 2018/04/22 08:56:54 [INFO] Reloading 2018/04/22 08:56:54 [INFO] plugin/reload: Running configuration MD5 = 9e2bfdd85bdc9cceb740ba9c80f34c1a 2018/04/22 08:56:54 [INFO] Reloading complete * update docs * better doc
53 lines
1.1 KiB
Go
53 lines
1.1 KiB
Go
package proxy
|
|
|
|
import (
|
|
"github.com/coredns/coredns/core/dnsserver"
|
|
"github.com/coredns/coredns/plugin"
|
|
"github.com/coredns/coredns/plugin/metrics"
|
|
clog "github.com/coredns/coredns/plugin/pkg/log"
|
|
|
|
"github.com/mholt/caddy"
|
|
)
|
|
|
|
var log = clog.NewWithPlugin("proxy")
|
|
|
|
func init() {
|
|
caddy.RegisterPlugin("proxy", caddy.Plugin{
|
|
ServerType: "dns",
|
|
Action: setup,
|
|
})
|
|
}
|
|
|
|
func setup(c *caddy.Controller) error {
|
|
upstreams, err := NewStaticUpstreams(&c.Dispenser)
|
|
if err != nil {
|
|
return plugin.Error("proxy", err)
|
|
}
|
|
|
|
t := dnsserver.GetConfig(c).Handler("trace")
|
|
P := &Proxy{Trace: t}
|
|
dnsserver.GetConfig(c).AddPlugin(func(next plugin.Handler) plugin.Handler {
|
|
P.Next = next
|
|
P.Upstreams = &upstreams
|
|
return P
|
|
})
|
|
|
|
c.OnStartup(func() error {
|
|
once.Do(func() { metrics.MustRegister(c, RequestCount, RequestDuration) })
|
|
return nil
|
|
})
|
|
|
|
for i := range upstreams {
|
|
u := upstreams[i]
|
|
c.OnStartup(func() error {
|
|
return u.Exchanger().OnStartup(P)
|
|
})
|
|
c.OnShutdown(func() error {
|
|
return u.Exchanger().OnShutdown(P)
|
|
})
|
|
// Register shutdown handlers.
|
|
c.OnShutdown(u.Stop)
|
|
}
|
|
|
|
return nil
|
|
}
|