Coding Applications for Astra in JAVA - datastaxdevs/awesome-astra GitHub Wiki

import java.nio.file.Paths;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.datastax.oss.driver.api.core.CqlSession;

public class AstraDriver4x {
 
 static final String ASTRA_ZIP_FILE = "<path_to_secureConnectBundle.zip>";
 static final String ASTRA_USERNAME = "<provide_a_clientId>";
 static final String ASTRA_PASSWORD = "<provide_a_clientSecret>";
 static final String ASTRA_KEYSPACE = "<provide_your_keyspace>";
    
 public static void main(String[] args) {
   Logger logger = LoggerFactory.getLogger(AstraDriver4x.class);
   try (CqlSession cqlSession = CqlSession.builder()
     .withCloudSecureConnectBundle(Paths.get(ASTRA_ZIP_FILE))
     .withAuthCredentials(ASTRA_USERNAME, ASTRA_PASSWORD)
     .withKeyspace(ASTRA_KEYSPACE)
     .build()) {
     logger.info("[OK] Welcome to ASTRA. Connected to Keyspace {}", cqlSession.getKeyspace().get());
   }
   logger.info("[OK] Success");
   System.exit(0);
 }

}

import java.io.File;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.datastax.driver.core.Cluster;
import com.datastax.driver.core.Session;

public class AstraDriver3x {    
 
 // Define inputs
 static final String ASTRA_ZIP_FILE = "<path_to_secureConnectBundle.zip>";
 static final String ASTRA_USERNAME = "<provide_a_clientId>";
 static final String ASTRA_PASSWORD = "<provide_a_clientSecret>";
 static final String ASTRA_KEYSPACE = "<provide_your_keyspace>";    
    
 public static void main(String[] args) {
  Logger logger = LoggerFactory.getLogger(AstraDriver3x.class);
  try(Cluster cluster = Cluster.builder()
    .withCloudSecureConnectBundle(new File(ASTRA_ZIP_FILE))
    .withCredentials(ASTRA_USERNAME, ASTRA_PASSWORD)
    .build() ) {
      Session session = cluster.connect(ASTRA_KEYSPACE);
      logger.info("[OK] Welcome to ASTRA. Connected to Keyspace {}", session.getLoggedKeyspace());
  }
  logger.info("[OK] Success");
  System.exit(0);
 }

}

public class AstraRestApiHttpClient {
    
    static final String ASTRA_TOKEN       = "<change_with_your_token>";
    static final String ASTRA_DB_ID       = "<change_with_your_database_identifier>";
    static final String ASTRA_DB_REGION   = "<change_with_your_database_region>";
    static final String ASTRA_DB_KEYSPACE = "<change_with_your_keyspace>";
    
    static  Logger logger = LoggerFactory.getLogger(AstraRestApiHttpClient.class);
    
    public static void main(String[] args) throws Exception {
        
        String apiRestEndpoint = new StringBuilder("https://")
                .append(ASTRA_DB_ID).append("-")
                .append(ASTRA_DB_REGION)
                .append(".apps.astra.datastax.com/api/rest")
                .toString();
        logger.info("Rest Endpoint is {}", apiRestEndpoint);
        
        try (CloseableHttpClient httpClient = HttpClients.createDefault()) {
            listKeyspaces(httpClient, apiRestEndpoint);
            createTable(httpClient, apiRestEndpoint);
            insertRow(httpClient, apiRestEndpoint);
            retrieveRow(httpClient, apiRestEndpoint);
        }
        logger.info("[OK] Success");
        System.exit(0);
    }

private static void listKeyspaces(CloseableHttpClient httpClient, String apiRestEndpoint) 
throws Exception {
 // Build Request
 HttpGet listKeyspacesReq = new HttpGet(apiRestEndpoint + "/v2/schemas/keyspaces");
 listKeyspacesReq.addHeader("X-Cassandra-Token", ASTRA_TOKEN);
 
 // Execute Request
 try(CloseableHttpResponse res = httpClient.execute(listKeyspacesReq)) {
   if (200 == res.getCode()) {
     logger.info("[OK] Keyspaces list retrieved");
     logger.info("Returned message: {}", EntityUtils.toString(res.getEntity()));
   }
 }
}

private static void createTable(CloseableHttpClient httpClient, String apiRestEndpoint) 
throws Exception {
  HttpPost createTableReq = new HttpPost(apiRestEndpoint 
       + "/v2/schemas/keyspaces/" + ASTRA_DB_KEYSPACE + "/tables");
  createTableReq.addHeader("X-Cassandra-Token", ASTRA_TOKEN);
  String createTableJson = "{...JSON.....}";
  createTableReq.setEntity(new StringEntity(createTableJson, ContentType.APPLICATION_JSON));
  
  // Execute Request
  try(CloseableHttpResponse res = httpClient.execute(createTableReq)) {
    if (201 == res.getCode()) {
      logger.info("[OK] Table Created (if needed)");
      logger.info("Returned message: {}", EntityUtils.toString(res.getEntity()));
    }
  }
}

private static void insertRow(CloseableHttpClient httpClient, String apiRestEndpoint) 
throws Exception {
  HttpPost insertCedrick = new HttpPost(apiRestEndpoint + "/v2/keyspaces/" 
   + ASTRA_DB_KEYSPACE + "/users" );
  insertCedrick.addHeader("X-Cassandra-Token", ASTRA_TOKEN);
  insertCedrick.setEntity(new StringEntity("{"
    + " \"firstname\": \"Cedrick\","
    + " \"lastname\" : \"Lunven\","
    + " \"email\"    : \"[email protected]\","
    + " \"color\"    : \"blue\" }", ContentType.APPLICATION_JSON));
        
  // Execute Request
  try(CloseableHttpResponse res = httpClient.execute(insertCedrick)) {
    if (201 == res.getCode()) {
      logger.info("[OK] Row inserted");
      logger.info("Returned message: {}", EntityUtils.toString(res.getEntity()));
    }
  }
}

private static void retrieveRow(CloseableHttpClient httpClient, String apiRestEndpoint) 
throws Exception {

  // Build Request
  HttpGet rowReq = new HttpGet(apiRestEndpoint + "/v2/keyspaces/" 
    + ASTRA_DB_KEYSPACE + "/users/Cedrick/Lunven" );
  rowReq.addHeader("X-Cassandra-Token", ASTRA_TOKEN);
        
  // Execute Request
  try(CloseableHttpResponse res = httpClient.execute(rowReq)) {
    if (200 == res.getCode()) {
      String payload =  EntityUtils.toString(res.getEntity());
      logger.info("[OK] Row retrieved");
      logger.info("Row retrieved : {}", payload);
    }
  }
}