Add option to change to camelcase

blueapron · Mar 2, 2023 · 7e63959 · 7e63959
1 parent 44641a1
commit 7e63959
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/main/java/com/blueapron/connect/protobuf/ProtobufData.java b/src/main/java/com/blueapron/connect/protobuf/ProtobufData.java
@@ -50,6 +50,7 @@ class ProtobufData {
   private final Schema schema;
   private final String legacyName;
   private final boolean useConnectSchemaMap;
+  private final boolean useCamelCase;
   public static final Descriptors.FieldDescriptor.Type[] PROTO_TYPES_WITH_DEFAULTS = new Descriptors.FieldDescriptor.Type[] { INT32, INT64, SINT32, SINT64, FLOAT, DOUBLE, BOOL, STRING, BYTES, ENUM };
   private HashMap<String, String> connectProtoNameMap = new HashMap<String, String>();
 
@@ -74,7 +75,7 @@ private String getProtoMapKey(String descriptorContainingTypeName, String connec
   }
 
   private String getConnectFieldName(Descriptors.FieldDescriptor descriptor) {
-    String name = descriptor.getName();
+    String name = useCamelCase ? descriptor.getJsonName() : descriptor.getName();
     for (Map.Entry<Descriptors.FieldDescriptor, Object> option: descriptor.getOptions().getAllFields().entrySet()) {
       if (option.getKey().getFullName().equalsIgnoreCase(this.legacyName)) {
         name = option.getValue().toString();
@@ -93,9 +94,14 @@ private String getProtoFieldName(String descriptorForTypeName, String connectFie
     this(clazz, legacyName, false);
   }
 
-  ProtobufData(Class<? extends com.google.protobuf.GeneratedMessageV3> clazz, String legacyName, boolean useConnectSchemaMap ) {
+  ProtobufData(Class<? extends com.google.protobuf.GeneratedMessageV3> clazz, String legacyName, boolean useConnectSchemaMap) {
+    this(clazz, legacyName, useConnectSchemaMap, false);
+  }
+
+  ProtobufData(Class<? extends com.google.protobuf.GeneratedMessageV3> clazz, String legacyName, boolean useConnectSchemaMap, boolean useCamelCase) {
     this.legacyName = legacyName;
     this.useConnectSchemaMap = useConnectSchemaMap;
+    this.useCamelCase = useCamelCase;
 
     try {
       this.newBuilder = clazz.getDeclaredMethod("newBuilder");