12.6 用切片读写文件

    1. package main
    2. import (
    3. "flag"
    4. "fmt"
    5. "os"
    6. )
    7. func cat(f *os.File) {
    8. var buf [NBUF]byte
    9. for {
    10. switch nr, err := f.Read(buf[:]); true {
    11. fmt.Fprintf(os.Stderr, "cat: error reading: %s\n", err.Error())
    12. os.Exit(1)
    13. case nr == 0: // EOF
    14. return
    15. case nr > 0:
    16. if nw, ew := os.Stdout.Write(buf[0:nr]); nw != nr {
    17. fmt.Fprintf(os.Stderr, "cat: error writing: %s\n", ew.Error())
    18. }
    19. }
    20. }
    21. }
    22. func main() {
    23. flag.Parse() // Scans the arg list and sets up flags
    24. cat(os.Stdin)
    25. }
    26. for i := 0; i < flag.NArg(); i++ {
    27. f, err := os.Open(flag.Arg(i))
    28. if f == nil {
    29. fmt.Fprintf(os.Stderr, "cat: can't open %s: error %s\n", flag.Arg(i), err)
    30. os.Exit(1)
    31. }
    32. cat(f)
    33. f.Close()
    34. }
    35. }