Closes issue #1
parent
929d1d4f03
commit
0b931d4f6c
10
alias.go
10
alias.go
|
@ -1,22 +1,22 @@
|
|||
package alias
|
||||
|
||||
import (
|
||||
"github.com/coredns/coredns/middleware"
|
||||
"github.com/coredns/coredns/plugin"
|
||||
|
||||
"github.com/miekg/dns"
|
||||
|
||||
"golang.org/x/net/context"
|
||||
)
|
||||
|
||||
// Rewrite is middleware to rewrite requests internally before being handled.
|
||||
// Rewrite is plugin to rewrite requests internally before being handled.
|
||||
type Alias struct {
|
||||
Next middleware.Handler
|
||||
Next plugin.Handler
|
||||
}
|
||||
|
||||
// ServeDNS implements the middleware.Handler interface.
|
||||
// ServeDNS implements the plugin.Handler interface.
|
||||
func (al Alias) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) {
|
||||
mw := NewResponseModifier(w)
|
||||
return middleware.NextOrFailure(al.Name(), al.Next, ctx, mw, r)
|
||||
return plugin.NextOrFailure(al.Name(), al.Next, ctx, mw, r)
|
||||
}
|
||||
|
||||
// Name implements the Handler interface.
|
||||
|
|
6
setup.go
6
setup.go
|
@ -2,7 +2,7 @@ package alias
|
|||
|
||||
import (
|
||||
"github.com/coredns/coredns/core/dnsserver"
|
||||
"github.com/coredns/coredns/middleware"
|
||||
"github.com/coredns/coredns/plugin"
|
||||
|
||||
"github.com/mholt/caddy"
|
||||
)
|
||||
|
@ -17,10 +17,10 @@ func init() {
|
|||
func setup(c *caddy.Controller) error {
|
||||
c.Next()
|
||||
if c.NextArg() {
|
||||
return middleware.Error("alias", c.ArgErr())
|
||||
return plugin.Error("alias", c.ArgErr())
|
||||
}
|
||||
|
||||
dnsserver.GetConfig(c).AddMiddleware(func(next middleware.Handler) middleware.Handler {
|
||||
dnsserver.GetConfig(c).AddPlugin(func(next plugin.Handler) plugin.Handler {
|
||||
return Alias{Next: next}
|
||||
})
|
||||
|
||||
|
|
Loading…
Reference in New Issue