每个进程都拥有自己的标准输入、输出和错误文件描述符。stdin作为进程的输入。本文将介绍如何从stdin读取数据。

读取标准输入

Golang 版本

1.12.1

前言

每个进程都拥有自己的标准输入、输出和错误文件描述符。stdin作为进程的输入。本文将介绍如何从stdin读取数据。

实现

1.创建文件fmt.go，代码如下：

package main

import "fmt"

func main() {

	var name string
	fmt.Println("你叫什么名字？")
	fmt.Scanf("%s\n", &name)

	var age int
	fmt.Println("你几岁？")
	fmt.Scanf("%d\n", &age)

	fmt.Printf("你好! %s, 你的年龄是 %d\n", name, age)

}

$ go run fmt.go
你叫什么名字？
GanymedeNil
你几岁？
25
你好! GanymedeNil, 你的年龄是 25

2.创建文件scanner.go，代码如下：

package main

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

func main() {

	// scan能够按行扫描输入
	sc := bufio.NewScanner(os.Stdin)

	for sc.Scan() {
		txt := sc.Text()
		fmt.Printf("Echo: %s\n", txt)
	}

}

$ go run scanner.go
Hello
Echo: Hello

3.创建文件reader.go，代码如下：

package main

import (
	"fmt"
	"os"
)

func main() {

	for {
		data := make([]byte, 8)
		n, err := os.Stdin.Read(data)
		if err == nil && n > 0 {
			process(data)
		} else {
			break
		}
	}

}

func process(data []byte) {
	fmt.Printf("Received: %X %s\n", data, string(data))
}

$  echo 'Go is awesome!' | go run reader.go
Received: 476F206973206177 Go is aw
Received: 65736F6D65210A00 esome!

原理

Go进程的stdin可以通过os包的Stdin获取。 实际上，它是一个实现Reader接口的File类型。 从Reader里读取是非常容易的。 前面的代码展示了如何从Stdin读取的三种非常常见的方法。

第一个例子演示了fmt包的使用，它提供了Scan、Scanf和Scanln函数。Scanf函数将输入写入给定变量。Scanf的优点是你可以确定扫描值的格式。Scan函数只是将输入写入变量（没有预定义格式），Scanln顾名思义，读取以换行符结束的输入。

Scanner是第二个例子，它提供了扫描较大输入的便捷方式。Scanner 包含了Split功能，通过该功能可以自定义分割功能。 例如，要扫描stdin中的单词，可以使用bufio.ScanWords作为预定义的SplitFunc。

通过Reader API读取是最后提出的方法。 这个可以让你更好地控制输入的读取方式。