大家好，又见面了，我是你们的朋友全栈君。

基于springboot自定义事务注解

1.开启注解支持(springboot默认支持注解)
2.自定义注解接口
3.写事务类
4.写切面类

2.自定义注解接口

import java.lang.annotation.*;


/** * 注解类 */
@Target(ElementType.METHOD)             //定义注解用在方法上
@Retention(RetentionPolicy.RUNTIME)     //运行时注解
@Documented
public @interface CustomTransaction { 
   
    String value() default "";
}

3.写事务类

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Scope;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.stereotype.Component;
import org.springframework.transaction.TransactionStatus;
import org.springframework.transaction.interceptor.DefaultTransactionAttribute;

@Component
@Scope(value = "prototype")
public class TransactionUtils { 
   


    @Autowired
    private DataSourceTransactionManager dataSourceTransactionManager;

    /** *初始化创建TransactionStatus对象 * @return */
    public TransactionStatus init(){ 
   
        System.out.println("创建事务了...");
        TransactionStatus transactionStatus = dataSourceTransactionManager.getTransaction(new DefaultTransactionAttribute());
        return transactionStatus;
    }

    /** * 提交事务 * @param transactionStatus */
    public void commit(TransactionStatus transactionStatus){ 
   
        System.out.println("提交事务...");
        dataSourceTransactionManager.commit(transactionStatus);
    }


    public void rollback(TransactionStatus transactionStatus){ 
   
        System.out.println("事务回滚了....");
        dataSourceTransactionManager.rollback(transactionStatus);
    }
}

4.写切面类

import io.tianma.common.annotation.CustomTransaction;
import io.tianma.common.utils.TransactionUtils;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.transaction.TransactionStatus;
import java.lang.reflect.Method;
@Aspect
@Component
public class CustomTransactionAspect { 

@Autowired
TransactionUtils transactionUtils;
/** * 选择切面的注解CustomTransaction */
@Pointcut("@annotation(io.tianma.common.annotation.CustomTransaction)")
public void transactionPointCut() { 

}
/** * 方法增强@Arounbd * @param point */
@Around("transactionPointCut()")
public void around(ProceedingJoinPoint point){ 

MethodSignature signature = (MethodSignature) point.getSignature();
Method method = signature.getMethod();//获取到方法对象
//获取到注解类
CustomTransaction annotation = method.getAnnotation(CustomTransaction.class);
if(annotation != null){ 

System.out.println(annotation.value());//打印注解上value的内容
}
//请求的类名以及方法名
String className = point.getTarget().getClass().getName();
String methodName = signature.getName();
System.out.println("执行的方法为:" + className + "." + methodName + "()");
//开启事务
TransactionStatus transactionStatus = transactionUtils.init();
try { 

//执行方法
point.proceed();
//执行成功提交事务
transactionUtils.commit(transactionStatus);
} catch (Throwable throwable) { 

//执行方法出错则回滚事务
transactionUtils.rollback(transactionStatus);
}
}
}

5.项目service

    @Override
@CustomTransaction("新增用户开启事务")
public void saveUser() { 

SysUserEntity user = new SysUserEntity();
user.setUserId("D123456");
user.setName("张三");
user.setCreateUserId("D123456");
user.setCreateTime(new Date());
this.save(user);
// System.out.println(1/0);
SysUserRoleEntity sysUserRoleEntity = new SysUserRoleEntity();
sysUserRoleEntity.setRoleId(123456L);
sysUserRoleEntity.setUserId("123456");
sysUserRoleService.save(sysUserRoleEntity);
}

6.先查询数据库

查询数据库没有数据

7. 调用api

可以看出控制台已经执行成功
再查询一次数据库


可以看到已经有一条记录产生，并且切面也已经执行了

8.打开注释(System.out.println(1/0))，并且将user_id改为D1234567

SysUserEntity user = new SysUserEntity();
user.setUserId("D1234567");
user.setName("张三");
user.setCreateUserId("D1234567");
user.setCreateTime(new Date());
this.save(user);
System.out.println(1/0);
SysUserRoleEntity sysUserRoleEntity = new SysUserRoleEntity();
sysUserRoleEntity.setRoleId(1234567L);
sysUserRoleEntity.setUserId("1234567");
sysUserRoleService.save(sysUserRoleEntity);

可以看出事务进行回滚了，再查询一次数据库

可以看到报错后事务并没有进行提交，而是进行了回滚

结尾

自此基于springboot的自定义事务注解已完成