From e9bfa1396ff47d171b3052a606e0931e6f93cc9c Mon Sep 17 00:00:00 2001
From: houzhongjian <houzhongyi@126.com>
Date: 星期四, 10 四月 2025 12:05:23 +0800
Subject: [PATCH] 还原framework代码
---
iailab-framework/iailab-common-mybatis/src/main/java/com/iailab/framework/mybatis/core/util/JdbcUtils.java | 61 ++++++++++++++++++++++++++++++
1 files changed, 61 insertions(+), 0 deletions(-)
diff --git a/iailab-framework/iailab-common-mybatis/src/main/java/com/iailab/framework/mybatis/core/util/JdbcUtils.java b/iailab-framework/iailab-common-mybatis/src/main/java/com/iailab/framework/mybatis/core/util/JdbcUtils.java
new file mode 100644
index 0000000..0275079
--- /dev/null
+++ b/iailab-framework/iailab-common-mybatis/src/main/java/com/iailab/framework/mybatis/core/util/JdbcUtils.java
@@ -0,0 +1,61 @@
+package com.iailab.framework.mybatis.core.util;
+
+import com.baomidou.dynamic.datasource.DynamicRoutingDataSource;
+import com.baomidou.mybatisplus.annotation.DbType;
+import com.iailab.framework.common.util.spring.SpringUtils;
+import com.iailab.framework.mybatis.core.enums.DbTypeEnum;
+
+import javax.sql.DataSource;
+import java.sql.Connection;
+import java.sql.DriverManager;
+import java.sql.SQLException;
+
+/**
+ * JDBC 工具类
+ *
+ * @author iailab
+ */
+public class JdbcUtils {
+
+ /**
+ * 判断连接是否正确
+ *
+ * @param url 数据源连接
+ * @param username 账号
+ * @param password 密码
+ * @return 是否正确
+ */
+ public static boolean isConnectionOK(String url, String username, String password) {
+ try (Connection ignored = DriverManager.getConnection(url, username, password)) {
+ return true;
+ } catch (Exception ex) {
+ return false;
+ }
+ }
+
+ /**
+ * 获得 URL 对应的 DB 类型
+ *
+ * @param url URL
+ * @return DB 类型
+ */
+ public static DbType getDbType(String url) {
+ return com.baomidou.mybatisplus.extension.toolkit.JdbcUtils.getDbType(url);
+ }
+
+ /**
+ * 通过当前数据库连接获得对应的 DB 类型
+ *
+ * @return DB 类型
+ */
+ public static DbType getDbType() {
+ DynamicRoutingDataSource dynamicRoutingDataSource = SpringUtils.getBean(DynamicRoutingDataSource.class);
+ DataSource dataSource = dynamicRoutingDataSource.determineDataSource();
+ try (Connection conn = dataSource.getConnection()) {
+ return DbTypeEnum.find(conn.getMetaData().getDatabaseProductName());
+ } catch (SQLException e) {
+ throw new IllegalArgumentException(e.getMessage());
+ }
+ }
+
+}
--
Gitblit v1.9.3