Merge pull request #340 from rashtao/feature/arangodb-graph

jnosql-arangodb: Graph API

Author: otaviojava

jnosql-arangodb/pom.xml

@@ -41,6 +41,11 @@
             <groupId>org.eclipse.jnosql.mapping</groupId>
             <artifactId>jnosql-mapping-key-value</artifactId>
         </dependency>
+        <dependency>
+            <groupId>org.eclipse.jnosql.mapping</groupId>
+            <artifactId>jnosql-mapping-graph</artifactId>
+            <version>${project.version}</version>
+        </dependency>
         <dependency>
             <groupId>${project.groupId}</groupId>
             <artifactId>jnosql-database-commons</artifactId>

jnosql-arangodb/src/main/java/org/eclipse/jnosql/databases/arangodb/communication/ArangoDBCommunicationEdge.java

@@ -0,0 +1,25 @@
+/*
+ *
+ *  Copyright (c) 2025 Contributors to the Eclipse Foundation
+ *   All rights reserved. This program and the accompanying materials
+ *   are made available under the terms of the Eclipse Public License v1.0
+ *   and Apache License v2.0 which accompanies this distribution.
+ *   The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
+ *   and the Apache License v2.0 is available at http://www.opensource.org/licenses/apache2.0.php.
+ *
+ *   You may elect to redistribute this code under either of these licenses.
+ *
+ *   Contributors:
+ *
+ *   Otavio Santana
+ *
+ */
+package org.eclipse.jnosql.databases.arangodb.communication;
+
+import org.eclipse.jnosql.communication.graph.CommunicationEdge;
+import org.eclipse.jnosql.communication.semistructured.CommunicationEntity;
+
+import java.util.Map;
+
+record ArangoDBCommunicationEdge(String id, CommunicationEntity source, CommunicationEntity target, String label, Map<String, Object> properties) implements CommunicationEdge {
+}

jnosql-arangodb/src/main/java/org/eclipse/jnosql/databases/arangodb/communication/ArangoDBDocumentManager.java

@@ -14,6 +14,7 @@
  */
 package org.eclipse.jnosql.databases.arangodb.communication;
 
+import org.eclipse.jnosql.communication.graph.GraphDatabaseManager;
 import org.eclipse.jnosql.communication.semistructured.CommunicationEntity;
 import org.eclipse.jnosql.communication.semistructured.DatabaseManager;
 
@@ -23,7 +24,7 @@
  * The ArangoDB implementation of {@link DatabaseManager}. This implementation does not support TTL methods in the context of
  * {@link DatabaseManager#insert(org.eclipse.jnosql.communication.semistructured.CommunicationEntity)}.
  */
-public interface ArangoDBDocumentManager extends DatabaseManager, ArangoDBAccessor {
+public interface ArangoDBDocumentManager extends DatabaseManager, GraphDatabaseManager, ArangoDBAccessor {
 
     /**
      * Executes an ArangoDB query using the ArangoDB Query Language (AQL).

jnosql-arangodb/src/main/java/org/eclipse/jnosql/databases/arangodb/communication/ArangoDBUtil.java

@@ -16,8 +16,10 @@
 package org.eclipse.jnosql.databases.arangodb.communication;
 
 
+import com.arangodb.ArangoCollection;
 import com.arangodb.ArangoDB;
-import com.arangodb.entity.CollectionEntity;
+import com.arangodb.entity.CollectionType;
+import com.arangodb.model.CollectionCreateOptions;
 import jakarta.json.Json;
 import jakarta.json.JsonArray;
 import jakarta.json.JsonArrayBuilder;
@@ -30,13 +32,10 @@
 import org.eclipse.jnosql.communication.semistructured.CommunicationEntity;
 import org.eclipse.jnosql.communication.semistructured.Element;
 
-import java.util.Collection;
-import java.util.Collections;
-import java.util.List;
-import java.util.Map;
-import java.util.Objects;
+import java.util.*;
 import java.util.logging.Level;
 import java.util.logging.Logger;
+import java.util.stream.Collectors;
 
 import static java.util.stream.Collectors.toList;
 import static java.util.stream.StreamSupport.stream;
@@ -49,12 +48,17 @@ public final class ArangoDBUtil {
     public static final String KEY = "_key";
     public static final String ID = "_id";
     public static final String REV = "_rev";
+    public static final String FROM = "_from";
+    public static final String TO = "_to";
+    private static final Set<String> META_FIELDS = Collections.unmodifiableSet(new HashSet<>(Arrays.asList(
+            ID, KEY, REV, FROM, TO
+    )));
+
     private static final Logger LOGGER = Logger.getLogger(ArangoDBUtil.class.getName());
 
     private ArangoDBUtil() {
     }
 
-
     static void checkDatabase(String database, ArangoDB arangoDB) {
         Objects.requireNonNull(database, "database is required");
         try {
@@ -67,14 +71,21 @@ static void checkDatabase(String database, ArangoDB arangoDB) {
         }
     }
 
-    public static void checkCollection(String bucketName, ArangoDB arangoDB, String namespace) {
-        checkDatabase(bucketName, arangoDB);
-        List<String> collections = arangoDB.db(bucketName)
-                .getCollections().stream()
-                .map(CollectionEntity::getName)
-                .toList();
-        if (!collections.contains(namespace)) {
-            arangoDB.db(bucketName).createCollection(namespace);
+    public static void checkCollection(String dbName, ArangoDB arangoDB, String collectionName) {
+        checkDatabase(dbName, arangoDB);
+        ArangoCollection collection = arangoDB.db(dbName).collection(collectionName);
+        if (!collection.exists()) {
+            collection.create();
+        }
+    }
+
+    public static void checkEdgeCollection(String dbName, ArangoDB arangoDB, String collectionName) {
+        checkDatabase(dbName, arangoDB);
+        ArangoCollection collection = arangoDB.db(dbName).collection(collectionName);
+        if (!collection.exists()) {
+            collection.create(new CollectionCreateOptions().type(CollectionType.EDGES));
+        } else if (collection.getInfo().getType() != CollectionType.EDGES) {
+            throw new IllegalStateException(String.format("The collection %s is not an edge collection", collectionName));
         }
     }
 
@@ -85,10 +96,27 @@ static CommunicationEntity toEntity(JsonObject jsonObject) {
         documents.add(Element.of(KEY, jsonObject.getString(KEY)));
         documents.add(Element.of(ID, id));
         documents.add(Element.of(REV, jsonObject.getString(REV)));
+        if (jsonObject.containsKey(FROM)) {
+            documents.add(Element.of(FROM, jsonObject.getString(FROM)));
+        }
+        if (jsonObject.containsKey(TO)) {
+            documents.add(Element.of(TO, jsonObject.getString(TO)));
+        }
         String collection = id.split("/")[0];
         return CommunicationEntity.of(collection, documents);
     }
 
+    static ArangoDBCommunicationEdge toEdge(JsonObject edge, JsonObject from, JsonObject to) {
+        String id = edge.getString(ID);
+        String label = id.split("/")[0];
+        Map<String, Object> properties = edge.entrySet().stream()
+                .filter(e -> !META_FIELDS.contains(e.getKey()))
+                .map(e -> new AbstractMap.SimpleEntry<>(e.getKey(), toDocuments(e.getValue())))
+                .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
+
+        return new ArangoDBCommunicationEdge(id, toEntity(edge), toEntity(from), label, properties);
+    }
+
     static JsonObject toJsonObject(CommunicationEntity entity) {
         return toJsonObject(entity.elements());
     }