Golang 非阻塞 channel 不起作用

Question

我是第一次在 go 中使用 goroutines 和 channels，之后就不再来了。

我有一个 websocket 连接，每次用户连接时都会生成一个新的 goroutine。现在我想在用户断开 websocket 连接时停止这个 goroutine。

为了管理停止信号，我创建了一个 channel 图。每个条目都可以通过用户的 websocket 连接来识别。我将 websocket 连接、停止信号的 channel 映射和其他两个参数传递给 goroutine。但是 goroutine 没有从 quit channel 接收到任何值，我不知道为什么。

下面是main包的相关代码:

package main

import (
    "net/http"
    "time"

    "github.com/gorilla/websocket"
)

func wsHandler(w http.ResponseWriter, r *http.Request) {
    ...
    ws, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        return
    }
    defer ws.Close()

    data.Quit[ws] = make(chan bool)

    data.DB.ListenToTable(data.GetTableName(source), channel, data.Quit, ws)

    for {
        if _, _, err := ws.NextReader(); err != nil {
            data.Quit[ws] <- true

            ws.Close()
            break
        }
    }
}

以及创建goroutine的data包的代码:

package data

var Quit = make(map[*websocket.Conn](chan bool))

func (db *rethinkDB) ListenToTable(name string, ch chan Data, quit map[*websocket.Conn](chan bool), ws *websocket.Conn) {
    go func(name string, ws *websocket.Conn) {
        for {
            select {
            case <-quit[ws]:
                fmt.Println("QUIT GOROUTINE")
                break
            default:
                res, err := r.Table(name).Changes().Run(db.session)
                if err != nil {
                    log.Fatalln(err)
                }

                var response DataFeed
                for res.Next(&response) {
                    response.NewVal.WS = ws
                    ch <- response.NewVal
                }

                if res.Err() != nil {
                    log.Println(res.Err())
                }
            }
        }
    }(name, ws)
}

我也尝试过缓冲 channel 或将 channel 而不是 channel 映射传递给 goroutine，但没有成功。 fmt.Println("QUIT GOROUTINE") 命令永远不会被调用，goroutine 也不会被删除。

我希望有人能帮助我，如果这个问题已经被问到但我还没有找到解决我的问题的解决方案，我很抱歉。

Answer 1

首先让事情变得更简单:

据我所知，您不需要为退出 channel 进行全局注册。只需在 main 中创建一个 ch := make(chan bool)，将其传递给 ListenToTable(而不是整个 channel 映射)并在选择中使用它。如果您想退出，请在 main close(ch) 中关闭它。但正如您所说，这并不能解决您的问题。

从理论上讲，您在关闭 go 例程方面走在了正确的轨道上。我使用了您的示例代码并从中生成了以下可运行代码:

package main

import (
    "fmt"
    "time"
)

func main() {

    chClose := make(chan bool)
    channel := make(chan string)

    ListenToTable("somestring", channel, chClose)

    time.Sleep(3 * time.Second)
    chClose <- true
    time.Sleep(1 * time.Second)
}

func ListenToTable(name string, ch chan string, chClose chan bool) {
    go func(name string) {
        for {
            select {
            case <-chClose:
                fmt.Println("QUIT GOROUTINE")
                return  // VERY IMPORTANT: not break!
            default:
            }
        }
    }(name)
}

问题一定出在您的代码中，可能被 default 部分中的某些内容阻止，甚至没有执行选择。尝试在 select { 之前打印 fmt.Println("something")。如果没有定期打印，那么您就有了答案。

还有一件事:如上面代码中所述，您不能使用单个 break 来中断 for { select { ... } } 。您需要使用 return (退出函数)或其他策略(如 Adrian 在评论中建议的带有标签的中断)。 break 只会退出 select，而不会退出 for 循环。