org.zodiac.plugin.integration.pf4j.DefaultPf4jFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of zodiac-plugin Show documentation
Zodiac Plugin
There is a newer version: 1.6.8
package org.zodiac.plugin.integration.pf4j;

import org.pf4j.*;
import org.zodiac.plugin.integration.IntegrationConfiguration;
import org.zodiac.plugin.integration.pf4j.descriptor.ManifestPluginDescriptorFinderExtend;
import org.zodiac.plugin.integration.pf4j.descriptor.ResolvePropertiesPluginDescriptorFinder;
import org.zodiac.plugin.integration.pf4j.descriptor.ResourcesPluginDescriptorFinder;

import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.List;
import java.util.stream.Collectors;

/**
 * 默认的插件集成工厂。
 */
public class DefaultPf4jFactory implements Pf4jFactory {

    private final IntegrationConfiguration configuration;

    public DefaultPf4jFactory(IntegrationConfiguration configuration) {
        this.configuration = configuration;
    }

    @Override
    public PluginManager getPluginManager() {
        if (configuration == null) {
            throw new NullPointerException("IntegrationConfiguration is null");
        }
        RuntimeMode environment = configuration.environment();
        if (environment == null) {
            throw new RuntimeException("Configuration RuntimeMode is null" + configuration.environment());
        }
        List sortInitPluginIds = configuration.sortInitPluginIds();
        DefaultPluginManager defaultPluginManager = null;

        List pluginDir = configuration.pluginPath();
        List pluginDirPath = pluginDir.stream().map(Paths::get).collect(Collectors.toList());

        if (RuntimeMode.DEVELOPMENT == environment) {
            // 开发环境下的插件管理者
            defaultPluginManager = new DefaultPluginManager(pluginDirPath) {

                @Override
                protected void initialize() {
                    super.initialize();
                    dependencyResolver = new SortDependencyResolver(sortInitPluginIds, versionManager);
                }

                @Override
                public RuntimeMode getRuntimeMode() {
                    System.setProperty("pf4j.mode", RuntimeMode.DEVELOPMENT.toString());
                    return RuntimeMode.DEVELOPMENT;
                }

                @Override
                protected PluginDescriptorFinder createPluginDescriptorFinder() {
                    return DefaultPf4jFactory.getPluginDescriptorFinder(RuntimeMode.DEVELOPMENT);
                }

                @Override
                protected PluginLoader createPluginLoader() {
                    return new CompoundPluginLoader().add(new DevelopmentPluginLoader(this), this::isDevelopment);
                }

                @Override
                protected PluginStatusProvider createPluginStatusProvider() {
                    return new ConfigPluginStatusProvider(configuration.enablePluginIds(),
                        configuration.disablePluginIds());
                }

                @Override
                public PluginState stopPlugin(String pluginId) {
                    return stopPlugin(pluginId, configuration.stopDependents());
                }
            };
        } else if (RuntimeMode.DEPLOYMENT == environment) {
            // 运行环境下的插件管理者
            defaultPluginManager = new DefaultPluginManager(pluginDirPath) {

                @Override
                protected void initialize() {
                    super.initialize();
                    dependencyResolver = new SortDependencyResolver(sortInitPluginIds, versionManager);
                }

                @Override
                protected PluginDescriptorFinder createPluginDescriptorFinder() {
                    return DefaultPf4jFactory.getPluginDescriptorFinder(RuntimeMode.DEPLOYMENT);
                }

                @Override
                protected PluginStatusProvider createPluginStatusProvider() {
                    return new ConfigPluginStatusProvider(configuration.enablePluginIds(),
                        configuration.disablePluginIds());
                }

                @Override
                protected PluginLoader createPluginLoader() {
                    return new CompoundPluginLoader().add(new JarPluginLoader(this), this::isNotDevelopment)
                        .add(new DefaultPluginLoader(this), this::isNotDevelopment);
                }

                @Override
                public PluginState stopPlugin(String pluginId) {
                    return stopPlugin(pluginId, configuration.stopDependents());
                }

            };
        }
        if (defaultPluginManager == null) {
            throw new RuntimeException("Not found run environment " + configuration.environment());
        }
        defaultPluginManager.setSystemVersion(configuration.version());
        defaultPluginManager.setExactVersionAllowed(configuration.exactVersionAllowed());
        return defaultPluginManager;
    }

    public static PluginDescriptorFinder getPluginDescriptorFinder(RuntimeMode runtimeMode) {
        if (runtimeMode == RuntimeMode.DEPLOYMENT) {
            // 生产
            return new CompoundPluginDescriptorFinder().add(new ResourcesPluginDescriptorFinder(runtimeMode))
                .add(new ManifestPluginDescriptorFinderExtend());
        } else {
            // 开发
            return new CompoundPluginDescriptorFinder().add(new ResourcesPluginDescriptorFinder(runtimeMode))
                .add(new ResolvePropertiesPluginDescriptorFinder()).add(new ManifestPluginDescriptorFinderExtend());
        }
    }

}