CodeCow/querydsl
1
0
Fork 0
mirror of https://github.com/querydsl/querydsl.git synced 2026-06-13 21:01:01 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

#623933 : fixed executeWithKeys for batches

Browse Source
This commit is contained in:
Timo Westkämper 2010-08-25 12:59:55 +00:00
parent 06c0a84166
commit 2a9456971a
1 changed files with 15 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
querydsl-sql/src/main/java/com/mysema/query/sql/dml/SQLInsertClause.java
View File

@ -148,22 +148,12 @@ public class SQLInsertClause extends AbstractSQLClause implements InsertClause<S
PreparedStatement stmt = null;
if (batches.isEmpty()){
serializer.serializeForInsert(metadata, entity, columns, values, subQuery);
queryString = serializer.toString();
logger.debug(queryString);
if (withKeys){
stmt = connection.prepareStatement(queryString, Statement.RETURN_GENERATED_KEYS);
}else{
stmt = connection.prepareStatement(queryString);
}
setParameters(stmt, serializer.getConstants(),Collections.<Param<?>,Object>emptyMap());
stmt = prepareStatementAndSetParameters(serializer, withKeys);
}else{
serializer.serializeForInsert(metadata, entity, batches.get(0).getColumns(), batches.get(0).getValues(), batches.get(0).getSubQuery());
queryString = serializer.toString();
logger.debug(queryString);
stmt = connection.prepareStatement(queryString);
stmt = prepareStatementAndSetParameters(serializer, withKeys);
// add first batch
setParameters(stmt, serializer.getConstants(),Collections.<Param<?>,Object>emptyMap());
stmt.addBatch();
// add other batches
@ -178,6 +168,19 @@ public class SQLInsertClause extends AbstractSQLClause implements InsertClause<S
}
return stmt;
}
private PreparedStatement prepareStatementAndSetParameters(SQLSerializer serializer, boolean withKeys) throws SQLException {
queryString = serializer.toString();
logger.debug(queryString);
PreparedStatement stmt;
if (withKeys){
stmt = connection.prepareStatement(queryString, Statement.RETURN_GENERATED_KEYS);
}else{
stmt = connection.prepareStatement(queryString);
}
setParameters(stmt, serializer.getConstants(),Collections.<Param<?>,Object>emptyMap());
return stmt;
}
public ResultSet executeWithKeys(){
try {