generator.server.springboot.database.cassandra.TestCassandraManager.mustache Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jhlite Show documentation
JHipster Lite
There is a newer version: 1.22.0
package {{ packageName }};

import com.datastax.oss.driver.api.core.CqlSession;
import org.springframework.boot.SpringApplication;
import org.springframework.core.Ordered;
import org.springframework.boot.context.event.ApplicationEnvironmentPreparedEvent;
import org.springframework.context.ApplicationListener;
import org.testcontainers.containers.CassandraContainer;

import java.net.InetAddress;
import java.net.UnknownHostException;

class TestCassandraManager implements ApplicationListener, Ordered  {

  private static final String KEYSPACE = "{{ baseName }}";
  private static CassandraContainer cassandraContainer;

  @Override
  public int getOrder() {
    return 0;
  }

  @Override
  public void onApplicationEvent(ApplicationEnvironmentPreparedEvent event) {
    if (cassandraContainer != null) {
      return;
    }
    createCassandraContainer();
    cassandraContainer.start();

    try (CqlSession session = getCqlSession()) {
      createTestKeyspace(session);
    }
    registerEnvironmentVariables();

    SpringApplication.getShutdownHandlers()
      .add(new Thread(stopContainer()));
  }

  private void registerEnvironmentVariables() {
    System.setProperty("TEST_CASSANDRA_PORT", String.valueOf(cassandraContainer.getMappedPort(CassandraContainer.CQL_PORT)));
    System.setProperty("TEST_CASSANDRA_CONTACT_POINT", getContainerIpAddress());
    System.setProperty("TEST_CASSANDRA_DC", cassandraContainer.getLocalDatacenter());
    System.setProperty("TEST_CASSANDRA_KEYSPACE", KEYSPACE);
  }

  /**
   *  For the driver contact point, an IP address should be used, and not "localhost" which can resolve to two addresses: ipv4 and ipv6.
   *  If it happens, the driver creates 2 contact points, and the second one throws a WARN in logs.
   *  Currently, cassandraContainer.getHost() returns "localhost", so it can't be used directly.
   * @see basic.contact-points comments
   * @return cassandra container's ip address, or 127.0.0.1 as fallback
   */
  private String getContainerIpAddress() {
    try {
      return InetAddress.getByName(cassandraContainer.getHost()).getHostAddress();
    } catch(UnknownHostException e) {
      return "127.0.0.1";
    }
  }

  private void createTestKeyspace(CqlSession session) {
    String createQuery = "CREATE KEYSPACE " + KEYSPACE + " WITH replication={'class' : 'SimpleStrategy', 'replication_factor':1}";
    session.execute(createQuery);
  }

  private CqlSession getCqlSession() {
    return CqlSession
      .builder()
      .addContactPoint(cassandraContainer.getContactPoint())
      .withLocalDatacenter(cassandraContainer.getLocalDatacenter())
      .build();
  }

  private void createCassandraContainer() {
    cassandraContainer = new CassandraContainer<>("{{ cassandraDockerImage }}");
  }

  private Runnable stopContainer() {
    return cassandraContainer::stop;
  }
}