Skip to content

Commit cb96217

Browse files
committed
merge master and update
1 parent 309de31 commit cb96217

File tree

3 files changed

+20
-58
lines changed

3 files changed

+20
-58
lines changed

test/e2e/env/src/test/java/org/apache/shardingsphere/test/e2e/env/container/atomic/storage/StorageContainerFactory.java

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@
2222
import org.apache.shardingsphere.database.connector.core.type.DatabaseType;
2323
import org.apache.shardingsphere.test.e2e.env.container.atomic.storage.option.StorageContainerConfigurationOption;
2424
import org.apache.shardingsphere.test.e2e.env.container.atomic.storage.type.docker.DockerStorageContainer;
25-
import org.apache.shardingsphere.test.e2e.env.container.atomic.storage.type.docker.impl.HiveContainer;
2625
import org.apache.shardingsphere.test.e2e.env.runtime.E2ETestEnvironment;
2726

2827
/**
@@ -62,7 +61,7 @@ public static StorageContainer newInstance(final DatabaseType databaseType, fina
6261
case "MariaDB":
6362
return new DockerStorageContainer(storageContainerImage, option, scenario);
6463
case "Hive":
65-
return new HiveContainer(storageContainerImage, option, scenario);
64+
return new DockerStorageContainer(storageContainerImage, option, scenario);
6665
default:
6766
throw new RuntimeException(String.format("Database `%s` is unknown.", databaseType.getType()));
6867
}

test/e2e/env/src/test/java/org/apache/shardingsphere/test/e2e/env/container/atomic/storage/type/docker/DockerStorageContainer.java

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,8 @@
3636
import org.apache.shardingsphere.test.e2e.env.runtime.scenario.path.ScenarioDataPath.Type;
3737

3838
import javax.sql.DataSource;
39+
import java.io.IOException;
40+
import lombok.extern.slf4j.Slf4j;
3941
import java.sql.DriverManager;
4042
import java.time.Duration;
4143
import java.util.Collection;
@@ -50,6 +52,7 @@
5052
/**
5153
* Docker storage container.
5254
*/
55+
@Slf4j
5356
public class DockerStorageContainer extends DockerITContainer implements StorageContainer {
5457

5558
private final StorageContainerConfigurationOption option;
@@ -114,10 +117,26 @@ private String getURL() {
114117

115118
@Override
116119
protected void postStart() {
120+
if ("Hive".equalsIgnoreCase(option.getDatabaseType())) {
121+
executeHiveMountedSQLScripts();
122+
}
117123
actualDataSourceMap.putAll(createAccessDataSources(getDataSourceNames(getDataSourceNameAndTypeMap(Type.ACTUAL))));
118124
expectedDataSourceMap.putAll(createAccessDataSources(getDataSourceNames(getDataSourceNameAndTypeMap(Type.EXPECTED))));
119125
}
120126

127+
private void executeHiveMountedSQLScripts() {
128+
try {
129+
execInContainer("bash", "-c",
130+
"if [ -f /docker-entrypoint-initdb.d/50-scenario-actual-init.sql ]; then beeline -u \"jdbc:hive2://localhost:10000/default\" -f "
131+
+ "/docker-entrypoint-initdb.d/50-scenario-actual-init.sql; fi");
132+
execInContainer("bash", "-c",
133+
"if [ -f /docker-entrypoint-initdb.d/60-scenario-expected-init.sql ]; then beeline -u \"jdbc:hive2://localhost:10000/default\" -f "
134+
+ "/docker-entrypoint-initdb.d/60-scenario-expected-init.sql; fi");
135+
} catch (final InterruptedException | IOException ex) {
136+
log.warn("Hive init scripts execution failed.", ex);
137+
}
138+
}
139+
121140
private Map<String, DatabaseType> getDataSourceNameAndTypeMap(final Type type) {
122141
return null == scenario ? Collections.emptyMap() : DatabaseEnvironmentManager.getDatabaseTypes(scenario, option.getType(), type);
123142
}

test/e2e/env/src/test/java/org/apache/shardingsphere/test/e2e/env/container/atomic/storage/type/docker/impl/HiveContainer.java

Lines changed: 0 additions & 56 deletions
This file was deleted.

0 commit comments

Comments
 (0)