codecs/encoding/framing/
character_delimited.rs1use bytes::{BufMut, BytesMut};
2use tokio_util::codec::Encoder;
3use vector_config::configurable_component;
4
5use super::BoxedFramingError;
6
7#[configurable_component]
9#[derive(Debug, Clone, Eq, PartialEq)]
10pub struct CharacterDelimitedEncoderConfig {
11 pub character_delimited: CharacterDelimitedEncoderOptions,
13}
14
15impl CharacterDelimitedEncoderConfig {
16 pub const fn new(delimiter: u8) -> Self {
18 Self {
19 character_delimited: CharacterDelimitedEncoderOptions { delimiter },
20 }
21 }
22
23 pub const fn build(&self) -> CharacterDelimitedEncoder {
25 CharacterDelimitedEncoder::new(self.character_delimited.delimiter)
26 }
27}
28
29#[configurable_component]
31#[derive(Clone, Debug, Eq, PartialEq)]
32pub struct CharacterDelimitedEncoderOptions {
33 #[configurable(metadata(docs::type_override = "ascii_char"))]
35 #[serde(with = "vector_core::serde::ascii_char")]
36 pub delimiter: u8,
37}
38
39#[derive(Debug, Clone)]
41pub struct CharacterDelimitedEncoder {
42 pub delimiter: u8,
44}
45
46impl CharacterDelimitedEncoder {
47 pub const fn new(delimiter: u8) -> Self {
49 Self { delimiter }
50 }
51}
52
53impl Encoder<()> for CharacterDelimitedEncoder {
54 type Error = BoxedFramingError;
55
56 fn encode(&mut self, _: (), buffer: &mut BytesMut) -> Result<(), BoxedFramingError> {
57 buffer.put_u8(self.delimiter);
58 Ok(())
59 }
60}
61
62#[cfg(test)]
63mod tests {
64 use super::*;
65
66 #[test]
67 fn encode() {
68 let mut codec = CharacterDelimitedEncoder::new(b'\n');
69
70 let mut buffer = BytesMut::from("abc");
71 codec.encode((), &mut buffer).unwrap();
72
73 assert_eq!(b"abc\n", &buffer[..]);
74 }
75}