如何解决如何捕获标准输出并显示进度
我有一个名为print()
的函数,该函数每2秒打印一次数字,该函数在goroutine中运行。
我需要将其stdout打印传递给一个变量并将其打印出来,但是直到完成时才打印一次。
我需要在无限循环中安装一些扫描仪,该扫描仪将扫描stdout并将其打印出来,一旦完成扫描仪的功能也将完成。
我尝试使用this answer,但是它不能打印任何内容。
这是我试图做的:
package main
import (
"bufio"
"fmt"
"os"
"sync"
"time"
)
func print() {
for i := 0; i < 50; i++ {
time.Sleep(2 * time.Second)
fmt.Printf("hello number: %d\n",i)
}
}
func main() {
old := os.Stdout // keep backup of the real stdout
defer func() { os.Stdout = old }()
r,w,_ := os.Pipe()
os.Stdout = w
go print()
var wg sync.WaitGroup
c := make(chan struct{})
wg.Add(1)
defer wg.Done()
for {
<-c
scanner := bufio.NewScanner(r)
for scanner.Scan() {
m := scanner.Text()
fmt.Println("output: " + m)
}
}
c <- struct{}{}
wg.Wait()
fmt.Println("DONE")
}
我还尝试使用io.Copy
来读取这样的缓冲区,但是它也不起作用:
package main
import (
"bytes"
"fmt"
"io"
"os"
"time"
)
func print() {
for i := 0; i < 50; i++ {
time.Sleep(2 * time.Second)
fmt.Printf("hello number: %d\n",i)
}
}
// https://blog.kowalczyk.info/article/wOYk/advanced-command-execution-in-go-with-osexec.html
func main() {
old := os.Stdout // keep backup of the real stdout
defer func() { os.Stdout = old }()
r,_ := os.Pipe()
os.Stdout = w
go print()
fmt.Println("DONE 1")
outC := make(chan string)
for {
var buf bytes.Buffer
io.Copy(&buf,r)
outC <- buf.String()
out := <-outC
fmt.Println("output: " + out)
}
// back to normal state
w.Close()
fmt.Println("DONE")
}
解决方法
可以将print()
作为“黑匣子”运行并捕获其输出,尽管这有点棘手并且在Go Playground上不起作用。
package main
import (
"bufio"
"fmt"
"os"
"runtime"
"time"
)
func print() {
for i := 0; i < 50; i++ {
time.Sleep(100 * time.Millisecond)
fmt.Printf("hello number: %d\n",i)
}
}
func main() {
var ttyName string
if runtime.GOOS == "windows" {
fmt.Println("*** Using `con`")
ttyName = "con"
} else {
fmt.Println("*** Using `/dev/tty`")
ttyName = "/dev/tty"
}
f,err := os.OpenFile(ttyName,os.O_WRONLY,0644)
if err != nil {
panic(err)
}
defer f.Close()
r,w,_ := os.Pipe()
oldStdout := os.Stdout
os.Stdout = w
defer func() {
os.Stdout = oldStdout
fmt.Println("*** DONE")
}()
fmt.Fprintln(f,"*** Stdout redirected")
go func(){
print()
w.Close()
r.Close()
}()
c := make(chan struct{})
go func(){c <- struct{}{}}()
defer close(c)
<-c
scanner := bufio.NewScanner(r)
for scanner.Scan() {
m := scanner.Text()
fmt.Fprintln(f,"output: " + m)
}
}
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。