Go原生的pkg中有一些核心的interface，其中io.Reader/Writer是比較常用的接口。很多原生的結構都圍繞這個系列的接口展開，在實際的開發過程中，你會發現通過這個接口可以在多種不同的io類型之間進行過渡和轉化。本文結合實際場景來總結一番。

總覽

圍繞io.Reader/Writer，有幾個常用的實現：

• net.Conn, os.Stdin, os.File: 網絡、標準輸入輸出、文件的流讀取
• strings.Reader: 把字符串抽象成Reader
• bytes.Reader: 把[]byte抽象成Reader
• bytes.Buffer: 把[]byte抽象成Reader和Writer
• bufio.Reader/Writer: 抽象成帶緩沖的流讀取（比如按行讀寫）

這些實現對于初學者來說其實比較難去記憶，在遇到實際問題的時候更是一臉蒙圈，不知如何是好。下面用實際的場景來舉例

場景舉例

0. base64編碼成字符串

encoding/base64包中：

func NewEncoder(enc *Encoding, w io.Writer) io.WriteCloser

這個用來做base64編碼，但是仔細觀察發現，它需要一個io.Writer作為輸出目標，并用返回的WriteCloser的Write方法將結果寫入目標，下面是Go官方文檔的例子

input := []byte("foo\x00bar") encoder := base64.NewEncoder(base64.StdEncoding, os.Stdout) encoder.Write(input)

這個例子是將結果寫入到Stdout，如果我們希望得到一個字符串呢？觀察上面的圖，不然發現可以用bytes.Buffer作為目標io.Writer：

input := []byte("foo\x00bar") buffer := new(bytes.Buffer) encoder := base64.NewEncoder(base64.StdEncoding, buffer) encoder.Write(input) fmt.Println(string(buffer.Bytes())

1. []byte和struct之間正反序列化

這種場景經常用在基于字節的協議上，比如有一個具有固定長度的結構：

type Protocol struct {Version uint8BodyLen uint16Reserved [2]byteUnit uint8Value uint32 }

通過一個[]byte來反序列化得到這個Protocol，一種思路是遍歷這個[]byte，然后逐一賦值。其實在encoding/binary包中有個方便的方法：

func Read(r io.Reader, order ByteOrder, data interface{}) error

這個方法從一個io.Reader中讀取字節，并已order指定的端模式，來給填充data（data需要是fixed-sized的結構或者類型）。要用到這個方法首先要有一個io.Reader，從上面的圖中不難發現，我們可以這么寫：

var p Protocol var bin []byte //... binary.Read(bytes.NewReader(bin), binary.LittleEndian, &p)

換句話說，我們將一個[]byte轉成了一個io.Reader。

反過來，我們需要將Protocol序列化得到[]byte，使用encoding/binary包中有個對應的Write方法：

func Write(w io.Writer, order ByteOrder, data interface{}) error

通過將[]byte轉成一個io.Writer即可：

var p Protocol buffer := new(bytes.Buffer) //... binary.Writer(buffer, binary.LittleEndian, p) bin := buffer.Bytes()

2. 從流中按行讀取

比如對于常見的基于文本行的HTTP協議的讀取，我們需要將一個流按照行來讀取。本質上，我們需要一個基于緩沖的讀寫機制（讀一些到緩沖，然后遍歷緩沖中我們關心的字節或字符）。在Go中有一個bufio的包可以實現帶緩沖的讀寫：

func NewReader(rd io.Reader) *Reader func (b *Reader) ReadString(delim byte) (string, error)

這個ReadString方法從io.Reader中讀取字符串，直到delim，就返回delim和之前的字符串。如果將delim設置為\n，相當于按行來讀取了：

var conn net.Conn //... reader := NewReader(conn) for {line, err := reader.ReadString([]byte('\n'))//... }

花式技（zuo）巧（si）

string轉[]byte

a := "Hello, playground" fmt.Println([]byte(a))

等價于

a := "Hello, playground" buf := new(bytes.Buffer) buf.ReadFrom(strings.NewReader(a)) fmt.Println(buf.Bytes())

總結

以上是生活随笔為你收集整理的Go编程技巧--io.Reader/Writer的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。