feat(server): Rework transport architecture and add StreamableHttpTransport (#49)
* feat(server): Rework transport architecture and add StreamableHttpTransport - `Server::connect()` no longer contains a processing loop. - `TransportInterface` is updated with `setMessageHandler()` and `listen()`. - `StdioTransport` is updated to implement the new interface. - A new, minimal `StreamableHttpTransport` is added for stateless HTTP. * refactor(server): use event-driven message handling * feat(server): Introduce a formal session management system * fix: ensure session elements are preserved when building server (regression from #46) * refactor: consolidate HTTP example to use shared dependencies * feat(server): Enhance message handling with session support - Updated `TransportInterface` to use `onMessage` for handling incoming messages with session IDs. - Refactored `Server`, `Handler`, and transport classes to accommodate session management using `Uuid`. - Introduced methods for creating sessions with auto-generated and specific UUIDs in `SessionFactory` and `SessionFactoryInterface`. * feat(server): Integrate session management in message handling - Added session support to the `Server` and `Handler` classes, allowing for session data to be managed during message processing. - Updated `TransportInterface` to include session context in the `send` method. - Refactored various request handlers to utilize session information, ensuring proper session handling for incoming requests. - Introduced a file-based session store for persistent session data management * feat: Update session builder method to use set* instead of with* * feat: Fix test compatibility with new session management architecture * chore: apply code style fixes * fix: remove deprecated SSE transport and resolve PHPStan issues
This commit is contained in:
parent
b1e54f1f86
commit
719262b15d
1 changed files with 1614 additions and 406 deletions
|
|
@ -14,13 +14,20 @@ namespace Mcp\Server\Transport;
|
|||
use Mcp\Server\TransportInterface;
|
||||
use Psr\Log\LoggerInterface;
|
||||
use Psr\Log\NullLogger;
|
||||
use Symfony\Component\Uid\Uuid;
|
||||
|
||||
/**
|
||||
* Heavily inspired by https://jolicode.com/blog/mcp-the-open-protocol-that-turns-llm-chatbots-into-intelligent-agents.
|
||||
* @author Kyrian Obikwelu <koshnawaza@gmail.com>
|
||||
*/
|
||||
class StdioTransport implements TransportInterface
|
||||
{
|
||||
private string $buffer = '';
|
||||
/** @var callable(string, ?Uuid): void */
|
||||
private $messageListener;
|
||||
|
||||
/** @var callable(Uuid): void */
|
||||
private $sessionEndListener;
|
||||
|
||||
private ?Uuid $sessionId = null;
|
||||
|
||||
/**
|
||||
* @param resource $input
|
||||
|
|
@ -37,39 +44,67 @@ class StdioTransport implements TransportInterface
|
|||
{
|
||||
}
|
||||
|
||||
public function isConnected(): bool
|
||||
public function onMessage(callable $listener): void
|
||||
{
|
||||
return true;
|
||||
$this->messageListener = $listener;
|
||||
}
|
||||
|
||||
public function receive(): \Generator
|
||||
{
|
||||
$line = fgets($this->input);
|
||||
|
||||
$this->logger->debug('Received message on StdioTransport.', [
|
||||
'line' => $line,
|
||||
]);
|
||||
|
||||
if (false === $line) {
|
||||
return;
|
||||
}
|
||||
$this->buffer .= rtrim($line).\PHP_EOL;
|
||||
if (str_contains($this->buffer, \PHP_EOL)) {
|
||||
$lines = explode(\PHP_EOL, $this->buffer);
|
||||
$this->buffer = array_pop($lines);
|
||||
|
||||
yield from $lines;
|
||||
}
|
||||
}
|
||||
|
||||
public function send(string $data): void
|
||||
public function send(string $data, array $context): void
|
||||
{
|
||||
$this->logger->debug('Sending data to client via StdioTransport.', ['data' => $data]);
|
||||
|
||||
if (isset($context['session_id'])) {
|
||||
$this->sessionId = $context['session_id'];
|
||||
}
|
||||
|
||||
fwrite($this->output, $data.\PHP_EOL);
|
||||
}
|
||||
|
||||
public function listen(): mixed
|
||||
{
|
||||
$this->logger->info('StdioTransport is listening for messages on STDIN...');
|
||||
|
||||
while (!feof($this->input)) {
|
||||
$line = fgets($this->input);
|
||||
if (false === $line) {
|
||||
break;
|
||||
}
|
||||
|
||||
$trimmedLine = trim($line);
|
||||
if (!empty($trimmedLine)) {
|
||||
$this->logger->debug('Received message on StdioTransport.', ['line' => $trimmedLine]);
|
||||
if (\is_callable($this->messageListener)) {
|
||||
\call_user_func($this->messageListener, $trimmedLine, $this->sessionId);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
$this->logger->info('StdioTransport finished listening.');
|
||||
|
||||
if (\is_callable($this->sessionEndListener) && null !== $this->sessionId) {
|
||||
\call_user_func($this->sessionEndListener, $this->sessionId);
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
|
||||
public function onSessionEnd(callable $listener): void
|
||||
{
|
||||
$this->sessionEndListener = $listener;
|
||||
}
|
||||
|
||||
public function close(): void
|
||||
{
|
||||
if (\is_callable($this->sessionEndListener) && null !== $this->sessionId) {
|
||||
\call_user_func($this->sessionEndListener, $this->sessionId);
|
||||
}
|
||||
|
||||
if (\is_resource($this->input)) {
|
||||
fclose($this->input);
|
||||
}
|
||||
|
||||
if (\is_resource($this->output)) {
|
||||
fclose($this->output);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in a new issue