例子程序 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()
函數(shù)的實(shí)際簽名
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()
依據(jù)指定的格式向第一個參數(shù)內(nèi)寫入字符串,第一個參數(shù)必須實(shí)現(xiàn)了 io.Writer
接口。Fprintf()
能夠?qū)懭肴魏晤愋?,只要其?shí)現(xiàn)了 Write
方法,包括 os.Stdout
,文件(例如 os.File),管道,網(wǎng)絡(luò)連接,通道等等,同樣的也可以使用 bufio 包中緩沖寫入。bufio 包中定義了 type Writer struct{...}
。
bufio.Writer 實(shí)現(xiàn)了 Write 方法:
func (b *Writer) Write(p []byte) (nn int, err error)
它還有一個工廠函數(shù):傳給它一個 io.Writer
類型的參數(shù),它會返回一個帶緩沖的 bufio.Writer
類型的 io.Writer
:
func NewWriter(wr io.Writer) (b *Writer)
其適合任何形式的緩沖寫入。
在緩沖寫入的最后千萬不要忘了使用 Flush()
,否則最后的輸出不會被寫入。
在 15.2-15.8 章節(jié),我們將使用 fmt.Fprint
函數(shù)向 http.ResponseWriter
寫入,其同樣實(shí)現(xiàn)了 io.Writer 接口。
練習(xí) 12.7:remove_3till5char.go
下面的代碼有一個輸入文件 goprogram
,然后以每一行為單位讀取,從讀取的當(dāng)前行中截取第 3 到第 5 的字節(jié)寫入另一個文件。然而當(dāng)你運(yùn)行這個程序,輸出的文件卻是個空文件。找出程序邏輯中的 bug,修正它并測試。
package main
import (
"bufio"
"fmt"
"os"
"io"
)
func main() {
inputFile, _ := os.Open("goprogram")
outputFile, _ := os.OpenFile("goprogramT", 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(inputString[2:5]) + "\r\n"
_, err := outputWriter.WriteString(outputString)
if err != nil {
fmt.Println(err)
return
}
}
fmt.Println("Conversion done")
}