bigml-parallel 0.9.2

CLI tool for executing multiple BigML scripts in parallel
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

//! A [`tokio::codec::Encoder`] that outputs a line-delimited JSON stream.

use anyhow::Error;
use bytes::{BufMut, BytesMut};
use serde::Serialize;
use std::marker::PhantomData;
use tokio_util::codec::Encoder;

/// A [`tokio::codec::Encoder`] that outputs a [line-delimited JSON
/// stream][json]. This can be used to output a `Stream` of values implementing
/// `Serialize` to an `AsyncWrite`.
///
/// [json]: https://en.wikipedia.org/wiki/JSON_streaming
pub struct LineDelimitedJsonCodec<T: Serialize> {
    _placeholder: PhantomData<T>,
}

impl<T: Serialize> LineDelimitedJsonCodec<T> {
    /// Create a new `LineDelimitedJsonCodec`.
    pub fn new() -> Self {
        Self {
            _placeholder: PhantomData,
        }
    }
}

impl<T: Serialize> Encoder<T> for LineDelimitedJsonCodec<T> {
    type Error = Error;

    fn encode(&mut self, item: T, buf: &mut BytesMut) -> Result<(), Error> {
        let json = serde_json::to_vec(&item)?;
        buf.reserve(json.len() + 1);
        buf.put(&json[..]);
        buf.put_u8(b'\n');
        Ok(())
    }
}