Question

让我们说我的golang函数类似于：

conn, _ := ln.Accept()
r := bufio.NewReader(conn)

func waiter(r *bufio.Reader) {
    r.ReadString('\n')
}

go waiter(r)
time.Sleep(time.Second)
fmt.Println("hello")

我有没有办法取消服务员，所以如果它从未到来，它就不会等待字符串输入？问题是bufio读卡器正在阻塞，我想让它在一个自动收报机上等待两秒钟，如果它没有读取任何数据到缓冲区以逃避goroutine。

Answer 1

Conn接口提供了一个方法SetReadDeadline来在特定时间中断固定操作：

for {
    // Set a deadline for reading. Read operation will fail if no data
    // is received after deadline.
    conn.SetReadDeadline(time.Now().Add(timeoutDuration))

    // Read tokens delimited by newline
    bytes, err := bufReader.ReadBytes('\n')
    if err != nil {
        fmt.Println(err)
        return
    }

    fmt.Printf("%s", bytes)
}

http://www.mrleong.net/post/130329994134/go-tcp-connection-listening-and-timeout

Answer 2

编辑：上下文无法阻止读取功能，这不是问题的正确解决方案。在这种情况下，最好为连接设置空闲超时。

您可以使用context包来控制goroutines。上下文主要用于停止goroutine，以防它被取消，超时等。要使用它，你必须再接收一个参数ctx context，并在你的goroutine中运行select。

Godoc的例子：

package main

import (
"context"
"fmt"
"time"
)

func main() {
// Pass a context with a timeout to tell a blocking function that it
// should abandon its work after the timeout elapses.
ctx, cancel := context.WithTimeout(context.Background(), 50*time.Millisecond)
defer cancel()

select {
case <-time.After(1 * time.Second):
    fmt.Println("overslept")
case <-ctx.Done():
    fmt.Println(ctx.Err()) // prints "context deadline exceeded"
}

}

在官方博客文章中阅读有关上下文用例的更多信息： https://blog.golang.org/context

取消go func（）

2 个答案: