卓越飞翔博客卓越飞翔博客

卓越飞翔 - 您值得收藏的技术分享站
技术文章34772本站已运行393

主 goroutine Golang 中无法接收沿通道发送的最后一个值

主 goroutine golang 中无法接收沿通道发送的最后一个值

主 goroutine Golang 中无法接收沿通道发送的最后一个值。这是由于当通道被关闭后,goroutine 无法再次接收到新的值。相反,它将一直阻塞在接收操作上,直到通道中的所有值都被接收完毕。这是 Golang 中的一种设计选择,旨在避免在接收操作过程中可能出现的死锁情况。因此,在编写 Golang 程序时,我们需要特别注意这一点,以避免潜在的问题和错误。

问题内容

给定 golang 中的 TCP 端口扫描器。 2个实现,第一个是我的,第二个来自golang书。假设第二个是 100% 可行的,正如许多读者之前测试的那样。但似乎两者都有相同的问题:在结果通道中发送的最后一个值无法在主协程中接收,它陷入无限等待来自通道的值,尽管该值实际上已发送。 一些观察:当端口数量少于 21 个时,它按预期工作;当金额超过1000时,未收到的金额增加到10左右。 我不明白为什么。

书中的实现

func worker(ports, results chan int) {
    for p := range ports {
        address := fmt.Sprintf("scanme.nmap.org:%d", p)
        conn, err := net.Dial("tcp", address)
        if err != nil {
            results <- 0
            fmt.Println("sent", p)
            continue
        }
        conn.Close()
        results <- p
        fmt.Println("sent", p)
    }
}

func main() {
    ports := make(chan int, 100)
    results := make(chan int)

    var openports []int

    for i := 0; i < cap(ports); i++ {
        go worker(ports, results)
    }

    go func() {
        for i := 1; i <= 50; i++ {
            ports <- i
        }
    }()

    for i := 0; i < 50; i++ {
        port := <-results // after 49 it gets stuck infinitely, never proceed further
        fmt.Println("received", port, i)
        if port != 0 {
            openports = append(openports, port)
        }
    }

    close(ports)
    close(results)

    sort.Ints(openports)

    fmt.Println(openports)

}

解决方法

通过向 net.Dialer 添加超时已解决该问题

func worker(ports, results chan int) {
    dialer := net.Dialer{Timeout: time.Second}

    for p := range ports {
        address := fmt.Sprintf("scanme.nmap.org:%d", p)
        conn, err := dialer.Dial("tcp", address)
        if err != nil {
            results <- 0
            continue
        }

        conn.Close()
        results <- p
    }
}
卓越飞翔博客
上一篇: 将字节文件读入 int64 切片的最有效方法是什么?
下一篇: 返回列表
留言与评论(共有 0 条评论)
   
验证码:
隐藏边栏