pbbson 0.1.8

Utilities for pbjson to BSON conversion
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

use bson::{Bson, DateTime, Document};
use prost::Message;
use regex::Regex;
use serde::Serialize;
use tonic::Status;

pub fn pb_to_bson<T: Message + Serialize>(message: &T) -> Result<Document, Status> {
    let bson_value = bson::to_bson(message).map_err(|e| Status::invalid_argument(e.to_string()))?;
    let mut doc = match bson_value.as_document() {
        Some(doc) => doc.clone(),
        None => {
            return Err(Status::invalid_argument(
                "Protobuf message must be able to parse into a BSON Document",
            ));
        }
    };

    // Don't serialize empty id's
    if let Ok(id) = doc.get_str("id") {
        if id.is_empty() {
            doc.remove("id");
        }
    }

    // Convert ISO8601 strings to BSON DateTime
    let re =
        Regex::new(r#"^[0-9][0-9][0-9][0-9]-[0-9][0-9]-[0-9][0-9]T"#).map_err(|e| Status::internal(e.to_string()))?;
    for (k, v) in doc.clone().iter() {
        if let Bson::String(s) = v {
            if re.is_match(s) {
                if let Ok(dt) = DateTime::parse_rfc3339_str(s) {
                    doc.insert(k, dt);
                }
            }
        }
    }

    Ok(doc)
}