core/mcp
8
0
Fork 0
Code Issues 4 Pull requests 2 Projects Releases Packages Wiki Activity Actions

refactor(mcp): centralize session broadcast iteration

Browse source
This commit is contained in:
Virgil 2026-04-02 17:50:49 +00:00
parent d4de2b4cd7
commit 1373a6d296
1 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
pkg/mcp/notify.go
View file

@ -147,9 +147,9 @@ func (s *Service) SendNotificationToAllClients(ctx context.Context, level mcp.Lo
return
}
ctx = normalizeNotificationContext(ctx)
for _, session := range snapshotSessions(s.server) {
s.broadcastToSessions(func(session *mcp.ServerSession) {
s.sendLoggingNotificationToSession(ctx, session, level, logger, data)
}
})
}
// SendNotificationToSession sends a log-level notification to one connected
@ -245,10 +245,19 @@ func (s *Service) sendChannelNotificationToAllClients(ctx context.Context, paylo
return
}
ctx = normalizeNotificationContext(ctx)
for _, session := range snapshotSessions(s.server) {
s.broadcastToSessions(func(session *mcp.ServerSession) {
if err := sendSessionNotification(ctx, session, ChannelNotificationMethod, payload); err != nil {
s.debugNotify("channel: failed to send to session", "session", session.ID(), "error", err)
}
})
}
func (s *Service) broadcastToSessions(fn func(*mcp.ServerSession)) {
if s == nil || s.server == nil || fn == nil {
return
}
for _, session := range snapshotSessions(s.server) {
fn(session)
}
}