feat(mcp): Top5#3 对齐 MCP Go SDK——协议版本协商 + list_changed 差量刷新 + keepalive + Streamable HTTP
- 协议版本协商:initialize 先发最高版本(2025-11-25)并声明能力集合,遇 UnsupportedProtocolVersionError(-32022)/服务器不支持版本 → negotiateMutuallySupportedVersion 求出双方最高共同版本降级重试(最多2次);无共同版本→ErrUnsupportedProtocolVersion。 协商结果记 negotiatedVersion,HTTP 传输回填 Mcp-Protocol-Version 头。 - tools/list_changed 订阅:capabilities 声明 tools.listChanged:true;传输读无 id 帧路由到 handleServerNotification;refreshPending 原子旗标合并突发 + 新 goroutine 调 onToolListChanged (避免 stdio lineRPC 重入自锁);Registry 真差量刷新(重新 DiscoverTools→移除消失/注册新增, 跳过同名冲突保外来工具)。 - keepalive:WithKeepAlive(interval,failures) 起 goroutine 定时 Call(ping),连续失败≥阈值关 session(默认3,可配);服务器不支持 ping(-32601)停 keepalive 不误关;ctx+Close 取消不泄漏。 - Streamable HTTP 传输:newTransport 支持 TransportType=http|streamable(保留 stdio/SSE 推断, 空值走旧逻辑不破坏);Call POST+Accept json/event-stream 响应可单 JSON 或 SSE 流找匹配 id; Mcp-Session-Id 会话 + 独立 GET SSE 流收 server 通知;config 读 MCP_TRANSPORT。 - schema 缓存判定不做:客户端透传 map[string]any 无重复编译,SDK SchemaCache 是服务端 按 Go 类型反射场景,客户端无对应。 go build/vet/test 全绿(count=2 无 flaky) + windows/darwin 交叉编译。
This commit is contained in:
1 parent
464ed2cad3
commit
b0bd93b15c
13 files changed
+1517
-66
No files matched your search
@@ -19,18 +19,27 @@ import (
|
||||
// requests to the message endpoint (discovered via an endpoint event, falling
|
||||
// back to the configured URL).
|
||||
type sseTransport struct {
|
||||
cfg MCPConfig
|
||||
log *slog.Logger
|
||||
url string
|
||||
endpoint string
|
||||
mu sync.Mutex
|
||||
nextID int
|
||||
pending map[string]chan []byte
|
||||
resp *http.Response
|
||||
body io.ReadCloser
|
||||
closeCh chan struct{}
|
||||
closeOnce sync.Once
|
||||
started bool
|
||||
cfg MCPConfig
|
||||
log *slog.Logger
|
||||
url string
|
||||
endpoint string
|
||||
mu sync.Mutex
|
||||
nextID int
|
||||
pending map[string]chan []byte
|
||||
resp *http.Response
|
||||
body io.ReadCloser
|
||||
closeCh chan struct{}
|
||||
closeOnce sync.Once
|
||||
started bool
|
||||
onNotification func(method string)
|
||||
}
|
||||
|
||||
// setOnNotification records the handler invoked when the server sends a
|
||||
// server-initiated notification (a message with no id) over the SSE stream.
|
||||
func (s *sseTransport) setOnNotification(fn func(method string)) {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
s.onNotification = fn
|
||||
}
|
||||
|
||||
// maxSSEResponse bounds how much of a POST reply body is read before falling
|
||||
@@ -120,6 +129,19 @@ func (s *sseTransport) dispatch(event, data string) {
|
||||
if err := json.Unmarshal([]byte(data), &resp); err != nil {
|
||||
return
|
||||
}
|
||||
if resp.ID == nil {
|
||||
// A server-initiated notification; hand it to the client's handler.
|
||||
// The handler must not re-enter this transport synchronously.
|
||||
if resp.Method != "" {
|
||||
s.mu.Lock()
|
||||
fn := s.onNotification
|
||||
s.mu.Unlock()
|
||||
if fn != nil {
|
||||
fn(resp.Method)
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
id, ok := resp.ID.(string)
|
||||
if !ok {
|
||||
return
|
||||
|
||||
Reference in new issue
Block a user