Skip to content

Commit 9a44d14

Browse files
Update server/sse.go
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
1 parent 103410e commit 9a44d14

File tree

1 file changed

+16
-22
lines changed

1 file changed

+16
-22
lines changed

server/sse.go

Lines changed: 16 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -275,28 +275,22 @@ func (s *SSEServer) handleSSE(w http.ResponseWriter, r *http.Request) {
275275
}()
276276

277277

278-
// Start keep alive : ping
279-
if s.keepAlive {
280-
go func() {
281-
ticker := time.NewTicker(s.keepAliveInterval)
282-
defer ticker.Stop()
283-
for {
284-
select {
285-
case <-ticker.C:
286-
select {
287-
case session.eventQueue <- fmt.Sprintf(":ping - %s\n\n", time.Now().Format(time.RFC3339)):
288-
// Ping sent successfully
289-
default:
290-
log.Printf("Keep-alive ping dropped: event queue is full")
291-
}
292-
case <-session.done:
293-
return
294-
case <-r.Context().Done():
295-
return
296-
}
297-
}
298-
}()
299-
}
278+
import (
279+
"context"
280+
"encoding/json"
281+
"fmt"
282+
"log"
283+
"net/http"
284+
"net/http/httptest"
285+
"net/url"
286+
"strings"
287+
"sync"
288+
"sync/atomic"
289+
"time"
290+
291+
"github.com/google/uuid"
292+
"github.com/mark3labs/mcp-go/mcp"
293+
)
300294

301295
messageEndpoint := fmt.Sprintf("%s?sessionId=%s", s.CompleteMessageEndpoint(), sessionID)
302296

0 commit comments

Comments
 (0)