Go言語で標準入力を1文字ずつ取得したい

Question

こんにちは。
Go言語で標準入力から1行ごとではなく、1文字ずつ取得する方法を知りたいです。Enterではなく、1文字入力するごとにfmt.Printlnで出力しようと思っているのですが、どのようにすればいいのかわかりません。

ご教授、お願いいたします。

Answer 1

注意：ここに書いたのは連続する標準入力を一文字づつ処理する方法です。
ターミナルは基本の挙動がEnter押すごとに標準入力に投入します。

ASCII文字しか扱わないなら１バイト長のbyteスライスを引数にしたReadでOKです。

b := []byte{0}
_, err := os.Stdio.Read(b)
...

ただこれだと、ASCII以外の文字で１文字づつになりません。
こういった場合でも正しく「１文字」ずつ読みたいのなら、
例えばbufio.ReaderのReadRuneを使います。

package main

import (
	"bufio"
	"fmt"
	"io"
	"log"
	"os"
)

func main() {
	reader := bufio.NewReader(os.Stdin)
	for {
		r, _, err := reader.ReadRune()
		if err != nil {
			if err == io.EOF {
				return
			}
			log.Fatal(err)
		}
		fmt.Println(string(r))
	}
}

追記

質問者のコメントにあるような文字単位の入力をプログラムで受け取るには
OSのAPIを利用するか、以下のようにターミナルの動作モードを変えるしかなさそうです。
（以下の例はmacOSまたはLinux向けの方法です）
また、このモードではターミナルがCtrl+Cを受け取ってプロセスにSIGINTを送る機能も無効になるので
止めるには外部からシグナルを投げるかプログラム自身に終了処理が必要です。

https://play.golang.org/p/75GhzqTfV4_L

package main

import (
	"bufio"
	"fmt"
	"io"
	"log"
	"os"
	"os/exec"
	"os/signal"
)

func main() {
	cmd := exec.Command("stty", "raw")
	cmd.Stdin = os.Stdin
	cmd.Run()
	defer func() {
		cmd := exec.Command("stty", "-raw")
		cmd.Stdin = os.Stdin
		cmd.Run()
	}()
	reader := bufio.NewReader(os.Stdin)
	sig := make(chan os.Signal, 1)
	signal.Notify(sig, os.Interrupt)
	go func() {
		defer close(sig)
		for {
			r, _, err := reader.ReadRune()
			if err != nil {
				if err == io.EOF {
					return
				}
				log.Fatal(err)
			}
			if r == 'q' {
				break
			}
			fmt.Println(string(r))
		}
	}()
	<-sig
}

Comments

[退会済みユーザー]

ご回答ありがとうございます。Enterを押さないと以下のように読まれないません。

```
go run ./main.go
hello(Enter)
h
e
l
l
o
```

以下のように、１文字ずつやりたいのですが、どのようにしたらできますでしょうか。

```
go run ./main.go
h
=> h
e
=> e
...
```

[nobonobo]

おお。それは標準入力では処理できません。ターミナルの機能またはGUI機能を呼ぶ必要があります。この場合OSがなんなのかに依存したりするのでなんのOSをお使いなのか追記ください。

[退会済みユーザー]

失礼いたしました。私はmacOSを使用しております。

Answer 2

こういうことでしょうか。

package main

import (
	"fmt"
	"log"

	"github.com/mattn/go-tty"
)

func main() {
	tty, err := tty.Open()
	if err != nil {
		log.Fatal(err)
	}
	defer tty.Close()

	for {
		r, err := tty.ReadRune()
		if err != nil {
			log.Fatal(err)
		}
		fmt.Println("Key press => " + string(r))
	}
}

//go.mod
module sample/tty

go 1.16

require github.com/mattn/go-tty v0.0.3

$ go run .
Key press => a
Key press => b
Key press => c
Key press => d
Key press => e
...(ctrl+c)
signal: interrupt
$

かの mattn さんの go-tty を使って、私の x86_64 系 *nix でビルドされたバイナリ動きました。未検証ですが Windows でもイケるらしいですよ。

• https://github.com/mattn/go-tty by mattn さん @ GitHub

動作確認済み環境:

• macOS Catalina(10.15.7), Linux 上で darwin 用にビルドしたバイナリ実行
• Debian GNU/Linux 10 (buster), go version go1.16.2 linux/amd64
• Alpine Linux v3.13, go version go1.16 linux/amd64