From f45b3e167af38d4cf5382652276dafd0cf181215 Mon Sep 17 00:00:00 2001
From: KeKe <wangxingmiao@huawei.com>
Date: Thu, 24 Apr 2025 19:53:29 +0800
Subject: [PATCH] =?UTF-8?q?SQL=20Server=E6=95=B0=E6=8D=AE=E5=BA=93?=
 =?UTF-8?q?=E9=A9=B1=E5=8A=A8=E5=8A=A0=E8=BD=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/org/full/migration/jdbc/SqlServerConnection.java    | 5 +++--
 .../main/java/org/full/migration/source/SourceDatabase.java  | 1 +
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/migration-tool/src/main/java/org/full/migration/jdbc/SqlServerConnection.java b/migration-tool/src/main/java/org/full/migration/jdbc/SqlServerConnection.java
index 5263b82..758ae08 100644
--- a/migration-tool/src/main/java/org/full/migration/jdbc/SqlServerConnection.java
+++ b/migration-tool/src/main/java/org/full/migration/jdbc/SqlServerConnection.java
@@ -38,11 +38,12 @@ public Connection getConnection(DatabaseConfig dbConfig) throws SQLException {
         String url = String.format(Locale.ROOT, SQLSERVER_URL, dbConfig.getHost(), dbConfig.getPort(),
             dbConfig.getDatabase());
         try {
+            Class.forName("com.microsoft.sqlserver.jdbc.SQLServerDriver");
             return DriverManager.getConnection(url, dbConfig.getUser(), dbConfig.getPassword());
-        } catch (SQLException e) {
+        } catch (SQLException | ClassNotFoundException e) {
             LOGGER.error("fail to create sql server connection, host:{}, port:{}, please check.", dbConfig.getHost(),
                 dbConfig.getPort());
-            throw e;
+            throw new SQLException(e.getMessage());
         }
     }
 
diff --git a/migration-tool/src/main/java/org/full/migration/source/SourceDatabase.java b/migration-tool/src/main/java/org/full/migration/source/SourceDatabase.java
index d5252fe..a43fcba 100644
--- a/migration-tool/src/main/java/org/full/migration/source/SourceDatabase.java
+++ b/migration-tool/src/main/java/org/full/migration/source/SourceDatabase.java
@@ -250,6 +250,7 @@ public void queryTables(Set<String> schemaSet) {
             if (isDumpJson) {
                 ProgressTracker.getInstance().recordTableProgress();
             }
+            LOGGER.info("success to query all tables...");
         } catch (SQLException e) {
             LOGGER.error("fail to query table list, error message:{}.", e.getMessage());
         }
-- 
Gitee