ArangoDB: expose driver from ArangoDBBucketManager and ArangoDBDocumentManager

Author: rashtao

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

@@ -0,0 +1,28 @@
+/*
+ *  Copyright (c) 2022 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:
+ *
+ *   Michele Rastelli
+ */
+package org.eclipse.jnosql.databases.arangodb.communication;
+
+import com.arangodb.ArangoDB;
+
+public interface ArangoDBAccessor {
+
+    /**
+     * Accessor for the underlying ArangoDB driver instance.
+     *
+     * @return the {@link ArangoDB} instance
+     */
+    ArangoDB getArangoDB();
+
+}

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

@@ -40,7 +40,7 @@
  * <p>{@link BucketManager#put(Iterable, Duration)}</p>
  * <p>{@link BucketManager#put(Iterable, Duration)}</p>
  */
-public class ArangoDBBucketManager implements BucketManager {
+public class ArangoDBBucketManager implements BucketManager, ArangoDBAccessor {
 
 
     private static final String KEY = "_key";
@@ -145,4 +145,9 @@ public void put(KeyValueEntity entity, Duration ttl) throws NullPointerException
         throw new UnsupportedOperationException("ArangoDB does not support TTL");
     }
 
+    @Override
+    public ArangoDB getArangoDB() {
+        return arangoDB;
+    }
+
 }

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

@@ -23,7 +23,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 {
+public interface ArangoDBDocumentManager extends DatabaseManager, ArangoDBAccessor {
 
     /**
      * Executes an ArangoDB query using the ArangoDB Query Language (AQL).

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

@@ -201,14 +201,15 @@ public Iterable<CommunicationEntity> insert(Iterable<CommunicationEntity> entiti
                 .collect(Collectors.toList());
     }
 
+    @Override
+    public ArangoDB getArangoDB() {
+        return arangoDB;
+    }
+
     private void updateEntity(CommunicationEntity entity, String key, String id, String rev) {
         entity.add(Element.of(KEY, key));
         entity.add(Element.of(ID, id));
         entity.add(Element.of(REV, rev));
     }
 
-    ArangoDB getArangoDB() {
-        return arangoDB;
-    }
-
 }

jnosql-arangodb/src/test/java/org/eclipse/jnosql/databases/arangodb/communication/ArangoDBDocumentManagerTest.java

@@ -322,6 +322,13 @@ void shouldIncludeSkip() {
         org.assertj.core.api.Assertions.assertThat(indexes).hasSize(15);
     }
 
+    @Test
+    void shouldExposeArangoDB() {
+        ArangoDB adb = entityManager.getArangoDB();
+        assertThat(adb).isNotNull();
+        assertThat(adb.getVersion()).isNotNull();
+    }
+
     private CommunicationEntity getEntity() {
         CommunicationEntity entity = CommunicationEntity.of(COLLECTION_NAME);
         Map<String, Object> map = new HashMap<>();

jnosql-arangodb/src/test/java/org/eclipse/jnosql/databases/arangodb/communication/ArangoDBKeyValueEntityManagerTest.java

@@ -15,6 +15,7 @@
 
 package org.eclipse.jnosql.databases.arangodb.communication;
 
+import com.arangodb.ArangoDB;
 import org.eclipse.jnosql.communication.Value;
 import org.eclipse.jnosql.communication.keyvalue.BucketManager;
 import org.eclipse.jnosql.communication.keyvalue.BucketManagerFactory;
@@ -118,4 +119,14 @@ public void shouldRemoveMultiKey() {
         Iterable<Value> users = values;
         assertEquals(0L, StreamSupport.stream(keyValueEntityManager.get(keys).spliterator(), false).count());
     }
+
+    @Test
+    void getArangoDB() {
+        assertThat(keyValueEntityManager).isInstanceOf(ArangoDBBucketManager.class);
+        ArangoDBBucketManager adbAccessor = (ArangoDBBucketManager) keyValueEntityManager;
+        ArangoDB adb = adbAccessor.getArangoDB();
+        assertThat(adb).isNotNull();
+        assertThat(adb.getVersion()).isNotNull();
+    }
+
 }