From 4525630abb98df8b62d6f279c95195dcf1186dd3 Mon Sep 17 00:00:00 2001 From: ginuerzh Date: Mon, 9 Oct 2023 21:26:41 +0800 Subject: [PATCH] add network for node --- chain/node.go | 7 +++++++ common/net/dialer/dialer.go | 7 +++++++ 2 files changed, 14 insertions(+) diff --git a/chain/node.go b/chain/node.go index fa4c84b..4389de8 100644 --- a/chain/node.go +++ b/chain/node.go @@ -26,6 +26,7 @@ type NodeOptions struct { HostMapper hosts.HostMapper Metadata metadata.Metadata Host string + Network string Protocol string HTTP *HTTPNodeSettings TLS *TLSNodeSettings @@ -64,6 +65,12 @@ func HostNodeOption(host string) NodeOption { } } +func NetworkNodeOption(network string) NodeOption { + return func(o *NodeOptions) { + o.Network = network + } +} + func ProtocolNodeOption(protocol string) NodeOption { return func(o *NodeOptions) { o.Protocol = protocol diff --git a/common/net/dialer/dialer.go b/common/net/dialer/dialer.go index 8bcf6b1..6112e4c 100644 --- a/common/net/dialer/dialer.go +++ b/common/net/dialer/dialer.go @@ -47,6 +47,13 @@ func (d *NetDialer) Dial(ctx context.Context, network, addr string) (conn net.Co log = logger.Default() } + switch network { + case "unix": + netd := net.Dialer{} + return netd.DialContext(ctx, network, addr) + default: + } + deadline := time.Now().Add(timeout) ifces := strings.Split(d.Interface, ",") for _, ifce := range ifces {