stacklok / toolhive / 17495147585

60%

Build Type

push

github

Committed by web-flow

Commit Message

Enhance Streamable HTTP proxy with session management and ID routing (#1738)

Refactor the existing Streamable HTTP proxy to improve request/response
correlation and add proper session management per MCP specification.

Key improvements:
- Add composite ID routing (sessID|idKey) for reliable request correlation
- Extract response dispatching into dedicated dispatcher with waiter-based routing
- Introduce session management with Manager and StreamableSession types
- Implement spec-compliant HTTP method handlers (GET/DELETE/POST)
- Fix session header handling with automatic Mcp-Session-Id assignment
- Support both synchronous batch processing and async SSE streaming
- Add 30-second default timeout for response handling
- Include comprehensive test coverage with mcp-go client integration
- Add spec compliance tests validating HTTP status codes
- Test concurrent clients and stress scenarios

Also clean up logging by replacing fmt.Printf with proper logger in
authz middleware and remove debug print statement from stdio transport.

The refactoring ensures proper message correlation between multiple
concurrent clients and container backends while maintaining full MCP
specification compliance.

Signed-off-by: Juan Antonio Osorio <ozz@stacklok.com>

Run Details

287 of 479 new or added lines in 5 files covered. (59.92%)

8 existing lines in 1 file now uncovered.

10917 of 28788 relevant lines covered (37.92%)

17.25 hits per line