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

Go to download
package com.jeesuite.mybatis.plugin;

import java.util.ArrayList;
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.springframework.beans.factory.DisposableBean;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.util.StringUtils;

import com.jeesuite.mybatis.core.InterceptorHandler;
import com.jeesuite.mybatis.core.InterceptorType;
import com.jeesuite.mybatis.parser.MybatisMapperParser;
import com.jeesuite.mybatis.plugin.cache.CacheHandler;
import com.jeesuite.mybatis.plugin.rwseparate.RwRouteHandler;
import com.jeesuite.mybatis.plugin.shard.DatabaseRouteHandler;

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

 * @author vakin
 * @date 2015年12月7日
 * @Copyright (c) 2015, jwww
 */
@Intercepts({  
    @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 JeesuiteMybatisPluginContext implements Interceptor,InitializingBean,DisposableBean{

	//CRUD框架驱动 default，mapper3
	private String crudDriver = "default";
	private List interceptorHandlers = new ArrayList<>();
	
	//cache,rwRoute,dbShard
	public void setInterceptorHandlers(String interceptorHandlers) {
		String[] handlerNames = StringUtils.tokenizeToStringArray(interceptorHandlers, ConfigurableApplicationContext.CONFIG_LOCATION_DELIMITERS);
	    
		for (String name : handlerNames) {
			if("cache".equals(name)){
				this.interceptorHandlers.add(new CacheHandler());
			}else if("rwRoute".equals(name)){
				this.interceptorHandlers.add(new RwRouteHandler());
			}else if("dbShard".equals(name)){
				this.interceptorHandlers.add(new DatabaseRouteHandler());
			}
		}
	}
	
	public void setMapperLocations(String mapperLocations){
		MybatisMapperParser.setMapperLocations(mapperLocations);
	}

	public void setCrudDriver(String crudDriver) {
		this.crudDriver = crudDriver;
	}

	public String getCrudDriver() {
		return crudDriver;
	}

	@Override
	public Object intercept(Invocation invocation) throws Throwable {
		
		boolean proceed = false;
		Object result = null;
		try {
			for (InterceptorHandler handler : interceptorHandlers) {
				Object object = handler.onInterceptor(invocation);
				if(handler.getInterceptorType().equals(InterceptorType.around)){
					result = object;
					//查询缓存命中，则不执行分库和读写分离的处理器
					if(result != null && handler instanceof CacheHandler){
						break;
					}
				}
			}
			if(result == null){
				result = invocation.proceed();
				proceed = true;
			}
			return result;
		} finally {
			for (InterceptorHandler handler : interceptorHandlers) {
				handler.onFinished(invocation,proceed ? result : null);
			}
		}
	}

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


	}

	@Override
	public void setProperties(Properties properties) {}


	@Override
	public void afterPropertiesSet() throws Exception {
		for (InterceptorHandler handler : interceptorHandlers) {
			handler.start(this);
		}
	}

	@Override
	public void destroy() throws Exception {
		for (InterceptorHandler handler : interceptorHandlers) {
			handler.close();
		}
	}

}