sjs/vibetunnel
1
0
Fork 0
mirror of https://github.com/samsonjs/vibetunnel.git synced 2026-04-24 14:47:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add session validation and retry

Browse source
This commit is contained in:
Peter Steinberger 2025-06-20 07:39:30 +02:00
parent aa60d9ef7d
commit 4bbdaa963c
1 changed files with 39 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
linux/pkg/api/websocket.go
View file

@ -151,10 +151,39 @@ func (h *BufferWebSocketHandler) streamSession(sessionID string, send chan []byt
streamPath := sess.StreamOutPath()
// Check if stream file exists, wait a bit if it doesn't
maxRetries := 5
for i := 0; i < maxRetries; i++ {
if _, err := os.Stat(streamPath); err == nil {
break
}
if i == maxRetries-1 {
log.Printf("[WebSocket] Stream file not found after retries: %s", streamPath)
errorMsg, _ := json.Marshal(map[string]string{
"type": "error",
"message": "Session stream not available",
})
select {
case send <- errorMsg:
case <-done:
}
return
}
time.Sleep(100 * time.Millisecond)
}
// Create file watcher
watcher, err := fsnotify.NewWatcher()
if err != nil {
log.Printf("[WebSocket] Failed to create watcher: %v", err)
errorMsg, _ := json.Marshal(map[string]string{
"type": "error",
"message": "Failed to create file watcher",
})
select {
case send <- errorMsg:
case <-done:
}
return
}
defer watcher.Close()
@ -163,6 +192,14 @@ func (h *BufferWebSocketHandler) streamSession(sessionID string, send chan []byt
err = watcher.Add(streamPath)
if err != nil {
log.Printf("[WebSocket] Failed to watch file: %v", err)
errorMsg, _ := json.Marshal(map[string]string{
"type": "error",
"message": fmt.Sprintf("Failed to watch session stream: %v", err),
})
select {
case send <- errorMsg:
case <-done:
}
return
}
@ -211,7 +248,8 @@ func (h *BufferWebSocketHandler) streamSession(sessionID string, send chan []byt
func (h *BufferWebSocketHandler) processAndSendContent(sessionID, streamPath string, headerSent *bool, seenBytes *int64, send chan []byte, done chan struct{}) {
file, err := os.Open(streamPath)
if err != nil {
log.Printf("[WebSocket] Failed to open stream file: %v", err)
log.Printf("[WebSocket] Failed to open stream file %s: %v", streamPath, err)
// Don't panic, just return gracefully
return
}
defer file.Close()