Merge pull request wso2#5525 from sadilchamishka/db2-issue

Fix db2 issue
Thumimku · Feb 19, 2024 · bfa9848 · bfa9848
2 parents 6e10788 + 0a263ba
commit bfa9848
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/....identity.core/src/main/java/org/wso2/carbon/identity/core/util/IdentityDatabaseUtil.java b/....identity.core/src/main/java/org/wso2/carbon/identity/core/util/IdentityDatabaseUtil.java
@@ -31,6 +31,8 @@
 import java.sql.SQLException;
 import javax.sql.DataSource;
 
+import static org.wso2.carbon.identity.core.util.IdentityCoreConstants.DB2;
+
 /**
  * Utility class for database operations.
  */
@@ -205,6 +207,9 @@ public static boolean isTableExists(String tableName) {
             if (metaData.storesLowerCaseIdentifiers()) {
                 tableName = tableName.toLowerCase();
             }
+            if (connection.getMetaData().getDatabaseProductName().toUpperCase().contains(DB2)) {
+                return connection.getMetaData().getTables(null, null, tableName, new String[]{"TABLE"}).next();
+            }
             String schemaName = connection.getSchema();
             String catalogName = connection.getCatalog();
             try (ResultSet resultSet = metaData.getTables(catalogName, schemaName, tableName, new String[]{"TABLE"})) {