From 77c9ca03e8853fefcaf3031affffccbff036f946 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Joakim=20Frosteg=C3=A5rd?= <joakim.frostegard@gmail.com>
Date: Mon, 1 Nov 2021 19:42:52 +0100
Subject: [PATCH] aquatic_ws_protocol: attempt to parse ws message from binary
 as well

---
 aquatic_ws_protocol/src/request/mod.rs | 22 ++++++++++++++--------
 1 file changed, 14 insertions(+), 8 deletions(-)

diff --git a/aquatic_ws_protocol/src/request/mod.rs b/aquatic_ws_protocol/src/request/mod.rs
index 8f067c3..af4acf7 100644
--- a/aquatic_ws_protocol/src/request/mod.rs
+++ b/aquatic_ws_protocol/src/request/mod.rs
@@ -23,14 +23,20 @@ impl InMessage {
 
     #[inline]
     pub fn from_ws_message(ws_message: tungstenite::Message) -> ::anyhow::Result<Self> {
-        use tungstenite::Message::Text;
+        use tungstenite::Message;
 
-        let mut text = if let Text(text) = ws_message {
-            text
-        } else {
-            return Err(anyhow::anyhow!("Message is not text"));
-        };
-
-        return ::simd_json::serde::from_str(&mut text).context("deserialize with serde");
+        match ws_message {
+            Message::Text(mut text) => {
+                ::simd_json::serde::from_str(&mut text)
+                    .context("deserialize from text xwith serde")
+            },
+            Message::Binary(mut bytes) => {
+                ::simd_json::serde::from_slice(&mut bytes[..])
+                    .context("deserialize from binary with serde")
+            },
+            _ => {
+                Err(anyhow::anyhow!("Message is neither text nor binary"))
+            }
+        }
     }
 }