com.jeesuite.mybatis.plugin.JeesuiteMybatisInterceptor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jeesuite-mybatis Show documentation
There is a newer version: 1.4.0
package com.jeesuite.mybatis.plugin;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.Iterator;
import java.util.List;
import java.util.Properties;

import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.plugin.Interceptor;
import org.apache.ibatis.plugin.Intercepts;
import org.apache.ibatis.plugin.Invocation;
import org.apache.ibatis.plugin.Plugin;
import org.apache.ibatis.plugin.Signature;
import org.apache.ibatis.session.ResultHandler;
import org.apache.ibatis.session.RowBounds;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.DisposableBean;

import com.jeesuite.mybatis.core.InterceptorHandler;
import com.jeesuite.mybatis.plugin.cache.CacheHandler;
import com.jeesuite.mybatis.plugin.dataprofile.DataProfileHandler;
import com.jeesuite.mybatis.plugin.pagination.PaginationHandler;
import com.jeesuite.mybatis.plugin.rwseparate.RwRouteHandler;

/**
 * mybatis 插件入口
 * @description 

 * @author vakin
 * @date 2015年12月7日
 * @Copyright (c) 2015, jwww
 */
@Intercepts({ 
	//@Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class, Integer.class}),
    @Signature(type = Executor.class, method = "update", args = { MappedStatement.class, Object.class }),  
    @Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class }) })  
public class JeesuiteMybatisInterceptor implements Interceptor,DisposableBean{

	protected static final Logger logger = LoggerFactory.getLogger("com.jeesuite.mybatis");
	
	private String groupName;
	private List interceptorHandlers = new ArrayList<>();
	
	private static boolean cacheEnabled,rwRouteEnabled;
	
	public JeesuiteMybatisInterceptor(String groupName, String[] hanlderNames) {
		this.groupName = groupName;
		this.setInterceptorHandlers(hanlderNames);
	}

	private void setInterceptorHandlers(String[] hanlderNames) {
		for (String name : hanlderNames) {
			if(org.apache.commons.lang3.StringUtils.isBlank(name))continue;
			if(CacheHandler.NAME.equals(name)){
				this.interceptorHandlers.add(new CacheHandler());
				cacheEnabled = true;
			}else if(RwRouteHandler.NAME.equals(name)){
				this.interceptorHandlers.add(new RwRouteHandler());
				rwRouteEnabled = true;
			}else if(PaginationHandler.NAME.equals(name)){
				this.interceptorHandlers.add(new PaginationHandler());
			}else if(DataProfileHandler.NAME.equals(name)){
				this.interceptorHandlers.add(new DataProfileHandler());
			}else{
				//自定义的拦截器
				try {
					Class clazz = Class.forName(name);
					InterceptorHandler handler = (InterceptorHandler) clazz.newInstance();
					this.interceptorHandlers.add(handler);
					logger.info("registered cumstom InterceptorHandler:{}",name);
				} catch (Exception e) {
					logger.error("registered cumstom InterceptorHandler error",e);
				}
			}
		}
		
		//排序 分页和数据权限会重写sql，所以不走缓存，所以放在缓存之前
		Collections.sort(this.interceptorHandlers, new Comparator() {
			@Override
			public int compare(InterceptorHandler o1, InterceptorHandler o2) {
				return Integer.compare(o1.interceptorOrder(), o2.interceptorOrder());
			}
		});
		
	}

	@Override
	public Object intercept(Invocation invocation) throws Throwable {
		
		InvocationVals invocationVal = new InvocationVals(invocation);
		
		Object result = null;
		boolean proceed = false;
		try {
			for (InterceptorHandler handler : interceptorHandlers) {
				result = handler.onInterceptor(invocationVal);
				if(result != null) {
					break;
				}
			}
			
			if(result == null){
				result = invocation.proceed();
				proceed = true;
			}

			return result;
		} finally {
			for (InterceptorHandler handler : interceptorHandlers) {
				try {					
					handler.onFinished(invocationVal,proceed ? result : null);
				} catch (Exception e) {
					e.printStackTrace();
				}
			}
		}
		
	}

	@Override
	public Object plugin(Object target) {
		if (target instanceof Executor) {
            return Plugin.wrap(target, this);
        } else {
            return target;
        }


	}

	@Override
	public void setProperties(Properties properties) {}


	public void afterRegister()  {
		Iterator it = interceptorHandlers.iterator();
		while(it.hasNext()){
			InterceptorHandler handler = it.next();
			handler.start(this);
		}
	}

	@Override
	public void destroy() throws Exception {
		for (InterceptorHandler handler : interceptorHandlers) {
			handler.close();
		}
	}
	
	public static boolean isCacheEnabled() {
		return cacheEnabled;
	}

	public static boolean isRwRouteEnabled() {
		return rwRouteEnabled;
	}
	
	public String getGroupName() {
		return groupName;
	}
}