Struct ToolTimeoutManager 

pub struct ToolTimeoutManager { /* private fields */ }

Tool timeout and cancellation manager

Manages per-tool timeout policies and provides cancellation-safe timeout enforcement for tool execution with comprehensive audit logging.

Implementations

impl ToolTimeoutManager

pub fn new(config: TimeoutConfig, metrics: Arc<ServerMetrics>) -> Self

Create a new timeout manager with the given configuration and metrics

pub fn get_tool_timeout(&self, tool_name: &str) -> Duration

Get the timeout duration for a specific tool

Returns per-tool override if configured, otherwise returns default timeout.

pub async fn execute_with_timeout_and_cancellation<F, T>( &self, tool_name: &str, operation: F, ) -> Result<(T, CancellationToken), ToolTimeoutError>

where F: Future<Output = Result<T, ServerError>>, T: Send,

Execute a tool with timeout and cooperative cancellation support

This is the primary method for executing tools with comprehensive timeout and cancellation handling following Tokio best practices.

Returns both the result and the cancellation token for context propagation.

pub async fn execute_with_external_token<F, T>( &self, tool_name: &str, operation: F, cancellation_token: CancellationToken, ) -> Result<T, ToolTimeoutError>

where F: Future<Output = Result<T, ServerError>>, T: Send,

Execute a tool with a provided cancellation token

This method allows external code to provide the cancellation token, enabling tight integration with RequestContext and other systems.

pub async fn execute_with_timeout<F, T>( &self, tool_name: &str, operation: F, ) -> Result<T, ToolTimeoutError>

where F: Future<Output = Result<T, ServerError>>, T: Send,

Execute a tool with timeout (backward compatible API)

This method provides backward compatibility for existing code that doesn’t need access to the cancellation token. New code should use execute_with_timeout_and_cancellation for cooperative cancellation support.

pub async fn get_active_executions(&self) -> Vec<ActiveExecutionInfo>

Get statistics about active tool executions

Returns information about currently running tools for monitoring and debugging purposes.

pub async fn cancel_all_executions(&self)

Cancel all active executions (for graceful shutdown)

Signals cooperative cancellation to all active tool executions. Tools that check their cancellation tokens will receive the signal and can perform graceful cleanup before terminating.

Trait Implementations

impl Clone for ToolTimeoutManager

fn clone(&self) -> ToolTimeoutManager

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ToolTimeoutManager

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.