App Engine Go：如何终止运行 goroutine

Question

我想运行 App Engine 中的 2 个 goroutine 并行，这样当第一个 goroutine 完成它的工作时，处理程序不需要等待第二个 goroutine - 它停止第二个 goroutine 并且 returns 结果给客户端。这可能吗？我用context.WithCancel()试了一下，没用（我用的是go1.6）。

这是我的代码：

package mytest

import (
    "net/http"
    "sync"
    "time"

    "golang.org/x/net/context"
    "google.golang.org/appengine"
    "google.golang.org/appengine/log"
    "google.golang.org/appengine/urlfetch"
)

func init() {
    http.HandleFunc("/test", handlerTest)
    http.HandleFunc("/testwait10s", handlerTest10s)
    http.HandleFunc("/testwait5s", handlerTest5s)
}

func handlerTest(w http.ResponseWriter, r *http.Request) {
    c := appengine.NewContext(r)
    ctx, _ := context.WithTimeout(c, 30*time.Second)

    ctx1, ctx1Cancel := context.WithCancel(ctx)
    ctx2, ctx2Cancel := context.WithCancel(ctx)

    var wg sync.WaitGroup
    wg.Add(2)

    go func() {
        defer wg.Done()
        log.Infof(ctx1, "Go1 begin ...")
        client1 := urlfetch.Client(ctx1)
        _, err := client1.Get("http://APP_NAME.appspot.com/testwait5s")
        if err != nil {
            log.Errorf(ctx1, "Go1 failed:  %v", err)
        }
        ctx2Cancel()
        log.Infof(ctx1, "Go1 over ...")
    }()

    go func() {
        defer wg.Done()
        log.Infof(ctx2, "Go2 begin ...")
        client2 := urlfetch.Client(ctx2)
        _, err := client2.Get("http://APP_NAME.appspot.com/testwait10s")
        if err != nil {
            log.Errorf(ctx2, "Go2 failed %v", err)
        }
        ctx1Cancel()
        log.Infof(ctx2, "Go2 over ...")
    }()

    wg.Wait()
    log.Infof(ctx1, "Go1 and GO2 over")
}

func handlerTest10s(w http.ResponseWriter, r *http.Request) {
    time.Sleep(10 * time.Second)
    return
}

func handlerTest5s(w http.ResponseWriter, r *http.Request) {
    time.Sleep(5 * time.Second)
    return
}

有什么想法吗？谢谢！

Answer 1

您可以尝试将 wg.Add() 的值减少到 wg.Add(1) 而不是 wg.Add(2)。

当一个 go-routine 完成时，wg.Done() 会将计数器值减少 1。因此，在这种情况下，WaitGroup (wg) 计数器值将变为 ZERO。因此，最后一行的 wg.Wait() 不会等待其他 go-routine 完成。

请注意，如果 wg 计数器的值低于零，它将在剩余的 go-routine 秒内发生恐慌。因此，go-routines 将被强行退出。

Answer 2

只需创建一个通知通道并向其发送一个信号，表明其中一个计算已结束，您可以继续进行而无需等待另一个。

func handlerTest(w http.ResponseWriter, r *http.Request) {
    c := appengine.NewContext(r)
    ctx, cancel := context.WithTimeout(c, 30*time.Second)
    done := make(chan error, 2)

    work := func(url, name string) {
        log.Infof(ctx, "%s begin ...", name)
        client := urlfetch.Client(ctx)
        req, err := http.NewRequest(http.MethodGet, url, nil)
        if err != nil {
            log.Errorf(ctx, "%s failed:  %v", name, err)
            done <- err
            return
        }
        req = req.WithContext(ctx)
        _, err = client.Do(req)
        done <- err
        if err != nil {
            log.Errorf(ctx, "%s failed:  %v", name, err)
            return
        }
        cancel()
        log.Infof(ctx, "%s over ...", name)
    }
    go work("go1", "http://APP_NAME.appspot.com/testwait5s")
    go work("go2", "http://APP_NAME.appspot.com/testwait10s")

    for i := 0; i < cap(done); i++ {
        if err := <-done; err == nil {
            log.Infof(ctx, "executed without errors")
            return
        }
    }
    log.Error(ctx, "both computations have failed")
}

App Engine Go：如何终止运行 goroutine

App Engine Go: How to kill a running goroutine

google-app-engine

go

goroutine

google-app-engine-go

App Engine Go：如何终止 运行 goroutine

App Engine Go: How to kill a running goroutine

google-app-engine

go

goroutine

google-app-engine-go

App Engine Go：如何终止运行 goroutine