All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • com.tsc9526.monalisa.tools.agent.AgentEnhancer Maven / Gradle / Ivy

    There is a newer version: 2.2.0
    Show newest version
    /*******************************************************************************************
 *	Copyright (c) 2016, zzg.zhou([email protected])
 * 
 *  Monalisa is free software: you can redistribute it and/or modify
 *	it under the terms of the GNU Lesser General Public License as published by
 *	the Free Software Foundation, either version 3 of the License, or
 *	(at your option) any later version.

 *	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 program.  If not, see .
 *******************************************************************************************/
package com.tsc9526.monalisa.tools.agent;

import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.List;

import net.sf.cglib.proxy.Enhancer;
import net.sf.cglib.proxy.MethodInterceptor;
import net.sf.cglib.proxy.MethodProxy;

import com.tsc9526.monalisa.orm.annotation.Interceptors;
import com.tsc9526.monalisa.orm.annotation.Tx;
import com.tsc9526.monalisa.tools.Tasks;

/**
 * 
 * @author zzg.zhou([email protected])
 */
public class AgentEnhancer implements MethodInterceptor {

	@SuppressWarnings({"unchecked","rawtypes"})
	public  T createProxyInstance(Class theClass) {
		Enhancer enhancer = new Enhancer() {
			protected void filterConstructors(Class sc, List constructors) {
				// don't filter, so that even classes without
				// visible constructors will work
			}
		};
		
		enhancer.setSuperclass(theClass);
		 
		enhancer.setCallback(new AgentEnhancer());
		 
		return (T) enhancer.create();
	}

	

	public Object intercept(final Object obj, final Method method, final Object[] args, final MethodProxy proxy) throws Throwable {
		Tx tx=method.getAnnotation(Tx.class);
		if (tx!= null) {
			return com.tsc9526.monalisa.orm.Tx.execute(new com.tsc9526.monalisa.orm.Tx.Atom() {
				public Object execute() throws Throwable {
					Object v = doInterceptors(obj,method,args,proxy);
					return v;
				}
			},tx.level());
		} else {
			if(Tasks.instance.isDestoried() && method.getName().equals("finalize")){
				return null;
			}else{
				return doInterceptors(obj,method,args,proxy);
			}
		}
	}
	
	private Object doInterceptors(final Object obj, final Method method, final Object[] args, final MethodProxy proxy)throws Throwable {
		List calls=new ArrayList();
		
		Interceptors interceptors=method.getAnnotation(Interceptors.class);
		if(interceptors!=null){
			for(Class cx:interceptors.value()){
				calls.add(cx.newInstance());
			}
		}
		
		for(Interceptor i:calls){
			i.before(obj, method, args);
		}
		
		Object v = proxy.invokeSuper(obj, args);
		
		for(Interceptor i:calls){
			Object rx=i.after(obj, method, args,v);
			if(rx!=null){
				return rx;
			}
		}
		
		return v;
	}
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy