fix: mjpeg server sleep

SonicCloudOrg · Nov 25, 2022 · f11dcbd · f11dcbd
1 parent 99e24dd
commit f11dcbd
Showing 1 changed file with 18 additions and 4 deletions.
diff --git a/src/main/java/org/cloud/sonic/agent/websockets/IOSScreenWSServer.java b/src/main/java/org/cloud/sonic/agent/websockets/IOSScreenWSServer.java
@@ -85,10 +85,24 @@ public void onOpen(Session session, @PathParam("key") String secretKey,
                 return;
             }
             MjpegInputStream mjpegInputStream = null;
-            try {
-                mjpegInputStream = new MjpegInputStream(url.openStream());
-            } catch (IOException e) {
-                log.info(e.getMessage());
+            int waitMjpeg = 0;
+            while (mjpegInputStream == null) {
+                try {
+                    mjpegInputStream = new MjpegInputStream(url.openStream());
+                } catch (IOException e) {
+                    log.info(e.getMessage());
+                }
+                try {
+                    Thread.sleep(1000);
+                } catch (InterruptedException e) {
+                    log.info(e.getMessage());
+                    return;
+                }
+                waitMjpeg++;
+                if (waitMjpeg >= 20) {
+                    log.info("mjpeg server connect fail");
+                    return;
+                }
             }
             ByteBuffer bufferedImage;
             int i = 0;