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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
use std::collections::HashMap;
use crate::descriptor::EnumDescriptorProto;
use crate::descriptor::EnumValueDescriptorProto;
use crate::descriptor::FileDescriptorProto;
use crate::descriptorx::find_enum_by_rust_name;
use crate::reflect::find_message_or_enum::find_message_or_enum;
use crate::reflect::find_message_or_enum::MessageOrEnum;
use crate::ProtobufEnum;
/// Description for enum variant.
///
/// Used in reflection.
#[derive(Clone, Debug)]
pub struct EnumValueDescriptor {
proto: &'static EnumValueDescriptorProto,
}
impl Copy for EnumValueDescriptor {}
impl EnumValueDescriptor {
/// Name of enum variant as specified in proto file
pub fn name(&self) -> &'static str {
self.proto.get_name()
}
/// `i32` value of the enum variant
pub fn value(&self) -> i32 {
self.proto.get_number()
}
}
/// Dynamic representation of enum type.
///
/// Can be used in reflective operations.
pub struct EnumDescriptor {
proto: &'static EnumDescriptorProto,
values: Vec<EnumValueDescriptor>,
index_by_name: HashMap<String, usize>,
index_by_number: HashMap<i32, usize>,
}
impl EnumDescriptor {
/// Enum name as given in `.proto` file
pub fn name(&self) -> &'static str {
self.proto.get_name()
}
/// `EnumDescriptor` for enum type
pub fn for_type<E: ProtobufEnum>() -> &'static EnumDescriptor {
E::enum_descriptor_static()
}
/// Create new enum descriptor.
///
/// This function is called by generated code, and should not be called manually.
#[deprecated(
since = "2.12",
note = "Please regenerate .rs files from .proto files to use newer APIs"
)]
pub fn new(rust_name: &'static str, file: &'static FileDescriptorProto) -> EnumDescriptor {
let proto = find_enum_by_rust_name(file, rust_name);
let mut index_by_name = HashMap::new();
let mut index_by_number = HashMap::new();
for (i, v) in proto.en.get_value().iter().enumerate() {
index_by_number.insert(v.get_number(), i);
index_by_name.insert(v.get_name().to_string(), i);
}
EnumDescriptor {
proto: proto.en,
values: proto
.en
.get_value()
.iter()
.map(|v| EnumValueDescriptor { proto: v })
.collect(),
index_by_name: index_by_name,
index_by_number: index_by_number,
}
}
/// Create new enum descriptor.
///
/// This function is called by generated code, and should not be called manually.
pub fn new_pb_name<E>(
name_in_file: &'static str,
file: &'static FileDescriptorProto,
) -> EnumDescriptor
where
E: ProtobufEnum,
{
let (_path_to_package, proto) = match find_message_or_enum(file, name_in_file) {
(path_to_package, MessageOrEnum::Enum(e)) => (path_to_package, e),
(_, MessageOrEnum::Message(_)) => panic!("not an enum"),
};
let mut index_by_name = HashMap::new();
let mut index_by_number = HashMap::new();
for (i, v) in proto.get_value().iter().enumerate() {
index_by_number.insert(v.get_number(), i);
index_by_name.insert(v.get_name().to_string(), i);
}
EnumDescriptor {
proto,
values: proto
.get_value()
.iter()
.map(|v| EnumValueDescriptor { proto: v })
.collect(),
index_by_name: index_by_name,
index_by_number: index_by_number,
}
}
/// Find enum value by name
pub fn value_by_name<'a>(&'a self, name: &str) -> &'a EnumValueDescriptor {
// TODO: clone is weird
let &index = self.index_by_name.get(&name.to_string()).unwrap();
&self.values[index]
}
/// Find enum value by number
pub fn value_by_number<'a>(&'a self, number: i32) -> &'a EnumValueDescriptor {
let &index = self.index_by_number.get(&number).unwrap();
&self.values[index]
}
}
|
