protobuf/reflect/oneof/
mod.rs

1pub(crate) mod generated;
2
3use crate::descriptor::OneofDescriptorProto;
4use crate::reflect::file::index::OneofIndices;
5use crate::reflect::file::FileDescriptorImpl;
6use crate::reflect::oneof::generated::GeneratedOneofDescriptor;
7use crate::reflect::FieldDescriptor;
8use crate::reflect::FileDescriptor;
9use crate::reflect::MessageDescriptor;
10
11/// Oneof descriptor.
12#[derive(Eq, PartialEq, Clone, Debug)]
13pub struct OneofDescriptor {
14    pub(crate) file_descriptor: FileDescriptor,
15    pub(crate) index: usize,
16}
17
18pub(crate) enum OneofDescriptorImplRef {
19    #[allow(dead_code)]
20    Generated(&'static GeneratedOneofDescriptor),
21    Dynamic,
22}
23
24impl OneofDescriptor {
25    fn index_entry(&self) -> &OneofIndices {
26        &self.file_descriptor.common().oneofs[self.index]
27    }
28
29    /// `.proto` part associated with this descriptor
30    pub fn proto(&self) -> &OneofDescriptorProto {
31        let index_entry = self.index_entry();
32        let message_descriptor = self
33            .file_descriptor
34            .message_proto_by_index(index_entry.containing_message);
35        &message_descriptor.oneof_decl[index_entry.index_in_containing_message]
36    }
37
38    /// Oneof name as specified in `.proto` file.
39    pub fn name(&self) -> &str {
40        self.proto().name()
41    }
42
43    #[allow(dead_code)]
44    pub(crate) fn _get_impl(&self) -> OneofDescriptorImplRef {
45        match &self.file_descriptor.imp {
46            FileDescriptorImpl::Generated(g) => {
47                OneofDescriptorImplRef::Generated(&g.oneofs[self.index])
48            }
49            FileDescriptorImpl::Dynamic(..) => OneofDescriptorImplRef::Dynamic,
50        }
51    }
52
53    /// Message which contains this oneof.
54    pub fn containing_message(&self) -> MessageDescriptor {
55        MessageDescriptor {
56            file_descriptor: self.file_descriptor.clone(),
57            index: self.index_entry().containing_message,
58        }
59    }
60
61    /// This oneof is not present in sources.
62    pub fn is_synthetic(&self) -> bool {
63        self.index_entry().synthetic
64    }
65
66    /// Fully qualified name of oneof (fully qualified name of enclosing message
67    /// followed by oneof name).
68    pub fn full_name(&self) -> String {
69        format!("{}.{}", self.containing_message(), self.name())
70    }
71
72    /// Fields in this oneof.
73    pub fn fields<'a>(&'a self) -> impl Iterator<Item = FieldDescriptor> + 'a {
74        let message = self.containing_message();
75        self.index_entry()
76            .fields
77            .iter()
78            .map(move |&i| message.field_by_index(i))
79    }
80}