alexey.rassokhin/tet123
1
0
Fork 0
Code Issues Pull Requests 1 Packages Projects Releases Wiki Activity

DBZ-7693 Support reading DDL SQL resources from test-jar file

Browse Source
This commit is contained in:
Chris Cranford 2024-03-28 19:57:08 -04:00 committed by Jiri Pechanec
parent 9d8cb8e534
commit 2181d8b7a3
1 changed files with 40 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
debezium-connector-binlog/src/test/java/io/debezium/connector/binlog/util/UniqueDatabase.java
View File

@ -7,7 +7,11 @@
import static org.junit.Assert.assertNotNull;
import java.io.IOException;
import java.net.URI;
import java.net.URL;
import java.nio.file.FileSystem;
import java.nio.file.FileSystems;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
@ -17,6 +21,7 @@
import java.util.List;
import java.util.Map;
import java.util.Random;
import java.util.function.Function;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Collectors;
@ -158,10 +163,10 @@ public void createAndInitialize(Map<String, Object> urlProperties) {
assertNotNull("Cannot locate " + ddlFile, ddlTestFile);
try {
try (JdbcConnection connection = forTestDatabase(DEFAULT_DATABASE, urlProperties)) {
final List<String> statements = Arrays.stream(
final List<String> statements = readFileContents(ddlTestFile.toURI(), (data) -> Arrays.stream(
Stream.concat(
Arrays.stream(charset != null ? CREATE_DATABASE_WITH_CHARSET_DDL : CREATE_DATABASE_DDL),
Files.readAllLines(Paths.get(ddlTestFile.toURI())).stream())
data)
.map(String::trim)
.filter(x -> !x.startsWith("--") && !x.isEmpty())
.map(x -> {
@ -171,7 +176,7 @@ public void createAndInitialize(Map<String, Object> urlProperties) {
.map(this::convertSQL)
.collect(Collectors.joining("\n")).split(";"))
.map(x -> x.replace("$$", ";"))
.collect(Collectors.toList());
.collect(Collectors.toList()));
connection.execute(statements.toArray(new String[statements.size()]));
}
}
@ -180,6 +185,38 @@ public void createAndInitialize(Map<String, Object> urlProperties) {
}
}
/**
* Supports reading the contents of the SQL file, regardless if its bundled in a jar or not.
*
* @param uri the file URI
* @param handler the handler to receive the file contents
* @return the contents
* @throws IOException if there was an error reading the file contents
*/
private List<String> readFileContents(URI uri, Function<Stream<String>, List<String>> handler) throws IOException {
if ("jar".equals(uri.getScheme())) {
// Open the JAR file
String[] parts = uri.toString().split("!");
URI jarUri = URI.create(parts[0]);
try (FileSystem fs = FileSystems.newFileSystem(jarUri, Collections.emptyMap())) {
try {
return handler.apply(Files.readAllLines(fs.getPath(parts[1])).stream());
}
catch (IOException e) {
throw new IOException("Failed to read contents", e);
}
}
catch (IOException e) {
throw new IOException("Failed to open file system", e);
}
}
else {
// Read from the file system
Path path = Paths.get(uri);
return handler.apply(Files.readAllLines(path).stream());
}
}
/**
* @param dbHistoryPath - directory where to store database schema history
* @see io.debezium.storage.file.history.FileSchemaHistory