-
Notifications
You must be signed in to change notification settings - Fork 480
/
AbstractContainerDataSourceProvider.java
39 lines (32 loc) · 1.07 KB
/
AbstractContainerDataSourceProvider.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
package com.vladmihalcea.hpjp.util.providers;
import org.testcontainers.containers.JdbcDatabaseContainer;
import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.SQLException;
/**
* @author Vlad Mihalcea
*/
public abstract class AbstractContainerDataSourceProvider implements DataSourceProvider {
@Override
public DataSource dataSource() {
DataSource dataSource = newDataSource();
try(Connection connection = dataSource.getConnection()) {
return dataSource;
} catch (SQLException e) {
Database database = database();
if(database.getContainer() == null) {
database.initContainer(username(), password());
}
return newDataSource();
}
}
@Override
public String url() {
JdbcDatabaseContainer container = database().getContainer();
return container != null ?
container.getJdbcUrl() :
defaultJdbcUrl();
}
protected abstract String defaultJdbcUrl();
protected abstract DataSource newDataSource();
}