mirror of
https://github.com/AlexxIT/go2rtc.git
synced 2025-09-27 04:36:12 +08:00
31 lines
590 B
Go
31 lines
590 B
Go
package exec
|
|
|
|
import (
|
|
"bufio"
|
|
"io"
|
|
"os/exec"
|
|
|
|
"github.com/AlexxIT/go2rtc/pkg/core"
|
|
)
|
|
|
|
// PipeCloser - return StdoutPipe that Kill cmd on Close call
|
|
func PipeCloser(cmd *exec.Cmd) (io.ReadCloser, error) {
|
|
stdout, err := cmd.StdoutPipe()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// add buffer for pipe reader to reduce syscall
|
|
return pipeCloser{bufio.NewReaderSize(stdout, core.BufferSize), stdout, cmd}, nil
|
|
}
|
|
|
|
type pipeCloser struct {
|
|
io.Reader
|
|
io.Closer
|
|
cmd *exec.Cmd
|
|
}
|
|
|
|
func (p pipeCloser) Close() error {
|
|
return core.Any(p.Closer.Close(), p.cmd.Process.Kill(), p.cmd.Wait())
|
|
}
|