使用Spring JDBC新增記錄如何返回自增主鍵值

在專案中，我們經常遇到這樣的情況，表主鍵值設定為自增長，也就是主鍵值不在應用層產生，而是在新增記錄時，由資料庫產生。當我們新增加一條記錄時，為了唯一定位到該記錄，我們需要獲得該記錄的主鍵值，而這需要在資料庫儲存資料後才能從資料庫中返回主鍵值。當我們使用Hibernate作為持久化框架時，新物件在持久化之後，主鍵值會自動繫結到物件上，但是當我們使用JDBC的時候要如何獲得主鍵值。

在JDBC3.0規範中，當新增記錄時，允許將資料庫自動產生的主鍵值繫結到Statement或PreparedStatement中。使用Statement時，可以透過以下方法繫結主鍵值:

int executeUpdate(String sql,int autoGeneratedKeys)

也可以透過Connection建立繫結自增值的PreparedStatement:

PreparedStatement prepareStatement(String sql,int autoGeneratedKeys)

當autoGeneratedKeys引數設定為Statement.RETURN_GENERATED_KEYS值時即可繫結資料庫產生的主鍵值，設定為Statement.NO_GENERATED_KEYS時，不繫結主鍵值。如下程式碼所示：

Statement stmt = conn.createStatement();
String sql = "insert into user(username,age) values('tom',22)";
stmt.executeUpdate(sql,Statement.RETURN_GENERATED_KEYS);
ResultSet rs = stmt.getGeneratedKeys();
if(rs.next()){
    int key = rs.getInt(0);
}

Spring利用這一技術，提供了一個可以返回新增記錄對應主鍵值的方法；

瞭解springcloud架構可以加求求：三五三六二四七二五九

　　int update(PreparedStatementCreator psc,KeyHolder generatedKeyHolder)

org.springframework.jdbc.support.KeyHolder是一個回撥介面，Spring使用它儲存新增記錄對應的主鍵，該介面的介面方法描述為：

Number getKey() throws InvalidDataAccessApiUsageException

　　當僅插入一行記錄時，主鍵不是複合鍵而是數字型別時，透過該方法可以直接返回新的主鍵值。如果是複合主鍵，或者有多個主鍵返回時，該方法丟擲InvalidDataAccessApiUsageException。

Map<String,Object> getKeys() throws InvalidDataAccessApiUsageException

　　如果是複合主鍵，則列名和列值構成Map中的一個Entry。如果返回的是多個主鍵，則該方法丟擲InvalidDataAccessApiUsageException異常。

List<Map<String ,Object>> getKeyList()

　　如果返回多個主鍵，即PreparedStatement新增了多條記錄，則每一個主鍵對應一個Map，多個Map構成一個List。

Spring為KeyHolder介面指代了一個通用的實現類GenerateKeyHolder，該類返回新增記錄時的自增長主鍵值。我們可以修改上面的程式碼使用Spring jdbc實現：