AfflictionNetworks/depot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Ok, this works on both mysql and postgres.

Browse Source 
On mysql the column name ends up being "GENERATED_KEY".
This commit is contained in:
Ray Greenwell
2014-05-02 01:31:58 +00:00
parent e2a9702c46
commit 1a9fc404ba
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/samskivert/depot/impl/IdentityValueGenerator.java
+8 -5
View File
@@ -6,6 +6,7 @@ package com.samskivert.depot.impl;
import java.sql.Connection; import java.sql.Connection;
import java.sql.ResultSet; import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException; import java.sql.SQLException;
import java.sql.Statement; import java.sql.Statement;
@@ -45,11 +46,13 @@ public class IdentityValueGenerator extends ValueGenerator
// if this JDBC driver supports getGeneratedKeys, use it! // if this JDBC driver supports getGeneratedKeys, use it!
if (stmt != null && conn.getMetaData().supportsGetGeneratedKeys()) { if (stmt != null && conn.getMetaData().supportsGetGeneratedKeys()) {
ResultSet rs = stmt.getGeneratedKeys(); ResultSet rs = stmt.getGeneratedKeys();
if (rs.next()) { while (rs.next()) {
try { ResultSetMetaData rsMetaData = rs.getMetaData();
return rs.getInt(column); int columnCount = rsMetaData.getColumnCount();
} catch (SQLException e) { for (int ii = 1; ii <= columnCount; ii++) {
log.info("Unable to get generated key, falling back", "errmsg", e.getMessage()); if (column.equals(rsMetaData.getColumnName(ii))) {
return rs.getInt(ii);
}
} }
} }
} }