Spring编程式事务

TransactionWrap

import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.annotation.Propagation;
import org.springframework.transaction.support.TransactionTemplate;

import java.util.function.Supplier;

@Component
@RequiredArgsConstructor
public class TransactionWrap {

    private final PlatformTransactionManager platformTransactionManager;

    /**
     * 执行目标代码使用指定的事务传播级别，运行方法无返回值
     */
    public void doInTransaction(Propagation propagation, Runnable runnable) {
        TransactionTemplate template = new TransactionTemplate(platformTransactionManager);
        template.setPropagationBehavior(propagation.value());
        template.executeWithoutResult(run -> {
            runnable.run();
        });
    }

    /**
     * 执行目标代码使用指定的事务传播级别，运行方法有返回值
     */
    public <T> T doInTransaction(Propagation propagation, Supplier<T> supplier) {
        TransactionTemplate template = new TransactionTemplate(platformTransactionManager);
        template.setPropagationBehavior(propagation.value());
        return template.execute((status) -> supplier.get());
    }

}