Apache
/
openjpa
mirror of https://github.com/apache/openjpa.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

OPENJPA-736: Change the way generated keys are retrieved so that the feature works also with databases other than DB2 

git-svn-id: https://svn.apache.org/repos/asf/openjpa/trunk@772756 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Milosz Tylenda 2009-05-07 19:57:34 +00:00
parent eb359de6a5
commit 195525392c
2 changed files with 12 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
openjpa-jdbc/src/main/java/org/apache/openjpa/jdbc/kernel/PreparedStatementManagerImpl.java
View File

@ -125,8 +125,7 @@ public class PreparedStatementManagerImpl
sql).getMessage());
}
if (autoAssignColNames != null)
populateAutoAssignCols(stmnt, autoAssign, autoAssignColNames,
row);
populateAutoAssignCols(stmnt, autoAssign, row);
else {
StateManagerImpl sm = (StateManagerImpl)row.getPrimaryKey();
if (sm != null) {
@ -168,19 +167,18 @@ public class PreparedStatementManagerImpl
* sql to select the key will be issued from DBDictionary.
*/
protected List populateAutoAssignCols(PreparedStatement stmnt,
Column[] autoAssign, String[] autoAssignColNames, RowImpl row)
Column[] autoAssign, RowImpl row)
throws SQLException {
List vals = null;
if (_dict.supportsGetGeneratedKeys) {
// set auto assign values to id col
vals = getGeneratedKeys(stmnt, autoAssignColNames);
vals = getGeneratedKeys(stmnt, autoAssign);
}
setObjectId(vals, autoAssign, autoAssignColNames, row);
setObjectId(vals, autoAssign, row);
return vals;
}
protected void setObjectId(List vals, Column[] autoAssign,
String[] autoAssignColNames, RowImpl row)
protected void setObjectId(List vals, Column[] autoAssign, RowImpl row)
throws SQLException{
OpenJPAStateManager sm = row.getPrimaryKey();
ClassMapping mapping = (ClassMapping) sm.getMetaData();
@ -203,13 +201,13 @@ public class PreparedStatementManagerImpl
* getGeneratedKeys.
*/
protected List getGeneratedKeys(PreparedStatement stmnt,
String[] autoAssignColNames)
Column[] autoAssign)
throws SQLException {
ResultSet rs = stmnt.getGeneratedKeys();
List vals = new ArrayList();
List<Object> vals = new ArrayList<Object>();
while (rs.next()) {
for (int i = 0; i < autoAssignColNames.length; i++)
vals.add(rs.getObject(autoAssignColNames[i]));
for (int i = 0; i < autoAssign.length; i++)
vals.add(rs.getObject(i + 1));
}
rs.close();
return vals;
@ -228,7 +226,8 @@ public class PreparedStatementManagerImpl
&& row.getPrimaryKey() != null) {
autoAssignColNames = new String[autoAssign.length];
for (int i = 0; i < autoAssign.length; i++)
autoAssignColNames[i] = autoAssign[i].getName();
autoAssignColNames[i] =
_dict.convertSchemaCase(autoAssign[i].getName());
}
return autoAssignColNames;
}

2
openjpa-jdbc/src/main/java/org/apache/openjpa/jdbc/sql/DBDictionary.java
View File

@ -3990,7 +3990,7 @@ public class DBDictionary
* Convert the specified schema name to a name that the database will
* be able to understand.
*/
protected String convertSchemaCase(String objectName) {
public String convertSchemaCase(String objectName) {
if (objectName == null)
return null;