io.inugami.commons.tools.ProxyBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of inugami_commons Show documentation
There is a newer version: 3.3.5
/* --------------------------------------------------------------------
 *  Inugami
 * --------------------------------------------------------------------
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, version 3.
 *
 * 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
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see .
 */
package io.inugami.commons.tools;

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

import java.lang.reflect.Method;

/**
 * ProxyBuilder
 *
 * @author patrick_guillerm
 * @since 11 janv. 2018
 */
@SuppressWarnings({"java:S1068", "java:S1450"})
public final class ProxyBuilder {

    // =========================================================================
    // ATTRIBUTES
    // =========================================================================
    private final Enhancer enhancer = new Enhancer();

    private Class superClass;

    private Class[] interfaces = null;

    private boolean callbackDefine;

    private final Callback callback = new MethodInterceptor() {
        @Override
        public Object intercept(final Object obj, final Method method,
                                final Object[] args,
                                final MethodProxy proxy) throws Throwable {
            // Nothing to do
            return null;
        }
    };

    // =========================================================================
    // METHODS
    // =========================================================================
    public ProxyBuilder addSuperClass(final Class superClass) {
        enhancer.setSuperclass(superClass);
        return this;
    }

    public ProxyBuilder addCallback(final Callback... callbacks) {
        callbackDefine = true;
        enhancer.setCallbacks(callbacks);
        return this;
    }

    public ProxyBuilder addInterface(final Class... classes) {
        enhancer.setInterfaces(classes);
        this.interfaces = classes;
        return this;
    }

    public T build() {
        if (!callbackDefine) {
            addCallback(callback);
        }
        return (T) enhancer.create();
    }
}