tests: Remove bridge dependency on unix

This commit is contained in:
Marek Siarkowicz 2021-09-22 15:37:23 +02:00
parent 3df2727746
commit 391d662f77
2 changed files with 31 additions and 23 deletions

View File

@ -15,20 +15,20 @@
package integration package integration
import ( import (
"fmt"
"io" "io"
"io/ioutil" "io/ioutil"
"net" "net"
"sync" "sync"
"go.etcd.io/etcd/client/pkg/v3/transport"
) )
// bridge creates a unix socket bridge to another unix socket, making it possible type Dialer interface {
Dial() (net.Conn, error)
}
// bridge proxies connections between listener and dialer, making it possible
// to disconnect grpc network connections without closing the logical grpc connection. // to disconnect grpc network connections without closing the logical grpc connection.
type bridge struct { type bridge struct {
inaddr string dialer Dialer
outaddr string
l net.Listener l net.Listener
conns map[*bridgeConn]struct{} conns map[*bridgeConn]struct{}
@ -40,30 +40,22 @@ type bridge struct {
mu sync.Mutex mu sync.Mutex
} }
func newBridge(addr string) (*bridge, error) { func newBridge(dialer Dialer, listener net.Listener) (*bridge, error) {
b := &bridge{ b := &bridge{
// bridge "port" is ("%05d%05d0", port, pid) since go1.8 expects the port to be a number // bridge "port" is ("%05d%05d0", port, pid) since go1.8 expects the port to be a number
inaddr: addr + "0", dialer: dialer,
outaddr: addr, l: listener,
conns: make(map[*bridgeConn]struct{}), conns: make(map[*bridgeConn]struct{}),
stopc: make(chan struct{}), stopc: make(chan struct{}),
pausec: make(chan struct{}), pausec: make(chan struct{}),
blackholec: make(chan struct{}), blackholec: make(chan struct{}),
} }
close(b.pausec) close(b.pausec)
l, err := transport.NewUnixListener(b.inaddr)
if err != nil {
return nil, fmt.Errorf("listen failed on socket %s (%v)", addr, err)
}
b.l = l
b.wg.Add(1) b.wg.Add(1)
go b.serveListen() go b.serveListen()
return b, nil return b, nil
} }
func (b *bridge) URL() string { return "unix://" + b.inaddr }
func (b *bridge) Close() { func (b *bridge) Close() {
b.l.Close() b.l.Close()
b.mu.Lock() b.mu.Lock()
@ -127,7 +119,7 @@ func (b *bridge) serveListen() {
case <-pausec: case <-pausec:
} }
outc, oerr := net.Dial("unix", b.outaddr) outc, oerr := b.dialer.Dial()
if oerr != nil { if oerr != nil {
inc.Close() inc.Close()
return return

View File

@ -736,20 +736,36 @@ func (m *member) listenGRPC() error {
if m.useIP { // for IP-only TLS certs if m.useIP { // for IP-only TLS certs
m.grpcAddr = "127.0.0.1:" + m.Name m.grpcAddr = "127.0.0.1:" + m.Name
} }
l, err := transport.NewUnixListener(m.grpcAddr) grpcListener, err := transport.NewUnixListener(m.grpcAddr)
if err != nil { if err != nil {
return fmt.Errorf("listen failed on grpc socket %s (%v)", m.grpcAddr, err) return fmt.Errorf("listen failed on grpc socket %s (%v)", m.grpcAddr, err)
} }
m.grpcBridge, err = newBridge(m.grpcAddr) bridgeAddr := m.grpcAddr + "0"
bridgeListener, err := transport.NewUnixListener(bridgeAddr)
if err != nil { if err != nil {
l.Close() grpcListener.Close()
return fmt.Errorf("listen failed on bridge socket %s (%v)", m.grpcAddr, err)
}
m.grpcBridge, err = newBridge(dialer{network: "unix", addr: m.grpcAddr}, bridgeListener)
if err != nil {
bridgeListener.Close()
grpcListener.Close()
return err return err
} }
m.grpcAddr = schemeFromTLSInfo(m.ClientTLSInfo) + "://" + m.grpcBridge.inaddr m.grpcAddr = schemeFromTLSInfo(m.ClientTLSInfo) + "://" + bridgeAddr
m.grpcListener = l m.grpcListener = grpcListener
return nil return nil
} }
type dialer struct {
network string
addr string
}
func (d dialer) Dial() (net.Conn, error) {
return net.Dial(d.network, d.addr)
}
func (m *member) ElectionTimeout() time.Duration { func (m *member) ElectionTimeout() time.Duration {
return time.Duration(m.s.Cfg.ElectionTicks*int(m.s.Cfg.TickMs)) * time.Millisecond return time.Duration(m.s.Cfg.ElectionTicks*int(m.s.Cfg.TickMs)) * time.Millisecond
} }