Go語言使用切片讀寫檔案

切片提供了 Go語言中處理 I/O 緩衝的標準方式，下面 cat 函數的第二版中，在一個切片緩衝內使用無限 for 迴圈（直到檔案尾部 EOF）讀取檔案，並寫入到標準輸出（os.Stdout）。

func cat(f *os.File) {
    const NBUF = 512
    var buf [NBUF]byte
    for {
        switch nr, err := f.Read(buf[:]); true {
            case nr < 0:
                fmt.Fprintf(os.Stderr, "cat: error reading: %sn", err.Error())
                os.Exit(1)
            case nr == 0: // EOF
                return
            case nr > 0:
                if nw, ew := os.Stdout.Write(buf[0:nr]); nw != nr {
                    fmt.Fprintf(os.Stderr, "cat: error writing: %sn", ew.Error())
                }
        }
    }
}

下面的程式碼使用了 os 包中的 os.file 和 Read 方法；與上一節範例程式碼具有同樣的功能。

package main
import (
    "flag"
    "fmt"
    "os"
)
func cat(f *os.File) {
    const NBUF = 512
    var buf [NBUF]byte
    for {
        switch nr, err := f.Read(buf[:]); true {
            case nr < 0:
                fmt.Fprintf(os.Stderr, "cat: error reading: %sn", err.Error())
                os.Exit(1)
            case nr == 0: // EOF
                return
            case nr > 0:
                if nw, ew := os.Stdout.Write(buf[0:nr]); nw != nr {
                    fmt.Fprintf(os.Stderr, "cat: error writing: %sn", ew.Error())
                }
        }
    }
}
func main() {
    flag.Parse() // 掃描arg列表並設定標誌
    if flag.NArg() == 0 {
        cat(os.Stdin)
    }
    for i := 0; i < flag.NArg(); i++ {
        f, err := os.Open(flag.Arg(i))
        if f == nil {
            fmt.Fprintf(os.Stderr, "cat: can't open %s: error %sn", flag.Arg(i), err)
            os.Exit(1)
        }
        cat(f)
        f.Close()
    }
}