实战之jvm-sandbox动态加载插件实现

在实际应用中,当我们某些功能点开发完成的时候,需要重启部署才能够让功能得到应用。但这个功能比较适合插件开发,将功能拆分成一个个独立的jar来提供功能点的拆组。

简单场景

假设我们现在有发短信和发送邮件的功能,这个时候我们需要再加一个发送微信或者钉钉消息的功能。

我们希望这两部分对接第三方的功能插件式开发,分别是两个独立的jar,各自负责各自的功能。

在开发完成之后,无需重启应用,直接放在特定的位置,让应用直接去刷新加载这两个jar就行了。

实际上确实有方法,最近开发jvm-sandbox的时候,发现它就有一个这样的功能。

它是如何去做的呢?

实现思路

  1. 插件jar开发完成之后,直接放到特定的位置。
  2. 应用程序去特定的位置读取jar
  3. 通过classload去加载jar中的类
  4. 通过SPI的方式去找特定的接口,并加入到应用容器中。

实现方案

实例对象版本

给定一个jar的路径,然后去扫描以jar结尾的包路径。

import com.google.common.collect.Lists;
import com.lkx.jvm.sandbox.core.classloader.ManagerClassLoader;
import com.lkx.jvm.sandbox.core.compoents.DefaultInjectResource;
import com.lkx.jvm.sandbox.core.compoents.GroupContainerHelper;
import com.lkx.jvm.sandbox.core.compoents.InjectResource;
import com.lkx.jvm.sandbox.core.util.FileUtils;
import com.sandbox.manager.api.Components;
import com.sandbox.manager.api.PluginModule;
import org.apache.commons.lang3.reflect.FieldUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.annotation.Resource;
import java.io.File;
import java.io.IOException;
import java.lang.reflect.Field;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.*;

/**
 * 插件实例加载工厂
 *
 * @author liukaixiong
 * @Email liukx@elab-plus.com
 * @date 2021/12/7 - 18:36
 */
public class JarFactory {
    private Logger log = LoggerFactory.getLogger(getClass());
    private final static String JAR_FILE_SUFFIX = ".jar";
//    private InjectResource injectResource;
    private URLClassLoader urlClassLoader;

    public JarFactory(String jarFilePath) {
        File file = new File(jarFilePath);
        if (!file.exists()) {
            throw new IllegalArgumentException("jar file does not exist, path=" + jarFilePath);
        }
        final URL[] urLs = getURLs(jarFilePath);
        if (urLs.length == 0) {
            throw new IllegalArgumentException("does not have any available jar in path:" + jarFilePath);
        }
//        this.injectResource = new DefaultInjectResource();
        this.urlClassLoader = new URLClassLoader(urLs, this.getClass().getClassLoader());
    }

    /**
     * 允许自定义
     *
     * @param injectResource
     */
//    public void setInjectResource(InjectResource injectResource) {
//        this.injectResource = injectResource;
//    }

    /**
     * 获取对应的插件模块
     *
     * @return
     */
    public List<Components> getComponents() {
        return loadObjectList(Components.class);
    }

    public void loadComponents() {
        loadObjectList(Components.class);
    }

    /**
     * 加载对应的实例对象
     *
     * @param clazz
     * @param <T>
     * @return
     */
    public <T> List<T> loadObjectList(Class<T> clazz) {
        List<T> objList = new ArrayList<>();
        // 基于SPI查找
        final ServiceLoader<T> moduleServiceLoader = ServiceLoader.load(clazz, this.urlClassLoader);

        final Iterator<T> moduleIt = moduleServiceLoader.iterator();
        while (moduleIt.hasNext()) {

            final T module;
            try {
                module = moduleIt.next();
            } catch (Throwable cause) {
                log.error("error load jar", cause);
                continue;
            }

            final Class<?> classOfModule = module.getClass();

            // 如果有注入对象
//            if (injectResource != null) {
//                for (final Field resourceField : FieldUtils.getFieldsWithAnnotation(classOfModule, Resource.class)) {
//                    final Class<?> fieldType = resourceField.getType();
//                    Object fieldObject = injectResource.getFieldValue(fieldType);
//                    if (fieldObject != null) {
//                        try {
//                            FieldUtils.writeField(
//                                    resourceField,
//                                    module,
//                                    fieldObject,
//                                    true
//                            );
//                        } catch (Exception e) {
//                            log.warn(" set Value error : " + e.getMessage());
//                        }
//                    }
//                }
//                injectResource.afterProcess(module);
//            }
            objList.add(module);
        }
        return objList;
    }

    /**
     * 获取模块jar的urls
     *
     * @param jarFilePath 插件路径
     * @return 插件URL列表
     */
    private URL[] getURLs(String jarFilePath) {
        File file = new File(jarFilePath);
        List<URL> jarPaths = Lists.newArrayList();
        if (file.isDirectory()) {
            File[] files = file.listFiles();
            if (files == null) {
                return jarPaths.toArray(new URL[0]);
            }
            for (File jarFile : files) {
                if (isJar(jarFile)) {
                    try {
                        File tempFile = File.createTempFile("manager_plugin", ".jar");
                        tempFile.deleteOnExit();
                        FileUtils.copyFile(jarFile, tempFile);
                        jarPaths.add(new URL("file:" + tempFile.getPath()));
                    } catch (IOException e) {
                        log.error("error occurred when get jar file", e);
                    }
                } else {
                    jarPaths.addAll(Arrays.asList(getURLs(jarFile.getAbsolutePath())));
                }
            }
        } else if (isJar(file)) {
            try {
                File tempFile = File.createTempFile("manager_plugin", ".jar");
                FileUtils.copyFile(file, tempFile);
                jarPaths.add(new URL("file:" + tempFile.getPath()));
            } catch (IOException e) {
                log.error("error occurred when get jar file", e);
            }
            return jarPaths.toArray(new URL[0]);
        } else {
            log.error("plugins jar path has no available jar, use empty url, path={}", jarFilePath);
        }
        return jarPaths.toArray(new URL[0]);
    }

    /**
     * @param file
     * @return
     */
    private boolean isJar(File file) {
        return file.isFile() && file.getName().endsWith(JAR_FILE_SUFFIX);
    }

    public static void main(String[] args) {
        String jarFile = "E:\\study\\sandbox\\sandbox-module\\";

        JarFactory factory = new JarFactory(jarFile);
        List<Components> components = factory.getComponents();

        System.out.println(components);

    }
}

这只是一个实例版本的,如果还想基于属性注入的话,可以将注释那块解开。

以上的案例是基于Components接口来 扫描的,需要jar中定义META-INF\services\com.sandbox.manager.api._Components_ 中的实现类。比如

com.sandbox.application.plugin.cat.CatTransactionModule
com.sandbox.application.plugin.cat.listener.LogAdviceListener

你如果嫌麻烦可以使用kohsuke包,只需在类上要定义:(注意还需要实现该接口),无需手动去创建文件和实现。

@MetaInfServices(Components.class)
public class LogAdviceListener implements Components {

}

pom文件引入:

<dependency>
  <groupId>org.kohsuke.metainf-services</groupId>
  <artifactId>metainf-services</artifactId>
  <version>1.7</version>
  <scope>compile</scope>
</dependency>

属性注入

  1. 定义注入的接口规范
/**
 * 注入资源对象
 *
 * @author liukaixiong
 * @Email liukx@elab-plus.com
 * @date 2021/12/7 - 16:23
 */
public interface InjectResource {

    /**
     * 获取注入对象
     *
     * @param resourceField
     * @return
     */
    public Object getFieldValue(Class<?> resourceField);

    /**
     * 实例对象被返回的处理
     *
     * @param obj
     */
    public void afterProcess(Object obj);

}
  1. 基于一个默认实现

GroupContainerHelper 你可以理解为一个Map,前提是属性的对象在Map中存在,存在则将对象赋值出去

/**
 * 默认注入工厂
 *
 * @author liukaixiong
 * @Email liukx@elab-plus.com
 * @date 2021/12/8 - 13:40
 */
public class DefaultInjectResource implements InjectResource {

    @Override
    public Object getFieldValue(Class<?> resourceField) {
        return GroupContainerHelper.getInstance().getObject(resourceField);
    }

    @Override
    public void afterProcess(Object obj) {
        Class<?> clazz = obj.getClass();
        
        // 分析类模型,将类分组保存关系
        builderObjectCache(clazz, obj);

        GroupContainerHelper.getInstance().registerObject(obj);
    }

    
    public void builderObjectCache(Class<?> clazz, Object obj) {

        if (clazz == Object.class) {
            return;
        }

        GroupContainerHelper.getInstance().registerList(clazz, obj);

        Class<?>[] interfaces = clazz.getInterfaces();
        // 将接口类进行分组
        if (interfaces.length > 0) {
            for (int i = 0; i < interfaces.length; i++) {
                Class<?> anInterface = interfaces[i];
                GroupContainerHelper.getInstance().registerList(anInterface, obj);
            }
        }

        builderObjectCache(clazz.getSuperclass(), obj);
    }

}

功能差不多就这样实现的,如果是Spring的话,可以使用工厂解析SPI扫描到的类。

当然啦,后续的实现你想怎么玩都行。

至于怎么已经加载过的包或者刷新等功能本文就不过多赘述。

如果你有好的方式也可以留言交流喔。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,463评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,868评论 3 391
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 161,213评论 0 351
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,666评论 1 290
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,759评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,725评论 1 294
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,716评论 3 415
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,484评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,928评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,233评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,393评论 1 345
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,073评论 5 340
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,718评论 3 324
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,308评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,538评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,338评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,260评论 2 352

推荐阅读更多精彩内容