com.gitee.starblues.loader.launcher.runner.MethodRunner Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-brick-loader Show documentation
加载插件模块
There is a newer version: 3.1.2
/**
 * Copyright [2019-Present] [starBlues]
 *
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *
 *        http://www.apache.org/licenses/LICENSE-2.0
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */

package com.gitee.starblues.loader.launcher.runner;

import com.gitee.starblues.loader.utils.CompareClassTypeUtils;
import com.gitee.starblues.loader.utils.ObjectUtils;

import java.lang.reflect.Method;
import java.util.Objects;

/**
 * 反射运行方法
 *
 * @author starBlues
 * @since 3.0.0
 * @version 3.0.0
 */
public class MethodRunner {

    protected final String className;
    protected final String runMethodName;

    protected String[] args;

    public MethodRunner(String className, String runMethodName, String[] args) {
        this.className = checkEmpty(className, "className 不能为空");
        this.runMethodName = checkEmpty(runMethodName, "runMethod 不能为空");
        this.args = (args != null) ? args.clone() : null;
    }

    public Object run() throws Exception {
        return run(null);
    }

    public Object run(ClassLoader classLoader) throws Exception {
        Class runClass = loadRunClass(classLoader);
        return runMethod(runClass);
    }

    protected Class loadRunClass(ClassLoader classLoader) throws Exception{
        if(classLoader == null){
            classLoader = Thread.currentThread().getContextClassLoader();
        }
        return Class.forName(this.className, false, classLoader);
    }

    protected Object runMethod(Class runClass) throws Exception {
        Method runMethod = findRunMethod(runClass);
        if(runMethod == null) {
            throw new NoSuchMethodException(runClass.getName() + "." + runMethodName + "(String[] args)");
        }
        Object instance = getInstance(runClass);
        runMethod.setAccessible(true);
        runMethod.invoke(instance, new Object[] { this.args });
        return instance;
    }

    protected Object getInstance(Class runClass) throws Exception {
        return runClass.getConstructor().newInstance();
    }

    private String checkEmpty(String value, String msg){
        if(ObjectUtils.isEmpty(value)){
            throw new IllegalArgumentException(msg);
        }
        return value;
    }

    private Method findRunMethod(Class runClass){
        Class searchType = runClass;
        Class[] argClasses = new Class[]{ String[].class };
        while (searchType != null) {
            Method[] methods = searchType.isInterface() ? searchType.getMethods() : searchType.getDeclaredMethods();
            for (Method method : methods) {
                if(!Objects.equals(method.getName(), runMethodName)){
                    continue;
                }
                if(hasSameParams(method, argClasses)){
                    return method;
                }
            }
            searchType = searchType.getSuperclass();
        }
        return null;
    }

    private boolean hasSameParams(Method method, Class[] paramTypes) {
        if(paramTypes.length != method.getParameterCount()){
            return false;
        }
        Class[] parameterTypes = method.getParameterTypes();
        for (int i = 0; i < paramTypes.length; i++) {
            Class paramType = paramTypes[i];
            Class methodParamType = parameterTypes[i];
            if(CompareClassTypeUtils.compare(methodParamType, paramType)){
                return true;
            }
        }
        return false;
    }

}