取消使用 golang 的 os.Stdin 进行输入操作

在Golang实战开发的过程中，我们经常会遇到一些这样那样的问题，然后要卡好半天，等问题解决了才发现原来一些细节知识点还是没有掌握好。今天golang学习网就整理分享《取消使用 golang 的 os.Stdin 进行输入操作》，聊聊，希望可以帮助到正在努力赚钱的你。

我希望能够使用上下文取消 os.stdin 读取，这是不可能的。通常，您关闭文件句柄来完成取消，但我不想关闭 os.stdin

可能的解决方案可能是：

• 能否确定 stdin.read 是否会阻塞？
• 线程可以像 pthreads 一样终止吗？
• 是否应该将 os.stdin 转发到另一个可以关闭的文件句柄？

这就是我得到的，丑陋的是scannerthread在上下文取消时仍然运行：

// Keystrokes emits keystroke events
// on g0.Context() shutdown, scannerThread is left running until the next newline
// the lines channel is never closed
func Keystrokes(lines chan<- string, g0 parl.Go) {
  var err error
  defer g0.Done(err)
  defer parl.Recover(parl.Annotation(), &err, parl.NoOnError)

  // stdio.Scan cannot be terminated, so let that thread terminate whenever
  var scanLines parl.NBChan[string]
  go scannerThread(scanLines.Send, g0.Context())

  // consume scannerThread output
  scannerCh := scanLines.Ch()
  done := g0.Context().Done()
  for {
    select {
    case line := <-scannerCh:
      lines <- line
    case <-done:
      return // canceled by context exit
    }
  }
}

// scannerThread reads from os.Stdin and therefore cannot be cancelled.
// send is a non-blocking send function.
// ctx indicates shutdown effective on next os.Stdin newline.
func scannerThread(send func(string), ctx context.Context) {
  var err error
  defer parl.Recover(parl.Annotation(), &err, parl.Infallible)

  scanner := bufio.NewScanner(os.Stdin)
  for scanner.Scan() { // scanner is typically stuck here
    if ctx.Err() != nil {
      return // terminated via context
    }
    send(scanner.Text())
  }

  // scanner had error
  err = scanner.Err()
}

正确答案

答案是读取 os.stdin 非阻塞，适用于 linux macos

os.stdin 有一个阻止关闭它的块，我记得

一旦完整行可用，非阻塞 os.stdin.read() 返回字节，否则返回 n==0

以下是如何设置非阻塞：

err = unix.setnonblock(int(os.stdin.fd()), true)

非阻塞的状态还可以读取：

flags, err := unix.FcntlInt(os.Stdin.Fd(), unix.F_GETFL, 0)
wasSet = flags&unix.O_NONBLOCK != 0

好了，本文到此结束，带大家了解了《取消使用 golang 的 os.Stdin 进行输入操作》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！