diff --git a/eBook/12.1.md b/eBook/12.1.md index 32f5b34..8cdf709 100644 --- a/eBook/12.1.md +++ b/eBook/12.1.md @@ -2,7 +2,7 @@ 我们如何读取用户的键盘(控制台)输入呢?从键盘和标准输入 `os.Stdin` 读取输入,最简单的办法是使用 `fmt` 包提供的 Scan 和 Sscan 开头的函数。请看以下程序: -**Listing 12.1—readinput1.go:** +示例 12.1 [readinput1.go](examples/chapter_12/readinput1.go): ```go // 从控制台读取输入: @@ -32,7 +32,7 @@ func main() { 您也可以使用 `bufio` 包提供的缓冲读取(buffered reader)来读取数据,正如以下例子所示: -**Listing 12.2—readinput2.go:** +示例 12.2 [readinput2.go](examples/chapter_12/readinput2.go): ```go package main @@ -80,7 +80,7 @@ input, err := inputReader.ReadString('\n') 第二个例子从键盘读取输入,使用了 `switch` 语句: -**Listing 12.3—switch_input.go:** +示例 12.3 [switch_input.go](examples/chapter_12/switch_input.go): ```go package main import ( @@ -128,13 +128,15 @@ func main() { **练习** -**练习 12.1:** word_letter_count.go +**练习 12.1:** [word_letter_count.go](exercises/chapter_12/word_letter_count.go) + 编写一个程序,从键盘读取输入。当用户输入 'S' 的时候表示输入结束,这时程序输出 3 个数字: i) 输入的字符的个数,包括空格,但不包括 '\r' 和 '\n' ii) 输入的单词的个数 iii) 输入的行数 -**练习 12.2:** calculator.go +**练习 12.2:** [calculator.go](exercises/chapter_12/calculator.go) + 编写一个简单的逆波兰式计算器,它接受用户输入的整型数(最大值 999999)和运算符 +、-、*、/。 输入的格式为:number1 ENTER number2 ENTER operator ENTER --> 显示结果 当用户输入字符 'q' 时,程序结束。请使用您在练习11.3中开发的 `stack` 包。 diff --git a/eBook/12.2.md b/eBook/12.2.md new file mode 100644 index 0000000..a6b3a24 --- /dev/null +++ b/eBook/12.2.md @@ -0,0 +1,162 @@ +# 文件读写 + +## 12.2.1 读文件 + +在 Go 语言中,文件使用指向 `os.File` 类型的指针来表示的,也叫做文件句柄。我们在前面章节使用到过标准输入 `os.Stdin` 和标准输出 `os.Stdout`,他们的类型都是 `*os.File`。让我们来看看下面这个程序: + +示例 12.4 [fileinput.go](examples/chapter_12/fileinput.go): +```go +package main +import ( + "bufio" + "fmt" + "io" + "os" +) + +func main() { + inputFile, inputError := os.Open("input.dat") + if inputError != nil { + fmt.Printf("An error occurred on opening the inputfile\n" + + "Does the file exist?\n" + + "Have you got acces to it?\n") + return // exit the function on error + } + defer inputFile.Close() + + inputReader := bufio.NewReader(inputFile) + for { + inputString, readerError := inputReader.ReadString('\n') + if readerError == io.EOF { + return + } + fmt.Printf("The input was: %s", inputString) + } +} +``` + +变量 `inputFile` 是 `*os.File` 类型的。该类型是一个结构,表示一个打开文件的描述符(文件句柄)。然后,使用 `os` 包里的 `Open` 函数来打开一个文件。该函数的参数是文件名,类型为 `string`。在上面的程序中,我们以只读模式打开 input.dat 文件。 + +如果文件不存在或者程序没有足够的权限打开这个文件,Open函数会返回一个错误:`inputFile, inputError = os.Open("input.dat")`。如果文件打开正常,我们就使用 `defer.Close()` 语句确保在程序退出前关闭该文件。然后,我们使用 `bufio.NewReader` 来获得一个读取器变量。 + +通过使用 `bufio` 包提供的读取器(写入器也类似),如上面程序所示,我们可以很方便的操作相对高层的 string 对象,而避免了去操作比较底层的字节。 + +接着,我们在一个无限循环中使用 `ReadString('\n')` 或 `ReadBytes('\n')` 将文件的内容逐行(行结束符 '\n')读取出来。 + +**注意:** 在之前的例子中,我们看到,Unix和Linux的行结束符是 \n,而Windows的行结束符是 \r\n。在使用 `ReadString` 和 `ReadBytes` 方法的时候,我们不需要关心操作系统的类型,直接使用 \n 就可以了。另外,我们也可以使用 `ReadLine()` 方法来实现相同的功能。 + +一旦读取到文件末尾,变量 `readerError` 的值将变成非空(事实上,常亮 `io.EOF` 的值是 true),我们就会执行 `return` 语句从而退出循环。 + +**其他类似函数:** + +**1) 将整个文件的内容读到一个字符串里:** + +如果您想这么做,可以使用 `io/ioutil` 包里的 `ioutil.ReadFile()` 方法,该方法第一个返回值的类型是 `[]byte`,里面存放读取到的内容,第二个返回值是错误,如果没有错误发生,第二个返回值为 nil。请看示例 12.5。类似的,函数 `WriteFile()` 可以将 `[]byte` 的值写入文件。 + +示例 12.5 [read_write_file1.go](examples/chapter_12/read_write_file1.go): +```go +package main +import ( + "fmt" + "io/ioutil" + "os" +) + +func main() { + inputFile := "products.txt" + outputFile := "products_copy.txt" + buf, err := ioutil.ReadFile(inputFile) + if err != nil { + fmt.Fprintf(os.Stderr, "File Error: %s\n", err) + // panic(err.Error()) + } + fmt.Printf("%s\n", string(buf)) + err = ioutil.WriteFile(outputFile, buf, 0x644) + if err != nil { + panic(err. Error()) + } +} +``` + +**2) 带缓冲的读取** + +在很多情况下,文件的内容是不按行划分的,或者干脆就是一个二进制文件。在这种情况下,`ReadString()`就无法使用了,我们可以使用 `bufio.Reader` 的 `Read()`,它只接收一个参数: +```go +buf := make([]byte, 1024) +... +n, err := inputReader.Read(buf) +if (n == 0) { break} +``` +变量 n 的值表示读取到的字节数. + +**3) 按列读取文件中的数据** + +如果数据是按列排列并用空格分隔的,你可以使用 `fmt` 包提供的以 FScan 开头的一系列函数来读取他们。请看以下程序,我们将 3 列的数据分别读入变量 v1、v2 和 v3 内,然后分别把他们添加到切片的尾部。 + +示例 12.6 [read_file2.go](examples/chapter_12/read_file2.go): +```go +package main +import ( + "fmt" + "os" +) + +func main() { + file, err := os.Open("products2.txt") + if err != nil { + panic(err) + } + defer file.Close() + + var col1, col2, col3 []string + for { + var v1, v2, v3 string + _, err := fmt.Fscanln(file, &v1, &v2, &v3) + // scans until newline + if err != nil { + break + } + col1 = append(col1, v1) + col2 = append(col2, v2) + col3 = append(col3, v3) + } + + fmt.Println(col1) + fmt.Println(col2) + fmt.Println(col3) +} +``` + +输出结果: + +``` +[ABC FUNC GO] +[40 56 45] +[150 280 356] +``` + +**注意:** `path` 包里包含一个子包叫 `filepath`,这个子包提供了跨平台的函数,用于处理文件名和路径。例如 Base() 函数用于获得路径中的最后一个元素(不包含后面的分隔符): +```go +import "path/filepath" +filename := filepath.Base(path) +``` + +**练习 12.3**:[read_csv.go](exercises/chapter_12/read_csv.go) + +文件 products.txt 的内容如下: +``` +"The ABC of Go";25.5;1500 +"Functional Programming with Go";56;280 +"Go for It";45.9;356 +"The Go Way";55;500 +``` +每行的第一个字段为 title,第二个字段为 price,第三个字段为 quantity。内容的格式基本与 示例 12.3c 的相同,除了分隔符改成了分号。请读取出文件的内容,创建一个结构用于存取一行的数据,然后使用结构的切片,并把数据打印出来。 + +关于解析 CSV 文件,`encoding/csv` 包提供了相应的功能。具体请参考 [http://golang.org/pkg/encoding/csv/](http://golang.org/pkg/encoding/csv/) + + +## 链接 + +- [目录](directory.md) +- 上一节:[读取用户的输入](12.1.md) +- 下一节:[文件拷贝](12.3.md)