fix: memory leak (#83)

langgenius · Sep 13, 2024 · ef193b1 · ef193b1
1 parent a5ac72b
commit ef193b1
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/internal/core/runner/output_capture.go b/internal/core/runner/output_capture.go
@@ -56,16 +56,14 @@ func (s *OutputCaptureRunner) CaptureOutput(cmd *exec.Cmd) error {
 		timeout = 5 * time.Second
 	}
 
-	timer := time.NewTimer(timeout)
-	go func() {
-		<-timer.C
+	timer := time.AfterFunc(timeout, func() {
 		if cmd != nil && cmd.Process != nil {
 			// write the error
 			s.WriteError([]byte("error: timeout\n"))
 			// send a signal to the process
 			cmd.Process.Kill()
 		}
-	}()
+	})
 
 	// create a pipe for the stdout
 	stdout_reader, err := cmd.StdoutPipe()