re-export Utf8Bytes, Bytes at root

snapview · Dec 17, 2024 · 63db5e9 · 63db5e9
1 parent 6505e51
commit 63db5e9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -37,8 +37,10 @@ type ReadBuffer = buffer::ReadBuffer<READ_BUFFER_CHUNK_SIZE>;
 
 pub use crate::{
     error::{Error, Result},
-    protocol::{Message, WebSocket},
+    protocol::{frame::Utf8Bytes, Message, WebSocket},
 };
+// re-export bytes since used in `Message` API.
+pub use bytes::Bytes;
 
 #[cfg(feature = "handshake")]
 pub use crate::{

diff --git a/src/protocol/mod.rs b/src/protocol/mod.rs
@@ -13,7 +13,10 @@ use self::{
     },
     message::{IncompleteMessage, IncompleteMessageType},
 };
-use crate::error::{CapacityError, Error, ProtocolError, Result};
+use crate::{
+    error::{CapacityError, Error, ProtocolError, Result},
+    protocol::frame::Utf8Bytes,
+};
 use log::*;
 use std::{
     io::{self, Read, Write},
@@ -713,7 +716,7 @@ impl WebSocketContext {
                     if !frame.code.is_allowed() {
                         CloseFrame {
                             code: CloseCode::Protocol,
-                            reason: "Protocol violation".into(),
+                            reason: Utf8Bytes::from_static("Protocol violation"),
                         }
                     } else {
                         frame