handle null return from func.apply

NucleoTeam · Dec 28, 2023 · e53e829 · e53e829
1 parent 9f8c2d7
commit e53e829
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 11 deletions.
diff --git a/build.gradle b/build.gradle
@@ -6,7 +6,7 @@ plugins {
 }
 
 group = 'com.nucleodb'
-version = '1.13.21'
+version = '1.13.22'
 
 repositories {
     mavenCentral()

diff --git a/src/main/java/com/nucleodb/library/database/tables/table/DataTable.java b/src/main/java/com/nucleodb/library/database/tables/table/DataTable.java
@@ -8,6 +8,7 @@
 import com.google.common.cache.CacheBuilder;
 import com.google.common.cache.CacheLoader;
 import com.google.common.collect.Queues;
+import com.google.common.collect.Sets;
 import com.nucleodb.library.database.modifications.ConnectionCreate;
 import com.nucleodb.library.database.modifications.ConnectionDelete;
 import com.nucleodb.library.database.modifications.ConnectionUpdate;
@@ -281,22 +282,31 @@ public Set<DataEntry> handleIndexOperation(Object obj, DataEntryProjection dataE
     if (dataEntryProjection == null) {
       dataEntryProjection = new DataEntryProjection();
     }
-    Stream<DataEntry> process = dataEntryProjection.process(func.apply(obj).stream());
-    if (dataEntryProjection.isWritable()) {
-      return process.map(de -> (DataEntry) de.copy(this.getConfig().getDataEntryClass())).collect(Collectors.toSet());
+    Set<DataEntry> apply = func.apply(obj);
+    if(apply!=null) {
+      Stream<DataEntry> process = dataEntryProjection.process(apply.stream());
+
+      if (dataEntryProjection.isWritable()) {
+        return process.map(de -> (DataEntry) de.copy(this.getConfig().getDataEntryClass())).collect(Collectors.toSet());
+      }
+      return process.collect(Collectors.toSet());
     }
-    return process.collect(Collectors.toSet());
+    return Sets.newTreeSet();
   }
 
   public Set<DataEntry> handleIndexStringOperation(String obj, DataEntryProjection dataEntryProjection, Function<String, Set<DataEntry>> func) {
     if (dataEntryProjection == null) {
       dataEntryProjection = new DataEntryProjection();
     }
-    Stream<DataEntry> process = dataEntryProjection.process(func.apply(obj).stream());
-    if (dataEntryProjection.isWritable()) {
-      return process.map(de -> (DataEntry) de.copy(this.getConfig().getDataEntryClass())).collect(Collectors.toSet());
+    Set<DataEntry> apply = func.apply(obj);
+    if(apply!=null) {
+      Stream<DataEntry> process = dataEntryProjection.process(apply.stream());
+      if (dataEntryProjection.isWritable()) {
+        return process.map(de -> (DataEntry) de.copy(this.getConfig().getDataEntryClass())).collect(Collectors.toSet());
+      }
+      return process.collect(Collectors.toSet());
     }
-    return process.collect(Collectors.toSet());
+    return Sets.newTreeSet();
   }
 
   public Set<DataEntry> search(String key, Object searchObject, DataEntryProjection dataEntryProjection) {

diff --git a/src/test/java/com/nucleodb/library/helpers/models/AuthorDE.java b/src/test/java/com/nucleodb/library/helpers/models/AuthorDE.java
@@ -4,8 +4,6 @@
 import com.nucleodb.library.database.modifications.Create;
 import com.nucleodb.library.database.tables.table.DataEntry;
 
-import java.util.UUID;
-
 public class AuthorDE extends DataEntry<Author>{
   public AuthorDE(Author obj) {
     super(obj);