feat: add utitlity for support serialization of nullable value

bsorrentino · Aug 9, 2024 · 7a82029 · 7a82029
1 parent 9776cbd
commit 7a82029
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/core-jdk8/src/main/java/org/bsc/langgraph4j/serializer/Serializer.java b/core-jdk8/src/main/java/org/bsc/langgraph4j/serializer/Serializer.java
@@ -4,14 +4,28 @@
 
 import java.io.*;
 import java.util.Objects;
+import java.util.Optional;
 
 public interface Serializer<T> {
 
     void write(T object, ObjectOutput out) throws IOException;
-
-
     T read(ObjectInput in) throws IOException, ClassNotFoundException ;
 
+    default void writeNullable(T object, ObjectOutput out) throws IOException {
+        if( object == null ) {
+            out.writeBoolean(false);
+        } else {
+            out.writeBoolean(true);
+            write(object, out);
+        }
+    }
+    default Optional<T> readNullable(ObjectInput in) throws IOException, ClassNotFoundException {
+        if( in.readBoolean() ) {
+            return Optional.ofNullable(read(in));
+        }
+        return Optional.empty();
+    }
+
     default byte[] writeObject(T object) throws IOException {
         Objects.requireNonNull( object, "object cannot be null" );
         try( ByteArrayOutputStream baos = new ByteArrayOutputStream() ) {