legolas.postgre.infra.PostgreSQLStarter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of legolas-postgresql Show documentation

There is a newer version: 1.0.14

package legolas.postgre.infra;

import legolas.net.core.interfaces.Port;
import legolas.net.core.interfaces.SocketType;
import legolas.postgre.interfaces.PostgreSQLEntry;
import legolas.postgre.interfaces.PostgreSQLServiceId;
import legolas.runtime.core.interfaces.ServiceId;
import legolas.sql.interfaces.DatabaseConfiguration;
import legolas.sql.interfaces.SQLStarter;
import legolas.sql.interfaces.TargetDatabase;
import legolas.starter.api.interfaces.StarterComponent;
import org.testcontainers.containers.PostgreSQLContainer;

import java.util.Arrays;
import java.util.stream.Stream;

@StarterComponent
public class PostgreSQLStarter extends SQLStarter {
  static final String PASSWORD = "postgre";
  static final String JDBC_DRIVER_NAME = "org.postgresql.Driver";
  static final int DEFAULT_PORT = 5432;

  public PostgreSQLStarter() {
    String database = this.databaseName();
    String url = String.format("jdbc:postgresql://%s:%d/%s", this.dockerHost(), DEFAULT_PORT, database);
    this.configuration
      .set(PostgreSQLEntry.HOST, this.dockerHost())
      .set(PostgreSQLEntry.PORT, DEFAULT_PORT)
      .set(PostgreSQLEntry.USERNAME, this.username())
      .set(PostgreSQLEntry.PASSWORD, PASSWORD)
      .set(PostgreSQLEntry.DRIVER, JDBC_DRIVER_NAME)
      .set(PostgreSQLEntry.URL, url);
  }

  @Override
  protected PostgreSQLContainer container() {
    return new PostgreSQLContainer().withUsername(this.username())
      .withPassword(PASSWORD)
      .withDatabaseName(this.databaseName());
  }

  @Override
  protected void setConfiguration(PostgreSQLContainer container) {
    this.configuration.set(PostgreSQLEntry.URL, container.getJdbcUrl());
  }

  private String databaseName(){
    return this.username().toLowerCase();
  }

  @Override
  protected TargetDatabase targetDatabase() {
    return TargetDatabase.POSTGRESQL;
  }

  @Override
  public Stream ports() {
    return Arrays.asList(Port.create(DEFAULT_PORT)).stream();
  }

  @Override
  public SocketType socketType() {
    return SocketType.TCP;
  }

  @Override
  public ServiceId id() {
    return PostgreSQLServiceId.INSTANCE;
  }

  @Override
  public String name() {
    return "PostgreSQL Container";
  }
}