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

Question

我目前正在研究 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

ds我需要的是在不做的情况下获得tcp dialer。我正在尝试在没有任何实际 tcp 调用的情况下测试负载平衡器。所以，基本上，当我们输入handleConnectionwed 在 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())但失败了。

Answer 1

您可以重构代码的模式是创建一个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()函数的类型。在您的测试中，您可以使用 atestDialer其中使用 a bytes.Buffer。这回答了你的问题：无需输入网络拨号即可模拟网络连接