Don't kill SSE on exception (#2200)

mobile-dev-inc · Dec 17, 2024 · d1096a5 · d1096a5
1 parent 4848e9f
commit d1096a5
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/maestro-studio/server/src/main/java/maestro/studio/DeviceService.kt b/maestro-studio/server/src/main/java/maestro/studio/DeviceService.kt
@@ -79,9 +79,14 @@ object DeviceService {
             call.response.cacheControl(CacheControl.NoCache(null))
             call.respondBytesWriter(contentType = ContentType.Text.EventStream) {
                 while (true) {
-                    val deviceScreen = getDeviceScreen(maestro)
-                    writeStringUtf8("data: $deviceScreen\n\n")
-                    flush()
+                    try {
+                        val deviceScreen = getDeviceScreen(maestro)
+                        writeStringUtf8("data: $deviceScreen\n\n")
+                        flush()
+                    } catch (e: Exception) {
+                        // Ignoring the exception to prevent SSE stream from dying
+                        e.printStackTrace()
+                    }
                 }
             }
         }