fix: Add multi-statement capability logic [DHIS2-18417]

dhis-2/dhis-services/dhis-service-analytics/src/main/java/org/hisp/dhis/analytics/table/AbstractJdbcTableManager.java

@@ -230,19 +230,18 @@
     boolean skipMasterTable =
         params.isPartialUpdate() && tableExists && table.getTableType().isLatestPartition();
 
-    log.info(
-        "Swapping table, master table exists: '{}', skip master table: '{}'",
-        tableExists,
-        skipMasterTable);
-    List<Table> swappedPartitions = new UniqueArrayList<>();
+    log.info("Swapping table: '%s'", table.getMainName());
+    log.info("Master table exists: '{}', skip master table: '{}'", tableExists, skipMasterTable);
+
+    List<Table> swapPartitions = new UniqueArrayList<>();
     table.getTablePartitions().stream()
-        .forEach(p -> swappedPartitions.add(swapTable(p, p.getMainName())));
+        .forEach(p -> swapPartitions.add(swapTable(p, p.getMainName())));
 
     if (!skipMasterTable) {
       swapTable(table, table.getMainName());
     } else {
-      swappedPartitions.forEach(
-          partition -> swapInheritance(partition, table.getName(), table.getMainName()));
+      swapPartitions.forEach(
+          partition -> swapParentTable(partition, table.getName(), table.getMainName()));
       dropTable(table);
     }
   }
@@ -290,7 +289,12 @@
    * @param mainTableName the main table name.
    */
   private Table swapTable(Table stagingTable, String mainTableName) {
-    executeSilently(sqlBuilder.swapTable(stagingTable, mainTableName));
+    if (sqlBuilder.supportsMultiStatements()) {
+      executeSilently(sqlBuilder.swapTable(stagingTable, mainTableName));
+    } else {
+      executeSilently(sqlBuilder.dropTableIfExistsCascade(mainTableName));
+      executeSilently(sqlBuilder.renameTable(stagingTable, mainTableName));
+    }
     return stagingTable.swapFromStaging();
   }
 
@@ -301,8 +305,13 @@
    * @param stagingMasterName the staging master table name.
    * @param mainMasterName the main master table name.
    */
-  private void swapInheritance(Table partition, String stagingMasterName, String mainMasterName) {
-    executeSilently(sqlBuilder.swapParentTable(partition, stagingMasterName, mainMasterName));
+  private void swapParentTable(Table partition, String stagingMasterName, String mainMasterName) {
+    if (sqlBuilder.supportsMultiStatements()) {
+      executeSilently(sqlBuilder.swapParentTable(partition, stagingMasterName, mainMasterName));
+    } else {
+      sqlBuilder.removeParentTable(partition, stagingMasterName);
+      sqlBuilder.setParentTable(partition, mainMasterName);
+    }
   }
 
   /**

dhis-2/dhis-support/dhis-support-sql/src/main/java/org/hisp/dhis/db/sql/ClickHouseSqlBuilder.java

@@ -169,6 +169,11 @@ public boolean supportsCorrelatedSubquery() {
     return false;
   }
 
+  @Override
+  public boolean supportsMultiStatements() {
+    return false;
+  }
+
   @Override
   public boolean requiresIndexesForAnalytics() {
     return false;

dhis-2/dhis-support/dhis-support-sql/src/main/java/org/hisp/dhis/db/sql/DorisSqlBuilder.java

@@ -171,6 +171,11 @@ public boolean supportsCorrelatedSubquery() {
     return true;
   }
 
+  @Override
+  public boolean supportsMultiStatements() {
+    return true;
+  }
+
   @Override
   public boolean requiresIndexesForAnalytics() {
     return false;

dhis-2/dhis-support/dhis-support-sql/src/main/java/org/hisp/dhis/db/sql/PostgreSqlBuilder.java

@@ -190,6 +190,11 @@ public boolean supportsCorrelatedSubquery() {
     return true;
   }
 
+  @Override
+  public boolean supportsMultiStatements() {
+    return true;
+  }
+
   @Override
   public boolean requiresIndexesForAnalytics() {
     return true;

dhis-2/dhis-support/dhis-support-sql/src/main/java/org/hisp/dhis/db/sql/SqlBuilder.java

@@ -175,10 +175,15 @@ public interface SqlBuilder {
   boolean supportsVacuum();
 
   /**
-   * @return true if the DBMS supports corrected subqueries.
+   * @return true if the DBMS supports correlated subqueries.
    */
   boolean supportsCorrelatedSubquery();
 
+  /**
+   * @return true if the DMBS supports multiple statements in one operation.
+   */
+  boolean supportsMultiStatements();
+
   /**
    * @return true if the DBMS requires indexes for analytics tables for performance.
    */