Convert message to builder if necessary

angushe · Dec 10, 2014 · 84c836c · 84c836c
1 parent cb35d46
commit 84c836c
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/hive/src/main/java/com/twitter/elephantbird/hive/serde/ProtobufStructObjectInspector.java b/hive/src/main/java/com/twitter/elephantbird/hive/serde/ProtobufStructObjectInspector.java
@@ -201,7 +201,15 @@ public Object getStructFieldData(Object data, StructField structField) {
     if (data == null) {
       return null;
     }
-    Message.Builder builder = (Message.Builder) data;
+    Message.Builder builder;
+    if (data instanceof Message.Builder) {
+      builder = (Message.Builder)data;
+    } else if (data instanceof Message) {
+      builder = ((Message)data).toBuilder();
+    } else {
+      throw new RuntimeException("Type Message or Message.Builder expected: " +
+              data.getClass().getCanonicalName());
+    }
     ProtobufStructField psf = (ProtobufStructField) structField;
     FieldDescriptor fieldDescriptor = psf.getFieldDescriptor();
     Object result = builder.getField(fieldDescriptor);