Fix for #377 - ClassCastException when acquireHostList=true (#378)

* isuse #377 - use a separate (sync) HostHandler and HostResolver for the Execution used to call `/_api/cluster/endpoints`

* re-enabled acquireHostList in async tests

Co-authored-by: Michele Rastelli <rashtao@gmail.com>

Author: wajda

src/main/java/com/arangodb/async/ArangoDBAsync.java

@@ -44,6 +44,7 @@
 import com.arangodb.internal.util.ArangoSerializerImpl;
 import com.arangodb.internal.util.DefaultArangoSerialization;
 import com.arangodb.internal.velocystream.VstCommunicationSync;
+import com.arangodb.internal.velocystream.VstConnectionFactorySync;
 import com.arangodb.model.DBCreateOptions;
 import com.arangodb.model.LogOptions;
 import com.arangodb.model.UserCreateOptions;
@@ -775,12 +776,22 @@ public synchronized ArangoDBAsync build() {
 
             final int max = maxConnections != null ? Math.max(1, maxConnections)
                     : ArangoDefaults.MAX_CONNECTIONS_VST_DEFAULT;
-            final ConnectionFactory connectionFactory = new VstConnectionFactoryAsync(host, timeout, connectionTtl,
+            final ConnectionFactory syncConnectionFactory = new VstConnectionFactorySync(host, timeout, connectionTtl,
                     keepAliveInterval, useSsl, sslContext);
-            final HostResolver hostResolver = createHostResolver(createHostList(max, connectionFactory), max,
-                    connectionFactory);
-            final HostHandler hostHandler = createHostHandler(hostResolver);
-            return new ArangoDBAsyncImpl(asyncBuilder(hostHandler), util, syncBuilder(hostHandler), hostResolver,
+            final ConnectionFactory asyncConnectionFactory = new VstConnectionFactoryAsync(host, timeout, connectionTtl,
+                    keepAliveInterval, useSsl, sslContext);
+            final HostResolver syncHostResolver = createHostResolver(createHostList(max, syncConnectionFactory), max,
+                    syncConnectionFactory);
+            final HostResolver asyncHostResolver = createHostResolver(createHostList(max, asyncConnectionFactory), max,
+                    asyncConnectionFactory);
+            final HostHandler syncHostHandler = createHostHandler(syncHostResolver);
+            final HostHandler asyncHostHandler = createHostHandler(asyncHostResolver);
+            return new ArangoDBAsyncImpl(
+                    asyncBuilder(asyncHostHandler),
+                    util,
+                    syncBuilder(syncHostHandler),
+                    asyncHostResolver,
+                    syncHostResolver,
                     new ArangoContext());
         }
 

src/main/java/com/arangodb/async/internal/ArangoDBAsyncImpl.java

@@ -52,17 +52,24 @@ public class ArangoDBAsyncImpl extends InternalArangoDB<ArangoExecutorAsync> imp
 
     private final CommunicationProtocol cp;
 
-    public ArangoDBAsyncImpl(final VstCommunicationAsync.Builder commBuilder, final ArangoSerializationFactory util,
-                             final VstCommunicationSync.Builder syncbuilder, final HostResolver hostResolver, final ArangoContext context) {
+    public ArangoDBAsyncImpl(
+            final VstCommunicationAsync.Builder asyncCommBuilder,
+            final ArangoSerializationFactory util,
+            final VstCommunicationSync.Builder syncCommBuilder,
+            final HostResolver asyncHostResolver,
+            final HostResolver syncHostResolver,
+            final ArangoContext context
+    ) {
 
-        super(new ArangoExecutorAsync(commBuilder.build(util.get(Serializer.INTERNAL)), util, new DocumentCache()), util, context);
+        super(new ArangoExecutorAsync(asyncCommBuilder.build(util.get(Serializer.INTERNAL)), util, new DocumentCache()), util, context);
 
-        final VstCommunication<Response, VstConnectionSync> cacheCom = syncbuilder.build(util.get(Serializer.INTERNAL));
+        final VstCommunication<Response, VstConnectionSync> cacheCom = syncCommBuilder.build(util.get(Serializer.INTERNAL));
 
         cp = new VstProtocol(cacheCom);
 
         ArangoExecutorSync arangoExecutorSync = new ArangoExecutorSync(cp, util, new DocumentCache());
-        hostResolver.init(arangoExecutorSync, util.get(Serializer.INTERNAL));
+        asyncHostResolver.init(arangoExecutorSync, util.get(Serializer.INTERNAL));
+        syncHostResolver.init(arangoExecutorSync, util.get(Serializer.INTERNAL));
 
     }
 

src/test/java/com/arangodb/async/ArangoDBTest.java

@@ -68,7 +68,7 @@ public void connect() throws Exception {
         final SSLContext sc = SSLContext.getInstance("TLS");
         sc.init(kmf.getKeyManagers(), tmf.getTrustManagers(), null);
 
-        final ArangoDBAsync arangoDB = new ArangoDBAsync.Builder().acquireHostList(false)
+        final ArangoDBAsync arangoDB = new ArangoDBAsync.Builder()
                 .loadProperties(ArangoSslTest.class.getResourceAsStream("/arangodb-ssl.properties")).useSsl(true)
                 .sslContext(sc).build();
         final ArangoDBVersion version = arangoDB.getVersion().get();
@@ -79,7 +79,7 @@ public void connect() throws Exception {
     @Ignore
     public void connectWithoutValidSslContext() throws Exception {
         try {
-            final ArangoDBAsync arangoDB = new ArangoDBAsync.Builder().acquireHostList(false)
+            final ArangoDBAsync arangoDB = new ArangoDBAsync.Builder()
                     .loadProperties(ArangoSslTest.class.getResourceAsStream("/arangodb-ssl.properties")).useSsl(true)
                     .build();
             arangoDB.getVersion().get();

src/test/java/com/arangodb/async/ArangoSslTest.java

@@ -40,13 +40,10 @@ public abstract class BaseTest {
     static ArangoDBAsync arangoDB;
     static ArangoDatabaseAsync db;
 
-    @ClassRule
-    public static TestRule acquireHostListRule = TestUtils.acquireHostListRule;
-
     @BeforeClass
     public static void init() throws InterruptedException, ExecutionException {
         if (arangoDB == null) {
-            arangoDB = new ArangoDBAsync.Builder().acquireHostList(false).build();
+            arangoDB = new ArangoDBAsync.Builder().build();
         }
 
         if (arangoDB.db(TEST_DB).exists().get()) {

src/test/java/com/arangodb/async/BaseTest.java

@@ -36,7 +36,7 @@ public class CommunicationTest {
     @Test
     @Ignore
     public void disconnect() {
-        final ArangoDBAsync arangoDB = new ArangoDBAsync.Builder().acquireHostList(false).build();
+        final ArangoDBAsync arangoDB = new ArangoDBAsync.Builder().build();
         final CompletableFuture<ArangoCursorAsync<Object>> result = arangoDB.db().query("return sleep(1)", null, null,
                 null);
         arangoDB.shutdown();

src/test/java/com/arangodb/async/CommunicationTest.java

@@ -42,7 +42,7 @@ public class ConcurrencyTest {
 
     @Before
     public void initialize() {
-        arangoDB = new ArangoDBAsync.Builder().acquireHostList(false).build();
+        arangoDB = new ArangoDBAsync.Builder().build();
     }
 
     /**

src/test/java/com/arangodb/async/ConcurrencyTest.java

@@ -48,7 +48,7 @@ public class ConsolidationIntervalMsec extends BaseTest {
     public void consolidationIntervalMsec() throws ExecutionException, InterruptedException {
         assumeTrue(isAtLeastVersion(3, 4));
 
-        ArangoDBAsync arango = new ArangoDBAsync.Builder().acquireHostList(false)
+        ArangoDBAsync arango = new ArangoDBAsync.Builder()
                 .user("root")
                 .password("test")
                 .build();

src/test/java/com/arangodb/async/TestUtils.java

@@ -41,7 +41,7 @@ public class ExampleBase {
 
     @BeforeClass
     public static void setUp() throws InterruptedException, ExecutionException {
-        arangoDB = new ArangoDBAsync.Builder().acquireHostList(false).build();
+        arangoDB = new ArangoDBAsync.Builder().build();
         if (arangoDB.db(DB_NAME).exists().get()) {
             arangoDB.db(DB_NAME).drop().get();
         }