我正在嘗試在 Go 中自動化一個過程。我已經能夠實現線程并相應地執行過程，但是輸出是混合和匹配的。我想知道是否有一種方法可以顯示程序根據程序的過程生成的輸出。因此，如果任務 A 在任務 B 之前完成，我們會在 B 之前顯示 A 的輸出，反之亦然。package mainimport ( "fmt" "log" "os" "os/exec" "sync")var url stringvar wg sync.WaitGroupfunc nikto() { cmd := exec.Command("nikto", "-h", url) cmd.Stdout = os.Stdout err := cmd.Run() if err != nil { log.Fatal(err) } wg.Done()}func whois() { cmd := exec.Command("whois", "google.co") cmd.Stdout = os.Stdout err := cmd.Run() if err != nil { log.Fatal(err) } wg.Done()}func main() { fmt.Printf("Please input URL") fmt.Scanln(&url) wg.Add(1) go nikto() wg.Add(1) go whois() wg.Wait()}
1 回答
手掌心
TA貢獻1942條經驗 獲得超3個贊
在您的進程中，您將os.Stdout文件描述符直接傳遞給您調用以運行子進程的命令。這意味著子進程的 STDOUT 管道將直接連接到您的 Go 程序的標準輸出，并且如果兩個子進程同時寫入，則可能會交錯。
解決這個問題的最簡單方法需要您在 Go 程序中緩沖來自子進程的 STDOUT 管道的輸出，這樣您就可以攔截輸出并在打印時進行控制。
Cmd包中的類型提供os/exec了一個函數調用Output()，它將調用子進程并以字節片的形式返回 STDOUT 的內容。您的代碼可以輕松調整以實現此模式并處理結果，例如：
func whois() {
cmd := exec.Command("whois", "google.co")
out, err := cmd.Output()
if err != nil {
log.Fatal(err)
}
fmt.Println(out)
wg.Done()
}
交織輸出
如果您使用fmt包中的函數來打印輸出，則不能保證并發調用fmt.Println不會交錯。
為防止交錯，您可以選擇序列化對 STDOUT 的訪問，或使用可安全并發使用的記錄器（例如包log）。下面是一個在 Go 進程中序列化訪問 STDOUT 的例子：
package main
import (
"fmt"
"log"
"os/exec"
"sync"
)
var url string
func nikto(outChan chan<- []byte) {
cmd := exec.Command("nikto", "-h", url)
bs, err := cmd.Output()
if err != nil {
log.Fatal(err)
}
outChan <- bs
}
func whois(outChan chan<- []byte) {
cmd := exec.Command("whois", "google.com")
bs, err := cmd.Output()
if err != nil {
log.Fatal(err)
}
outChan <- bs
}
func main() {
outChan := make(chan []byte)
fmt.Printf("Please input URL")
fmt.Scanln(&url)
go nikto(outChan)
go whois(outChan)
for i := 0; i < 2; i++ {
bs := <-outChan
fmt.Println(string(bs))
}
}
- 1 回答
- 0 關注
- 288 瀏覽
添加回答
舉報
0/150
提交
取消