made BaseDocument and BaseEdgeDocument serializable (#596)

arangodb · Jan 27, 2025 · 6e52fa0 · 6e52fa0
1 parent bafed68
commit 6e52fa0
Show file tree

Hide file tree

Showing 4 changed files with 54 additions and 1 deletion.
diff --git a/core/src/main/java/com/arangodb/entity/AbstractBaseDocument.java b/core/src/main/java/com/arangodb/entity/AbstractBaseDocument.java
@@ -26,6 +26,7 @@
 import com.fasterxml.jackson.annotation.JsonIgnore;
 import com.fasterxml.jackson.annotation.JsonInclude;
 
+import java.io.Serializable;
 import java.util.Collections;
 import java.util.HashMap;
 import java.util.Map;
@@ -35,7 +36,9 @@
  * @author Mark Vollmary
  * @author Michele Rastelli
  */
-abstract class AbstractBaseDocument {
+abstract class AbstractBaseDocument implements Serializable {
+
+    private static final long serialVersionUID = 6985324876843525239L;
 
     private static final String[] META_PROPS = new String[]{
             DocumentFields.ID,

diff --git a/...sources/META-INF/native-image/com.arangodb/arangodb-java-driver/serialization-config.json b/...sources/META-INF/native-image/com.arangodb/arangodb-java-driver/serialization-config.json
@@ -10,5 +10,17 @@
   },
   {
     "name": "com.arangodb.internal.net.ArangoDBRedirectException"
+  },
+  {
+    "name": "com.arangodb.entity.AbstractBaseDocument"
+  },
+  {
+    "name": "com.arangodb.entity.BaseDocument"
+  },
+  {
+    "name": "com.arangodb.entity.BaseEdgeDocument"
+  },
+  {
+    "name": "java.util.HashMap"
   }
 ]
diff --git a/.../META-INF/native-image/com.arangodb/arangodb-java-driver-shaded/serialization-config.json b/.../META-INF/native-image/com.arangodb/arangodb-java-driver-shaded/serialization-config.json
@@ -10,5 +10,17 @@
   },
   {
     "name": "com.arangodb.internal.net.ArangoDBRedirectException"
+  },
+  {
+    "name": "com.arangodb.entity.AbstractBaseDocument"
+  },
+  {
+    "name": "com.arangodb.entity.BaseDocument"
+  },
+  {
+    "name": "com.arangodb.entity.BaseEdgeDocument"
+  },
+  {
+    "name": "java.util.HashMap"
   }
 ]
diff --git a/test-functional/src/test/java/com/arangodb/SerializableTest.java b/test-functional/src/test/java/com/arangodb/SerializableTest.java
@@ -1,5 +1,7 @@
 package com.arangodb;
 
+import com.arangodb.entity.BaseDocument;
+import com.arangodb.entity.BaseEdgeDocument;
 import com.arangodb.entity.ErrorEntity;
 import com.arangodb.internal.net.ArangoDBRedirectException;
 import com.fasterxml.jackson.databind.JsonNode;
@@ -50,6 +52,30 @@ void serializeArangoDBMultipleException() throws IOException, ClassNotFoundExcep
         assertThat(e2.getExceptions().iterator().next().getMessage()).isEqualTo("foo");
     }
 
+    @Test
+    void serializeBaseDocument() throws IOException, ClassNotFoundException {
+        BaseDocument doc = new BaseDocument();
+        doc.setKey("test");
+        doc.setId("id");
+        doc.setRevision("revision");
+        doc.addAttribute("foo", "bar");
+        BaseDocument doc2 = roundTrip(doc);
+        assertThat(doc2).isEqualTo(doc);
+    }
+
+    @Test
+    void serializeBaseEdgeDocument() throws IOException, ClassNotFoundException {
+        BaseEdgeDocument doc = new BaseEdgeDocument();
+        doc.setKey("test");
+        doc.setId("id");
+        doc.setRevision("revision");
+        doc.setFrom("from");
+        doc.setTo("to");
+        doc.addAttribute("foo", "bar");
+        BaseDocument doc2 = roundTrip(doc);
+        assertThat(doc2).isEqualTo(doc);
+    }
+
     private <T> T roundTrip(T input) throws IOException, ClassNotFoundException {
         ByteArrayOutputStream os = new ByteArrayOutputStream();
         ObjectOutputStream objectOutputStream = new ObjectOutputStream(os);