Skip to content

Commit

Permalink
[CALCITE-6243] Upgrade Cassandra to 4.1.3 and DataStax driver for Cas…
Browse files Browse the repository at this point in the history
…sandra to 4.17.0
  • Loading branch information
asolimando committed Feb 10, 2024
1 parent df54592 commit bb8b12b
Show file tree
Hide file tree
Showing 3 changed files with 31 additions and 29 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -46,9 +46,6 @@ class CassandraAdapterTest {
private static final ImmutableMap<String, String> TWISSANDRA =
CassandraExtension.getDataset("/model.json");

private static final ImmutableMap<String, String> TWISSANDRA_WITHOUT_KEYSPACE =
CassandraExtension.getDataset("/model-without-keyspace.json");

@BeforeAll
static void load(CqlSession session) {
new CQLDataLoader(session)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -17,16 +17,14 @@
package org.apache.calcite.test;

import org.apache.calcite.config.CalciteSystemProperty;
import org.apache.calcite.util.Bug;
import org.apache.calcite.util.Sources;
import org.apache.calcite.util.TestUtil;

import org.apache.cassandra.concurrent.Stage;
import org.apache.cassandra.config.DatabaseDescriptor;
import org.apache.cassandra.db.WindowsFailedSnapshotTracker;
import org.apache.cassandra.service.CassandraDaemon;
import org.apache.cassandra.service.StorageService;
import org.apache.cassandra.utils.FBUtilities;
import org.apache.commons.lang3.SystemUtils;

import com.datastax.oss.driver.api.core.CqlSession;
import com.google.common.collect.ImmutableMap;
Expand All @@ -43,8 +41,6 @@
import java.io.IOException;
import java.io.UncheckedIOException;
import java.lang.reflect.Field;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.time.Duration;
import java.util.Locale;
import java.util.Objects;
Expand All @@ -53,10 +49,11 @@
/**
* JUnit5 extension to start and stop embedded Cassandra server.
*
* <p>Note that tests will be skipped if running on JDK11+ or Eclipse OpenJ9 JVM
* (not supported by cassandra-unit and Cassandra, respectively) see
* <p>Note that tests will be skipped if running on JDK11+, Eclipse OpenJ9 JVM or Windows
* (the first isn't supported by cassandra-unit, the last two by Cassandra) see
* <a href="https://github.com/jsevellec/cassandra-unit/issues/294">cassandra-unit issue #294</a>
* and <a href="https://issues.apache.org/jira/browse/CASSANDRA-14883">CASSANDRA-14883</a>,
* <a href="https://issues.apache.org/jira/browse/CASSANDRA-14883">CASSANDRA-14883</a>,
* and <a href="https://issues.apache.org/jira/browse/CASSANDRA-16956">CASSANDRA-16956</a>,
* respectively.
*/
class CassandraExtension implements ParameterResolver, ExecutionCondition {
Expand Down Expand Up @@ -121,22 +118,39 @@ private static CassandraResource getOrCreate(ExtensionContext context) {
* {@link com.google.common.collect.ImmutableSet#builderWithExpectedSize(int)}
* and therefore Guava 23 or higher.
*
* <p>Cassandra dropped support for Windows in 4.1.x
* @see <a href="https://issues.apache.org/jira/browse/CASSANDRA-16956">CASSANDRA-16956</a>
*
* @return {@code true} if test is compatible with current environment,
* {@code false} otherwise
*/
@Override public ConditionEvaluationResult evaluateExecutionCondition(
final ExtensionContext context) {
boolean enabled = CalciteSystemProperty.TEST_CASSANDRA.value();
Bug.upgrade("remove JDK version check once cassandra-unit supports JDK11+");
// remove JDK version check once cassandra-unit supports JDK11+
boolean compatibleJdk = TestUtil.getJavaMajorVersion() < 11;
boolean compatibleGuava = TestUtil.getGuavaMajorVersion() >= 23;
Bug.upgrade("remove JVM check once Cassandra supports Eclipse OpenJ9 JVM");
// remove JVM check once Cassandra supports Eclipse OpenJ9 JVM
boolean compatibleJVM = !"Eclipse OpenJ9".equals(TestUtil.getJavaVirtualMachineVendor());
if (enabled && compatibleJdk && compatibleGuava && compatibleJVM) {
return ConditionEvaluationResult.enabled("Cassandra enabled");
} else {
return ConditionEvaluationResult.disabled("Cassandra tests disabled");
boolean compatibleOS = !SystemUtils.IS_OS_WINDOWS;
if (enabled && compatibleJdk && compatibleGuava && compatibleJVM && compatibleOS) {
return ConditionEvaluationResult.enabled("Cassandra tests enabled");
}

final StringBuilder sb = new StringBuilder("Cassandra tests disabled:");
if (!compatibleJdk) {
sb.append("\n - JDK11+ is not supported");
}
if (!compatibleGuava) {
sb.append("\n - Guava versions < 23 are not supported");
}
if (!compatibleJVM) {
sb.append("\n - Eclipse OpenJ9 JVM is not supported");
}
if (!compatibleOS) {
sb.append("\n - Cassandra doesn't support Windows");
}
return ConditionEvaluationResult.disabled(sb.toString());
}

/** Cassandra resource. */
Expand All @@ -157,7 +171,7 @@ private CassandraResource() {
* during close, clean shutdown (as part of unit test) is not
* straightforward.
*/
@Override public void close() throws IOException {
@Override public void close() {
session.close();

CassandraDaemon daemon = extractDaemon();
Expand All @@ -175,14 +189,6 @@ private CassandraResource() {
Thread.currentThread().interrupt();
}
Stage.shutdownNow();

if (FBUtilities.isWindows) {
// for some reason .toDelete stale folder is not deleted on cassandra shutdown
// doing it manually here
WindowsFailedSnapshotTracker.resetForTests();
// manually delete stale file(s)
Files.deleteIfExists(Paths.get(WindowsFailedSnapshotTracker.TODELETEFILE));
}
}

private static void startCassandra() {
Expand Down Expand Up @@ -223,5 +229,4 @@ private static CassandraDaemon extractDaemon() {
}
}
}

}
4 changes: 2 additions & 2 deletions gradle.properties
Original file line number Diff line number Diff line change
Expand Up @@ -85,8 +85,8 @@ aggdesigner-algorithm.version=6.0
apiguardian-api.version=1.1.2
asm.version=7.2
byte-buddy.version=1.9.3
cassandra-all.version=4.0.1
cassandra-java-driver-core.version=4.13.0
cassandra-all.version=4.1.3
cassandra-java-driver-core.version=4.17.0
cassandra-unit.version=4.3.1.0
chinook-data-hsqldb.version=0.2
commons-codec.version=1.16.0
Expand Down

0 comments on commit bb8b12b

Please sign in to comment.