我目前正在研究 vendor go balancer 代码。我需要删除 tcp 拨号调用并在没有调用的情况下模拟成功的连接。在下面的代码中,有 这一行:
ds, err := net.Dial("tcp", backend.String());if err != nil {
log.Printf("failed to dial %s: %s", backend, err)
us.Close()
return
}
它所做的是拨号到 tcp 服务器,然后返回连接
ds
中的响应,在此处定义:https://golang.org/pkg/net/#Dial
我需要的是在不执行 tcp dialer
的情况下获取 ds
。我是
尝试在没有任何实际 tcp 调用的情况下测试负载均衡器。所以,
本质上,
当我们输入 handleConnection
时,我们会在 tcp 拨号之前创建一个网络连接并使用这个网络连接,它应该在 tcp 拨号开始之前模拟 100% 的网络连接。
func copy(wc io.WriteCloser, r io.Reader) { defer wc.Close()
io.Copy(wc, r)
}
func handleConnection(us net.Conn, backend BA.Backend) {
if backend == nil {
log.Printf("no backend available for connection from %s",
us.RemoteAddr())
us.Close()
return
}
host, _, _ := net.SplitHostPort(us.RemoteAddr().String())
_, ok := dbAuthTokenData[host]; if !ok {
w := bufio.NewWriter(us)
w.WriteString("InvalidCredentials")
w.Flush()
us.Close()
return
}
ds, err := net.Dial("tcp", backend.String());if err != nil {
log.Printf("failed to dial %s: %s", backend, err)
us.Close()
return
}
// Ignore errors
go copy(ds, us)
go copy(us, ds)
}
func tcpBalance(bind string, backends BA.Backends) error {
log.Println("using tcp balancing")
ln, err := net.Listen("tcp", bind)
if err != nil {
return fmt.Errorf("failed to bind: %s", err)
}
log.Printf("listening on %s, balancing %d backends", bind, backends.Len())
for {
conn, err := ln.Accept()
if err != nil {
log.Printf("failed to accept: %s", err)
continue
}
go handleConnection(conn, backends.Choose())
}
return err
}
我尝试注释掉 go handleConnection(conn, backends.Choose())
但失败了。
最佳答案
您可以重构代码的模式是创建一个 Dialer
接口(interface)。在您的代码示例中,您将返回的 ds
用作 io.ReadWriteCloser
。所以你不需要实现整个 net.Conn
接口(interface)。因为 net.Conn
里面有读写方法,一切正常
type Dialer interface{
Dial(network, address string) (io.ReadWriteCloser, error)
}
现在让我们更改您的函数:
func handleConnection(us net.Conn, backend BA.Backend, d Dialer) {
// ...
// Code here stays
ds, err := d.Dial("tcp", backend.String());if err != nil {
log.Printf("failed to dial %s: %s", backend, err)
us.Close()
return
}
// ...
}
为了使您的生产代码能够正常工作,您现在需要定义一个 netDialer
类型,它包装了 net.Dial()
函数。在您的测试中,您可以使用 testDialer
,它使用 bytes.Buffer
。
这回答了你的问题:
Emulate net connection without entering net dial
关于go - 无需输入网络拨号即可模拟网络连接,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52862898/