micro/resolver/dns/dns.go

// Package dns resolves names to dns records
package dns

import (
	"context"
	"net"
	"sync"
	"time"

	"github.com/unistack-org/micro/v3/resolver"
)

// Resolver is a DNS network resolve
type Resolver struct {
	// The resolver address to use
	Address    string
	goresolver *net.Resolver
	sync.RWMutex
}

// Resolve tries to resolve endpoint address
func (r *Resolver) Resolve(name string) ([]*resolver.Record, error) {
	host, port, err := net.SplitHostPort(name)
	if err != nil {
		host = name
		port = "8085"
	}

	if len(host) == 0 {
		host = "localhost"
	}

	if len(r.Address) == 0 {
		r.Address = "1.1.1.1:53"
	}

	// parsed an actual ip
	if v := net.ParseIP(host); v != nil {
		rec := &resolver.Record{Address: net.JoinHostPort(host, port)}
		return []*resolver.Record{rec}, nil
	}

	r.RLock()
	goresolver := r.goresolver
	r.RUnlock()

	if goresolver == nil {
		r.Lock()
		r.goresolver = &net.Resolver{
			Dial: func(ctx context.Context, network, address string) (net.Conn, error) {
				d := net.Dialer{
					Timeout: time.Millisecond * time.Duration(100),
				}
				return d.DialContext(ctx, "udp", r.Address)
			},
		}
		r.Unlock()
	}

	addrs, err := goresolver.LookupIP(context.TODO(), "ip", host)
	if err != nil {
		return nil, err
	}

	if len(addrs) == 0 {
		rec := &resolver.Record{Address: net.JoinHostPort(host, port)}
		return []*resolver.Record{rec}, nil
	}

	records := make([]*resolver.Record, 0, len(addrs))
	for _, addr := range addrs {
		records = append(records, &resolver.Record{
			Address: net.JoinHostPort(addr.String(), port),
		})
	}

	return records, nil
}