使用 exec.Command 将输出重定向到另一个 Powershell

一分耕耘，一分收获！既然打开了这篇文章《使用 exec.Command 将输出重定向到另一个 Powershell》，就坚持看下去吧！文中内容包含等等知识点...希望你能在阅读本文后，能真真实实学到知识或者帮你解决心中的疑惑，也欢迎大佬或者新人朋友们多留言评论，多给建议！谢谢！

我是 go 新手，遇到以下问题。我需要使用 os.exec 与 powershell 交互并捕获其中每个命令/管道的输出。

例如我有以下命令

powershell /c cat somefile.md | powershell /c cat somefile.md |选择字符串-pattern someinput

我需要第一个命令的输出

powershell /c cat somefile.md

和管道的输出

powershell /c 选择字符串-pattern someinput

下面的代码在 linux 上运行良好，在 windows 上使用 cmd，我必须使用 powershell...

编辑：经过进一步调查，我得出的结论是，无论出于何种原因，select-string 和其他 powershell 特定命令都无法通过管道传输。是因为 powershell 需要 psobject 作为输入吗？

这是我现在的混乱代码：

func executeCmds{
    //save all exec.Commands in one array (I've no clue how many cmds there are later)
    var cmdList []exec.Cmds 

    cmdList = append(cmdList, *exec.Command("powershell", "/C", "cat somefile.md"))

    //*****EDIT*****//

    // not working $Input is missing
    cmdList = append(cmdList, *exec.Command("powershell", "/C", "Select-String -Pattern someinput"))

    // working command $Input is important
    cmdList = append(cmdList, exec.Command("powershell", "/C", "$Input | Select-String -Pattern someinput"))

    // this is performance wise NOT an option!
    // cmdList = append(cmdList, *exec.Command("powershell", "/C", "cat somefile.md | Select-String -Pattern someinput"))
    
    // save the latest stdout into tmp to stdin the next command (works fine on linux and cmd.exe)
    var tmp []byte

    for i, s := range cmdList {
        if i > 0 {
            s.Stdin = strings.NewReader(string(tmp)) //this does not work, powershell somehow can't use Stdin
        }

        stdout, err := s.CombinedOutput()
        tmp = stdout

        if err != nil {
            o := fmt.Sprint(err) + ": " + string(stdout)
            err = errors.New(o)
            panic(err)
            break
        }

        // Print the output
        fmt.Println(string(stdout))
    }
}

正确答案

如果您需要通过 go 应用程序传递数据，那么您可以执行如下操作（我不确定这将如何提高使用 powershell 管道的性能）：

package main

import (
    "fmt"
    "io"
    "os/exec"
)

func main() {
    // save all exec.commands in one array (i've no clue how many cmds there are later)
    var cmdlist []*exec.cmd

    cmdlist = append(cmdlist, exec.command("powershell", "/c", "cat somefile.md"))
    cmdlist = append(cmdlist, exec.command("powershell", "/c", "$input | select-string -pattern someinput"))

    var tmp []byte
    for i, s := range cmdlist {
        if i > 0 { // send the output of the previous command into the next command
            input, err := s.stdinpipe()
            if err != nil {
                panic(err)
            }
            go func(w io.writecloser, data []byte) {
                w.write(data) // should really check for errors here
                w.close()
            }(input, tmp)
        }
        var err error
        tmp, err = s.combinedoutput() // as you are taking standard error here too you should check for errors
        if err != nil {
            panic(err)
        }
    }
    // print the final output
    fmt.println(string(tmp))
}

您不需要使用 powershell，直接使用 go 即可：

package main
import "golang.org/x/sys/windows/registry"

func main() {
   a, e := registry.CURRENT_USER.ReadSubKeyNames(0)
   if e != nil {
      panic(e)
   }
   for _, s := range a {
      println(s)
   }
}

https://pkg.go.dev/golang.org/x/sys/windows/registry

到这里，我们也就讲完了《使用 exec.Command 将输出重定向到另一个 Powershell》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！