Add NativeDialer method to mtglib.Network

This commit is contained in:
9seconds
2026-03-12 19:25:53 +01:00
parent 6493688282
commit 94e4179fb7
7 changed files with 19 additions and 17 deletions
+4
View File
@@ -60,6 +60,10 @@ func (n *network) DialContext(ctx context.Context, protocol, address string) (es
return nil, fmt.Errorf("cannot dial to %s:%s: %w", protocol, address, err)
}
func (n *network) NativeDialer() *net.Dialer {
return &net.Dialer{}
}
func (n *network) MakeHTTPClient(dialFunc func(ctx context.Context,
network, address string) (essentials.Conn, error),
) *http.Client {
-9
View File
@@ -11,10 +11,7 @@ package network
import (
"errors"
"net"
"time"
"github.com/9seconds/mtg/v2/mtglib"
)
const (
@@ -37,9 +34,3 @@ const (
)
var ErrCannotDial = errors.New("cannot dial to any address")
type Network interface {
mtglib.Network
NativeDialer() *net.Dialer
}
+4 -3
View File
@@ -8,10 +8,11 @@ import (
"net/http"
"github.com/9seconds/mtg/v2/essentials"
"github.com/9seconds/mtg/v2/mtglib"
)
type multiNetwork struct {
networks []Network
networks []mtglib.Network
}
func (m multiNetwork) Dial(network, address string) (essentials.Conn, error) {
@@ -22,7 +23,7 @@ func (m multiNetwork) DialContext(ctx context.Context, network, address string)
networks := m.networks
if len(networks) > 1 {
networks = make([]Network, len(m.networks))
networks = make([]mtglib.Network, len(m.networks))
copy(networks, m.networks)
rand.Shuffle(len(m.networks), func(i, j int) {
@@ -59,7 +60,7 @@ func (m multiNetwork) MakeHTTPClient(
return m.networks[0].MakeHTTPClient(dialFunc)
}
func Join(networks ...Network) (Network, error) {
func Join(networks ...mtglib.Network) (mtglib.Network, error) {
if len(networks) == 0 {
return nil, errors.New("cannot join no networks")
}
+2 -1
View File
@@ -8,6 +8,7 @@ import (
"time"
"github.com/9seconds/mtg/v2/essentials"
"github.com/9seconds/mtg/v2/mtglib"
)
type network struct {
@@ -70,7 +71,7 @@ func New(
tcpTimeout,
httpTimeout,
idleTimeout time.Duration,
) Network {
) mtglib.Network {
if dnsResolver == nil {
dnsResolver = net.DefaultResolver
}
+3 -2
View File
@@ -6,11 +6,12 @@ import (
"net/url"
"github.com/9seconds/mtg/v2/essentials"
"github.com/9seconds/mtg/v2/mtglib"
"golang.org/x/net/proxy"
)
type proxyNetwork struct {
Network
mtglib.Network
client proxy.ContextDialer
}
@@ -23,7 +24,7 @@ func (p proxyNetwork) DialContext(ctx context.Context, network, address string)
return essentials.WrapNetConn(conn), nil
}
func NewProxyNetwork(base Network, proxyURL *url.URL) (*proxyNetwork, error) {
func NewProxyNetwork(base mtglib.Network, proxyURL *url.URL) (*proxyNetwork, error) {
socks, err := proxy.FromURL(proxyURL, base.NativeDialer())
if err != nil {
return nil, fmt.Errorf("cannot build proxy dialer: %w", err)