(shardingsphere) branch master updated: Refactor DataSetEnvironmentManager (#30485)

[chengzhang]

This is an automated email from the ASF dual-hosted git repository.

chengzhang pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/shardingsphere.git


The following commit(s) were added to refs/heads/master by this push:
     new eb62b4f8c3d Refactor DataSetEnvironmentManager (#30485)
eb62b4f8c3d is described below

commit eb62b4f8c3d7b8064cf15706ab9dade5789cb02b
Author: Liang Zhang <zhangli...@apache.org>
AuthorDate: Thu Mar 14 21:09:42 2024 +0800

    Refactor DataSetEnvironmentManager (#30485)
---
 .../shardingsphere/test/e2e/env/DataSetEnvironmentManager.java   | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git 
a/test/e2e/sql/src/test/java/org/apache/shardingsphere/test/e2e/env/DataSetEnvironmentManager.java
 
b/test/e2e/sql/src/test/java/org/apache/shardingsphere/test/e2e/env/DataSetEnvironmentManager.java
index 8fb260e443d..37ea9de770d 100644
--- 
a/test/e2e/sql/src/test/java/org/apache/shardingsphere/test/e2e/env/DataSetEnvironmentManager.java
+++ 
b/test/e2e/sql/src/test/java/org/apache/shardingsphere/test/e2e/env/DataSetEnvironmentManager.java
@@ -219,13 +219,18 @@ public final class DataSetEnvironmentManager {
             try (Connection connection = dataSource.getConnection()) {
                 for (String each : tableNames) {
                     DatabaseType databaseType = 
DatabaseTypeFactory.get(connection.getMetaData().getURL());
-                    DialectDatabaseMetaData dialectDatabaseMetaData = new 
DatabaseTypeRegistry(databaseType).getDialectDatabaseMetaData();
-                    try (PreparedStatement preparedStatement = 
connection.prepareStatement(String.format("TRUNCATE TABLE %s", 
dialectDatabaseMetaData.getQuoteCharacter().wrap(each)))) {
+                    String quotedTableName = getQuotedTableName(each, 
databaseType);
+                    try (PreparedStatement preparedStatement = 
connection.prepareStatement(String.format("TRUNCATE TABLE %s", 
quotedTableName))) {
                         preparedStatement.execute();
                     }
                 }
             }
             return null;
         }
+        
+        private String getQuotedTableName(final String tableName, final 
DatabaseType databaseType) {
+            DialectDatabaseMetaData dialectDatabaseMetaData = new 
DatabaseTypeRegistry(databaseType).getDialectDatabaseMetaData();
+            return dialectDatabaseMetaData.getQuoteCharacter().wrap(tableName);
+        }
     }
 }