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
|
use crate::gen::field::elem::FieldElem;
use crate::gen::field::option_kind::OptionKind;
use crate::gen::file_and_mod::FileAndMod;
use crate::gen::rust_types_values::RustType;
use crate::Customize;
#[derive(Clone, PartialEq, Eq, Copy)]
pub enum SingularFieldFlag {
// proto2 or proto3 message
WithFlag {
required: bool,
option_kind: OptionKind,
},
// proto3
WithoutFlag,
}
impl SingularFieldFlag {
pub fn is_required(&self) -> bool {
match *self {
SingularFieldFlag::WithFlag { required, .. } => required,
SingularFieldFlag::WithoutFlag => false,
}
}
}
#[derive(Clone)]
pub(crate) struct SingularField<'a> {
pub flag: SingularFieldFlag,
pub elem: FieldElem<'a>,
}
impl<'a> SingularField<'a> {
pub(crate) fn rust_storage_type(&self, reference: &FileAndMod) -> RustType {
match self.flag {
SingularFieldFlag::WithFlag { option_kind, .. } => {
option_kind.wrap_element(self.elem.rust_storage_elem_type(reference))
}
SingularFieldFlag::WithoutFlag => self.elem.rust_storage_elem_type(reference),
}
}
pub(crate) fn default_value(
&self,
customize: &Customize,
reference: &FileAndMod,
const_expr: bool,
) -> String {
self.rust_storage_type(reference)
.default_value(customize, const_expr)
}
}
|