/**
* Assigns all fields in {@link JsonNode} instance to a {@link Builder}.
*
* @param builder the builder to be populated
* @param root the JSON object
* @return the original builder, populated with any fields that were discovered
*/
public static Builder toProto(final Builder builder, final JsonNode root) {
Descriptor type = builder.getDescriptorForType();
for (final FieldDescriptor fieldDesc : type.getFields()) {
final String attrName = fieldDesc.getName();
final JsonNode node = root.get(attrName);
if (node == null) {
continue;
}
if (node.isNull()) {
continue;
}
if (fieldDesc.isRepeated()) {
final Iterator<JsonNode> iter = node.getElements();
while (iter.hasNext()) {
builder.addRepeatedField(fieldDesc, toJavaObj(builder, fieldDesc, iter.next()));
}
} else {
builder.setField(fieldDesc, toJavaObj(builder, fieldDesc, node));
}
}
return builder;
}
/**
* Turn a Tuple into a Message with the given type.
*
* @param builder a builder for the Message type the tuple will be converted to
* @param tuple the tuple
* @return a message representing the given tuple
*/
public Message tupleToMessage(Builder builder, Tuple tuple) {
List<FieldDescriptor> fieldDescriptors = builder.getDescriptorForType().getFields();
if (tuple == null) {
return builder.build();
}
for (int i = 0; i < fieldDescriptors.size() && i < tuple.size(); i++) {
Object tupleField = null;
FieldDescriptor fieldDescriptor = fieldDescriptors.get(i);
try {
tupleField = tuple.get(i);
} catch (ExecException e) {
LOG.warn(
"Could not convert tuple field "
+ tupleField
+ " to field with descriptor "
+ fieldDescriptor);
continue;
}
if (tupleField != null) {
if (fieldDescriptor.isRepeated()) {
// Repeated fields are set with Lists containing objects of the fields' Java type.
builder.setField(
fieldDescriptor,
dataBagToRepeatedField(builder, fieldDescriptor, (DataBag) tupleField));
} else {
if (fieldDescriptor.getType() == FieldDescriptor.Type.MESSAGE) {
Builder nestedMessageBuilder = builder.newBuilderForField(fieldDescriptor);
builder.setField(
fieldDescriptor,
tupleToMessage((Builder) nestedMessageBuilder, (Tuple) tupleField));
} else {
builder.setField(fieldDescriptor, tupleFieldToSingleField(fieldDescriptor, tupleField));
}
}
}
}
return builder.build();
}
/**
* Wrapper around {@link #readFieldNoTag(CodedInputStream, FieldDescriptor, Builder)}. <br>
* same as <br>
* <code>
* builder.setField(fd, readFieldNoTag(input, fd, builder)); </code>
*/
public static void setFieldValue(CodedInputStream input, FieldDescriptor fd, Builder builder)
throws IOException {
builder.setField(fd, readFieldNoTag(input, fd, builder));
}