自动化运维 立即执行 fix

src/service/task_manage.go

@@ -258,64 +258,43 @@ func (t *TaskManageSvc) ExecScript(req request.ExecScriptReq) (err error) {
 	if err != nil {
 		fmt.Println("执行出错：", err)
 		fmt.Println("stdout：", stdout)
-		//return nil, resp.CmdExecError.WithError(err)
+		return resp.CmdExecError.WithError(err)
+	}
+	if err = cmd.Start(); err != nil {
+		fmt.Println("执行出错2：", err)
+		fmt.Println("stdout：", stdout)
+		return resp.CmdExecError.WithError(err)
 	}
-	readout := bufio.NewReader(stdout)
-	go func() {
-		GetOutput(readout)
-	}()
 
-	//执行命令
-	//err = cmd.Run()
-	//if err != nil {
-	//	cmd.Stderr = os.Stderr
-	//	fmt.Println(cmd.Stderr)
-	//	err = resp.FileExecError.WithError(err)
-	//	return
-	//}
+	outputBuf := bufio.NewReader(stdout)
+	for {
+		line, err2 := outputBuf.ReadString('\n')
+		if err2 != nil || io.EOF == err2 {
+			break
+		}
+		fmt.Println("输出：", line)
+	}
+	cmd.Wait()
+
 	fmt.Println("执行成功")
 	fmt.Println("stdout：", stdout)
-
-	//cmd.Stdout = os.Stdout
-	//cmd.Stderr = os.Stderr
-	//stdout, err := cmd.StdoutPipe()
-	//if err != nil {
-	//	fmt.Println("执行出错：", err)
-	//	fmt.Println("stdout：", stdout)
-	//	return nil, resp.CmdExecError.WithError(err)
-	//}
-	//if err = cmd.Start(); err != nil { // 运行命令
-	//	log.Fatal(err)
-	//}
-
-	//outputBuf := bufio.NewReader(stdout)
-	//output, isPrefix, err := outputBuf.ReadLine()
-	//output, err := cmd.Output()
-	//if err != nil {
-	//	err = resp.CmdExecError.WithError(err)
-	//	return
-	//}
-	//fmt.Println(string(output))
-	//data["step"] = isPrefix
-	//data["output"] = string(output)
-	//data["output"] = string(output)
 	return nil
 }
 
-func GetOutput(reader *bufio.Reader) {
-	var sumOutput string //统计屏幕的全部输出内容
-	outputBytes := make([]byte, 200)
-	for {
-		n, err := reader.Read(outputBytes) //获取屏幕的实时输出(并不是按照回车分割，所以要结合sumOutput)
-		if err != nil {
-			if err == io.EOF {
-				break
-			}
-			fmt.Println(err)
-			sumOutput += err.Error()
-		}
-		output := string(outputBytes[:n])
-		fmt.Println("输出屏幕内容:", output) //输出屏幕内容
-		sumOutput += output
-	}
-}
+//func GetOutput(reader *bufio.Reader) {
+//	var sumOutput string //统计屏幕的全部输出内容
+//	outputBytes := make([]byte, 200)
+//	for {
+//		n, err := reader.Read(outputBytes) //获取屏幕的实时输出(并不是按照回车分割，所以要结合sumOutput)
+//		if err != nil {
+//			if err == io.EOF {
+//				break
+//			}
+//			fmt.Println(err)
+//			sumOutput += err.Error()
+//		}
+//		output := string(outputBytes[:n])
+//		fmt.Println("输出屏幕内容:", output) //输出屏幕内容
+//		sumOutput += output
+//	}
+//}