resolve.go

// Copyright © by Jeff Foley 2017-2026. All rights reserved.
// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
// SPDX-License-Identifier: Apache-2.0

package resolve

import (
	"log"
	"runtime"
	"time"

	"github.com/owasp-amass/resolve/conn"
	"github.com/owasp-amass/resolve/pool"
	"github.com/owasp-amass/resolve/selectors"
	"github.com/owasp-amass/resolve/servers"
	"github.com/owasp-amass/resolve/types"
)

func NewServerPool(timeout time.Duration, logger *log.Logger, addrs ...string) *pool.Pool {
	var sel types.Selector

	if len(addrs) == 0 {
		sel = selectors.NewAuthoritative(timeout, servers.NewNameserver)
	} else {
		var servs []types.Nameserver
		for _, addrstr := range addrs {
			servs = append(servs, servers.NewNameserver(addrstr))
		}
		sel = selectors.NewRoundRobin(timeout, servs...)
	}

	conns := conn.New(runtime.NumCPU(), sel)
	return pool.New(0, sel, conns, logger)
}