• Home
• io.stargate.it
• testing
• 1.0.85
• source code
• DefaultContactPointResolver.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

io.stargate.it.driver.DefaultContactPointResolver Maven / Gradle / Ivy

package io.stargate.it.driver;

import io.stargate.it.storage.StargateConnectionInfo;
import io.stargate.it.storage.StargateEnvironmentInfo;
import io.stargate.it.storage.StargateExtension;
import java.net.InetSocketAddress;
import java.util.ArrayList;
import java.util.List;
import org.junit.jupiter.api.extension.ExtensionContext;

public class DefaultContactPointResolver implements ContactPointResolver {
  @Override
  public List resolve(ExtensionContext context) {
    StargateEnvironmentInfo stargate =
        (StargateEnvironmentInfo)
            context.getStore(ExtensionContext.Namespace.GLOBAL).get(StargateExtension.STORE_KEY);
    if (stargate == null) {
      throw new IllegalStateException(
          String.format(
              "%s can only be used in conjunction with %s (make sure it is declared last)",
              CqlSessionExtension.class.getSimpleName(), StargateExtension.class.getSimpleName()));
    }
    List contactPoints = new ArrayList<>();
    for (StargateConnectionInfo node : stargate.nodes()) {
      contactPoints.add(new InetSocketAddress(node.seedAddress(), node.cqlPort()));
    }
    return contactPoints;
  }
}