修复PostgreSQL、H2使用upsert字段大小写问题

This commit is contained in:
Looly 2023-12-28 18:50:29 +08:00
parent 51a66151f3
commit d1348eabcf
4 changed files with 32 additions and 28 deletions

View File

@ -3,7 +3,7 @@
------------------------------------------------------------------------------------------------------------- -------------------------------------------------------------------------------------------------------------
# 6.0.0-M10 (2023-12-23) # 6.0.0-M10 (2023-12-28)
### 计划实现 ### 计划实现
* 【poi 】 Markdown相关如HTML转换等基于commonmark-java * 【poi 】 Markdown相关如HTML转换等基于commonmark-java

View File

@ -12,8 +12,8 @@
package org.dromara.hutool.db.dialect.impl; package org.dromara.hutool.db.dialect.impl;
import org.dromara.hutool.core.lang.Assert;
import org.dromara.hutool.core.array.ArrayUtil; import org.dromara.hutool.core.array.ArrayUtil;
import org.dromara.hutool.core.lang.Assert;
import org.dromara.hutool.core.text.StrUtil; import org.dromara.hutool.core.text.StrUtil;
import org.dromara.hutool.db.Entity; import org.dromara.hutool.db.Entity;
import org.dromara.hutool.db.Page; import org.dromara.hutool.db.Page;
@ -23,7 +23,6 @@ import org.dromara.hutool.db.sql.SqlBuilder;
import java.sql.Connection; import java.sql.Connection;
import java.sql.PreparedStatement; import java.sql.PreparedStatement;
import java.sql.SQLException;
/** /**
* H2数据库方言 * H2数据库方言
@ -33,6 +32,9 @@ import java.sql.SQLException;
public class H2Dialect extends AnsiSqlDialect { public class H2Dialect extends AnsiSqlDialect {
private static final long serialVersionUID = 1490520247974768214L; private static final long serialVersionUID = 1490520247974768214L;
/**
* 构造
*/
public H2Dialect() { public H2Dialect() {
// wrapper = new Wrapper('"'); // wrapper = new Wrapper('"');
} }
@ -49,7 +51,7 @@ public class H2Dialect extends AnsiSqlDialect {
} }
@Override @Override
public PreparedStatement psForUpsert(final Connection conn, final Entity entity, final String... keys) throws SQLException { public PreparedStatement psForUpsert(final Connection conn, final Entity entity, String... keys) {
Assert.notEmpty(keys, "Keys must be not empty for H2 MERGE SQL."); Assert.notEmpty(keys, "Keys must be not empty for H2 MERGE SQL.");
SqlBuilder.validateEntity(entity); SqlBuilder.validateEntity(entity);
final SqlBuilder builder = SqlBuilder.of(quoteWrapper); final SqlBuilder builder = SqlBuilder.of(quoteWrapper);
@ -75,6 +77,7 @@ public class H2Dialect extends AnsiSqlDialect {
String tableName = entity.getTableName(); String tableName = entity.getTableName();
if (null != this.quoteWrapper) { if (null != this.quoteWrapper) {
tableName = this.quoteWrapper.wrap(tableName); tableName = this.quoteWrapper.wrap(tableName);
keys = quoteWrapper.wrap(keys);
} }
builder.append("MERGE INTO ").append(tableName) builder.append("MERGE INTO ").append(tableName)
// 字段列表 // 字段列表

View File

@ -17,21 +17,23 @@ import org.dromara.hutool.db.Entity;
import org.dromara.hutool.db.Page; import org.dromara.hutool.db.Page;
import org.dromara.hutool.db.StatementUtil; import org.dromara.hutool.db.StatementUtil;
import org.dromara.hutool.db.dialect.DialectName; import org.dromara.hutool.db.dialect.DialectName;
import org.dromara.hutool.db.sql.SqlBuilder;
import org.dromara.hutool.db.sql.QuoteWrapper; import org.dromara.hutool.db.sql.QuoteWrapper;
import org.dromara.hutool.db.sql.SqlBuilder;
import java.sql.Connection; import java.sql.Connection;
import java.sql.PreparedStatement; import java.sql.PreparedStatement;
import java.sql.SQLException;
/** /**
* MySQL方言 * MySQL方言
* @author loolly
* *
* @author loolly
*/ */
public class MysqlDialect extends AnsiSqlDialect { public class MysqlDialect extends AnsiSqlDialect {
private static final long serialVersionUID = -3734718212043823636L; private static final long serialVersionUID = -3734718212043823636L;
/**
* 构造
*/
public MysqlDialect() { public MysqlDialect() {
quoteWrapper = new QuoteWrapper('`'); quoteWrapper = new QuoteWrapper('`');
} }
@ -57,11 +59,10 @@ public class MysqlDialect extends AnsiSqlDialect{
* @param entity 数据实体类包含表名 * @param entity 数据实体类包含表名
* @param keys 此参数无效 * @param keys 此参数无效
* @return PreparedStatement * @return PreparedStatement
* @throws SQLException SQL执行异常
* @since 5.7.20 * @since 5.7.20
*/ */
@Override @Override
public PreparedStatement psForUpsert(final Connection conn, final Entity entity, final String... keys) throws SQLException { public PreparedStatement psForUpsert(final Connection conn, final Entity entity, final String... keys) {
SqlBuilder.validateEntity(entity); SqlBuilder.validateEntity(entity);
final SqlBuilder builder = SqlBuilder.of(quoteWrapper); final SqlBuilder builder = SqlBuilder.of(quoteWrapper);

View File

@ -12,19 +12,18 @@
package org.dromara.hutool.db.dialect.impl; package org.dromara.hutool.db.dialect.impl;
import org.dromara.hutool.core.lang.Assert;
import org.dromara.hutool.core.array.ArrayUtil; import org.dromara.hutool.core.array.ArrayUtil;
import org.dromara.hutool.core.lang.Assert;
import org.dromara.hutool.core.text.CharUtil; import org.dromara.hutool.core.text.CharUtil;
import org.dromara.hutool.core.text.StrUtil; import org.dromara.hutool.core.text.StrUtil;
import org.dromara.hutool.db.Entity; import org.dromara.hutool.db.Entity;
import org.dromara.hutool.db.StatementUtil; import org.dromara.hutool.db.StatementUtil;
import org.dromara.hutool.db.dialect.DialectName; import org.dromara.hutool.db.dialect.DialectName;
import org.dromara.hutool.db.sql.SqlBuilder;
import org.dromara.hutool.db.sql.QuoteWrapper; import org.dromara.hutool.db.sql.QuoteWrapper;
import org.dromara.hutool.db.sql.SqlBuilder;
import java.sql.Connection; import java.sql.Connection;
import java.sql.PreparedStatement; import java.sql.PreparedStatement;
import java.sql.SQLException;
/** /**
@ -48,7 +47,7 @@ public class PostgresqlDialect extends AnsiSqlDialect {
} }
@Override @Override
public PreparedStatement psForUpsert(final Connection conn, final Entity entity, final String... keys) throws SQLException { public PreparedStatement psForUpsert(final Connection conn, final Entity entity, String... keys) {
Assert.notEmpty(keys, "Keys must be not empty for Postgres."); Assert.notEmpty(keys, "Keys must be not empty for Postgres.");
SqlBuilder.validateEntity(entity); SqlBuilder.validateEntity(entity);
final SqlBuilder builder = SqlBuilder.of(quoteWrapper); final SqlBuilder builder = SqlBuilder.of(quoteWrapper);
@ -78,6 +77,7 @@ public class PostgresqlDialect extends AnsiSqlDialect {
String tableName = entity.getTableName(); String tableName = entity.getTableName();
if (null != this.quoteWrapper) { if (null != this.quoteWrapper) {
tableName = this.quoteWrapper.wrap(tableName); tableName = this.quoteWrapper.wrap(tableName);
keys = quoteWrapper.wrap(keys);
} }
builder.append("INSERT INTO ").append(tableName) builder.append("INSERT INTO ").append(tableName)
// 字段列表 // 字段列表