GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-06-07 16:59:34 +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

26
uixt/driver_handler.go
View File

@@ -194,11 +194,33 @@ func callMCPActionTool(driver IDriver,
// Get XTDriver from cache
dExt := getXTDriverFromCache(driver)
if dExt == nil {
log.Warn().Msg("XTDriver not found in cache, skipping MCP tool call")
return
}
dExt.CallMCPTool(context.Background(),
serverName, actionType, arguments)
// Create a context with timeout that can be cancelled
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
log.Debug().Str("server", serverName).Str("action", actionType).
Interface("arguments", arguments).Msg("calling MCP action tool")
// Call MCP tool with timeout context
result, err := dExt.CallMCPTool(ctx, serverName, actionType, arguments)
if err != nil {
// Classify error types for better debugging
if ctx.Err() == context.DeadlineExceeded {
log.Warn().Str("server", serverName).Str("action", actionType).
Msg("MCP action tool call timeout")
} else {
log.Warn().Err(err).Str("server", serverName).Str("action", actionType).
Msg("MCP action tool call failed")
}
return
}
log.Debug().Str("server", serverName).Str("action", actionType).
Interface("result", result).Msg("MCP action tool call succeeded")
}
// getAntiRisk_SetTouchInfo_Arguments gets arguments for SetTouchInfo MCP tool