lsm/hutool
1
0
Fork 0
mirror of https://gitee.com/dromara/hutool.git synced 2025-04-05 17:37:59 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

!1192 新增 数据库 wrapper 支持反解

Browse Source 
Merge pull request !1192 from 蒋小小/v5-dev
This commit is contained in:
Looly 2024-03-28 07:49:16 +00:00 committed by Gitee
commit c93e28d775
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 73 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
hutool-db/src/main/java/cn/hutool/db/sql/Wrapper.java
View File

@ -120,6 +120,36 @@ public class Wrapper implements Serializable {
return StrUtil.format("{}{}{}", preWrapQuote, field, sufWrapQuote);
}
/**
* 反解包装字段名<br>
*
* @param field 字段名
* @return 未包装的字段名
*/
public String unWrap(String field) {
if (preWrapQuote == null || sufWrapQuote == null || StrUtil.isBlank(field)) {
return field;
}
//如果已经包含包装的引号，返回原字符
if (!StrUtil.isSurround(field, preWrapQuote, sufWrapQuote)) {
return field;
}
//如果字段中包含通配符或者括号（字段通配符或者函数），不做包装
if (StrUtil.containsAnyIgnoreCase(field, "*", "(", " ", " as ")) {
return field;
}
//对于Oracle这类数据库，表名中包含用户名需要单独拆分包装
if (field.contains(StrUtil.DOT)) {
final Collection<String> target = CollUtil.edit(StrUtil.split(field, CharUtil.DOT, 2), t -> StrUtil.unWrap(t, preWrapQuote, sufWrapQuote));
return CollectionUtil.join(target, StrUtil.DOT);
}
return StrUtil.unWrap(field, preWrapQuote, sufWrapQuote);
}
/**
* 包装字段名<br>
* 有时字段与SQL的某些关键字冲突，导致SQL出错，因此需要将字段名用单引号或者反引号包装起来，避免冲突

43
hutool-db/src/test/java/cn/hutool/db/WrapperTest.java Normal file
View File

@ -0,0 +1,43 @@
package cn.hutool.db;
import cn.hutool.db.sql.Wrapper;
import org.junit.Assert;
import org.junit.Ignore;
import org.junit.Test;
/**
* @author bwcx_jzy
* @since 24/3/28 028
*/
public class WrapperTest {
@Test
@Ignore
public void test() {
Wrapper wrapper = new Wrapper('`');
String originalName = "name";
String wrapName = wrapper.wrap(originalName);
String unWrapName = wrapper.unWrap(wrapName);
Assert.assertEquals(unWrapName, originalName);
}
@Test
@Ignore
public void testDotWrap() {
Wrapper wrapper = new Wrapper('`');
String originalName = "name.age";
String wrapName = wrapper.wrap(originalName);
String unWrapName = wrapper.unWrap(wrapName);
Assert.assertEquals(unWrapName, originalName);
}
@Test
@Ignore
public void testError() {
Wrapper wrapper = new Wrapper('`');
String originalName = "name.age*";
String wrapName = wrapper.wrap(originalName);
String unWrapName = wrapper.unWrap(wrapName);
Assert.assertEquals(unWrapName, originalName);
}
}