gitee
/
roses
mirror of https://gitee.com/stylefeng/roses
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

【8.3.4】【db】更新获取数据库名称的工具

master
stylefeng 2025-05-07 22:08:57 +08:00
parent 36e9c3d094
commit c11d9fe93f
1 changed files with 108 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

128
kernel-d-db/db-api/src/main/java/cn/stylefeng/roses/kernel/db/api/util/DatabaseUtil.java
View File

@ -11,6 +11,7 @@ import cn.stylefeng.roses.kernel.db.api.sqladapter.database.CreateDatabaseSql;
import cn.stylefeng.roses.kernel.db.api.sqladapter.database.GetDatabasesSql;
import cn.stylefeng.roses.kernel.db.api.sqladapter.table.TableFieldListSql;
import cn.stylefeng.roses.kernel.db.api.sqladapter.table.TableListSql;
import cn.stylefeng.roses.kernel.rule.enums.DbTypeEnum;
import lombok.extern.slf4j.Slf4j;
import java.sql.Connection;
@ -19,6 +20,8 @@ import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
/**
* 数据库操作工具类，可用来获取一些元数据
@ -182,33 +185,118 @@ public class DatabaseUtil {
}
/**
* 获取数据库名称
* 根据数据库配置，获取数据库名称
* <p>
* oracle的数据库会直接返回username作为数据库名称，这里需要注意一下，如果用户名不是数据库名则返回不准确
*
* @author fengshuonan
* @since 2021/5/19 10:39
* @since 2025/5/7 21:45
*/
private static String getDbName(DruidProperties druidProperties) {
public static String getDbName(DruidProperties druidProperties) {
String url = druidProperties.getUrl().toLowerCase();
return getDbName(url);
}
if (druidProperties.getUrl().contains("oracle")) {
// 如果是oracle，直接返回username
return druidProperties.getUsername();
} else if (druidProperties.getUrl().contains("postgresql")) {
// postgresql，直接返回最后一个/后边的字符
int first = druidProperties.getUrl().lastIndexOf("/") + 1;
return druidProperties.getUrl().substring(first);
} else if (druidProperties.getUrl().contains("sqlserver")) {
// sqlserver，直接返回最后一个=后边的字符
int first = druidProperties.getUrl().lastIndexOf("=") + 1;
return druidProperties.getUrl().substring(first);
} else {
// mysql，返回/和?之间的字符
String jdbcUrl = druidProperties.getUrl();
int first = jdbcUrl.lastIndexOf("/") + 1;
int last = jdbcUrl.indexOf("?");
return jdbcUrl.substring(first, last);
/**
* 根据数据库URL获取数据库配置
*
* @author fengshuonan
* @since 2025/5/7 21:45
*/
public static String getDbName(String url) {
try {
// 如果是oracle
if (url.contains(DbTypeEnum.ORACLE.getUrlWords())) {
Pattern pattern = Pattern.compile(":(thin|oci|kprb):@(//)?[^:/]+(:\\d+)?[:/]([^?]+)");
Matcher matcher = pattern.matcher(url);
return matcher.find() ? matcher.group(4) : parseAfterLastSlash(url);
}
// 如果是PostgreSQL
else if (url.contains(DbTypeEnum.PG_SQL.getUrlWords())) {
return parseAfterLastSlash(url);
}
// 如果是sql server
else if (url.contains(DbTypeEnum.MS_SQL.getUrlWords())) {
// SQL Server: databaseName= 或 Initial Catalog=
return parseParameter(url, "databasename=", "initial catalog=");
}
// 如果是Mysql
else if (url.contains(DbTypeEnum.MYSQL.getUrlWords())) {
int first = url.lastIndexOf("/") + 1;
int last = url.indexOf("?");
return url.substring(first, last == -1 ? url.length() : last);
}
// 如果是达梦数据库
else if (url.contains(DbTypeEnum.DM.getUrlWords())) {
return parseParameter(url, "dbname=", "database=");
}
// 人大金仓、openGauss、瀚高数据库
else if (url.contains(DbTypeEnum.KING_BASE.getUrlWords()) || url.contains(DbTypeEnum.OPEN_GAUSS.getUrlWords()) || url.contains(DbTypeEnum.HIGH_GO.getUrlWords())) {
return parseAfterLastSlash(url);
}
} catch (Exception e) {
throw new IllegalArgumentException("Failed to parse database name from URL: " + url, e);
}
// 尝试通用解析作为最后手段
try {
return parseAfterLastSlash(url);
} catch (Exception e) {
throw new IllegalArgumentException("Unsupported or malformed database URL: " + url);
}
}
/**
* 解析URL中最后一个/之后的部分
*
* @author fengshuonan
* @since 2025/5/7 22:04
*/
private static String parseAfterLastSlash(String url) {
int lastSlashIndex = url.lastIndexOf('/');
if (lastSlashIndex == -1) {
return url; // 如果没有 /，返回整个 URL（或抛异常）
}
int paramStart = url.indexOf('?', lastSlashIndex);
if (paramStart == -1) {
return url.substring(lastSlashIndex + 1);
}
return url.substring(lastSlashIndex + 1, paramStart);
}
/**
* 解析URL参数
*
* @author fengshuonan
* @since 2025/5/7 22:04
*/
private static String parseParameter(String url, String... paramNames) {
for (String param : paramNames) {
// 匹配 databaseName=xxx 或 ;databaseName=xxx 或 &databaseName=xxx
Pattern pattern = Pattern.compile("(?:[;&?]|^)" + param + "([^;&?]+)");
Matcher matcher = pattern.matcher(url.toLowerCase());
if (matcher.find()) {
return matcher.group(1);
}
}
return parseAfterLastSlash(url);
}
public static void main(String[] args) {
System.out.println(getDbName("jdbc:oracle:thin:@//127.0.0.1:1521/right_oracle"));
System.out.println(getDbName("jdbc:postgresql://192.168.1.1:5432/right_pg"));
System.out.println(getDbName("jdbc:jtds:sqlserver://172.1.1.1:1433;DatabaseName=rightsqlserver"));
System.out.println(getDbName("jdbc:mysql://localhost:3306/right_mysql?autoReconnect=true&useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=CONVERT_TO_NULL&useSSL=false&nullCatalogMeansCurrent=true&allowPublicKeyRetrieval=true"));
System.out.println(getDbName("jdbc:dm://172.1.1.1:5236?useUnicode=true&useJDBCCompliantTimezoneShift=true&useLegacyDatetimeCode=false&serverTimezone=Asia/Shanghai&useSSL=true&characterEncoding=UTF-8"));
System.out.println(getDbName("jdbc:kingbase8://192.168.1.1:5432/right_king"));
System.out.println(getDbName("jdbc:opengauss://192.168.1.1:5432/right_open_gauss"));
System.out.println(getDbName("jdbc:highgo://localhost:5866/right_highgo"));
System.out.println(getDbName("jdbc:postgresql://172.1.1.1:5432/right_hailiang"));
}
}