首页  >  文章  >  后端开发  >  检查 URL 可达性时如何正确处理 Go 中通道的超时?

检查 URL 可达性时如何正确处理 Go 中通道的超时?

Mary-Kate Olsen
Mary-Kate Olsen原创
2024-11-20 02:18:02856浏览

How to Properly Handle Timeouts with Channels in Go When Checking URL Reachability?

在 Go 中使用通道处理超时

问题:

当使用 goroutine 和通道检查 URL 列表的可达性,超时条件似乎没有被执行。这会导致代码始终返回 true,即使某些 URL 无法访问。

分析:

问题在于 check(u) 函数的实现,其中导致超时永远不会被执行。在您的代码中, check(u) 函数在当前 goroutine 中执行,从而阻止 IsReachable 中 select 语句的执行。当 select 语句被解除阻塞时, check(u) 和 time.After(time.Second) 通道都已准备好被使用,运行时可以选择其中之一。

解决方案1:

要解决这个问题,你可以在单独的 goroutine 中执行 check(u) 函数:

package main

import "fmt"
import "time"

func check(u string, checked chan<- bool) {
    time.Sleep(4 * time.Second)
    checked <- true
}

func IsReachable(urls []string) bool {

    ch := make(chan bool, 1)
    for _, url := range urls {
        go func(u string) {
            checked := make(chan bool)
            go check(u, checked)
            select {
            case ret := <-checked:
                ch <- ret
            case <-time.After(1 * time.Second):
                ch <- false
            }
        }(url)
    }
    return <-ch
}

func main() {
    fmt.Println(IsReachable([]string{"url1"}))
}

在这个修改后的代码中, check(u) 函数在goroutine中执行,并将结果传递到检查的通道。 IsReachable 中的 select 语句现在等待已检查通道或超时通道准备就绪。这可确保正确处理超时条件。

解决方案 2:

或者,您可以通过为所有 URL 设置单个超时来简化代码。如果超时与创建 goroutine 所需的时间相比足够长,则此方法是可行的:

package main

import "fmt"
import "time"

func check(u string, ch chan<- bool) {
    time.Sleep(4 * time.Second)
    ch <- true
}

func IsReachable(urls []string) bool {
    ch := make(chan bool, len(urls))
    for _, url := range urls {
        go check(url, ch)
    }
    time.AfterFunc(time.Second, func() { ch <- false })
    return <-ch
}
func main() {
    fmt.Println(IsReachable([]string{"url1", "url2"}))
}

在此代码中,使用 time.AfterFunc 为所有 URL 设置单个超时。如果在超时期限内可以访问任何 URL,则 ch 通道将收到 true,并且该函数将返回 true。否则,将向 ch 发送 false,函数将返回 false。

以上是检查 URL 可达性时如何正确处理 Go 中通道的超时?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn