Search in sources :

Example 1 with Reactable

use of io.gravitee.gateway.reactor.Reactable in project gravitee-gateway by gravitee-io.

the class DefaultPolicyManager method initialize.

private void initialize() {
    PolicyPluginManager ppm = applicationContext.getBean(PolicyPluginManager.class);
    PolicyClassLoaderFactory pclf = applicationContext.getBean(PolicyClassLoaderFactory.class);
    ReactorHandler rh = applicationContext.getBean(ReactorHandler.class);
    ResourceLifecycleManager rm = applicationContext.getBean(ResourceLifecycleManager.class);
    Reactable reactable = rh.reactable();
    Set<Policy> requiredPlugins = reactable.dependencies(Policy.class);
    requiredPlugins.forEach(policy -> {
        final PolicyPlugin policyPlugin = ppm.get(policy.getName());
        if (policyPlugin == null) {
            logger.error("Policy [{}] can not be found in policy registry", policy.getName());
            throw new IllegalStateException("Policy [" + policy.getName() + "] can not be found in policy registry");
        }
        PluginClassLoader policyClassLoader = null;
        // Load dependant resources to enhance policy classloader
        Collection<? extends Resource> resources = rm.getResources();
        if (!resources.isEmpty()) {
            ClassLoader[] resourceClassLoaders = rm.getResources().stream().map(new Function<Resource, ClassLoader>() {

                @Override
                public ClassLoader apply(Resource resource) {
                    return resource.getClass().getClassLoader();
                }
            }).toArray(ClassLoader[]::new);
            DelegatingClassLoader parentClassLoader = new DelegatingClassLoader(rh.classloader(), resourceClassLoaders);
            policyClassLoader = pclf.getOrCreateClassLoader(policyPlugin, parentClassLoader);
        } else {
            policyClassLoader = pclf.getOrCreateClassLoader(policyPlugin, rh.classloader());
        }
        logger.debug("Loading policy {} for {}", policy.getName(), rh);
        PolicyMetadataBuilder builder = new PolicyMetadataBuilder();
        builder.setId(policyPlugin.id());
        try {
            // Prepare metadata
            Class<?> policyClass = ClassUtils.forName(policyPlugin.policy().getName(), policyClassLoader);
            builder.setPolicy(policyClass).setMethods(new PolicyMethodResolver().resolve(policyClass));
            if (policyPlugin.configuration() != null) {
                builder.setConfiguration((Class<? extends PolicyConfiguration>) ClassUtils.forName(policyPlugin.configuration().getName(), policyClassLoader));
            }
            // Prepare context if defined
            if (policyPlugin.context() != null) {
                Class<? extends PolicyContext> policyContextClass = (Class<? extends PolicyContext>) ClassUtils.forName(policyPlugin.context().getName(), policyClassLoader);
                // Create policy context instance and initialize context provider (if used)
                PolicyContext context = new PolicyContextFactory(reactable).create(policyContextClass);
                builder.setContext(context);
            }
            RegisteredPolicy registeredPolicy = new RegisteredPolicy();
            registeredPolicy.classLoader = policyClassLoader;
            registeredPolicy.metadata = builder.build();
            policies.put(policy.getName(), registeredPolicy);
        } catch (Exception ex) {
            logger.error("Unable to load policy metadata", ex);
            if (policyClassLoader != null) {
                try {
                    policyClassLoader.close();
                } catch (IOException ioe) {
                    logger.error("Unable to close classloader for policy", ioe);
                }
            }
        }
    });
}
Also used : Policy(io.gravitee.definition.model.Policy) Reactable(io.gravitee.gateway.reactor.Reactable) PolicyClassLoaderFactory(io.gravitee.plugin.policy.PolicyClassLoaderFactory) Function(java.util.function.Function) ReactorHandler(io.gravitee.gateway.reactor.handler.ReactorHandler) PolicyMethodResolver(io.gravitee.plugin.policy.internal.PolicyMethodResolver) PolicyContext(io.gravitee.policy.api.PolicyContext) PluginClassLoader(io.gravitee.plugin.core.api.PluginClassLoader) PluginClassLoader(io.gravitee.plugin.core.api.PluginClassLoader) PolicyPlugin(io.gravitee.plugin.policy.PolicyPlugin) Resource(io.gravitee.resource.api.Resource) IOException(java.io.IOException) IOException(java.io.IOException) ResourceLifecycleManager(io.gravitee.gateway.resource.ResourceLifecycleManager) PolicyPluginManager(io.gravitee.plugin.policy.PolicyPluginManager)

Example 2 with Reactable

use of io.gravitee.gateway.reactor.Reactable in project gravitee-gateway by gravitee-io.

the class ResourceManagerImpl method initialize.

private void initialize() {
    ResourcePluginManager rpm = applicationContext.getBean(ResourcePluginManager.class);
    ResourceClassLoaderFactory rclf = applicationContext.getBean(ResourceClassLoaderFactory.class);
    ResourceConfigurationFactory rcf = applicationContext.getBean(ResourceConfigurationFactory.class);
    ReactorHandler rh = applicationContext.getBean(ReactorHandler.class);
    Reactable reactable = rh.reactable();
    Set<Resource> resourceDeps = reactable.dependencies(Resource.class);
    resourceDeps.forEach(resource -> {
        final ResourcePlugin resourcePlugin = rpm.get(resource.getType());
        if (resourcePlugin == null) {
            logger.error("Resource [{}] can not be found in plugin registry", resource.getType());
            throw new IllegalStateException("Resource [" + resource.getType() + "] can not be found in plugin registry");
        }
        PluginClassLoader resourceClassLoader = rclf.getOrCreateClassLoader(resourcePlugin, rh.classloader());
        logger.debug("Loading resource {} for {}", resource.getName(), rh);
        try {
            Class<? extends io.gravitee.resource.api.Resource> resourceClass = (Class<? extends io.gravitee.resource.api.Resource>) ClassUtils.forName(resourcePlugin.resource().getName(), resourceClassLoader);
            Map<Class<?>, Object> injectables = new HashMap<>();
            if (resourcePlugin.configuration() != null) {
                Class<? extends ResourceConfiguration> resourceConfigurationClass = (Class<? extends ResourceConfiguration>) ClassUtils.forName(resourcePlugin.configuration().getName(), resourceClassLoader);
                injectables.put(resourceConfigurationClass, rcf.create(resourceConfigurationClass, resource.getConfiguration()));
            }
            io.gravitee.resource.api.Resource resourceInstance = new ResourceFactory().create(resourceClass, injectables);
            if (resourceInstance instanceof ApplicationContextAware) {
                ((ApplicationContextAware) resourceInstance).setApplicationContext(applicationContext);
            }
            resources.put(resource.getName(), resourceInstance);
        } catch (Exception ex) {
            logger.error("Unable to create resource", ex);
            if (resourceClassLoader != null) {
                try {
                    resourceClassLoader.close();
                } catch (IOException ioe) {
                    logger.error("Unable to close classloader for resource", ioe);
                }
            }
        }
    });
}
Also used : ApplicationContextAware(org.springframework.context.ApplicationContextAware) Reactable(io.gravitee.gateway.reactor.Reactable) ResourceClassLoaderFactory(io.gravitee.plugin.resource.ResourceClassLoaderFactory) ResourceConfiguration(io.gravitee.resource.api.ResourceConfiguration) ReactorHandler(io.gravitee.gateway.reactor.handler.ReactorHandler) ResourcePlugin(io.gravitee.plugin.resource.ResourcePlugin) PluginClassLoader(io.gravitee.plugin.core.api.PluginClassLoader) ResourceConfigurationFactory(io.gravitee.gateway.resource.ResourceConfigurationFactory) Resource(io.gravitee.definition.model.plugins.resources.Resource) IOException(java.io.IOException) IOException(java.io.IOException) ResourcePluginManager(io.gravitee.plugin.resource.ResourcePluginManager)

Aggregations

Reactable (io.gravitee.gateway.reactor.Reactable)2 ReactorHandler (io.gravitee.gateway.reactor.handler.ReactorHandler)2 PluginClassLoader (io.gravitee.plugin.core.api.PluginClassLoader)2 IOException (java.io.IOException)2 Policy (io.gravitee.definition.model.Policy)1 Resource (io.gravitee.definition.model.plugins.resources.Resource)1 ResourceConfigurationFactory (io.gravitee.gateway.resource.ResourceConfigurationFactory)1 ResourceLifecycleManager (io.gravitee.gateway.resource.ResourceLifecycleManager)1 PolicyClassLoaderFactory (io.gravitee.plugin.policy.PolicyClassLoaderFactory)1 PolicyPlugin (io.gravitee.plugin.policy.PolicyPlugin)1 PolicyPluginManager (io.gravitee.plugin.policy.PolicyPluginManager)1 PolicyMethodResolver (io.gravitee.plugin.policy.internal.PolicyMethodResolver)1 ResourceClassLoaderFactory (io.gravitee.plugin.resource.ResourceClassLoaderFactory)1 ResourcePlugin (io.gravitee.plugin.resource.ResourcePlugin)1 ResourcePluginManager (io.gravitee.plugin.resource.ResourcePluginManager)1 PolicyContext (io.gravitee.policy.api.PolicyContext)1 Resource (io.gravitee.resource.api.Resource)1 ResourceConfiguration (io.gravitee.resource.api.ResourceConfiguration)1 Function (java.util.function.Function)1 ApplicationContextAware (org.springframework.context.ApplicationContextAware)1