Struct AgentPool 

pub struct AgentPool { /* private fields */ }

Manages a pool of long-lived agent processes keyed by auth token

Implementations

impl AgentPool

pub fn new(config: PoolConfig) -> Self

pub fn with_working_dir(self, dir: PathBuf) -> Self

Set the working directory for spawned agent processes.

pub fn with_push_relay(self, push_relay: Arc<PushRelayClient>) -> Self

Set the push relay client for sending notifications

pub async fn get_or_spawn( &mut self, token: &str, agent_command: &str, ) -> Result<(Sender<String>, Receiver<String>, Vec<String>, bool, Option<String>, Option<String>, Sender<String>)>

Get an existing agent or spawn a new one for the given token. Returns (ws_to_agent_tx, agent_to_ws_rx, buffered_messages, was_reused, cached_init_response, cached_session_response, broadcast_tx)

pub fn mark_disconnected(&mut self, token: &str)

Mark a client as disconnected. The agent stays alive for idle_timeout.

pub fn cache_init_response(&mut self, token: &str, response: String)

Cache the agent’s initialize response so reconnections can skip re-initialization

pub fn cache_session_response(&mut self, token: &str, response: String)

Cache the agent’s createSession response so reconnections reuse the same session ID

pub fn clear_session_response(&mut self, token: &str)

Clear the cached session response (e.g., when agent reports “Session not found”)

pub async fn remove_agent(&mut self, token: &str)

Remove and kill an agent

pub async fn reap_idle_agents(&mut self)

Check for idle agents that have exceeded the timeout and kill them

pub fn stats(&self) -> PoolStats

Get pool statistics

pub fn contains(&self, token: &str) -> bool

Check if the pool contains an agent for the given token

pub async fn kill_agent(&mut self, token: &str) -> bool

Kill a specific agent’s process (for testing). Returns true if the agent existed.

pub fn buffer_message(&mut self, token: &str, message: String)

Buffer a message for a disconnected agent

pub async fn shutdown_all(&mut self)

Shut down all agents in the pool