假設(shè)我們需要在Go中運(yùn)行下面的命令：

PS -A | grep wget

這里需要寫成兩個(gè)exec.Command，如下，第一個(gè)命令為cmd，第二個(gè)為cmd2：

cmd := exec.Command("PS", "-A")
cmd2 := exec.Command("grep", "wget")

然后使用管道連接二者的標(biāo)準(zhǔn)輸出和標(biāo)準(zhǔn)輸入，需要注意第一個(gè)命令cmd的標(biāo)準(zhǔn)輸出應(yīng)該使用cmd.StdoutPipe()，而不是Stdout，如下（忽略了錯(cuò)誤err和其處理）：

cmd2.Stdin, _ = cmd.StdoutPipe()

因?yàn)?code>cmd.Stdout是一個(gè)io.Writer，是一個(gè)寫入器，因?yàn)檫@個(gè)輸出是要寫入某些地方的。而同理，cmd2.Stdin是一個(gè)io.Reader，是一個(gè)讀取器，用來讀取一些地方的內(nèi)容。二者直接賦值的話會(huì)出現(xiàn)類型不匹配的錯(cuò)誤。所以需要使用StdoutPipe()函數(shù)，這個(gè)函數(shù)會(huì)返回一個(gè)io.Reader。（這里比較繞，所以可能需要想一下）

在獲取了輸出之后，需要將其轉(zhuǎn)換成字符串的話，可以使用bytes.Buffer來獲取cmd2.Stdout的標(biāo)準(zhǔn)輸出（記住這是個(gè)io.Writer），然后再轉(zhuǎn)換成字符串。

我們是是無法直接將io.Writer直接寫入到bytes.Buffer之中的，你可能會(huì)說bytes.Buffer不是有兩個(gè)方法ReadFrom和WriteTo嗎？

前者只能讀取io.Reader的，后者只能寫入io.Writer，所以我們需要一個(gè)管道來將io.Writer轉(zhuǎn)換成io.Reader，然后才能讀取或復(fù)制其內(nèi)容。而這個(gè)轉(zhuǎn)換就是再使用一次管道，如下：

var buf bytes.Buffer
r, w, _ := os.Pipe()
cmd2.Stdout = w
go buf.ReadFrom(r)

這里buf.ReadFrom(r)必須使用 goroutine，也就是讓這個(gè)代碼并行運(yùn)行，所以在前面加上go。

這里的go buf.ReadFrom(r)也可以使用go io.Copy(&buf, r)替代，效果一樣。

因?yàn)槊顖?zhí)行的順序是先啟動(dòng)cmd2，然后運(yùn)行cmd，cmd運(yùn)行完之后，數(shù)據(jù)流通過管道傳遞給cmd2，cmd2再運(yùn)行。不然cmd運(yùn)行的時(shí)候的標(biāo)準(zhǔn)輸出是空的，就會(huì)一直等。

buf.ReadFrom(r)對(duì)cmd2也是同理，不過由于這是行代碼，無法使用start啟動(dòng)它，所以并行就行了。

接下來的命令如下：

cmd2.Start()
cmd.Run()
cmd2.Wait()

這里就是前面說的流程：cmd2啟動(dòng)，運(yùn)行cmd，讓cmd2等待cmd的輸出。

需要注意一點(diǎn)：go buf.ReadFrom(r)其實(shí)可以放在上面代碼中，除了最后一行之外的任何地方。之所以不能放在最后是因?yàn)檫@時(shí)候都運(yùn)行完了，再讀取就是空的了。

然后將其轉(zhuǎn)換成字符串：

str := buf.String()

打印看看：

fmt.Println("123" + str + "123")

之所以要前后都加上"123"是為了避免調(diào)試的時(shí)候把輸出到標(biāo)準(zhǔn)輸出文件的內(nèi)容當(dāng)成這里打印的。

結(jié)果如下：

% go run main.go
12360651 ttys010    0:00.00 grep wget
123

這里分成兩行是因?yàn)楂@取的時(shí)候grep wget后面有個(gè)\n，這里看不出來，如果%#v格式化打印就能看到了。

完整代碼如下：

func main() {
	cmd := exec.Command("PS", "-A")
	cmd2 := exec.Command("grep", "wget")
	cmd2.Stdin, _ = cmd.StdoutPipe()
	var buf bytes.Buffer
	r, w, _ := os.Pipe()
	cmd2.Stdout = w
	// 下面這行代碼可以替換為：go io.Copy(&buf, r)
	go buf.ReadFrom(r)
	cmd2.Start()
	cmd.Run()
	cmd2.Wait()
	str := buf.String()
	fmt.Println("123" + str + "123")
}

到此這篇關(guān)于Go實(shí)現(xiàn)將io.Writer轉(zhuǎn)換成字符串的文章就介紹到這了,更多相關(guān)Go io.Writer轉(zhuǎn)換成字符串內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論