Revert change, causing NPE on runTestViaConfig() test

aerospike · Jun 19, 2024 · 2717ff2 · 2717ff2
1 parent 345a2c2
commit 2717ff2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/main/java/com/aerospike/mapper/tools/configuration/ClassConfig.java b/src/main/java/com/aerospike/mapper/tools/configuration/ClassConfig.java
@@ -232,8 +232,13 @@ public Builder withKeyField(String fieldName) {
         }
 
         public Builder withKeyFieldAndStoreAsBin(String fieldName, boolean storeAsBin) {
+            if (this.classConfig.getKey() == null) {
+                this.classConfig.setKey(new KeyConfig());
+            }
+            this.validateFieldExists(fieldName);
+            this.classConfig.getKey().setField(fieldName);
             this.classConfig.getKey().setStoreAsBin(storeAsBin);
-            return withKeyField(fieldName);
+            return this;
         }
 
         public Builder withKeyGetterAndSetterOf(String getterName, String setterName) {