refactor(netxlite): add more functions to resolver (#455)
We would like to refactor the code so that a DoH resolver owns the connections of its underlying HTTP client. To do that, we need first to incorporate CloseIdleConnections into the Resolver model. Then, we need to add the same function to all netxlite types that wrap a Resolver type. At the same time, we want the rest of the code for now to continue with the simpler definition of a Resolver, now called ResolverLegacy. We will eventually propagate this change to the rest of the tree and simplify the way in which we manage Resolvers. To make this possible, we introduce a new factory function that adapts a ResolverLegacy to become a Resolver. See https://github.com/ooni/probe/issues/1591.
This commit is contained in:
parent
2e0118d1a6
commit
a3654f60b7
22 changed files with 278 additions and 118 deletions
|
|
@ -1,6 +1,7 @@
|
|||
package netxlite
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"strings"
|
||||
|
||||
|
|
@ -59,3 +60,65 @@ type (
|
|||
TLSHandshakerConfigurable = tlsHandshakerConfigurable
|
||||
TLSHandshakerLogger = tlsHandshakerLogger
|
||||
)
|
||||
|
||||
// ResolverLegacy performs domain name resolutions.
|
||||
//
|
||||
// This definition of Resolver is DEPRECATED. New code should use
|
||||
// the more complete definition in the new Resolver interface.
|
||||
//
|
||||
// Existing code in ooni/probe-cli is still using this definition.
|
||||
type ResolverLegacy interface {
|
||||
// LookupHost behaves like net.Resolver.LookupHost.
|
||||
LookupHost(ctx context.Context, hostname string) (addrs []string, err error)
|
||||
}
|
||||
|
||||
// NewResolverLegacyAdapter adapts a ResolverLegacy to
|
||||
// become compatible with the Resolver definition.
|
||||
func NewResolverLegacyAdapter(reso ResolverLegacy) Resolver {
|
||||
return &ResolverLegacyAdapter{reso}
|
||||
}
|
||||
|
||||
// ResolverLegacyAdapter makes a ResolverLegacy behave like
|
||||
// it was a Resolver type. If ResolverLegacy is actually also
|
||||
// a Resolver, this adapter will just forward missing calls,
|
||||
// otherwise it will implement a sensible default action.
|
||||
type ResolverLegacyAdapter struct {
|
||||
ResolverLegacy
|
||||
}
|
||||
|
||||
var _ Resolver = &ResolverLegacyAdapter{}
|
||||
|
||||
type resolverLegacyNetworker interface {
|
||||
Network() string
|
||||
}
|
||||
|
||||
// Network implements Resolver.Network.
|
||||
func (r *ResolverLegacyAdapter) Network() string {
|
||||
if rn, ok := r.ResolverLegacy.(resolverLegacyNetworker); ok {
|
||||
return rn.Network()
|
||||
}
|
||||
return "adapter"
|
||||
}
|
||||
|
||||
type resolverLegacyAddresser interface {
|
||||
Address() string
|
||||
}
|
||||
|
||||
// Address implements Resolver.Address.
|
||||
func (r *ResolverLegacyAdapter) Address() string {
|
||||
if ra, ok := r.ResolverLegacy.(resolverLegacyAddresser); ok {
|
||||
return ra.Address()
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
type resolverLegacyIdleConnectionsCloser interface {
|
||||
CloseIdleConnections()
|
||||
}
|
||||
|
||||
// CloseIdleConnections implements Resolver.CloseIdleConnections.
|
||||
func (r *ResolverLegacyAdapter) CloseIdleConnections() {
|
||||
if ra, ok := r.ResolverLegacy.(resolverLegacyIdleConnectionsCloser); ok {
|
||||
ra.CloseIdleConnections()
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in a new issue