go - 无需输入网络拨号即可模拟网络连接
问题描述
我目前正在研究供应商去平衡器代码。我需要删除 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
我需要的是在ds
不做的情况下获得tcp dialer
。我正在尝试在没有任何实际 tcp 调用的情况下测试负载均衡器。所以,本质上,当我们输入handleConnection
wed 在 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
接口。在您的代码示例中,您使用返回的ds
只是一个io.ReadWriteCloser
. 所以你不需要实现整个net.Conn
接口。就像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()
函数的类型。在您的测试中,您可以使用 a testDialer
which uses a bytes.Buffer
。
这回答了你的问题:
无需输入网络拨号即可模拟网络连接
推荐阅读
- sql - SQL Server 子分区
- c# - WCF 何时从 app.config 加载值?
- android - Android Room - 将@Entity 拆分为 Key 和 DataObject
- python - 引发错误后有什么方法可以自动运行脚本
- html - 如何在搜索栏旁边对齐图像
- typescript - 如何从类型初始化 BehaviorSubject?
- postgresql - pg_ctl促进不快速暂停复制
- javascript - 如何将字符串附加到会话存储密钥?
- sql - SQL Server检查值是否在下一行重复
- laravel-5 - Laravel 雄辩的模型,其中日期时间小于或等于