Merge pull request #1522 from data-integrations/bug/null-pointer

[PLUGIN-1843] Handle Null pointer when using BQ Multi sink without reference name

Author: bhardwaj-priyanshu

src/main/java/io/cdap/plugin/gcp/bigquery/util/BigQueryUtil.java

@@ -802,18 +802,27 @@ public static String generateTimePartitionCondition(StandardTableDefinition tabl
    *
    * @param datasetProject Name of the BQ project
    * @param datasetName Name of the BQ dataset
-   * @param tableName Name of the BQ table
+   * @param tableName Name of the BQ table, If null, only project and dataset are included.
    * @return String fqn
    */
-  public static String getFQN(String datasetProject, String datasetName, String tableName) {
+  public static String getFQN(String datasetProject, String datasetName,
+      @Nullable String tableName) {
 
     String formattedProject = GCPUtils.formatAsFQNComponent(datasetProject);
     String formattedDataset = GCPUtils.formatAsFQNComponent(datasetName);
-    String formattedTable = GCPUtils.formatAsFQNComponent(tableName);
+    StringBuilder fqnBuilder = new StringBuilder(BigQueryConstants.BQ_FQN_PREFIX)
+        .append(":")
+        .append(formattedProject)
+        .append(".")
+        .append(formattedDataset);
+
+    if (tableName != null) {
+      String formattedTable = GCPUtils.formatAsFQNComponent(tableName);
+      fqnBuilder.append(".").append(formattedTable);
+    }
 
-    String fqn = String.format("%s:%s.%s.%s", BigQueryConstants.BQ_FQN_PREFIX, formattedProject,
-        formattedDataset, formattedTable);
-    LOG.trace("Formatted Fully-Qualified Name (FQN): {}", fqn);
+    String fqn = fqnBuilder.toString();
+    LOG.trace("Constructed Fully-Qualified Name (FQN): {}", fqn);
     return fqn;
   }
 

src/test/java/io/cdap/plugin/gcp/bigquery/util/BigQueryUtilTest.java

@@ -183,6 +183,17 @@ public void testGetFQNWithReservedCharacters() {
     Assert.assertEquals(result, expectedFQN);
   }
 
+  @Test
+  public void testGetFQNWithNullTableName() {
+    String datasetProject = "project";
+    String datasetName = "dataset";
+    String tableName = null;
+    String expectedFQN = "bigquery:project.dataset";
+
+    String result = BigQueryUtil.getFQN(datasetProject, datasetName, tableName);
+    Assert.assertEquals(result, expectedFQN);
+  }
+
   @Test
   public void testGetFQNWithoutReservedCharacters() {
     String datasetProject = "project";