初始化项目
This commit is contained in:
zeaslity
257
agent-wdd/op/Excutor.go
Normal file
257
agent-wdd/op/Excutor.go
Normal file
@@ -0,0 +1,257 @@
|
||||
package op
|
||||
|
||||
import (
|
||||
"agent-wdd/log"
|
||||
"bufio"
|
||||
"bytes"
|
||||
"fmt"
|
||||
"io"
|
||||
"os/exec"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// SingleLineCommandExecutor 执行单行shell命令，返回执行结果和输出内容
|
||||
// singleLineCommand: 命令及参数，如 []string{"ls", "-l"}
|
||||
// 返回值:
|
||||
//
|
||||
// bool - 命令是否执行成功（true为成功，false为失败）
|
||||
// []string - 合并后的标准输出和标准错误内容（按行分割）
|
||||
func SingleLineCommandExecutor(singleLineCommand []string) (ok bool, resultLog []string) {
|
||||
|
||||
if len(singleLineCommand) == 0 {
|
||||
return false, nil
|
||||
}
|
||||
|
||||
log.Info("start => %v", strings.Join(singleLineCommand, " "))
|
||||
|
||||
// 创建命令实例
|
||||
cmd := exec.Command(singleLineCommand[0], singleLineCommand[1:]...)
|
||||
|
||||
// 创建输出缓冲区
|
||||
var stdoutBuf, stderrBuf bytes.Buffer
|
||||
cmd.Stdout = &stdoutBuf
|
||||
cmd.Stderr = &stderrBuf
|
||||
|
||||
// 执行命令并获取错误信息
|
||||
err := cmd.Run()
|
||||
|
||||
// 合并输出结果
|
||||
output := mergeOutput(&stdoutBuf, &stderrBuf)
|
||||
|
||||
// 判断执行结果
|
||||
return err == nil, output
|
||||
}
|
||||
|
||||
// 若追求极致性能，可优化为流式合并
|
||||
func mergeOutput(stdout, stderr *bytes.Buffer) []string {
|
||||
combined := make([]string, 0, bytes.Count(stdout.Bytes(), []byte{'\n'})+bytes.Count(stderr.Bytes(), []byte{'\n'}))
|
||||
combined = append(combined, strings.Split(stdout.String(), "\n")...)
|
||||
combined = append(combined, strings.Split(stderr.String(), "\n")...)
|
||||
return combined
|
||||
}
|
||||
|
||||
// mergeOutput 合并标准输出和标准错误，按行分割
|
||||
//func mergeOutput(stdoutBuf, stderrBuf *bytes.Buffer) []string {
|
||||
// var output []string
|
||||
//
|
||||
// // 处理标准输出
|
||||
// scanner := bufio.NewScanner(stdoutBuf)
|
||||
// for scanner.Scan() {
|
||||
// output = append(output, scanner.Text())
|
||||
// }
|
||||
//
|
||||
// // 处理标准错误
|
||||
// scanner = bufio.NewScanner(stderrBuf)
|
||||
// for scanner.Scan() {
|
||||
// output = append(output, scanner.Text())
|
||||
// }
|
||||
//
|
||||
// return output
|
||||
//}
|
||||
|
||||
// PipeLineCommandExecutor 执行管道命令，返回执行结果和合并后的输出内容
|
||||
// pipeLineCommand: 管道命令组，如 [][]string{{"ps", "aux"}, {"grep", "nginx"}, {"wc", "-l"}}
|
||||
// 返回值:
|
||||
//
|
||||
// bool - 所有命令是否全部执行成功
|
||||
// []string - 合并后的输出内容（中间命令的stderr + 最后一个命令的stdout/stderr）
|
||||
func PipeLineCommandExecutor(pipeLineCommand [][]string) (ok bool, resultLog []string) {
|
||||
if len(pipeLineCommand) == 0 {
|
||||
return false, nil
|
||||
}
|
||||
|
||||
// 转换为 管道命令的字符串格式
|
||||
// [][]string{{"ps", "aux"}, {"grep", "nginx"}, {"wc", "-l"}} 转换为 ps aux | grep nginx | wc -l
|
||||
pipeLineCommandStr := ""
|
||||
|
||||
// 预检所有子命令
|
||||
for _, cmd := range pipeLineCommand {
|
||||
if len(cmd) == 0 {
|
||||
return false, nil
|
||||
}
|
||||
pipeLineCommandStr += fmt.Sprintf("%s | ", strings.Join(cmd, " "))
|
||||
}
|
||||
|
||||
pipeLineCommandStr = strings.TrimSuffix(pipeLineCommandStr, " | ")
|
||||
log.Info("start => %v", pipeLineCommandStr)
|
||||
|
||||
// 创建命令组
|
||||
cmds := make([]*exec.Cmd, len(pipeLineCommand))
|
||||
for i, args := range pipeLineCommand {
|
||||
cmds[i] = exec.Command(args[0], args[1:]...)
|
||||
}
|
||||
|
||||
// 建立管道连接
|
||||
for i := 0; i < len(cmds)-1; i++ {
|
||||
stdoutPipe, err := cmds[i].StdoutPipe()
|
||||
if err != nil {
|
||||
return false, []string{err.Error()}
|
||||
}
|
||||
cmds[i+1].Stdin = stdoutPipe
|
||||
}
|
||||
|
||||
// 准备输出捕获
|
||||
stderrBuffers := make([]bytes.Buffer, len(cmds)) // 所有命令的stderr
|
||||
var lastStdout bytes.Buffer // 最后一个命令的stdout
|
||||
|
||||
// 绑定输出
|
||||
for i, cmd := range cmds {
|
||||
cmd.Stderr = &stderrBuffers[i] // 每个命令单独捕获stderr
|
||||
}
|
||||
cmds[len(cmds)-1].Stdout = &lastStdout // 仅捕获最后一个命令的stdout
|
||||
|
||||
// 启动所有命令
|
||||
started := make([]*exec.Cmd, 0, len(cmds))
|
||||
defer func() {
|
||||
// 异常时清理已启动进程
|
||||
for _, cmd := range started {
|
||||
if cmd.Process != nil {
|
||||
cmd.Process.Kill()
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
for _, cmd := range cmds {
|
||||
if err := cmd.Start(); err != nil {
|
||||
return false, []string{err.Error()}
|
||||
}
|
||||
started = append(started, cmd)
|
||||
}
|
||||
|
||||
// 等待所有命令完成
|
||||
success := true
|
||||
for _, cmd := range cmds {
|
||||
if err := cmd.Wait(); err != nil {
|
||||
success = false
|
||||
}
|
||||
}
|
||||
|
||||
// 合并输出内容
|
||||
output := make([]string, 0)
|
||||
|
||||
// 合并中间命令的stderr（按命令顺序）
|
||||
for i := 0; i < len(cmds)-1; i++ {
|
||||
scanner := bufio.NewScanner(&stderrBuffers[i])
|
||||
for scanner.Scan() {
|
||||
output = append(output, scanner.Text())
|
||||
}
|
||||
}
|
||||
|
||||
// 合并最后一个命令的输出（stdout在前 + stderr在后）
|
||||
scanner := bufio.NewScanner(&lastStdout)
|
||||
for scanner.Scan() {
|
||||
output = append(output, scanner.Text())
|
||||
}
|
||||
scanner = bufio.NewScanner(&stderrBuffers[len(cmds)-1])
|
||||
for scanner.Scan() {
|
||||
output = append(output, scanner.Text())
|
||||
}
|
||||
|
||||
return success, output
|
||||
}
|
||||
|
||||
// RealTimeCommandExecutor 执行命令，需要实时打印输出执行命令的日志，并返回执行结果和输出内容
|
||||
// realTimeCommand: 实时命令，如 "docker load -i /root/wdd/harbor/harbor-offline-installer-v2.10.1.tgz"
|
||||
// 返回值:
|
||||
//
|
||||
// bool - 命令是否执行成功（true为成功，false为失败）
|
||||
// []string - 合并后的标准输出和标准错误内容（按行分割）
|
||||
func RealTimeCommandExecutor(realTimeCommand []string) (ok bool, resultLog []string) {
|
||||
if len(realTimeCommand) == 0 {
|
||||
return false, nil
|
||||
}
|
||||
|
||||
log.Info("start real time command => %v", strings.Join(realTimeCommand, " "))
|
||||
|
||||
cmd := exec.Command(realTimeCommand[0], realTimeCommand[1:]...)
|
||||
|
||||
stdout, err := cmd.StdoutPipe()
|
||||
if err != nil {
|
||||
return false, []string{err.Error()}
|
||||
}
|
||||
|
||||
stderr, err := cmd.StderrPipe()
|
||||
if err != nil {
|
||||
return false, []string{err.Error()}
|
||||
}
|
||||
|
||||
if err := cmd.Start(); err != nil {
|
||||
return false, []string{err.Error()}
|
||||
}
|
||||
|
||||
output := make([]string, 0)
|
||||
scanner := bufio.NewScanner(io.MultiReader(stdout, stderr))
|
||||
for scanner.Scan() {
|
||||
line := scanner.Text()
|
||||
fmt.Println(line)
|
||||
output = append(output, line)
|
||||
}
|
||||
|
||||
if err := cmd.Wait(); err != nil {
|
||||
return false, append(output, err.Error())
|
||||
}
|
||||
|
||||
return true, output
|
||||
}
|
||||
|
||||
func main() {
|
||||
// 成功案例
|
||||
success, output := SingleLineCommandExecutor([]string{"ls", "-l"})
|
||||
fmt.Println("执行成功:", success)
|
||||
fmt.Println("输出内容:", output)
|
||||
|
||||
// 失败案例（命令存在但参数错误）
|
||||
success, output = SingleLineCommandExecutor([]string{"ls", "/nonexistent"})
|
||||
fmt.Println("执行成功:", success)
|
||||
fmt.Println("输出内容:", output)
|
||||
|
||||
// 失败案例（命令不存在）
|
||||
success, output = SingleLineCommandExecutor([]string{"invalid_command"})
|
||||
fmt.Println("执行成功:", success)
|
||||
fmt.Println("输出内容:", output)
|
||||
|
||||
// 成功案例（三阶管道）
|
||||
success, output = PipeLineCommandExecutor([][]string{
|
||||
{"ps", "aux"},
|
||||
{"grep", "nginx"},
|
||||
{"wc", "-l"},
|
||||
})
|
||||
fmt.Println("执行结果:", success)
|
||||
fmt.Println("合并输出:", output)
|
||||
|
||||
// 失败案例（命令不存在）
|
||||
success, output = PipeLineCommandExecutor([][]string{
|
||||
{"invalid_cmd"},
|
||||
{"wc", "-l"},
|
||||
})
|
||||
fmt.Println("执行结果:", success)
|
||||
fmt.Println("合并输出:", output)
|
||||
|
||||
// 失败案例（参数错误）
|
||||
success, output = PipeLineCommandExecutor([][]string{
|
||||
{"ls", "/nonexistent"},
|
||||
{"grep", "test"},
|
||||
})
|
||||
fmt.Println("执行结果:", success)
|
||||
fmt.Println("合并输出:", output)
|
||||
}
|
||||
Reference in New Issue
Block a user