1
0
Fork 0

Closes issue #1

pull/2/head
Ambrose Chua 2017-09-16 18:20:21 +08:00
parent 929d1d4f03
commit 0b931d4f6c
2 changed files with 8 additions and 8 deletions

View File

@ -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.

View File

@ -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}
})