feat: 增加流量限制功能

backend/utils/cmd/cmd.go

@@ -1,50 +1,18 @@
 package cmd
 
 import (
-	"bufio"
-	"context"
-	"io"
+	"bytes"
 	"os/exec"
-	"sync"
 )
 
-func Exec(cmdStr string) (out string, err error) {
-	command := exec.CommandContext(context.Background(), "bash", "-c", cmdStr)
-
-	var wg sync.WaitGroup
-	wg.Add(1)
-
-	stdout, err := command.StdoutPipe()
+func Exec(cmdStr string) (string, error) {
+	cmd := exec.Command("bash", "-c", cmdStr)
+	var stdout, stderr bytes.Buffer
+	cmd.Stdout = &stdout
+	cmd.Stderr = &stderr
+	err := cmd.Run()
 	if err != nil {
-		return
+		return string(stderr.Bytes()), err
 	}
-	readout := bufio.NewReader(stdout)
-	go func() {
-		defer wg.Done()
-		out = getOutput(readout)
-	}()
-
-	err = command.Run()
-	if err != nil {
-		return
-	}
-	wg.Wait()
-	return
-}
-
-func getOutput(reader *bufio.Reader) string {
-	var sumOutput string
-	outputBytes := make([]byte, 200)
-	for {
-		n, err := reader.Read(outputBytes)
-		if err != nil {
-			if err == io.EOF {
-				break
-			}
-			sumOutput += err.Error()
-		}
-		output := string(outputBytes[:n])
-		sumOutput += output
-	}
-	return sumOutput
+	return string(stdout.Bytes()), nil
 }