制限時間付き標準入力

プログラムを書いていて一度は制限時間のある、標準入力を作りたいと思ったことはあると思います。因みに僕はありません。
以外と簡単に Go で実現できたのでメモがてら書きます。いつかの自分のためですね。

方法

方法はいたって簡単で、標準入力を受け取るサブルーチンを作って、入力テキストをチャンネルで受け取るだけです。

in := make(chan string, 1)
go func() {
	sc := bufio.NewScanner(os.Stdin)
	sc.Scan()
	in <- sc.Text()
}()

上記のチァンネルと time.NewTimer や time.NewTicker を組み合わせることで簡単に実現できます。

timer := time.NewTimer(time.Second * 10)
select {
  case text := <-in:
    // 入力を使った処理
  case <-timer.C:
    // タイムアウト時の処理
}

適当に作ったサンプル

上記の方法で適当に作ったサンプルです。ローカルで実行してみると雰囲気を掴めると思います。

package main

import (
	"bufio"
	"fmt"
	"os"
	"time"
)

func main() {
	in := make(chan string, 1)
	go func() {
		sc := bufio.NewScanner(os.Stdin)
		sc.Scan()
		in <- sc.Text()
	}()

	ticker := time.NewTicker(time.Second)
	defer ticker.Stop()
	fmt.Println("制限時間は5秒。坊っちゃんの作者を入力して")
	for i := 0; i < 5; i++ {
		select {
		case text := <-in:
			if text == "夏目漱石" {
				fmt.Printf("入力できてすごい\n")
			} else {
				fmt.Printf("間違えているよ\n%s\n", text)
			}
			i = 5
		case <-ticker.C:
			switch i {
			case 0:
				fmt.Println("制限時間あるので早めに")
			case 1:
				fmt.Println("はやくはやく")
			case 2:
				fmt.Println("邪魔だと思う？")
			case 3:
				fmt.Println("僕もそう思う")
			case 4:
				fmt.Println("ゲームオーバー(笑)")
			}
		}
	}
}

締め

簡単でしたが、いつ使うんでしょうね。