added code to create keyspace

ollionorg · Jan 26, 2025 · 1f583d5 · 1f583d5
1 parent 186a8ea
commit 1f583d5
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 17 deletions.
diff --git a/...urcedb/src/test/java/com/google/cloud/teleport/v2/templates/SpannerToCassandraLTBase.java b/...urcedb/src/test/java/com/google/cloud/teleport/v2/templates/SpannerToCassandraLTBase.java
@@ -68,8 +68,8 @@ public void setupResourceManagers(
     spannerResourceManager = createSpannerDatabase(spannerDdlResource);
     spannerMetadataResourceManager = createSpannerMetadataDatabase();
 
-    LOG.info("Test Name: {}", testName);
-    LOG.info("Class Name: {}", getClass().getSimpleName());
+    System.out.println("Test Name: " + testName);
+    System.out.println("Class Name: " + getClass().getSimpleName());
 
     gcsResourceManager =
         GcsResourceManager.builder(artifactBucket, getClass().getSimpleName(), CREDENTIALS).build();
@@ -106,9 +106,7 @@ private CassandraResourceManager generateKeyspaceAndBuildCassandraResource() {
     }
 
     System.out.println("Cassandra keyspaceName :: " + keyspaceName);
-    return CassandraResourceManager.builder("rr-spdr-csdr-loadtest-" + testName)
-        //        .setKeyspaceName(keyspaceName)
-        .build();
+    return CassandraResourceManager.builder(testName).setKeyspaceName(keyspaceName).build();
     //    return CassandraSharedResourceManager.builder("rr-spdr-csdr-loadtest-" + testName)
     //        .setKeyspaceName(keyspaceName)
     //        .sePreGeneratedKeyspaceName(true)
@@ -184,9 +182,10 @@ public void createAndUploadCassandraConfigToGcs(
     String host = cassandraResourceManagers.getHost();
     int port = cassandraResourceManagers.getPort();
     String keyspaceName = cassandraResourceManagers.getKeyspaceName();
-    LOG.info("Cassandra keyspaceName :: {}", keyspaceName);
-    LOG.info("Cassandra host :: {}", host);
-    LOG.info("Cassandra port :: {}", port);
+    System.out.println("Cassandra keyspaceName :: " + keyspaceName);
+    System.out.println("Cassandra host :: " + host);
+    System.out.println("Cassandra port :: " + port);
+
     String cassandraConfigContents;
     try (InputStream inputStream =
         Thread.currentThread()
@@ -205,7 +204,7 @@ public void createAndUploadCassandraConfigToGcs(
             .replace("##port##", Integer.toString(port))
             .replace("##keyspace##", keyspaceName);
 
-    LOG.info("Cassandra file contents: {}", cassandraConfigContents);
+    System.out.println("Cassandra file contents: " + cassandraConfigContents);
 
     gcsResourceManager.createArtifact("input/cassandra-config.conf", cassandraConfigContents);
   }

diff --git a/...cedb/src/test/java/com/google/cloud/teleport/v2/templates/SpannerToCassandraSourceLT.java b/...cedb/src/test/java/com/google/cloud/teleport/v2/templates/SpannerToCassandraSourceLT.java
@@ -83,8 +83,9 @@ public void teardown() {
   public void reverseReplicationCassandra1KTpsLoadTest()
       throws IOException, ParseException, InterruptedException {
     // Start data generator
-    LOG.info("SpannerInstanceID: {}", spannerResourceManager.getInstanceId());
-    LOG.info("SpannerDatabaseID: {}", spannerResourceManager.getDatabaseId());
+    System.out.println("SpannerInstanceID :: " + spannerResourceManager.getInstanceId());
+    System.out.println("SpannerDatabaseID :: " + spannerResourceManager.getDatabaseId());
+
     DataGenerator dataGenerator =
         DataGenerator.builderWithSchemaLocation(testName, generatorSchemaPath)
             .setQPS("100") // 1000
@@ -114,35 +115,43 @@ public void reverseReplicationCassandra1KTpsLoadTest()
         pipelineOperator.waitForCondition(
             createConfig(jobInfo, Duration.ofMinutes(10), Duration.ofSeconds(30)), check);
 
-    LOG.info("Job wait: {}", result);
+    System.out.println("Job wait: " + result);
     // Assert Conditions
     assertThatResult(result).meetsConditions();
 
     PipelineOperator.Result result1 =
         pipelineOperator.cancelJobAndFinish(createConfig(jobInfo, Duration.ofMinutes(20)));
 
-    LOG.info("Job finished: {}", result1);
+    System.out.println("Job finished: " + result1);
     assertThatResult(result1).isLaunchFinished();
 
     exportMetrics(jobInfo, numShards);
   }
 
   private void createCassandraSchema(CassandraResourceManager cassandraResourceManager) {
     String keyspace = cassandraResourceManager.getKeyspaceName();
+    System.out.println("Creating keyspace: " + keyspace);
+    cassandraResourceManager.executeStatement(
+        String.format(
+            "CREATE KEYSPACE IF NOT EXISTS %s WITH REPLICATION = { 'class' : 'SimpleStrategy', 'replication_factor' : 1 };",
+            keyspace));
+    System.out.println("Keyspace Created");
+
     String createTableStatement =
         String.format(
-            "CREATE TABLE IF NOT EXISTS %s.%s ("
+            "USE %s; "
+                + "CREATE TABLE IF NOT EXISTS %s.%s ("
                 + "id uuid PRIMARY KEY, "
                 + "first_name1 text, "
                 + "last_name1 text, "
                 + "first_name2 text, "
                 + "last_name2 text, "
                 + "first_name3 text, "
                 + "last_name3 text);",
-            keyspace, table);
+            keyspace, keyspace, table);
 
-    LOG.info("Creating table: {}", createTableStatement);
+    System.out.println("Creating table: " + createTableStatement);
     cassandraResourceManager.executeStatement(createTableStatement);
-    LOG.info("Table Created");
+    System.out.println("Table Created");
   }
 }