在Spring中用select last_insert_id()時遇到問題

一直使用的Oracle數(shù)據(jù)庫,通過序列來實現(xiàn)自增字段,插入之前就已經(jīng)獲得了自增id,保存下來即可在后來的操作中使用

今天在使用MySQL時卻不知如何處理,插入記錄后不知怎樣獲得剛剛插入的id,查過文檔后發(fā)現(xiàn)了select last_insert_id(),在插入之后執(zhí)行此查詢,即可獲得自增id,喜出望外。
可用到自己的程序中之后卻得不到想要的結果,于是就懷疑到了Spring頭上,因為通過基本JDBC測試是沒有任何問題的,所以就去跟蹤Spring JDBC, 看過源碼之后才豁然開朗,原來Spring中如此獲得數(shù)據(jù)庫的: con = DataSourceUtils.getConnection(getDataSource());, 哎,只能怪自己Spring掌握不夠好,所以就不能在執(zhí)行insert之后去執(zhí)行select last_insert_id()了,因為select last_insert_id()是真對當前 在一般情況下,在新增領域對象后,都需要獲取對應的主鍵值。使用應用層來維護主鍵,在一定程度上有利于程序性能的優(yōu)化和應用移植性的提高。在采用數(shù)據(jù)庫自增主鍵的方案里,如果JDBC驅動不能綁定新增記錄對應的主鍵,就需要手工執(zhí)行查詢語句以獲取對應的主鍵值,對于高并發(fā)的系統(tǒng),這很容易返回錯誤的主鍵。通過帶緩存的DataFieldMaxValueIncrementer,可以一次獲取批量的主鍵值,供多次插入領域對象時使用,它的執(zhí)行性能是很高的。

使用數(shù)據(jù)庫的自增主鍵
我們經(jīng)常使用數(shù)據(jù)的自增字段作為表主鍵,也即主鍵值不在應用層產(chǎn)生,而是在新增記錄時,由數(shù)據(jù)庫產(chǎn)生。這樣,應用層在保存對象前并不知道對象主鍵值,而必須在保存數(shù)據(jù)后才能從數(shù)據(jù)庫中返回主鍵值。在很多情況下,我們需要獲取新對象持久化后的主鍵值。在Hibernate等ORM框架,新對象持久化后,Hibernate會自動將主鍵值綁定到對象上,給程序的開發(fā)帶來了很多方便。

在JDBC 3.0規(guī)范中,當新增記錄時,允許將數(shù)據(jù)庫自動產(chǎn)生的主鍵值綁定到Statement或PreparedStatement中。使用Statement時,可以通過以下方法綁定主鍵值:
int executeUpdate(String sql,int autoGeneratedKeys)
也可以通過Connection創(chuàng)建綁定自增值的PreparedStatement:
PreparedStatement prepareStatement(String sql, int autoGeneratedKeys)

當autoGeneratedKeys參數(shù)設置為Statement.RETURN_GENERATED_KEYS值時即可綁定數(shù)據(jù)庫產(chǎn)生的主鍵值,設置為Statement.NO_GENERATED_KEYS時,不綁定主鍵值。下面的代碼演示了Statement綁定并獲取數(shù)據(jù)庫產(chǎn)生的主鍵值的過程:

Spring利用這一技術,提供了一個可以返回新增記錄對應主鍵值的方法:
int update(PreparedStatementCreator psc, KeyHolder generatedKeyHolder)
org.springframework.jdbc.support.KeyHolder是一個回調(diào)接口,Spring使用它保存新增記錄對應的主鍵,該接口的接口方法描述如下:
Number getKey() throws InvalidDataAccessApiUsageException
當 僅插入一行數(shù)據(jù),主鍵不是復合鍵且是數(shù)字類型時,通過該方法可以直接返回新的主鍵值。如果是復合主鍵,或者有多個主鍵返回時,該方法拋出 InvalidDataAccessApiUsageException。該方法是最常用的方法,因為一般情況下,我們一次僅插入一條數(shù)據(jù)并且主鍵字段類型為數(shù)字類型;

Map getKeys() throws InvalidDataAccessApiUsageException
如果是復合主鍵,則列名和列值構成Map中的一個Entry。如果返回的是多個主鍵,則該方法拋出InvalidDataAccessApiUsageException異常;

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

Spring為KeyHolder接口指代了一個通用的實現(xiàn)類GeneratedKeyHolder,該類返回新增記錄時的自增長主鍵值。假設我們希望在新增論壇板塊對象后,希望將主鍵值加載到對象中,則可以按以下代碼進行調(diào)整:

 

這樣,在調(diào)用addForum(final Forum forum)新增forum領域對象后,forum將擁有對應的主鍵值,方便后繼的使用。
在JDBC 3.0之前的版本中,PreparedStatement不能綁定主鍵,如果采用表自增鍵(如MySql的auto increment或SqlServer的identity)將給獲取正確的主鍵值帶來挑戰(zhàn)——因為你必須在插入數(shù)據(jù)后,馬上執(zhí)行另一條獲取新增主鍵的查詢語句。表 1給出了不同數(shù)據(jù)庫獲取最新自增主鍵值的查詢語句:
表 1 不同數(shù)據(jù)庫獲取新增加的主鍵值

Spring JDBC提供了自增鍵以及行集的支持,自增鍵對象讓我們可以不依賴數(shù)據(jù)庫的自增鍵,在應用層為新記錄提供主鍵值。在JDK 1.4中引入了RowSet,它允許在連接斷開的情況下操作數(shù)據(jù),在這節(jié)里,我們將介紹如何在Spring JDBC中使用RowSet。

自增鍵的使用
一般數(shù)據(jù)庫都提供了自增鍵的功能,如MySql的auto_increment、SqlServerr的identity字段等。Spring允許你在應用層產(chǎn)生主鍵值,為此定義了 org.springframework.jdbc.support.incrementer.DataFieldMaxValueIncrementer 接口,提供兩種產(chǎn)生主鍵的方案:第一,通過序列產(chǎn)生主鍵;第二,通過表產(chǎn)生主鍵。根據(jù)主鍵產(chǎn)生方式和數(shù)據(jù)庫的不同,Spring提供了若干實現(xiàn)類,如圖 1所示:

圖 1 DateFieldValueIncrementer繼承類圖
根據(jù)不同的主鍵產(chǎn)生方式,可能需要配置表名、主鍵字段名或序列名等信息。下面,我們以Oracle和MySql為例分別講解使用序列及表字段產(chǎn)生主鍵值的方式。

DataFieldMaxValueIncrementer接口定義了3個獲取下一個主鍵值的方法:
? int nextIntValue():獲取下一個主鍵值,主鍵數(shù)據(jù)類型為int;
? long nextLongValue():獲取下一個主鍵值,主鍵數(shù)據(jù)類型為long;
? String nextStringValue():獲取下一個主鍵值,主鍵數(shù)據(jù)類型為String;
在其抽象實現(xiàn)類AbstractDataFieldMaxValueIncrementer中,提供了幾個重要的屬性,其中 incrementerName定義序列或主鍵表的名稱;如果返回的主鍵是String類型,則paddingLength屬性可能會派上用場,它允許你指定返回主鍵的長度,不足的部分前面補0。

HsqlMaxValueIncrementer和MySQLMaxValueIncrementer兩個主鍵值產(chǎn)生器基于表進行工作。通過 columnName屬性定義主鍵列的名字,通過cacheSize屬性定義緩存的主鍵個數(shù),當內(nèi)存中的主鍵值用完后,產(chǎn)生器將一次性獲取 cacheSize個主鍵,這樣可以減少數(shù)據(jù)訪問的次數(shù),提高應用的性能。

我們通過DateFieldValueIncrementer從數(shù)據(jù)庫中獲取主鍵值來彌補這個缺陷。首先,調(diào)整PostJdbcDao的代碼,添加DateFieldValueIncrementer屬性,并通過它從序列中得到下一個主鍵值:
代碼清單 13 使用DateFieldValueIncrementer產(chǎn)生主鍵

 

在②處,我們通過incre.nextIntValue()獲取下一個主鍵值。

以序列方式產(chǎn)生主鍵值
在Oracle數(shù)據(jù)庫中創(chuàng)建一個seq_post_id序列,使用這個序列為t_post提供主鍵值,以下是創(chuàng)建seq_post_id的腳本:

 

接著,調(diào)整Spring的配置,使用OracleSequenceMaxValueIncrementer作為主鍵產(chǎn)生器:

 

以表方式產(chǎn)生主鍵值
在Mysql中創(chuàng)建一張用于維護t_post主鍵的t_post_id表,以下是創(chuàng)建該表及插入初始化的SQL腳本:

 

由于主鍵維護表的并發(fā)訪問量很大,所以最好將其聲明為MYISAM類型,此外需要為該表提供初始值,以便后續(xù)主鍵值在此之上進行遞增。
調(diào)整為MySql數(shù)據(jù)庫后,我們僅需要對Spring配置進行小小的調(diào)整就可以了:

 

incrementerName和columnName都很容易理解,cacheSize決定一次返回的主鍵個數(shù),這里我們設置為10。當?shù)谝淮瓮ㄟ^ MySQLMaxValueIncrementer# nextIntValue()獲取主鍵值時,MySQLMaxValueIncrementer將使t_post_id. sequence_id遞增10,而后續(xù)9次調(diào)用nextIntValue()方法時,都從緩存中獲取主鍵值。直到第10次再次調(diào)用 nextIntValue()方法時,才會再次將t_post_id. sequence_id字段值遞增10,如此循環(huán)反復。

小結
主鍵的生產(chǎn)方式從產(chǎn)生地點上可以分為應用層產(chǎn)生和數(shù)據(jù)庫產(chǎn)生兩種方式。應用層借助數(shù)據(jù)庫的序列或表產(chǎn)生主鍵,這種方式可以保證程序的可移植性和安全性,同時可以通過緩存機制提高運行效率。有些數(shù)據(jù)庫支持數(shù)據(jù)表自增鍵的主鍵產(chǎn)生機制,在JDBC 3.0以前的版本中,無法通過Statement自動獲取新增記錄的對應主鍵。這時需要在插入數(shù)據(jù)后,馬上執(zhí)行一條數(shù)據(jù)庫相關的主鍵獲取SQL語句以得到對應的主鍵值,在數(shù)據(jù)庫高并發(fā)的情況下,有可能獲取到不正確的主鍵值。在這種情況下,在插入數(shù)據(jù)前事先在應用層準備好主鍵值是很好的備選方案。

另外補充一點在SqlUpdate執(zhí)行update之前需設置setReturnGeneratedKeys(true);

? 版權聲明
THE END
喜歡就支持一下吧
點贊10 分享