gproxy-transform 3.0.0-alpha.0

GPROXY v3 pairwise protocol transforms
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
35
36
37
38
39
40
41

use bytes::Bytes;
use serde_json::Value;

use crate::ResponseStream;

pub(super) fn bytes_text(frames: &[Bytes]) -> String {
    String::from_utf8_lossy(
        &frames
            .iter()
            .flat_map(|frame| frame.iter().copied())
            .collect::<Vec<_>>(),
    )
    .into_owned()
}

pub(super) fn drive(mut stream: ResponseStream, wire: &str, chunk: usize) -> Vec<u8> {
    let mut output = Vec::new();
    for part in wire.as_bytes().chunks(chunk) {
        for frame in stream.push(Bytes::copy_from_slice(part)).unwrap() {
            output.extend_from_slice(&frame);
        }
    }
    for frame in stream.finish().unwrap() {
        output.extend_from_slice(&frame);
    }
    output
}

pub(super) fn data_frames(wire: &[u8]) -> Vec<Value> {
    String::from_utf8_lossy(wire)
        .split("\n\n")
        .filter_map(|frame| {
            let data = frame
                .lines()
                .filter_map(|line| line.strip_prefix("data: "))
                .collect::<Vec<_>>()
                .join("\n");
            serde_json::from_str(&data).ok()
        })
        .collect()
}