Docker 容器中的服务器连接被拒绝，我应该在测试中添加 time.Sleep(100 * time.Millisecond) 吗？

积累知识，胜过积蓄金银！毕竟在Golang开发的过程中，会遇到各种各样的问题，往往都是一些细节知识点还没有掌握好而导致的，因此基础知识点的积累是很重要的。下面本文《Docker 容器中的服务器连接被拒绝，我应该在测试中添加 time.Sleep(100 * time.Millisecond) 吗？》，就带大家讲解一下知识点，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

我目前正在开发一个旨在在 docker 容器中运行的服务器。

这是我的测试设置方法：

func testmain(m *testing.m) {
    schedulername := "scheduler1"
    ip, err := container.startnewscheduler(schedulername)
    if err != nil {
        log.println("could not create container.")
        log.fatal(err)
    }
    serverip = ip
    code := m.run()
    cleanupcontainer(schedulername)
    os.exit(code)
}

行 container.startnewscheduler(schedulername) 启动一个名为“scheduler1”的新 docker 容器，并告诉它运行其中的服务器。

接下来，我使用在后台运行的容器来运行测试，现在我只有一个测试。

func testnewscheduler(t *testing.t) {
    testcodeinput := "this is a test"
    requestbody, err := json.marshal(map[string]string{
        "code": fmt.sprintf("print(\"%s\")", testcodeinput),
    })

    if err != nil {
        t.fatal(err)
    }

    url := fmt.sprintf("http://%s:%d/execute/python", serverip, 3000)
    contenttype := "application/json"
    body := bytes.newbuffer(requestbody)

    response := post(url, contenttype, body, t)
    actual := parseoutput(response.body, t)
    response.body.close()

    expected := fmt.sprintf("{\"stdout\":\"%s\\n\"}", testcodeinput)
    if actual != expected {
        t.fatalf("expected %s, but got %s", expected, actual)
    }
}

我遇到的问题是有时连接被拒绝，有时却没有。

server_container_test.go:51: post http://172.20.0.2:3000/execute/python: dial tcp 172.20.0.2:3000: connect: connection refused

我注意到，每当我尝试调试问题时，一切似乎都工作正常。我的运行理论是因为当我单步执行代码时，容器有更多时间启动并让服务器在其中运行。

为了测试我的假设，我在 post 方法中添加了第二个 post 调用，并在调用它之前设置了计时器。

func post(url string, contentType string, body io.Reader, t *testing.T) *http.Response {
    t.Helper()
    response, err := http.Post(url, contentType, body)
    if err != nil {
        //There is an error where the container takes a second to boot up and so
        //the scheduler isn't running when the first request is sent, so we try
        //one more time here and  check again.
        time.Sleep(100 * time.Millisecond) <--- Right here
        response, err = http.Post(url, contentType, body)
        if err != nil {
            t.Fatal(err)
        }
    }
    return response
}

还有其他人对导致我出现此问题的原因有任何猜测吗？

如果我的假设是正确的，这是解决这个问题的最佳方法吗？在测试中添加 time.sleep 是不是一个坏主意？

谢谢！

解决方案

好吧，经过一番思考后，我更改了源代码，如果您认为这是解决我的问题的好方法，请告诉我。我仍在学习 go 和 http 服务器，因此任何意见都会受到欢迎。

这是我的修复/想法：

以前，一旦创建了容器，我只是返回了它的 ip 地址，然后就忘记了。

现在我创建一个 go 例程，反复尝试向服务器发送 post 请求。如果没有失败，那么我通过通道发送 true 并关闭该函数。

ip := info.networksettings.networks[networkname].ipaddress
    works := make(chan bool)
    ctx, canelroutine := context.withcancel(context.background())
    defer canelroutine()

    go func(ctx context.context) {
        requestbody, _ := json.marshal(map[string]string{
            "code": "print()",
        })
        select {
        case <-ctx.done():
            return
        default:
            for {
                _, err := http.post(
                    fmt.sprintf("http://%s:%d/execute/python", ip, 3000),
                    "application/json",
                    bytes.newbuffer(requestbody),
                )
                if err == nil {
                    works <- true
                    return
                }
            }
        }
    }(ctx)

发送 goroutine 后，我创建一个计时器并等待计时器或 goroutine 返回。

timer := time.after(500 * time.millisecond)
    select {
    case <-works:
        return ip, nil
    case <-timer:
        return ip, &unreachablecontainererror{name: schedulername}
    }

这个解决方案的优点是我现在引入了一个 unreachablecontainererror ，它允许我更具体地了解我的错误消息，并且可以在接收端进行检查。我还会以任何一种方式将 ip 地址发回，以防客户端因其他原因需要它。

这里是完整的 startnewscheduler 方法，以防您想查看它。

//StartNewScheduler starts a new scheduler with the given options.
//returns the IP address for the given scheduler.
func StartNewScheduler(schedulerName string) (string, error) {
    ///Defaults
    dockerfile := "Dockerfile_standard"
    networkName := "scheduler-cluster"
    imageID := "lkelly93/scheduler_image:latest"

    cli, err := client.NewEnvClient()
    if err != nil {
        return "", err
    }

    err = createDefaultImageIfNeeded(
        cli,
        imageID,
        dockerfile)

    if err != nil {
        return "", err
    }

    err = createSchedulerClusterNetworkIfNeeded(cli, networkName)
    if err != nil {
        return "", err
    }

    ctx := context.Background()
    resp, err := cli.ContainerCreate(
        ctx,
        &container.Config{Image: imageID},
        &container.HostConfig{
            NetworkMode: container.NetworkMode(networkName),
            Privileged:  true,
        },
        nil,
        schedulerName,
    )

    if err != nil {
        return "", err
    }

    err = cli.ContainerStart(ctx, resp.ID, types.ContainerStartOptions{})
    if err != nil {
        return "", err
    }

    //Get container IP
    info, err := cli.ContainerInspect(ctx, resp.ID)
    if err != nil {
        return "", err
    }

    IP := info.NetworkSettings.Networks[networkName].IPAddress
    works := make(chan bool)
    ctx, canelRoutine := context.WithCancel(context.Background())
    defer canelRoutine()

    go func(ctx context.Context) {
        requestBody, _ := json.Marshal(map[string]string{
            "Code": "print()",
        })
        select {
        case <-ctx.Done():
            return
        default:
            for {
                _, err := http.Post(
                    fmt.Sprintf("http://%s:%d/execute/python", IP, 3000),
                    "application/json",
                    bytes.NewBuffer(requestBody),
                )
                if err == nil {
                    works <- true
                    return
                }
            }
        }
    }(ctx)

    timer := time.After(500 * time.Millisecond)
    select {
    case <-works:
        return IP, nil
    case <-timer:
        return IP, &UnreachableContainerError{name: schedulerName}
    }
}

您可以在容器内运行测试代码，通过 docker compose 使用选项 depends_on 启动。

本篇关于《Docker 容器中的服务器连接被拒绝，我应该在测试中添加 time.Sleep(100 * time.Millisecond) 吗？》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！