读取 os.stdout

目前golang学习网上已经有很多关于Golang的文章了，自己在初次阅读这些文章中，也见识到了很多学习思路；那么本文《读取 os.stdout》，也希望能帮助到大家，如果阅读完后真的对你学习Golang有帮助，欢迎动动手指，评论留言并分享~

我想将日志信息发送到套接字，为此，我需要首先捕获 os.stdout。我知道我可以使用 os.pipe 重定向 os.stdout。但是有没有办法使用 bufio.newreader 或 bufio.newscanner 直接从 os.stdout 读取？

func Start() {
   //dataChan := make(chan string)

   outC := make(chan string, 3)
   defer close(outC)
   conn, err := net.Dial("tcp", "localhost:9090")

   if err != nil {
       log.Fatal(err)
   }

   fmt.Println("first line!")
   fmt.Println("second line!")
   fmt.Println("third line!")
   // write to channel
   go func() {
       scanner := bufio.NewScanner(os.Stdout)
       for scanner.Scan() {
           outC <- scanner.Text() + "\n"
           err = scanner.Err()

           if err != nil {
               log.Fatal(err)
           }
       }
   }()

   // read from channel and print to connection
   go func() {
       out := <-outC

       for {
           conn.Write([]byte(out + "\n"))
       }

   }()

}

正确答案

您可以通过 os.Pipe 读取标准输出

    old := os.Stdout
    r, w, _ := os.Pipe()
    os.Stdout = w

    go func() {
        var buf bytes.Buffer
        io.Copy(&buf, r)
        outC <- buf.String()
    }()

    fmt.Println("first line!")
    fmt.Println("second line!")
    fmt.Println("third line!")

    w.Close()
    os.Stdout = old // restore stdout

    for {
        select {
        case out := <-outC:
            conn.Write([]byte("read " + out + "\n"))
        }
    }

理论要掌握，实操不能落！以上关于《读取 os.stdout》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！