record-query 1.0.4

A tool for doing record analysis and transformation
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
42
43
44
45

use std::fmt;

use crate::error;
use protobuf;
use serde;

use crate::value;
use serde_protobuf;
use serde_protobuf::descriptor;

pub struct Source<'a>(serde_protobuf::de::Deserializer<'a>, bool);

#[inline]
pub fn source<'a>(
    descriptors: &'a descriptor::Descriptors,
    message_name: &str,
    input: protobuf::CodedInputStream<'a>,
) -> error::Result<Source<'a>> {
    let de = serde_protobuf::de::Deserializer::for_named_message(descriptors, message_name, input)?;
    Ok(Source(de, true))
}

impl<'a> value::Source for Source<'a> {
    #[inline]
    fn read(&mut self) -> error::Result<Option<value::Value>> {
        if self.1 {
            self.1 = false;
            match serde::Deserialize::deserialize(&mut self.0)
                .map_err(serde_protobuf::error::CompatError::into_error)
            {
                Ok(v) => Ok(Some(v)),
                Err(serde_protobuf::error::Error::EndOfStream) => Ok(None),
                Err(e) => Err(error::Error::from(e)),
            }
        } else {
            Ok(None)
        }
    }
}

impl<'a> fmt::Debug for Source<'a> {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        f.debug_struct("ProtobufSource").finish()
    }
}