stuck task issue

This commit is contained in:
Ian Gulliver
2026-10-18 22:33:47 -07:00
parent c85266afd7
commit 131dfbecdc
4 changed files with 13 additions and 8 deletions
+4 -2
View File
@@ -16,6 +16,7 @@ import (
"os/exec"
"sync"
"sync/atomic"
"syscall"
"time"
"g.fc.run/ai/byre/internal/mcp"
@@ -251,6 +252,7 @@ func Start(ctx context.Context, opts Options) (*Session, error) {
s.cmd.Dir = opts.Cwd
s.cmd.Stderr = os.Stderr
s.cmd.Env = buildEnv(opts)
s.cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
s.stdin, err = s.cmd.StdinPipe()
if err != nil {
@@ -505,7 +507,7 @@ func (s *Session) Close() error {
select {
case <-s.done:
case <-time.After(5 * time.Second):
s.cmd.Process.Kill()
syscall.Kill(-s.cmd.Process.Pid, syscall.SIGKILL)
<-s.done
}
@@ -513,7 +515,7 @@ func (s *Session) Close() error {
}
func (s *Session) Kill() {
s.cmd.Process.Kill()
syscall.Kill(-s.cmd.Process.Pid, syscall.SIGKILL)
<-s.done
s.cmd.Wait()
}