io.github.jhipster.loaded.JHipsterReloaderAutoConfiguration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of core Show documentation
The newest version!
package io.github.jhipster.loaded;

import io.github.jhipster.loaded.condition.ConditionalOnSpringLoaded;
import io.github.jhipster.loaded.reloader.Reloader;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.context.annotation.ComponentScan;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.annotation.AnnotationAwareOrderComparator;
import org.springframework.core.env.Environment;
import org.springsource.loaded.agent.SpringLoadedAgent;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;

@Configuration
@ConditionalOnSpringLoaded
@ComponentScan("io.github.jhipster")
public class JHipsterReloaderAutoConfiguration implements ApplicationContextAware {

    private final Logger log = LoggerFactory.getLogger(JHipsterReloaderAutoConfiguration.class);

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        try {
            Environment env = applicationContext.getEnvironment();
            final ConfigurableApplicationContext configurableApplicationContext = (ConfigurableApplicationContext) applicationContext;

            if (env.getProperty("hotReload.enabled", Boolean.class, false)) {
                SpringLoadedAgent.getInstrumentation();
                log.info("Spring Loaded is running, registering hot reloading features");

                final Map allReloaderClasses = applicationContext.getBeansOfType(Reloader.class);

                for (Reloader reloader : allReloaderClasses.values()) {
                    reloader.init(configurableApplicationContext);
                }

                List orderedReloaders = new ArrayList<>();
                orderedReloaders.addAll(allReloaderClasses.values());
                Collections.sort(orderedReloaders, new AnnotationAwareOrderComparator());

                JHipsterPluginManagerReloadPlugin.register(configurableApplicationContext, orderedReloaders,
                    JHipsterReloaderAutoConfiguration.class.getClassLoader());

            }
        } catch (UnsupportedOperationException uoe) {
            log.info("Spring Loaded is not running, hot reloading is not enabled");
        }
    }
}