Merge pull request #189 from jesse-engineer/feat/streamable_http_heartbeat_cancel

jesse-engineer · web-flow · commit 7ffd953a2832 · 2025-08-24T15:12:28.000+08:00
feat: streamable http server heartbeat add cancel
diff --git a/transport/streamable_http_server.go b/transport/streamable_http_server.go
@@ -213,6 +213,8 @@ func (t *streamableHTTPServerTransport) handlePost(w http.ResponseWriter, r *htt
 	}
 
 	ctx := r.Context()
+	ctx, cancel := context.WithCancel(ctx)
+	defer cancel()
 
 	// For InitializeRequest HTTP response
 	if t.stateMode == Stateful {

Original file line number	Diff line number	Diff line change
`@@ -213,6 +213,8 @@ func (t streamableHTTPServerTransport) handlePost(w http.ResponseWriter, r htt`
`213`	`213`	`}`
`214`	`214`
`215`	`215`	`ctx := r.Context()`
	`216`	`+ ctx, cancel := context.WithCancel(ctx)`
	`217`	`+ defer cancel()`
`216`	`218`
`217`	`219`	`// For InitializeRequest HTTP response`
`218`	`220`	`if t.stateMode == Stateful {`