insert columns may need to be anti-escaped

apache · Feb 3, 2023 · 8e03b4f · 8e03b4f
1 parent b6becd0
commit 8e03b4f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/...source/src/main/java/io/seata/rm/datasource/exec/postgresql/PostgresqlInsertExecutor.java b/...source/src/main/java/io/seata/rm/datasource/exec/postgresql/PostgresqlInsertExecutor.java
@@ -31,6 +31,7 @@
 import io.seata.sqlparser.struct.SqlMethodExpr;
 import io.seata.sqlparser.struct.SqlSequenceExpr;
 import io.seata.sqlparser.struct.SqlDefaultExpr;
+import io.seata.sqlparser.util.ColumnUtils;
 import io.seata.sqlparser.util.JdbcConstants;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -109,8 +110,9 @@ public boolean containsAnyPk() {
         if (CollectionUtils.isEmpty(pkColumnNameList)) {
             return false;
         }
-        return pkColumnNameList.stream().anyMatch(pkColumn -> insertColumns.contains(pkColumn)
-            || CollectionUtils.toUpperList(insertColumns).contains(pkColumn.toUpperCase()));
+        List<String> newColumns = ColumnUtils.delEscape(insertColumns, getDbType());
+        return pkColumnNameList.stream().anyMatch(pkColumn -> newColumns.contains(pkColumn)
+            || CollectionUtils.toUpperList(newColumns).contains(pkColumn.toUpperCase()));
     }
 
     @Override