tokn-convert 0.2.0-rc.3

Request and response conversion pipeline across tokn endpoint formats
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

use super::event::SseEvent;
use bytes::Bytes;

pub fn encode_sse(event: Option<&str>, data: &str) -> Bytes {
  let mut out = String::new();
  if let Some(event) = event {
    out.push_str("event: ");
    out.push_str(event);
    out.push('\n');
  }
  if data.is_empty() {
    out.push_str("data:\n\n");
    return Bytes::from(out);
  }
  for line in data.lines() {
    out.push_str("data: ");
    out.push_str(line);
    out.push('\n');
  }
  out.push('\n');
  Bytes::from(out)
}

pub fn encode_done() -> Bytes {
  Bytes::from_static(b"data: [DONE]\n\n")
}

pub(crate) fn encode_event(event: &SseEvent) -> Bytes {
  if event.is_done() {
    encode_done()
  } else {
    encode_sse(event.event.as_deref(), &event.data)
  }
}