Fix oneof codegen when bytes feature enabled

Issue #362
stepancheg · Dec 26, 2018 · 9b6bf64 · 9b6bf64
1 parent 269f235
commit 9b6bf64
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 5 deletions.
diff --git a/protobuf-codegen/src/field.rs b/protobuf-codegen/src/field.rs
@@ -30,13 +30,23 @@ fn type_is_copy(field_type: FieldDescriptorProto_Type) -> bool {
 }
 
 trait FieldDescriptorProtoTypeExt {
-    fn read(&self, is: &str) -> String;
+    fn read(&self, is: &str, primitive_type_variant: PrimitiveTypeVariant) -> String;
     fn is_s_varint(&self) -> bool;
 }
 
 impl FieldDescriptorProtoTypeExt for FieldDescriptorProto_Type {
-    fn read(&self, is: &str) -> String {
-        format!("{}.read_{}()", is, protobuf_name(*self))
+    fn read(&self, is: &str, primitive_type_variant: PrimitiveTypeVariant) -> String {
+        match primitive_type_variant {
+            PrimitiveTypeVariant::Default => format!("{}.read_{}()", is, protobuf_name(*self)),
+            PrimitiveTypeVariant::Carllerche => {
+                let protobuf_name = match self {
+                    FieldDescriptorProto_Type::TYPE_STRING => "chars",
+                    _ => protobuf_name(*self),
+                };
+                format!("{}.read_carllerche_{}()", is, protobuf_name)
+            }
+        }
+
     }
 
     /// True if self is signed integer with zigzag encoding
@@ -1620,7 +1630,7 @@ impl<'a> FieldGen<'a> {
         self.write_assert_wire_type(wire_type_var, w);
 
         let typed = RustValueTyped {
-            value: format!("{}?", self.proto_type.read("is")),
+            value: format!("{}?", self.proto_type.read("is", f.elem.primitive_type_variant())),
             rust_type: self.full_storage_iter_elem_type(),
         };
 
@@ -1674,7 +1684,7 @@ impl<'a> FieldGen<'a> {
                 ));
             }
             _ => {
-                let read_proc = format!("{}?", self.proto_type.read("is"));
+                let read_proc = format!("{}?", self.proto_type.read("is", s.elem.primitive_type_variant()));
 
                 self.write_assert_wire_type(wire_type_var, w);
                 w.write_line(&format!("let tmp = {};", read_proc));

diff --git a/protobuf-test/src/common/v2/test_carllerche_bytes_pb.proto b/protobuf-test/src/common/v2/test_carllerche_bytes_pb.proto
@@ -11,4 +11,8 @@ message TestCarllercheBytes {
     optional string s1 = 2;
     repeated bytes br = 3;
     repeated string sr = 4;
+    oneof one {
+        bytes ob = 11;
+        string os = 12;
+    }
 }