ginuerzh
618cc7fd9c
tokenTransport now delegates CloseIdleConnections to the underlying *http.Transport so http.Client.CloseIdleConnections() works through the wrapper. Three plugin Close() methods updated to unwrap via HTTPClientTransport before calling CloseIdleConnections. RoundTrip now short-circuits when caller already set Authorization, avoiding an unnecessary req.Clone allocation on the pass-through path.
76 lines
2.0 KiB
Go
76 lines
2.0 KiB
Go
package resolver
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"net"
|
|
"strings"
|
|
|
|
"github.com/go-gost/core/logger"
|
|
"github.com/go-gost/core/resolver"
|
|
"github.com/go-gost/x/config"
|
|
"github.com/go-gost/x/internal/plugin"
|
|
"github.com/go-gost/x/registry"
|
|
xresolver "github.com/go-gost/x/resolver"
|
|
resolver_plugin "github.com/go-gost/x/resolver/plugin"
|
|
)
|
|
|
|
// ParseResolver converts a ResolverConfig into a resolver.Resolver. It
|
|
// supports plugin backends (HTTP or gRPC) and inline nameserver definitions
|
|
// with optional chain-based upstream routing, TTL, timeout, and
|
|
// prefer/async/only settings. Returns nil with no error when cfg is nil.
|
|
func ParseResolver(cfg *config.ResolverConfig) (resolver.Resolver, error) {
|
|
if cfg == nil {
|
|
return nil, nil
|
|
}
|
|
|
|
if cfg.Plugin != nil {
|
|
var tlsCfg *tls.Config
|
|
if cfg.Plugin.TLS != nil {
|
|
tlsCfg = &tls.Config{
|
|
ServerName: cfg.Plugin.TLS.ServerName,
|
|
InsecureSkipVerify: !cfg.Plugin.TLS.Secure,
|
|
}
|
|
}
|
|
switch strings.ToLower(cfg.Plugin.Type) {
|
|
case "http":
|
|
return resolver_plugin.NewHTTPPlugin(
|
|
cfg.Name, cfg.Plugin.Addr,
|
|
plugin.TokenOption(cfg.Plugin.Token),
|
|
plugin.TLSConfigOption(tlsCfg),
|
|
plugin.TimeoutOption(cfg.Plugin.Timeout),
|
|
), nil
|
|
default:
|
|
return resolver_plugin.NewGRPCPlugin(
|
|
cfg.Name, cfg.Plugin.Addr,
|
|
plugin.TokenOption(cfg.Plugin.Token),
|
|
plugin.TLSConfigOption(tlsCfg),
|
|
)
|
|
}
|
|
}
|
|
|
|
var nameservers []xresolver.NameServer
|
|
for _, server := range cfg.Nameservers {
|
|
nameservers = append(nameservers, xresolver.NameServer{
|
|
Addr: server.Addr,
|
|
Chain: registry.ChainRegistry().Get(server.Chain),
|
|
TTL: server.TTL,
|
|
Timeout: server.Timeout,
|
|
ClientIP: net.ParseIP(server.ClientIP),
|
|
Prefer: server.Prefer,
|
|
Hostname: server.Hostname,
|
|
Async: server.Async,
|
|
Only: server.Only,
|
|
})
|
|
}
|
|
|
|
return xresolver.NewResolver(
|
|
nameservers,
|
|
xresolver.LoggerOption(
|
|
logger.Default().WithFields(map[string]any{
|
|
"kind": "resolver",
|
|
"resolver": cfg.Name,
|
|
}),
|
|
),
|
|
)
|
|
}
|