diff --git a/metadata-integration/java/datahub-protobuf/src/main/java/datahub/protobuf/model/ProtobufField.java b/metadata-integration/java/datahub-protobuf/src/main/java/datahub/protobuf/model/ProtobufField.java index 477aefea9c..584fdc77a9 100644 --- a/metadata-integration/java/datahub-protobuf/src/main/java/datahub/protobuf/model/ProtobufField.java +++ b/metadata-integration/java/datahub-protobuf/src/main/java/datahub/protobuf/model/ProtobufField.java @@ -73,6 +73,10 @@ public class ProtobufField implements ProtobufElement { return nativeType(); } + public int getNumber() { + return fieldProto.getNumber(); + } + @Override public String nativeType() { return Optional.ofNullable(nativeType).orElseGet(() -> { diff --git a/metadata-integration/java/datahub-protobuf/src/main/java/datahub/protobuf/model/ProtobufGraph.java b/metadata-integration/java/datahub-protobuf/src/main/java/datahub/protobuf/model/ProtobufGraph.java index 3b3eea42af..c9d9709aa2 100644 --- a/metadata-integration/java/datahub-protobuf/src/main/java/datahub/protobuf/model/ProtobufGraph.java +++ b/metadata-integration/java/datahub-protobuf/src/main/java/datahub/protobuf/model/ProtobufGraph.java @@ -306,7 +306,7 @@ public class ProtobufGraph extends DefaultDirectedGraph { + wrappedPrimitiveFields.stream().filter(fld -> fld.getNumber() == 1).forEach(primitiveField -> { // remove incoming old edges to primitive removeEdges.addAll(incomingEdgesOf(primitiveField));