12.8 使用接口的實際例子：fmt.Fprintf

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

示例 12.15 io_interfaces.go：

// interfaces being used in the GO-package fmt
package main

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

func main() {
    // unbuffered
    fmt.Fprintf(os.Stdout, "%s\n", "hello world! - unbuffered")
    // buffered: os.Stdout implements io.Writer
    buf := bufio.NewWriter(os.Stdout)
    // and now so does buf.
    fmt.Fprintf(buf, "%s\n", "hello world! - buffered")
    buf.Flush()
}

輸出：

hello world! - unbuffered
hello world! - buffered

下面是 fmt.Fprintf() 函數的實際簽名

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

其不是寫入一個文件，而是寫入一個 io.Writer 接口類型的變量，下面是 Writer 接口在 io 包中的定義：

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

fmt.Fprintf() 依據指定的格式向第一個參數內寫入字符串，第一參數必須實現了 io.Writer 接口。Fprintf() 能夠寫入任何類型，只要其實現了 Write 方法，包括 os.Stdout,文件（例如 os.File），管道，網絡連接，通道等等，同樣的也可以使用 bufio 包中緩衝寫入。bufio 包中定義了 type Writer struct{...}。

bufio.Writer 實現了 Write 方法：

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

它還有一個工廠函數：傳給它一個 io.Writer 類型的參數，它會返回一個緩衝的 bufio.Writer 類型的 io.Writer:

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

其適合任何形式的緩衝寫入。

在緩衝寫入的最後千萬不要忘了使用 Flush()，否則最後的輸出不會被寫入。

在 15.2-15.8 章節，我們將使用 fmt.Fprint 函數向 http.ResponseWriter 寫入，其同樣實現了 io.Writer 接口。

練習 12.7：remove_3till5char.go

下面的代碼有一個輸入文件 goprogram.go，然後以每一行爲單位讀取，從讀取的當前行中截取第 3 到第 5 的字節寫入另一個文件。然而當你運行這個程序，輸出的文件卻是個空文件。找出程序邏輯中的 bug，修正它並測試。

package main

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

func main() {
    inputFile, _ := os.Open("goprogram.go")
    outputFile, _ := os.OpenFile("goprogramT.go", os.O_WRONLY|os.O_CREATE, 0666)
    defer inputFile.Close()
    defer outputFile.Close()
    inputReader := bufio.NewReader(inputFile)
    outputWriter := bufio.NewWriter(outputFile)
    for {
        inputString, _, readerError := inputReader.ReadLine()
        if readerError == io.EOF {
            fmt.Println("EOF")
            return
        }
        outputString := string([]byte(inputString)[2:5]) + "\r\n"
        n, err := outputWriter.WriteString(outputString)
        if err != nil {
            fmt.Println(err)
            return
        }
    }
    fmt.Println("Conversion done")
}

鏈接

• 目錄
• 上一節：用 defer 關閉文件
• 下一節：格式化 Json 數據