Go基礎系列:讀取標准輸入


fmt包中提供了3類讀取輸入的函數:

  • Scan家族:從標准輸入os.Stdin中讀取數據,包括Scan()、Scanf()、Scanln()
  • SScan家族:從字符串中讀取數據,包括Sscan()、Sscanf()、Sscanln()
  • Fscan家族:從io.Reader中讀取數據,包括Fscan()、Fscanf()、Fscanln()

其中:

  • Scanln、Sscanln、Fscanln在遇到換行符的時候停止
  • Scan、Sscan、Fscan將換行符當作空格處理
  • Scanf、Sscanf、Fscanf根據給定的format格式讀取,就像Printf一樣

這3家族的函數都返回讀取的記錄數量,並會設置報錯信息,例如讀取的記錄數量不足、超出或者類型轉換失敗等。

以下是它們的定義方式:

$ go doc fmt | grep -Ei "func [FS]*Scan"
func Fscan(r io.Reader, a ...interface{}) (n int, err error)
func Fscanf(r io.Reader, format string, a ...interface{}) (n int, err error)
func Fscanln(r io.Reader, a ...interface{}) (n int, err error)
func Scan(a ...interface{}) (n int, err error)
func Scanf(format string, a ...interface{}) (n int, err error)
func Scanln(a ...interface{}) (n int, err error)
func Sscan(str string, a ...interface{}) (n int, err error)
func Sscanf(str string, format string, a ...interface{}) (n int, err error)
func Sscanln(str string, a ...interface{}) (n int, err error)

因為還沒介紹io.Reader,所以Fscan家族的函數暫且略過,但用法和另外兩家族的scan類函數是一樣的。

Scan、Scanf和Scanln

Scan家族函數從標准輸入讀取數據時,將以空格為分隔符分隔標准輸入中的內容,並將分隔后的各個記錄保存到給定的變量中。其中Scanf()可以指定分隔符。

例如,使用Scanln函數等待用戶輸入數據,或從管道中讀取數據。下面的代碼將等待用戶輸入,且將讀取的內容分別保存到name變量和age變量中:

package main

import (
	"fmt"
)

func main() {
	var (
		name string
		age  int
	)
	fmt.Print("輸入姓名和年齡,使用空格分隔:")
	fmt.Scanln(&name, &age)
	fmt.Printf("name: %s\nage: %d\n", name, age)
}

因為Scanln()遇到換行符或EOF的時候終止讀取,所以在輸入的時候只要按下回車鍵就會結束讀取。

運行它,將提示輸入姓名:

請輸入姓名和年齡,空格分隔:
周伯通 69
name: 周伯通
age: 69

同理Scanf()也在遇到換行符或EOF的時候終止讀取行為。使用Scanf()的時候,需要給定格式化字符串形式:
例如:

fmt.Scanf("%s %d",&name,&age)

輸入時,第一個字段會轉換成字符串格式保存到name變量中,第二個記錄會轉換成整數保存到age中,如果轉換失敗,將不會進行保存。例如輸入malongshuai aaa,由於aaa無法轉換成int,所以age變量的值仍然為初始化的數值0。

Scanf可指定分隔符,其中上面的是%s %d中間的空格就是分隔符。例如下面指定:作為分隔符:

fmt.Scanf("%s : %d",&name,&age)

在輸入時,必須按照以下格式進行輸入:首先至少一個空格,然后一個冒號,再至少一個空格:

周伯通 : 23    // 或者連續多個空格  "周伯通     :   23"
name: 周伯通
age: 23

如果使用的是fmt.Scan(),則輸入數據時可以換行輸入,Scan()會將換行符作為空格進行處理,直到讀取到了2個記錄之后自動終止讀取操作:

fmt.Scan(&name, &age)

請輸入姓名和年齡,空格分隔:周伯通
87
name: 周伯通
age: 87

一般來說,只使用Scanf類函數比較好。

返回值

這些函數都有返回值:讀取的記錄數量和err信息。

以Scanln()為例:

func main() {
	var (
		name string
		age  int
	)
	fmt.Print("輸入姓名和年齡,使用空格分隔:")
	n, err := fmt.Scanln(&name, &age)

	fmt.Printf("name: %s\nage: %d\n", name, age)

	fmt.Println("records count:",n)
	fmt.Println("err or not:",err)
}

輸入:

malongshuai 23     // n = 2, err = nil
malongshuai        // n = 1, err != nil
malongshuai long   // n = 2, err != nil
malongshuai 23 23  // n = 2, err != nil

Sscan、Sscanf和Scanln

Sscan家族的函數用於從給定字符串中讀取數據,用法和Scan家族類似。

func Sscan(str string, a ...interface{}) (n int, err error)
func Sscanln(str string, a ...interface{}) (n int, err error)
func Sscanf(str string, format string, a ...interface{}) (n int, err error)

例如:

package main

import (
	"fmt"
)

func main() {
	var (
		name string
		age  int
	)
	input := "malongshuai 23"

	fmt.Sscan(input, &name, &age)
	fmt.Printf("name: %s\nage: %d\n", name, age)
}

使用Sscanf()可以指定分隔符:

input := "malongshuai : 23"

fmt.Sscanf(input, "%s : %d", &name, &age)

使用bufio中讀取標准輸入

除了fmt包的Scan類函數,bufio包也可以讀取標准輸入。當然,讀取標准輸入只是它的一個功能示例,它的作用是操作緩沖IO。

package main

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

var inputReader *bufio.Reader
var input string
var err error

func main() {
	inputReader = bufio.NewReader(os.Stdin)
	fmt.Println("輸入姓名:")
	input, err = inputReader.ReadString('\n')
	if err == nil {
		fmt.Printf("The input was: %s\n", input)
	}
}

其中NewReader()創建一個bufio.Reader實例,表示創建一個從給定文件中讀取數據的讀取器對象。然后調用讀取器對象(Reader實例)的ReadString()方法,這個方法以\n作為分隔符,它的分隔符必須只能是單字符,且必須使用單引號包圍,因為它會作為byte讀取。ReadString()讀取來自os.Stdin的內容后將其保存到input變量中,同時返回是否出錯的信息。ReadString()只有一種情況會返回err:沒有遇到分隔符。

ReadString會將讀取的內容包括分隔符都一起放進緩沖中,如果讀取文件時讀到了結尾,則會將整個文件內容放進緩沖,並將文件終止標識符io.EOF放進設置為err。

通常無需單獨定義這些變量,下面是更常見的用法。

inputReader := bufio.NewReader(os.Stdin)
input, err := inputReader.ReadString('\n')


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM