From 4f0064912d9d2fab5ae1f230103d6e41d8601333 Mon Sep 17 00:00:00 2001 From: tmlx1990 Date: Wed, 28 Feb 2024 17:23:27 +0800 Subject: [PATCH] =?UTF-8?q?fix=EF=BC=9A1186=20=E4=BF=AE=E5=A4=8DMySQL?= =?UTF-8?q?=E4=B8=8D=E8=83=BD=E8=8E=B7=E5=8F=96=E8=A1=A8=E6=B3=A8=E9=87=8A?= =?UTF-8?q?=E7=9A=84=E9=97=AE=E9=A2=98=E3=80=82=20=E5=A2=9E=E5=8A=A0useInf?= =?UTF-8?q?ormationSchema=E5=B1=9E=E6=80=A7=E4=B8=BAtrue=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../chat2db/plugin/mysql/MysqlDBManage.java | 20 ++++++++++++++++--- 1 file changed, 17 insertions(+), 3 deletions(-) diff --git a/chat2db-server/chat2db-plugins/chat2db-mysql/src/main/java/ai/chat2db/plugin/mysql/MysqlDBManage.java b/chat2db-server/chat2db-plugins/chat2db-mysql/src/main/java/ai/chat2db/plugin/mysql/MysqlDBManage.java index 2322dd669..205d8671b 100644 --- a/chat2db-server/chat2db-plugins/chat2db-mysql/src/main/java/ai/chat2db/plugin/mysql/MysqlDBManage.java +++ b/chat2db-server/chat2db-plugins/chat2db-mysql/src/main/java/ai/chat2db/plugin/mysql/MysqlDBManage.java @@ -1,13 +1,16 @@ package ai.chat2db.plugin.mysql; -import java.sql.Connection; -import java.sql.SQLException; - import ai.chat2db.spi.DBManage; import ai.chat2db.spi.jdbc.DefaultDBManage; +import ai.chat2db.spi.model.KeyValue; +import ai.chat2db.spi.sql.ConnectInfo; import ai.chat2db.spi.sql.SQLExecutor; import org.springframework.util.StringUtils; +import java.sql.Connection; +import java.sql.SQLException; +import java.util.List; + public class MysqlDBManage extends DefaultDBManage implements DBManage { @Override public void connectDatabase(Connection connection, String database) { @@ -29,6 +32,17 @@ public void dropTable(Connection connection, String databaseName, String schemaN SQLExecutor.getInstance().execute(connection,sql, resultSet -> null); } + @Override + public Connection getConnection(ConnectInfo connectInfo) { + KeyValue keyValue = new KeyValue(); + keyValue.setKey("useInformationSchema"); + keyValue.setValue("true"); + List extendInfoList = connectInfo.getExtendInfo(); + extendInfoList.add(keyValue); + connectInfo.setExtendInfo(extendInfoList); + return super.getConnection(connectInfo); + } + public static String format(String tableName) { return "`" + tableName + "`"; }