无需网络连接直接模拟网络通信

从现在开始，努力学习吧！本文《无需网络连接直接模拟网络通信》主要讲解了等等相关知识点，我会在golang学习网中持续更新相关的系列文章，欢迎大家关注并积极留言建议。下面就先一起来看一下本篇正文内容吧，希望能帮到你！

我目前正在研究供应商 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 时，我们会在 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() 函数。在您的测试中，您可以使用 testdialer，它使用 bytes.buffer。

这回答了您的问题：

无需输入网络拨号即可模拟网络连接

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《无需网络连接直接模拟网络通信》文章吧，也可关注golang学习网公众号了解相关技术文章。