• 12.8 使用接口的实际例子:fmt.Fprintf

    12.8 使用接口的实际例子:fmt.Fprintf

    例子程序 io_interfaces.go 很好的阐述了 io 包中的接口概念。

    示例 12.15 io_interfaces.go:

    1. // interfaces being used in the GO-package fmt
    2. package main
    3. import (
    4. "bufio"
    5. "fmt"
    6. "os"
    7. )
    8. func main() {
    9. // unbuffered
    10. fmt.Fprintf(os.Stdout, "%s\n", "hello world! - unbuffered")
    11. // buffered: os.Stdout implements io.Writer
    12. buf := bufio.NewWriter(os.Stdout)
    13. // and now so does buf.
    14. fmt.Fprintf(buf, "%s\n", "hello world! - buffered")
    15. buf.Flush()
    16. }

    输出:

    1. hello world! - unbuffered
    2. hello world! - buffered

    下面是 fmt.Fprintf() 函数的实际签名

    1. func Fprintf(w io.Writer, format string, a ...interface{}) (n int, err error)

    其不是写入一个文件,而是写入一个 io.Writer 接口类型的变量,下面是 Writer 接口在 io 包中的定义:

    1. type Writer interface {
    2. Write(p []byte) (n int, err error)
    3. }

    fmt.Fprintf() 依据指定的格式向第一个参数内写入字符串,第一参数必须实现了 io.Writer 接口。Fprintf() 能够写入任何类型,只要其实现了 Write 方法,包括 os.Stdout,文件(例如 os.File),管道,网络连接,通道等等,同样的也可以使用 bufio 包中缓冲写入。bufio 包中定义了 type Writer struct{...}

    bufio.Writer 实现了 Write 方法:

    1. func (b *Writer) Write(p []byte) (nn int, err error)

    它还有一个工厂函数:传给它一个 io.Writer 类型的参数,它会返回一个缓冲的 bufio.Writer 类型的 io.Writer:

    1. func NewWriter(wr io.Writer) (b *Writer)

    其适合任何形式的缓冲写入。

    在缓冲写入的最后千万不要忘了使用 Flush(),否则最后的输出不会被写入。

    在 15.2-15.8 章节,我们将使用 fmt.Fprint 函数向 http.ResponseWriter 写入,其同样实现了 io.Writer 接口。

    练习 12.7:remove_3till5char.go

    下面的代码有一个输入文件 goprogram,然后以每一行为单位读取,从读取的当前行中截取第 3 到第 5 的字节写入另一个文件。然而当你运行这个程序,输出的文件却是个空文件。找出程序逻辑中的 bug,修正它并测试。

    1. package main
    2. import (
    3. "bufio"
    4. "fmt"
    5. "os"
    6. "io"
    7. )
    8. func main() {
    9. inputFile, _ := os.Open("goprogram")
    10. outputFile, _ := os.OpenFile("goprogramT", os.O_WRONLY|os.O_CREATE, 0666)
    11. defer inputFile.Close()
    12. defer outputFile.Close()
    13. inputReader := bufio.NewReader(inputFile)
    14. outputWriter := bufio.NewWriter(outputFile)
    15. for {
    16. inputString, _, readerError := inputReader.ReadLine()
    17. if readerError == io.EOF {
    18. fmt.Println("EOF")
    19. return
    20. }
    21. outputString := string(inputString[2:5]) + "\r\n"
    22. _, err := outputWriter.WriteString(outputString)
    23. if err != nil {
    24. fmt.Println(err)
    25. return
    26. }
    27. }
    28. fmt.Println("Conversion done")
    29. }