org.wildfly.swarm.springboot.SpringApplicationContextService Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-boot Show documentation

WildFly Swarm: Spring Boot

The newest version!

package org.wildfly.swarm.springboot;

import java.util.concurrent.CountDownLatch;

import org.jboss.msc.service.Service;
import org.jboss.msc.service.ServiceController;
import org.jboss.msc.service.ServiceName;
import org.jboss.msc.service.ServiceTarget;
import org.jboss.msc.service.StartContext;
import org.jboss.msc.service.StartException;
import org.jboss.msc.service.StopContext;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ConfigurableApplicationContext;

/**
 * @author Bob McWhirter
 */
public class SpringApplicationContextService implements Service {

    public static final ServiceName SERVICE_NAME = ServiceName.of("swarm", "spring-boot", "context");

    public static void addService(ServiceTarget target, ClassLoader classLoader, Class source) {
        target.addService(SERVICE_NAME, new SpringApplicationContextService(classLoader, source))
                .install();
    }

    private final ClassLoader classLoader;

    private final Class source;

    private ConfigurableApplicationContext context;

    public SpringApplicationContextService(ClassLoader classLoader, Class source) {
        this.classLoader = classLoader;
        this.source = source;
    }

    @Override
    public void start(StartContext context) throws StartException {

        context.asynchronous();

        //Thread thread = new Thread(() -> {
        Thread.currentThread().setContextClassLoader(this.classLoader);
        try {
            this.context = org.springframework.boot.SpringApplication.run(new Object[]{
                            this.source
                    },
                    new String[]{});

            context.complete();
        } catch (Throwable t) {
            context.failed(new StartException(t));
            t.printStackTrace();
        }
    }

    @Override
    public void stop(StopContext context) {
        this.context.stop();
    }

    @Override
    public ApplicationContext getValue() throws IllegalStateException, IllegalArgumentException {
        return this.context;
    }
}