From 94e2ae71ae4baec723f652b6a686b453a532b97b Mon Sep 17 00:00:00 2001
From: SwallowGG <1558143046@qq.com>
Date: Tue, 17 Oct 2023 18:19:17 +0800
Subject: [PATCH] Add operation log

---
 .../src/main/java/ai/chat2db/spi/jdbc/DefaultDBManage.java      | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/jdbc/DefaultDBManage.java b/chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/jdbc/DefaultDBManage.java
index 3e14efe2..fee2c924 100644
--- a/chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/jdbc/DefaultDBManage.java
+++ b/chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/jdbc/DefaultDBManage.java
@@ -67,7 +67,7 @@ public class DefaultDBManage implements DBManage {
         }
         connectInfo.setSession(session);
         connectInfo.setConnection(connection);
-        if (StringUtils.isNotBlank(connectInfo.getDatabaseName())) {
+        if (StringUtils.isNotBlank(connectInfo.getDatabaseName()) || StringUtils.isNotBlank(connectInfo.getSchemaName())) {
             connectDatabase(connection, connectInfo.getDatabaseName());
         }
         return connection;