Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 7 additions & 2 deletions pom.xml
Original file line number Diff line number Diff line change
Expand Up @@ -82,6 +82,12 @@
<groupId>com.datastax.cassandra</groupId>
<artifactId>cassandra-driver-core</artifactId>
<version>3.0.0</version>
<exclusions>
<exclusion>
<groupId>com.google.guava</groupId>
<artifactId>guava</artifactId>
</exclusion>
</exclusions>
</dependency>
<dependency>
<groupId>junit</groupId>
Expand Down Expand Up @@ -124,8 +130,7 @@
<dependency>
<groupId>com.google.guava</groupId>
<artifactId>guava</artifactId>
<version>18.0</version>
<scope>test</scope>
<version>19.0</version>
</dependency>
</dependencies>

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -20,13 +20,8 @@
import com.contrastsecurity.cassandra.migration.config.ScriptsLocations;
import com.contrastsecurity.cassandra.migration.info.ResolvedMigration;
import com.contrastsecurity.cassandra.migration.resolver.cql.CqlMigrationResolver;
import com.contrastsecurity.cassandra.migration.resolver.java.ExternalJavaMigrationResolver;
import com.contrastsecurity.cassandra.migration.resolver.java.JavaMigrationResolver;

import java.io.File;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.*;

/**
Expand Down Expand Up @@ -59,11 +54,6 @@ public CompositeMigrationResolver(ClassLoader classLoader, ScriptsLocations loca
for (ScriptsLocation location : locations.getLocations()) {
migrationResolvers.add(new CqlMigrationResolver(classLoader, location, encoding));
migrationResolvers.add(new JavaMigrationResolver(classLoader, location));
try {
migrationResolvers.add(new ExternalJavaMigrationResolver(new URLClassLoader(new URL[]{new File(location.getPath()).toURI().toURL()}, this.getClass().getClassLoader()), location));
} catch (MalformedURLException e) {
throw new CassandraMigrationException("Unable to add external url for classes migrations in location: " + location, e);
}
}

migrationResolvers.addAll(Arrays.asList(customMigrationResolvers));
Expand Down

This file was deleted.

This file was deleted.

Original file line number Diff line number Diff line change
Expand Up @@ -15,10 +15,21 @@
*/
package com.contrastsecurity.cassandra.migration.resolver.java;

import com.contrastsecurity.cassandra.migration.CassandraMigrationException;
import com.contrastsecurity.cassandra.migration.api.MigrationChecksumProvider;
import com.contrastsecurity.cassandra.migration.api.MigrationInfoProvider;
import com.contrastsecurity.cassandra.migration.api.JavaMigration;
import com.contrastsecurity.cassandra.migration.config.MigrationType;
import com.contrastsecurity.cassandra.migration.config.ScriptsLocation;
import com.contrastsecurity.cassandra.migration.info.MigrationVersion;
import com.contrastsecurity.cassandra.migration.info.ResolvedMigration;
import com.contrastsecurity.cassandra.migration.resolver.MigrationInfoHelper;
import com.contrastsecurity.cassandra.migration.resolver.MigrationResolver;
import com.contrastsecurity.cassandra.migration.resolver.ResolvedMigrationComparator;
import com.contrastsecurity.cassandra.migration.utils.ClassUtils;
import com.contrastsecurity.cassandra.migration.utils.Pair;
import com.contrastsecurity.cassandra.migration.utils.StringUtils;
import com.contrastsecurity.cassandra.migration.utils.scanner.Scanner;

import java.util.ArrayList;
import java.util.Collections;
Expand All @@ -28,7 +39,7 @@
* Migration resolver for Java migrations. The classes must have a name like V1 or V1_1_3 or V1__Description
* or V1_1_3__Description.
*/
public class JavaMigrationResolver extends CommonJavaResolver implements MigrationResolver {
public class JavaMigrationResolver implements MigrationResolver {
/**
* The base package on the classpath where to migrations are located.
*/
Expand All @@ -51,15 +62,70 @@ public JavaMigrationResolver(ClassLoader classLoader, ScriptsLocation location)
}

public List<ResolvedMigration> resolveMigrations() {
List<ResolvedMigration> migrations = new ArrayList<ResolvedMigration>();

if (!location.isClassPath()) {
return Collections.emptyList();
return migrations;
}

List<ResolvedMigration> migrations = loadJavaMigrationFiles(classLoader, location);
try {
Class<?>[] classes = new Scanner(classLoader).scanForClasses(location, JavaMigration.class);
for (Class<?> clazz : classes) {
JavaMigration javaMigration = ClassUtils.instantiate(clazz.getName(), classLoader);

ResolvedMigration migrationInfo = extractMigrationInfo(javaMigration);
migrationInfo.setPhysicalLocation(ClassUtils.getLocationOnDisk(clazz));
migrationInfo.setExecutor(new JavaMigrationExecutor(javaMigration));

migrations.add(migrationInfo);
}
} catch (Exception e) {
throw new CassandraMigrationException("Unable to resolve Java migrations in location: " + location, e);
}

Collections.sort(migrations, new ResolvedMigrationComparator());
return migrations;
}

/**
* Extracts the migration info from this migration.
*
* @param javaMigration The migration to analyse.
* @return The migration info.
*/
ResolvedMigration extractMigrationInfo(JavaMigration javaMigration) {
Integer checksum = null;
if (javaMigration instanceof MigrationChecksumProvider) {
MigrationChecksumProvider checksumProvider = (MigrationChecksumProvider) javaMigration;
checksum = checksumProvider.getChecksum();
}

MigrationVersion version;
String description;
if (javaMigration instanceof MigrationInfoProvider) {
MigrationInfoProvider infoProvider = (MigrationInfoProvider) javaMigration;
version = infoProvider.getVersion();
description = infoProvider.getDescription();
if (!StringUtils.hasText(description)) {
throw new CassandraMigrationException("Missing description for migration " + version);
}
} else {
Pair<MigrationVersion, String> info =
MigrationInfoHelper.extractVersionAndDescription(
ClassUtils.getShortName(javaMigration.getClass()), "V", "__", "");
version = info.getLeft();
description = info.getRight();
}

String script = javaMigration.getClass().getName();


ResolvedMigration resolvedMigration = new ResolvedMigration();
resolvedMigration.setVersion(version);
resolvedMigration.setDescription(description);
resolvedMigration.setScript(script);
resolvedMigration.setChecksum(checksum);
resolvedMigration.setType(MigrationType.JAVA_DRIVER);
return resolvedMigration;
}
}