com.xunmo.config.TranAopInterceptor Maven / Gradle / Ivy

Go to download
package com.xunmo.config;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.thread.ThreadUtil;
import com.xunmo.utils.TranAfterUtil;
import lombok.extern.slf4j.Slf4j;
import org.noear.solon.Solon;
import org.noear.solon.SolonProps;
import org.noear.solon.core.aspect.Interceptor;
import org.noear.solon.core.aspect.Invocation;
import org.noear.solon.data.annotation.Tran;
import org.noear.solon.data.annotation.TranAnno;
import org.noear.solon.data.tran.TranUtils;

import java.lang.annotation.Annotation;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.atomic.AtomicReference;

/**
 * 事务aop拦截器
 *
 * @author zengyufei
 * @date 2022/11/19
 */
@Slf4j
public class TranAopInterceptor implements Interceptor {

	private static String appName;

	private static List prefixNameList = new ArrayList<>();

	private static List suffixNameList = new ArrayList<>();

	public TranAopInterceptor() {
		init();
	}

	public void init() {
		final SolonProps cfg = Solon.cfg();
		appName = cfg.get("solon.app.name");
		prefixNameList = cfg.getBean("xm.tran.method.prefix", List.class);
		suffixNameList = cfg.getBean("xm.tran.method.suffix", List.class);
	}

	@Override
	public Object doIntercept(Invocation inv) throws Throwable {
		final Method method = inv.method().getMethod();
		final String methodName = method.getName();
		boolean isHasTran = getIsHasTran(method);
		if (isHasTran) {
			log.info(appName + "应用, 调用 " + methodName + " 已存在事务");
			final Object o = inv.invoke();
			final List afterMethods = TranAfterUtil.getAfterMethods();
			if (CollUtil.isNotEmpty(afterMethods)) {
				for (TranAfterUtil.AfterFunction consumer : afterMethods) {
					consumer.test();
				}
			}
			final List afterSyncMethods = TranAfterUtil.getAfterSyncMethods();
			if (CollUtil.isNotEmpty(afterSyncMethods)) {
				for (TranAfterUtil.AfterFunction consumer : afterSyncMethods) {
					ThreadUtil.execute(consumer::test);
				}
			}
			return o;
		}

		boolean isAddTran = getIsAddTran(methodName);
		if (isAddTran) {
			// 自动给符合的方法名自动进入事务;
			log.info(appName + "应用, 调用 " + methodName + " 方法满足 xm.tran.method 匹配, 自动加事务");
			AtomicReference