GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-06-07 08:49:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: enhance signal handling and graceful shutdown for MCP integration

Browse Source
This commit is contained in:
lilong.129
2025-05-29 00:59:17 +08:00
parent c5fb391ef5
commit d3011d467e
7 changed files with 299 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
step_thinktime.go
View File

@@ -1,6 +1,7 @@
package hrp
import (
"fmt"
"time"
"github.com/rs/zerolog/log"
@@ -76,6 +77,19 @@ func (s *StepThinkTime) Run(r *SessionRunner) (*StepResult, error) {
}
}
time.Sleep(tt)
// Use interruptible sleep that can respond to signals
log.Debug().Float64("duration_ms", float64(tt.Milliseconds())).Msg("starting think time")
select {
case <-time.After(tt):
// Normal completion
log.Debug().Float64("duration_ms", float64(tt.Milliseconds())).Msg("think time completed normally")
case <-r.caseRunner.hrpRunner.interruptSignal:
// Interrupted by signal
log.Info().Float64("planned_duration_ms", float64(tt.Milliseconds())).
Msg("think time interrupted by signal")
return stepResult, fmt.Errorf("think time interrupted")
}
return stepResult, nil
}