Как отменить fmt.Scanf после определенного времени ожидания?

Question

У меня есть очень простая утилита командной строки, и в конце я жду, когда пользователь нажмет клавишу ввода, чтобы завершить программу:

fmt.Scanf("\n") // wait for hitting the enter key to end the program

Теперь я хочу изменить ее иесли клавиша ввода не была нажата в течение некоторого времени, я хочу отменить Scanf и сделать что-то еще.Можно ли отменить ожидание ввода пользователя?

Answer 1

Вы можете просто создать канал и запустить программу, которая выполняет fmt.Scanf("\n"), а затем что-то записывает в канал.Тогда select между этим каналом и time.After(3 * time.Second).

Вот чистое решение:

package main

import (
    "fmt"
    "time"
)

func main() {
    // Rest of the program...

    ch := make(chan int)

    go func() {
        fmt.Scanf("\n")
        ch <- 1
    }()

    select {
    case <-ch:
        fmt.Println("Exiting.")
    case <-time.After(3 * time.Second):
        fmt.Println("Timed out, exiting.")
    }
}

Answer 2

Вам просто нужно поместить его в бесконечный цикл

package main

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

// 3 seconds for example
var deadline = time.Second * 3

func main() {
    c := make(chan string, 1)
    go scan(c)

    ctx, _ := context.WithTimeout(context.Background(), deadline)

    select {
    case <-ctx.Done():
        // didnt type for deadline seconds
    case <-c:
        // did it in time
    }
}

func scan(in chan string) {
    var input string
    _, err := fmt.Scanln(&input)
    if err != nil {
        panic(err)
    }

    in <- input
}