Properly passing data on stdin to a command and receiving data from stdout of that command in golang

后端 未结 2 1414
遥遥无期
遥遥无期 2021-02-05 20:49

I have the following program:

package main

import \"bytes\"
import \"io\"
import \"log\"
import \"os\"
import \"os/exec\"
import \"time\"

func main() {
    run         


        
2条回答
  •  不知归路
    2021-02-05 21:01

    Here is a version of your first code which works. Note the addition of the sync.WaitGroup to make sure you finish with the sending and receiving go routines before closing the command.

    package main
    
    import (
        "bytes"
        "io"
        "log"
        "os"
        "os/exec"
        "sync"
        "time"
    )
    
    func main() {
        runCatFromStdinWorks(populateStdin("aaa\n"))
        runCatFromStdinWorks(populateStdin("bbb\n"))
    }
    
    func populateStdin(str string) func(io.WriteCloser) {
        return func(stdin io.WriteCloser) {
            defer stdin.Close()
            io.Copy(stdin, bytes.NewBufferString(str))
        }
    }
    
    func runCatFromStdinWorks(populate_stdin_func func(io.WriteCloser)) {
        cmd := exec.Command("cat")
        stdin, err := cmd.StdinPipe()
        if err != nil {
            log.Panic(err)
        }
        stdout, err := cmd.StdoutPipe()
        if err != nil {
            log.Panic(err)
        }
        err = cmd.Start()
        if err != nil {
            log.Panic(err)
        }
        var wg sync.WaitGroup
        wg.Add(2)
        go func() {
            defer wg.Done()
            populate_stdin_func(stdin)
        }()
        go func() {
            defer wg.Done()
            time.Sleep(5 * time.Second)
            io.Copy(os.Stdout, stdout)
        }()
        wg.Wait()
        err = cmd.Wait()
        if err != nil {
            log.Panic(err)
        }
    }
    

    (This is just another way of saying what @peterSO said though ;-)

提交回复
热议问题