vrl 0.32.0

Vector Remap Language
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

use prost_reflect::{DescriptorPool, MessageDescriptor};
use std::path::Path;

pub fn get_message_descriptor(
    descriptor_set_path: &Path,
    message_type: &str,
) -> Result<MessageDescriptor, String> {
    let bytes = std::fs::read(descriptor_set_path)
        .map_err(|e| format!("Failed to open protobuf desc file '{descriptor_set_path:?}': {e}"))?;
    let pool = DescriptorPool::decode(bytes.as_slice()).map_err(|e| {
        format!("Failed to parse protobuf desc file '{descriptor_set_path:?}': {e}")
    })?;
    resolve_message_descriptor(
        pool,
        message_type,
        &format!("'{}'", descriptor_set_path.display()),
    )
}

pub fn get_message_descriptor_from_bytes(
    descriptor_bytes: &[u8],
    message_type: &str,
) -> Result<MessageDescriptor, String> {
    let pool = DescriptorPool::decode(descriptor_bytes)
        .map_err(|e| format!("Failed to parse protobuf descriptor bytes: {e}"))?;
    resolve_message_descriptor(pool, message_type, "the provided descriptor bytes")
}

fn resolve_message_descriptor(
    descriptor_pool: DescriptorPool,
    message_type: &str,
    context: &str,
) -> Result<MessageDescriptor, String> {
    descriptor_pool
        .get_message_by_name(message_type)
        .ok_or_else(|| format!("The message type '{message_type}' could not be found in {context}"))
}