Spring框架AOP源碼分析(三):手寫Spring事務(wù)框架

1.編程事務(wù)實現(xiàn)

所謂編程式事務(wù)指的是通過編碼方式實現(xiàn)事務(wù),即類似于JDBC編程實現(xiàn)事務(wù)管理。管理使用TransactionTemplate或者直接使用底層的PlatformTransactionManager。對于編程式事務(wù)管理,spring推薦使用TransactionTemplate。

使用編程事務(wù)實現(xiàn)手動事務(wù)(重點)

@Component
public class TransactionUtils {

    @Autowired
    private DataSourceTransactionManager dataSourceTransactionManager;

    // 開啟事務(wù)
    public TransactionStatus begin() {
        TransactionStatus transaction = dataSourceTransactionManager.getTransaction(new DefaultTransactionAttribute());
        return transaction;
    }

    // 提交事務(wù)
    public void commit(TransactionStatus transactionStatus) {
        dataSourceTransactionManager.commit(transactionStatus);
    }

    // 回滾事務(wù)
    public void rollback(TransactionStatus transactionStatus) {
        dataSourceTransactionManager.rollback(transactionStatus);
    }
}

@Service
public class UserService {
    @Autowired
    private UserDao userDao;
    @Autowired
    private TransactionUtils transactionUtils;

    public void add() {
        TransactionStatus transactionStatus = null;
        try {
            transactionStatus = transactionUtils.begin();
            userDao.add("wangmazi", 27);
            int i = 1 / 0;
            System.out.println("我是add方法");
            userDao.add("zhangsan", 16);
            transactionUtils.commit(transactionStatus);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (transactionStatus != null) {
                transactionStatus.rollbackToSavepoint(transactionStatus);
            }
        }

    }

}

AOP技術(shù)封裝手動事務(wù)

@Component
@Aspect
public class AopTransaction {
    @Autowired
    private TransactionUtils transactionUtils;

    // // 異常通知
    @AfterThrowing("execution(* com.itmayiedu.service.UserService.add(..))")
    public void afterThrowing() {
        System.out.println("程序已經(jīng)回滾");
        // 獲取程序當(dāng)前事務(wù) 進(jìn)行回滾
        TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
    }

    // 環(huán)繞通知
    @Around("execution(* com.itmayiedu.service.UserService.add(..))")
    public void around(ProceedingJoinPoint proceedingJoinPoint) throws Throwable {
        System.out.println("開啟事務(wù)");
        TransactionStatus begin = transactionUtils.begin();
        proceedingJoinPoint.proceed();
        transactionUtils.commit(begin);
        System.out.println("提交事務(wù)");
    }
}

使用事務(wù)注意事項

事務(wù)是程序運行如果沒有錯誤,會自動提交事物,如果程序運行發(fā)生異常,則會自動回滾。
如果使用了try捕獲異常時.一定要在catch里面手動回滾。
事務(wù)手動回滾代碼
TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();

2.聲明事務(wù)實現(xiàn)

管理建立在AOP之上的。其本質(zhì)是對方法前后進(jìn)行攔截,然后在目標(biāo)方法開始之前創(chuàng)建或者加入一個事務(wù),在執(zhí)行完目標(biāo)方法之后根據(jù)執(zhí)行情況提交或者回滾事務(wù)。聲明式事務(wù)最大的優(yōu)點就是不需要通過編程的方式管理事務(wù),這樣就不需要在業(yè)務(wù)邏輯代碼中摻雜事務(wù)管理的代碼,只需在配置文件中做相關(guān)的事務(wù)規(guī)則聲明(或通過基于@Transactional注解的方式),便可以將事務(wù)規(guī)則應(yīng)用到業(yè)務(wù)邏輯中。
顯然聲明式事務(wù)管理要優(yōu)于編程式事務(wù)管理,這正是spring倡導(dǎo)的非侵入式的開發(fā)方式。

聲明式事務(wù)管理使業(yè)務(wù)代碼不受污染,一個普通的POJO對象,只要加上注解就可以獲得完全的事務(wù)支持。和編程式事務(wù)相比,聲明式事務(wù)唯一不足地方是,后者的最細(xì)粒度只能作用到方法級別,無法做到像編程式事務(wù)那樣可以作用到代碼塊級別。但是即便有這樣的需求,也存在很多變通的方法,比如,可以將需要進(jìn)行事務(wù)管理的代碼塊獨立為方法等等。

注解版本聲明

<beans xmlns="http://www.springframework.org/schema/beans"
    xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:p="http://www.springframework.org/schema/p"
    xmlns:context="http://www.springframework.org/schema/context"
    xmlns:aop="http://www.springframework.org/schema/aop" xmlns:tx="http://www.springframework.org/schema/tx"
    xsi:schemaLocation="http://www.springframework.org/schema/beans
         http://www.springframework.org/schema/beans/spring-beans.xsd
         http://www.springframework.org/schema/context
         http://www.springframework.org/schema/context/spring-context.xsd
         http://www.springframework.org/schema/aop
         http://www.springframework.org/schema/aop/spring-aop.xsd
         http://www.springframework.org/schema/tx
         http://www.springframework.org/schema/tx/spring-tx.xsd">


    <!-- 開啟注解 -->
    <context:component-scan base-package="com.itmayiedu"></context:component-scan>
    <!-- 1. 數(shù)據(jù)源對象: C3P0連接池 -->
    <bean id="dataSource" class="com.mchange.v2.c3p0.ComboPooledDataSource">
        <property name="driverClass" value="com.mysql.jdbc.Driver"></property>
        <property name="jdbcUrl" value="jdbc:mysql://localhost:3306/test"></property>
        <property name="user" value="root"></property>
        <property name="password" value="root"></property>
    </bean>

    <!-- 2. JdbcTemplate工具類實例 -->
    <bean id="jdbcTemplate" class="org.springframework.jdbc.core.JdbcTemplate">
        <property name="dataSource" ref="dataSource"></property>
    </bean>

    <!-- 配置事物 -->
    <bean id="dataSourceTransactionManager"
        class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
        <property name="dataSource" ref="dataSource"></property>
    </bean>
    <!-- 開啟注解事物 -->
    <tx:annotation-driven transaction-manager="dataSourceTransactionManager" />
</beans>

用法

    @Transactional
    public void add() {
        userDao.add("wangmazi", 27);
        int i = 1 / 0;
        System.out.println("我是add方法");
        userDao.add("zhangsan", 16);
    }

手寫Spring注解版本事務(wù)

Jdk1.5新增新技術(shù),注解。很多框架為了簡化代碼,都會提供有些注解??梢岳斫鉃椴寮?,是代碼級別的插件,在類的方法上寫:@XXX,就是在代碼上插入了一個插件。
注解不會也不能影響代碼的實際邏輯,僅僅起到輔助性的作用。

注解分類:內(nèi)置注解(也成為元注解 jdk 自帶注解)、自定義注解(Spring框架)

什么是內(nèi)置注解?

  • (1) @SuppressWarnings 再程序前面加上可以在javac編譯中去除警告--階段是SOURCE
  • (2) @Deprecated 帶有標(biāo)記的包,方法,字段說明其過時----階段是SOURCE
  • (3)@Overricle 打上這個標(biāo)記說明該方法是將父類的方法重寫--階段是SOURCE

實現(xiàn)自定義注解

元注解的作用就是負(fù)責(zé)注解其他注解。Java5.0定義了4個標(biāo)準(zhǔn)的meta-annotation類型,它們被用來提供對其它 annotation類型作說明。Java5.0定義的元注解: @Target

@Target說明了Annotation所修飾的對象范圍:Annotation可被用于 packages、types(類、接口、枚舉、Annotation類型)、類型成員(方法、構(gòu)造方法、成員變量、枚舉值)、方法參數(shù)和本地變量(如循環(huán)變量、catch參數(shù))。在Annotation類型的聲明中使用了target可更加明晰其修飾的目標(biāo)。

  1. CONSTRUCTOR:用于描述構(gòu)造器

  2. FIELD:用于描述域

  3. LOCAL_VARIABLE:用于描述局部變量

  4. METHOD:用于描述方法

  5. PACKAGE:用于描述包

  6. PARAMETER:用于描述參數(shù)

  7. TYPE:用于描述類、接口(包括注解類型) 或enum聲明

2.@Retention
表示需要在什么級別保存該注釋信息,用于描述注解的生命周期(即:被描述的注解在什么范圍內(nèi)有效)

3.@Documented

4.@Inherited

使用@interface 定義注解。

@Target(value = { ElementType.METHOD })
@Retention(RetentionPolicy.RUNTIME)
public @interface AddAnnotation {

    int userId() default;

    String userName() default "默認(rèn)名稱";

    String[]arrays();
}
// 反射讀取注解信息
    public static void main(String[] args) throws ClassNotFoundException {
        Class classInfo = Class.forName("com.itmayiedu.entity.User");
        // 獲取到所有方法
        Method[] methods = classInfo.getDeclaredMethods();
        for (Method method : methods) {
            System.out.println(method);
            AddAnnotation declaredAnnotation = method.getDeclaredAnnotation(AddAnnotation.class);
            if (declaredAnnotation == null) {
                // 結(jié)束本次循環(huán)
                continue;
            }
            // 獲取userId
            int userId = declaredAnnotation.userId();
            System.out.println("userId:" + userId);
            // 獲取userName
            String userName = declaredAnnotation.userName();
            System.out.println("userName:" + userName);
            // 獲取arrays
            String[] arrays = declaredAnnotation.arrays();
            for (String str : arrays) {
                System.out.println("str:" + str);
            }
        }
    }

自定義事務(wù)注解(重點)

//編程事務(wù)(需要手動begin 手動回滾  手都提交)
@Component()
@Scope("prototype") // 設(shè)置成原型解決線程安全
public class TransactionUtils {

    private TransactionStatus transactionStatus;
    // 獲取事務(wù)源
    @Autowired
    private DataSourceTransactionManager dataSourceTransactionManager;

    // 開啟事務(wù)
    public TransactionStatus begin() {
        transactionStatus = dataSourceTransactionManager.getTransaction(new DefaultTransactionAttribute());
        return transactionStatus;
    }

    // 提交事務(wù)
    public void commit(TransactionStatus transaction) {
        dataSourceTransactionManager.commit(transaction);
    }

    // 回滾事務(wù)
    public void rollback() {
        System.out.println("rollback");
        dataSourceTransactionManager.rollback(transactionStatus);
    }

}

//注解類

@Autowired
    private TransactionUtils transactionUtils;

    @AfterThrowing("execution(* com.itmayiedu.service.*.*.*(..))")
    public void afterThrowing() throws NoSuchMethodException, SecurityException {
        // isRollback(proceedingJoinPoint);
        System.out.println("程序發(fā)生異常");
        // TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
        // TransactionStatus currentTransactionStatus =
        // TransactionAspectSupport.currentTransactionStatus();
        // System.out.println("currentTransactionStatus:" +
        // currentTransactionStatus);
        transactionUtils.rollback();
    }

    // // 環(huán)繞通知 在方法之前和之后處理事情
    @Around("execution(* com.itmayiedu.service.*.*.*(..))")
    public void around(ProceedingJoinPoint proceedingJoinPoint) throws Throwable {

        // 調(diào)用方法之前執(zhí)行
        TransactionStatus transactionStatus = begin(proceedingJoinPoint);
        proceedingJoinPoint.proceed();// 代理調(diào)用方法 注意點: 如果調(diào)用方法拋出異常不會執(zhí)行后面代碼
        // 調(diào)用方法之后執(zhí)行
        commit(transactionStatus);
    }

    public TransactionStatus begin(ProceedingJoinPoint pjp) throws NoSuchMethodException, SecurityException {

        // // 判斷是否有自定義事務(wù)注解
        ExtTransaction declaredAnnotation = getExtTransaction(pjp);
        if (declaredAnnotation == null) {
            return null;
        }
        // 如果有自定義事務(wù)注解,開啟事務(wù)
        System.out.println("開啟事務(wù)");
        TransactionStatus transactionStatu = transactionUtils.begin();
        return transactionStatu;
    }

    public void commit(TransactionStatus transactionStatu) {
        if (transactionStatu != null) {
            // 提交事務(wù)
            System.out.println("提交事務(wù)");
            transactionUtils.commit(transactionStatu);
        }
    }

    public ExtTransaction getExtTransaction(ProceedingJoinPoint pjp) throws NoSuchMethodException, SecurityException {
        // 獲取方法名稱
        String methodName = pjp.getSignature().getName();
        // 獲取目標(biāo)對象
        Class<?> classTarget = pjp.getTarget().getClass();
        // 獲取目標(biāo)對象類型
        Class<?>[] par = ((MethodSignature) pjp.getSignature()).getParameterTypes();
        // 獲取目標(biāo)對象方法
        Method objMethod = classTarget.getMethod(methodName, par);
        // // 判斷是否有自定義事務(wù)注解
        ExtTransaction declaredAnnotation = objMethod.getDeclaredAnnotation(ExtTransaction.class);
        if (declaredAnnotation == null) {
            System.out.println("您的方法上,沒有加入注解!");
            return null;
        }
        return declaredAnnotation;

    }

    // 回滾事務(wù)
    public void isRollback(ProceedingJoinPoint pjp) throws NoSuchMethodException, SecurityException {
        // // 判斷是否有自定義事務(wù)注解
        ExtTransaction declaredAnnotation = getExtTransaction(pjp);
        if (declaredAnnotation != null) {
            System.out.println("已經(jīng)開始回滾事務(wù)");
            // 獲取當(dāng)前事務(wù) 直接回滾
            TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
            return;
        }
    }

  //使用自定義注解
  @ExtTransaction
  public void add() {
      userDao.add("test001", 20);
      int i = 1 / 0;
      System.out.println("################");
      userDao.add("test002", 21);
  }

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。
禁止轉(zhuǎn)載,如需轉(zhuǎn)載請通過簡信或評論聯(lián)系作者。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容