CompensableBeanPostProcessor.java

/**
 * Copyright 2014-2017 yangming.liu<bytefox@126.com>.
 *
 * This copyrighted material is made available to anyone wishing to use, modify,
 * copy, or redistribute it subject to the terms and conditions of the GNU
 * Lesser General Public License, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
 * for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this distribution; if not, see <http://www.gnu.org/licenses/>.
 */
package org.bytesoft.bytetcc.supports.spring;

import org.aopalliance.aop.Advice;
import org.bytesoft.compensable.Compensable;
import org.springframework.aop.Advisor;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.BeanPostProcessor;

public class CompensableBeanPostProcessor implements BeanPostProcessor {

	public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
		this.switchAdvisorOrderIfNecessary(bean);
		return bean;
	}

	public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
		this.switchAdvisorOrderIfNecessary(bean);
		return bean;
	}

	private void switchAdvisorOrderIfNecessary(Object bean) {
		if (org.springframework.aop.framework.Advised.class.isInstance(bean) == false) {
			return;
		}

		org.springframework.aop.framework.Advised advised = (org.springframework.aop.framework.Advised) bean;

		Class<?> targetClass = advised.getTargetClass();
		Compensable annotation = targetClass.getAnnotation(Compensable.class);
		if (annotation == null) {
			return;
		}

		Advisor[] advisors = advised.getAdvisors();

		int compensableIndex = -1;
		int transactionIndex = -1;
		for (int i = 0; i < advisors.length; i++) {
			Advisor advisor = advisors[i];
			Advice advice = advisor.getAdvice();
			if (org.bytesoft.bytetcc.supports.spring.CompensableMethodInterceptor.class.isInstance(advice)) {
				compensableIndex = i;
			} else if (org.springframework.transaction.interceptor.TransactionInterceptor.class.isInstance(advice)) {
				transactionIndex = i;
			}
		}

		if (transactionIndex != -1 && compensableIndex != -1 && transactionIndex < compensableIndex) {
			Advisor advisor = advisors[transactionIndex];
			advisors[transactionIndex] = advisors[compensableIndex];
			advisors[compensableIndex] = advisor;
		}

	}

}