實(shí)現(xiàn)類Spring聲明式事務(wù)

　　原生的jdbc 對事務(wù)管理也是比較繁瑣的, 需要手工進(jìn)行提交和回滾, 還要一堆try-catch. 而熟悉spring 的同學(xué)都知道, spring采用了聲明式事務(wù)方式來管理事務(wù), 使事務(wù)管理變得很簡單. Spring 事務(wù)很強(qiáng)大, 筆者這里僅使用jdbc 來模擬簡單的幾個屬性.

　　1. 聲明式事務(wù)方案設(shè)計(jì)

　　聲明式事務(wù)主要依據(jù)java 動態(tài)代理實(shí)現(xiàn)

　　通過將Connection 存放在ThreadLocal 變量中, 來解決并發(fā)問題. Spring 底層也是用的ThreadLocal.

　　通過記錄Connection 的創(chuàng)建者, 來解決事務(wù)的嵌套問題.

　　自定義注解@EnableTranscation: 用于標(biāo)明方法是否開啟事務(wù)

　　Service工廠: 用于模擬Spring容器創(chuàng)建Bean過程, 如果Service 中包含使用@EnableTranscation修飾的方法, 則創(chuàng)建Service的代理對象, 否則返回Service 實(shí)例

　　自定義Dao時(shí), 不能直接創(chuàng)建Connection, 需要獲取當(dāng)前線程中保存的Connection.

　　2. 連接池管理

　　筆者對數(shù)據(jù)庫的連接采用c3p0 連接池.

　　2.1 c3po 配置

　　root

　　root

　　com.mysql.jdbc.Driver

　　jdbc:mysql://localhost:3306/learn-jdbc?characterEncoding=UTF-8

　　10

　　5

　　5

　　50

　　100

　　10

　　2.2 數(shù)據(jù)庫連接工具類

　　封裝獲取數(shù)據(jù)庫連接和關(guān)閉數(shù)據(jù)庫連接資源的方法

　　public class DbConnUtil {

　　// c3P0配置名

　　private static final String c3p0PoolName = "myC3p0Pool";

　　// 配置數(shù)據(jù)源

　　private static final DataSource dataSource = new ComboPooledDataSource(c3p0PoolName);

　　// 配置本地連接

　　private static ThreadLocal txConnectionLocal = new ThreadLocal<>();

　　/** 獲取數(shù)據(jù)庫連接

　　* @param autoCommitTx 是否開啟提供提交事務(wù)

　　* @return Connection 數(shù)據(jù)庫連接

　　* @since 1.0

　　* @author zongf

　　* @created 2019-07-18

　　*/

　　public static Connection getConnection(boolean autoCommitTx) {

　　try {

　　Connection connection = dataSource.getConnection();

　　connection.setAutoCommit(autoCommitTx);

　　return connection;

　　} catch (SQLException e) {

　　e.printStackTrace();

　　}

　　return null;

　　}

　　/**

　　* @Description: 獲取本線程連接

　　* @return: Connection 數(shù)據(jù)庫連接

　　* @author: zongf

　　* @time: 2019-06-26 14:37:00

　　* @since 1.0

　　*/

　　public static TxConnection getTxConnection() {

　　TxConnection txConnection = null;

　　// 如果ThreadLocal 中連接為空, 則創(chuàng)建新的連接

　　if (txConnectionLocal.get() == null || txConnectionLocal.get().getConnection() == null) {

　　txConnection = new TxConnection(getConnection(true));

　　txConnectionLocal.set(txConnection);

　　} else {

　　txConnection = txConnectionLocal.get();

　　}

　　return txConnection;

　　}

　　/** 獲取當(dāng)前線程內(nèi)的數(shù)據(jù)庫連接

　　* @return Connection

　　* @since 1.0

　　* @author zongf

　　* @created 2019-07-18

　　*/

　　public static Connection getLocalConnection() {

　　return getTxConnection().getConnection();

　　}

　　/** 獲取當(dāng)前線程的數(shù)據(jù)庫連接對象

　　* @return ThreadLocal

　　* @since 1.0

　　* @author zongf

　　* @created 2019-07-18

　　*/

　　public static ThreadLocal getLocalTxConnection() {

　　return txConnectionLocal;

　　}

　　/** 當(dāng)歸還連接時(shí), 需要設(shè)置自動提交事務(wù)為true.

　　* @param connection

　　* @return null

　　* @since 1.0

　　* @author zongf

　　* @created 2019-07-18

　　*/

　　public static void release(Connection connection) throws SQLException {

　　try {

　　if (connection != null && !connection.isClosed()) {

　　connection.setAutoCommit(true);

　　}

　　} catch (SQLException e) {

　　e.printStackTrace();

　　} finally {

　　connection.close();

　　}

　　}

　　}

　　2.3 定義事務(wù)連接對象

　　由于在事務(wù)嵌套時(shí), 需要遵循哪一層動態(tài)代理開啟的事務(wù), 由哪一層動態(tài)代理負(fù)責(zé)事務(wù)的開啟和回滾, 因此需要記錄事務(wù)的開啟者. 因此筆者創(chuàng)建了一個TxConnneciton 對象.

　　public class TxConnection {

　　private Connection connection;

　　private String creator;

　　// 省略setter/getter 方法

　　}

　　3. 自定義開啟事務(wù)注解

　　定義一個類似于spring @Transcation 的注解, 用于開啟事務(wù).

　　openNewTx 用于模擬spring七種事務(wù)傳播行為之一的 Propagation.REQUIRES_NEW

　　@Documented

　　@Target(ElementType.METHOD)

　　@Retention(RetentionPolicy.RUNTIME)

　　public @interface EnableTranscation {

　　// 是否開啟新的事務(wù)

　　boolean openNewTx() default false;

　　}

　　4. 動態(tài)代理處理器

　　/**事務(wù)動態(tài)代理處理器

　　* @since 1.0

　　* @author zongf

　　* @created 2019-07-18

　　*/

　　public class TranscationHandler implements InvocationHandler {

　　private Object target;

　　public TranscationHandler(Object target) {

　　this.target = target;

　　}

　　public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {

　　// 獲取當(dāng)前數(shù)據(jù)庫連接

　　TxConnection txConnection = DbConnUtil.getTxConnection();

　　// 保存老的連接對象

　　TxConnection oldTxConnection = null;

　　try {

　　// 獲取目標(biāo)對象方法

　　Method targetMethod = target.getClass().getMethod(method.getName(), method.getParameterTypes());

　　// 看當(dāng)前方法是否開啟了事務(wù)

　　boolean enableTx = targetMethod.isAnnotationPresent(EnableTranscation.class);

　　// 如果開啟事務(wù), 則設(shè)置當(dāng)前連接為手動提交事務(wù)

　　if (enableTx) {

　　// 獲取注解信息

　　EnableTranscation annotation = targetMethod.getAnnotation(EnableTranscation.class);

　　// 獲取是否開啟新事務(wù)

　　boolean openNewTx = annotation.openNewTx();

　　if (!txConnection.getConnection().getAutoCommit()) { //為false, 表示已經(jīng)開啟了事務(wù)

　　if (openNewTx) { // 如果需要開啟的事務(wù)

　　// 保存原數(shù)據(jù)庫連接

　　oldTxConnection = txConnection;

　　// 獲取新的連接

　　txConnection = new TxConnection(DbConnUtil.getConnection(false), this.toString());

　　// 替換當(dāng)前線程中的數(shù)據(jù)庫連接

　　DbConnUtil.getLocalTxConnection().set(txConnection);

　　}

　　} else { // 為true, 表示未開啟事務(wù)

　　// 沒有開啟事務(wù), 設(shè)置自動提交為false. 表示已經(jīng)開始了事務(wù)

　　txConnection.getConnection().setAutoCommit(false);

　　txConnection.setCreator(this.toString());

　　}

　　}

　　// 執(zhí)行目標(biāo)方法

　　Object object = targetMethod.invoke(this.target, args);

　　// 如果事務(wù)是當(dāng)前handler對象創(chuàng)建, 那么提交事務(wù)

　　if (this.toString().equals(txConnection.getCreator())) {

　　txConnection.getConnection().commit();

　　}

　　return object;

　　} catch (Exception e) {

　　if (txConnection != null && this.toString().equals(txConnection.getCreator())) {

　　if (txConnection.getConnection() != null && !txConnection.getConnection().isClosed()) {

　　txConnection.getConnection().rollback();

　　txConnection.getConnection().setAutoCommit(true);

　　}

　　}

　　throw new RuntimeException("發(fā)生異常, 事務(wù)已回滾!", e);

　　} finally {

　　// 釋放數(shù)據(jù)庫連接

　　if (txConnection != null && this.toString().equals(txConnection.getCreator())) {

　　DbConnUtil.release(txConnection.getConnection());

　　}

　　// 如果新連接不為null, 則表示開啟了新事務(wù). 則回滾原連接

　　if (oldTxConnection != null) {

　　DbConnUtil.getLocalTxConnection().set(oldTxConnection);

　　}

　　}

　　}

　　}

　　5. ServiceFactory 工廠

　　創(chuàng)建Service 工廠類, 用于模擬Spring 容器. 當(dāng)目標(biāo)Service中包含@EnableTransaction 注解時(shí), 創(chuàng)建Service 的動態(tài)代理, 否則創(chuàng)建Service 對象.

　　/** Service工廠, 模擬spring 容器

　　* @since 1.0

　　* @author zongf

　　* @created 2019-07-18

　　*/

　　public class ServiceFactory {

　　/** 獲取Service 實(shí)例

　　* @param clz Service 實(shí)現(xiàn)類類型

　　* @return T Service 對象或動態(tài)代理對象

　　* @since 1.0

　　* @author zongf

　　* @created 2019-07-18

　　*/

　　public static T getService(Class clz) {

　　T t = null;

　　try {

　　t = clz.newInstance();

　　} catch (Exception e) {

　　e.printStackTrace();

　　throw new RuntimeException("創(chuàng)建對象失敗");

　　}

　　// 判斷不能是接口, 接口不能創(chuàng)建實(shí)現(xiàn)類

　　if(clz.isInterface()){

　　throw new RuntimeException("接口不能創(chuàng)建實(shí)例!");

　　}

　　// 是否開啟動態(tài)代理

　　boolean enableTx = false;

　　// 遍歷所有非私有方法, 如果方法有@EnableTx注解, 則說明需要創(chuàng)建代理

　　Method[] methods = clz.getMethods();

　　for (Method method : methods) {

　　if (method.getAnnotation(EnableTranscation.class) != null) {

　　enableTx = true;

　　break;

　　}

　　}

　　// 如果需要創(chuàng)建代理, 則返回代理對象

　　if (enableTx) {

　　return (T) Proxy.newProxyInstance(clz.getClassLoader(), clz.getInterfaces(), new TranscationHandler(t));

　　}

　　return t;

　　}

　　}

　　6. 聲明式事務(wù)測試

　　測試用例, 筆者借助于之前寫的BaseDao來簡化基本步驟的開發(fā).

　　1.1 定義接口

　　public interface IMixService {

　　// 模擬正常

　　void success();

　　// 模擬異常操作, 事務(wù)回滾

　　void error();

　　void show();

　　}

　　6.2 定義實(shí)現(xiàn)類

　　public class MixService implements IMixService {

　　private IUserService userService = ServiceFactory.getService(UserService.class);

　　private IPersonService personService = ServiceFactory.getService(PersonService.class);

　　@EnableTranscation

　　@Override

　　public void success() {

　　this.userService.save(new UserPO("user-01", "123456"));

　　this.personService.save(new PersonPO("person-01", "abcdefg"));

　　}無錫婦科醫(yī)院 http://www.bhnnk120.com/

　　@EnableTranscation

　　@Override

　　public void error() {

　　this.userService.save(new UserPO("user-01", "123456"));

　　this.personService.save(new PersonPO("person-01", "abcdefg"));

　　// 模擬異常會館

　　int a = 1/0;

　　}

　　@Override

　　public void show() {

　　List userPOS = this.userService.queryAll();

　　List personPOS = this.personService.queryAll();

　　System.out.println("\n****** t_user: *****");

　　userPOS.forEach(System.out::println);

　　System.out.println("\n****** t_person: *****");

　　personPOS.forEach(System.out::println);

　　}

　　}

　　6.3 測試用例

　　public class MixService implements IMixService {

　　private IUserService userService = ServiceFactory.getService(UserService.class);

　　private IPersonService personService = ServiceFactory.getService(PersonService.class);

　　@EnableTranscation

　　@Override

　　public void success() {

　　this.userService.save(new UserPO("user-01", "123456"));

　　this.personService.save(new PersonPO("person-01", "abcdefg"));

　　}

　　@EnableTranscation

　　@Override

　　public void error() {

　　this.userService.save(new UserPO("user-01", "123456"));

　　this.personService.save(new PersonPO("person-01", "abcdefg"));

　　// 模擬異常會館

　　int a = 1/0;

　　}

　　@Override

　　public void show() {

　　List userPOS = this.userService.queryAll();

　　List personPOS = this.personService.queryAll();

　　System.out.println("\n****** t_user: *****");

　　userPOS.forEach(System.out::println);

　　System.out.println("\n****** t_person: *****");

　　personPOS.forEach(System.out::println);

　　}

　　}