fix: add an extra flag to decide frame format

vectordotdev · Jul 24, 2023 · 6899035 · 6899035
1 parent 0ef5d5f
commit 6899035
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/sinks/websocket/sink.rs b/src/sinks/websocket/sink.rs
@@ -196,20 +196,26 @@ pub struct WebSocketSink {
     connector: WebSocketConnector,
     ping_interval: Option<NonZeroU64>,
     ping_timeout: Option<NonZeroU64>,
+    encode_as_binary: bool,
 }
 
 impl WebSocketSink {
     pub fn new(config: &WebSocketSinkConfig, connector: WebSocketConnector) -> crate::Result<Self> {
         let transformer = config.encoding.transformer();
         let serializer = config.encoding.build()?;
         let encoder = Encoder::<()>::new(serializer);
+        let encode_as_binary = match config.encoding.config() {
+            codecs::encoding::SerializerConfig::RawMessage => true,
+            _ => true,
+        };
 
         Ok(Self {
             transformer,
             encoder,
             connector,
             ping_interval: config.ping_interval,
             ping_timeout: config.ping_timeout,
+            encode_as_binary: encode_as_binary,
         })
     }
 
@@ -301,7 +307,10 @@ impl WebSocketSink {
                         Ok(()) => {
                             finalizers.update_status(EventStatus::Delivered);
 
-                            let message = Message::binary(bytes);
+                            let message = match self.encode_as_binary {
+                                true => Message::binary(bytes),
+                                false => Message::text(String::from_utf8_lossy(&bytes)),
+                            };
                             let message_len = message.len();
 
                             ws_sink.send(message).await.map(|_| {