mirror of
https://github.com/coredns/coredns.git
synced 2025-10-31 02:03:20 -04:00
995179a6c6
Add a test for this as well as it's annoying to point out in every code review. Fix all the import paths that are flagged by this new test. Fixes: #3634 Signed-off-by: Miek Gieben <miek@miek.nl>
37 lines
977 B
Go
37 lines
977 B
Go
// Package upstream abstracts a upstream lookups so that plugins can handle them in an unified way.
|
|
package upstream
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/coredns/coredns/core/dnsserver"
|
|
"github.com/coredns/coredns/plugin/pkg/nonwriter"
|
|
"github.com/coredns/coredns/request"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
// Upstream is used to resolve CNAME or other external targets via CoreDNS itself.
|
|
type Upstream struct{}
|
|
|
|
// New creates a new Upstream to resolve names using the coredns process.
|
|
func New() *Upstream { return &Upstream{} }
|
|
|
|
// Lookup routes lookups to our selves or forward to a remote.
|
|
func (u *Upstream) Lookup(ctx context.Context, state request.Request, name string, typ uint16) (*dns.Msg, error) {
|
|
server, ok := ctx.Value(dnsserver.Key{}).(*dnsserver.Server)
|
|
if !ok {
|
|
return nil, fmt.Errorf("no full server is running")
|
|
}
|
|
|
|
req := new(dns.Msg)
|
|
req.SetQuestion(name, typ)
|
|
|
|
nw := nonwriter.New(state.W)
|
|
|
|
server.ServeDNS(ctx, nw, req)
|
|
|
|
return nw.Msg, nil
|
|
}
|