chao.java.tools.servicepool.cache.AbsServiceCacheStrategy Maven / Gradle / Ivy

Go to download
package chao.java.tools.servicepool.cache;

import chao.java.tools.servicepool.IService;
import chao.java.tools.servicepool.IServiceInterceptor;
import chao.java.tools.servicepool.IServiceInterceptorCallback;
import chao.java.tools.servicepool.ServiceInterceptorStrategy;
import chao.java.tools.servicepool.ServicePool;
import chao.java.tools.servicepool.combine.CombineService;
import com.luqinx.interceptor.Interceptor;
import com.luqinx.interceptor.OnInvoke;
import java.lang.reflect.Method;

public abstract class AbsServiceCacheStrategy implements ServiceCacheStrategy {

    private static ServiceInterceptorStrategy strategy = new ServiceInterceptorStrategy();

    private boolean disableIntercept;

    public void setDisableIntercept(boolean disableIntercept) {
        this.disableIntercept = disableIntercept;
    }


    protected T getProxyService(final Class originClazz, final T instance) {
        if (disableIntercept) {
            return instance;
        }
        if (originClazz != null && originClazz.isInterface()) {
            return Interceptor.of(instance, originClazz).intercepted(true).invoke(new OnInvoke() {

                @Override
                public Object onInvoke(T source, final Method method, final Object[] args) {
                    final ResultHolder holder = new ResultHolder();

                    CombineService combineService = (CombineService) ServicePool.getCombineService(IServiceInterceptor.class, strategy);
                    if (combineService.size() == 0) {
                        try {
                            holder.result = method.invoke(instance, args);
                        } catch (Throwable e) {
                            if (ServicePool.exceptionHandler != null) {
                                ServicePool.exceptionHandler.onException(e, e.getMessage());
                            }
                            e.printStackTrace();
                        }
                    } else {
                        IServiceInterceptor interceptor = (IServiceInterceptor) combineService;
                        interceptor.intercept(originClazz, source, method, args, new IServiceInterceptorCallback() {
                            @Override
                            public void onContinue(Method interceptorMethod, Object... interceptorArgs) {
                                try {
                                    holder.result = method.invoke(instance, args);
                                } catch (Throwable e) {
                                    if (ServicePool.exceptionHandler != null) {
                                        ServicePool.exceptionHandler.onException(e, e.getMessage());
                                    }
                                    e.printStackTrace();
                                }
                            }

                            @Override
                            public void onInterrupt(Object lastResult) {
                                holder.result = lastResult;
                            }
                        });
                    }
                    return holder.result;
                }

            }).newInstance();
        } else {
            return instance;
        }
    }

    private class ResultHolder {
        Object result;
    }
}