mirror of
https://github.com/etcd-io/etcd.git
synced 2024-09-27 06:25:44 +00:00
tests: Remove bridge dependency on unix
This commit is contained in:
parent
3df2727746
commit
391d662f77
@ -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
|
||||||
|
@ -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
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user