Send websocket messages to Interceptor

thewca · Oct 4, 2020 · 053ba23 · 053ba23
1 parent 0de22a4
commit 053ba23
Show file tree

Hide file tree

Showing 4 changed files with 59 additions and 26 deletions.
diff --git a/tnoodle-ui/src/main/api/tnoodle.api.js b/tnoodle-ui/src/main/api/tnoodle.api.js
@@ -28,8 +28,7 @@ export const fetchZip = (
 
     return scrambleClient
         .loadScrambles(zipEndpoint, payload, wcif.id)
-        .then((result) => convertToBlob(result))
-        .catch((error) => console.error(error));
+        .then((result) => convertToBlob(result));
 };
 
 const convertToBlob = async (result) => {

diff --git a/tnoodle-ui/src/main/api/tnoodle.socket.js b/tnoodle-ui/src/main/api/tnoodle.socket.js
@@ -9,6 +9,8 @@ export class ScrambleClient {
 
         this.contentType = null;
         this.resultPayload = null;
+
+        this.errorPayload = null;
     }
 
     loadScrambles(endpoint, payload, targetMarker) {
@@ -20,8 +22,8 @@ export class ScrambleClient {
                 ws.send(JSON.stringify(payload));
             };
 
-            ws.onerror = (err) => {
-                reject(err);
+            ws.onerror = () => {
+                reject(this.errorPayload);
             };
 
             ws.onclose = (cls) => {
@@ -33,11 +35,20 @@ export class ScrambleClient {
 
                     resolve(resultObject);
                 } else {
-                    reject(cls);
+                    reject(this.errorPayload);
                 }
             };
 
             ws.onmessage = (msg) => {
+                if (this.state === MARKER_ERROR_MESSAGE) {
+                    let rawPayload = msg.data.toString();
+                    this.errorPayload = JSON.parse(rawPayload);
+                }
+
+                if (msg.data === MARKER_ERROR_MESSAGE) {
+                    this.state = MARKER_ERROR_MESSAGE; // purposefully go into an error state
+                }
+
                 if (this.state === SCRAMBLING_STATES.INITIATE) {
                     this.state = SCRAMBLING_STATES.SCRAMBLING;
 
@@ -78,3 +89,6 @@ const SCRAMBLING_STATES = {
     COMPUTED_DATA: "COMPUTED_DATA",
     DONE: "DONE",
 };
+
+// this has to be identical with backend value in JobSchedulingHandler.kt
+const MARKER_ERROR_MESSAGE = "%%ERROR%%";
diff --git a/tnoodle-ui/src/main/components/Main.jsx b/tnoodle-ui/src/main/components/Main.jsx
@@ -46,6 +46,8 @@ const Main = connect(
             this.state = {
                 competitionNameFileZip: "",
             };
+
+            this.interceptor = React.createRef();
         }
         onSubmit = (evt) => {
             evt.preventDefault();
@@ -77,11 +79,17 @@ const Main = connect(
                 this.props.mbld,
                 this.props.password,
                 this.props.translations
-            ).then((blob) => {
-                this.props.updateFileZipBlob(blob);
-                this.props.updateGeneratingScrambles(false);
-                this.props.resetScramblingProgressCurrent();
-            });
+            )
+                .then((blob) => {
+                    this.props.updateFileZipBlob(blob);
+                })
+                .catch((err) => {
+                    this.interceptor.current.updateMessage(err);
+                })
+                .finally(() => {
+                    this.props.updateGeneratingScrambles(false);
+                    this.props.resetScramblingProgressCurrent();
+                });
             this.props.updateGeneratingScrambles(true);
         };
 
@@ -160,7 +168,7 @@ const Main = connect(
             return (
                 <form onSubmit={this.onSubmit}>
                     <div className="sticky-top bg-light">
-                        <Interceptor />
+                        <Interceptor ref={this.interceptor} />
                         <VersionInfo />
                         <div className="container-fluid pt-2">
                             <div className="row">

diff --git a/.../org/worldcubeassociation/tnoodle/server/webscrambles/routing/job/JobSchedulingHandler.kt b/.../org/worldcubeassociation/tnoodle/server/webscrambles/routing/job/JobSchedulingHandler.kt
@@ -4,9 +4,7 @@ import io.ktor.application.ApplicationCall
 import io.ktor.application.call
 import io.ktor.http.ContentType
 import io.ktor.http.HttpStatusCode
-import io.ktor.http.cio.websocket.Frame
-import io.ktor.http.cio.websocket.close
-import io.ktor.http.cio.websocket.send
+import io.ktor.http.cio.websocket.*
 import io.ktor.response.respond
 import io.ktor.response.respondBytes
 import io.ktor.routing.*
@@ -16,6 +14,8 @@ import kotlinx.serialization.builtins.serializer
 import org.worldcubeassociation.tnoodle.server.RouteHandler
 import org.worldcubeassociation.tnoodle.server.crypto.StringEncryption.encodeBase64
 import org.worldcubeassociation.tnoodle.server.serial.JsonConfig
+import org.worldcubeassociation.tnoodle.server.webscrambles.serial.FrontendErrorMessage
+import org.worldcubeassociation.tnoodle.server.webscrambles.serial.FrontendErrorMessage.Companion.asFrontendError
 
 object JobSchedulingHandler : RouteHandler {
     private val JOBS = mutableMapOf<Int, MutableMap<String, Int>>()
@@ -24,6 +24,8 @@ object JobSchedulingHandler : RouteHandler {
 
     private const val JOB_ID_PARAM = "jobId"
 
+    const val MARKER_ERROR_MESSAGE = "%%ERROR%%";
+
     private suspend fun ApplicationCall.checkAndYieldJobId(): Int? {
         val jobId = parameters[JOB_ID_PARAM]?.toIntOrNull() ?: -1
 
@@ -106,24 +108,34 @@ object JobSchedulingHandler : RouteHandler {
         webSocket {
             for (frame in incoming) {
                 if (frame is Frame.Text) {
-                    val request = job.extractFrame(call, frame)
+                    try {
+                        val request = job.extractFrame(call, frame)
+
+                        val target = job.getTargetState(request)
+                        val targetEnc = JsonConfig.SERIALIZER.encodeToString(MapSerializer(String.serializer(), Int.serializer()), target)
+                        send(targetEnc)
+
+                        val (type, data) = job.channel(request, this)
 
-                    val target = job.getTargetState(request)
-                    val targetEnc = JsonConfig.SERIALIZER.encodeToString(MapSerializer(String.serializer(), Int.serializer()), target)
-                    send(targetEnc)
+                        val targetMarker = job.getResultMarker(request)
+                        val encodedData = data.encodeBase64()
 
-                    val (type, data) = job.channel(request, this)
+                        // signal that the computation result is about to start
+                        send(targetMarker)
 
-                    val targetMarker = job.getResultMarker(request)
-                    val encodedData = data.encodeBase64()
+                        send(type.toString())
+                        send(encodedData)
 
-                    // signal that the computation result is about to start
-                    send(targetMarker)
+                        close()
+                    } catch (e: Throwable) {
+                        val errorModel = e.asFrontendError()
+                        val errorSerial = JsonConfig.SERIALIZER.encodeToString(FrontendErrorMessage.serializer(), errorModel)
 
-                    send(type.toString())
-                    send(encodedData)
+                        send(MARKER_ERROR_MESSAGE)
+                        send(errorSerial)
 
-                    close()
+                        close(CloseReason(CloseReason.Codes.INTERNAL_ERROR, ""))
+                    }
                 }
             }
         }