-
Notifications
You must be signed in to change notification settings - Fork 67
Expand file tree
/
Copy pathSubscriptionManager.php
More file actions
96 lines (79 loc) · 2.65 KB
/
SubscriptionManager.php
File metadata and controls
96 lines (79 loc) · 2.65 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
<?php
namespace PhpMcp\Server\Session;
use Psr\Log\LoggerInterface;
class SubscriptionManager
{
/** @var array<string, array<string, true>> Key: URI, Value: array of session IDs */
private array $resourceSubscribers = [];
/** @var array<string, array<string, true>> Key: Session ID, Value: array of URIs */
private array $sessionSubscriptions = [];
public function __construct(
private readonly LoggerInterface $logger
) {
}
/**
* Subscribe a session to a resource
*/
public function subscribe(string $sessionId, string $uri): void
{
// Add to both mappings for efficient lookup
$this->resourceSubscribers[$uri][$sessionId] = true;
$this->sessionSubscriptions[$sessionId][$uri] = true;
$this->logger->debug('Session subscribed to resource', [
'sessionId' => $sessionId,
'uri' => $uri
]);
}
/**
* Unsubscribe a session from a resource
*/
public function unsubscribe(string $sessionId, string $uri): void
{
unset($this->resourceSubscribers[$uri][$sessionId]);
unset($this->sessionSubscriptions[$sessionId][$uri]);
// Clean up empty arrays
if (empty($this->resourceSubscribers[$uri])) {
unset($this->resourceSubscribers[$uri]);
}
$this->logger->debug('Session unsubscribed from resource', [
'sessionId' => $sessionId,
'uri' => $uri
]);
}
/**
* Get all sessions subscribed to a resource
*/
public function getSubscribers(string $uri): array
{
return array_keys($this->resourceSubscribers[$uri] ?? []);
}
/**
* Check if a session is subscribed to a resource
*/
public function isSubscribed(string $sessionId, string $uri): bool
{
return isset($this->sessionSubscriptions[$sessionId][$uri]);
}
/**
* Clean up all subscriptions for a session
*/
public function cleanupSession(string $sessionId): void
{
if (!isset($this->sessionSubscriptions[$sessionId])) {
return;
}
$uris = array_keys($this->sessionSubscriptions[$sessionId]);
foreach ($uris as $uri) {
unset($this->resourceSubscribers[$uri][$sessionId]);
// Clean up empty arrays
if (empty($this->resourceSubscribers[$uri])) {
unset($this->resourceSubscribers[$uri]);
}
}
unset($this->sessionSubscriptions[$sessionId]);
$this->logger->debug('Cleaned up all subscriptions for session', [
'sessionId' => $sessionId,
'count' => count($uris)
]);
}
}