org.wildfly.swarm.springboot.util.Bridge 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.util;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;

import org.jboss.msc.service.Service;
import org.jboss.msc.service.ServiceController;
import org.jboss.msc.service.ServiceName;
import org.jboss.msc.service.ValueService;
import org.jboss.msc.value.ImmediateValue;
import org.wildfly.swarm.container.internal.Deployer;
import org.wildfly.swarm.springboot.MSCBridgeActivator;

/**
 * @author Bob McWhirter
 */
public class Bridge {
    private static final ServiceName DEPLOYER_SERVICE = ServiceName.of("swarm", "deployer");

    public static Deployer getDeployer() {
        return (Deployer) MSCBridgeActivator.REGISTRY.getService(DEPLOYER_SERVICE).getValue();
    }

    public static void addObject(Object object, String... nameParts) {
        ServiceName name = ServiceName.of(nameParts);
        addObject( object, name );
    }

    public static void addObject(Object object, ServiceName name) {
        MSCBridgeActivator.TARGET.addService(
                name,
                new ValueService<>(
                        new ImmediateValue<>(object)
                )
        ).install();
    }

    public static  T lookup(ServiceName name) {
        try {
            ServiceController s = MSCBridgeActivator.REGISTRY.getService(name);
            s.setMode(ServiceController.Mode.ACTIVE);
            return (T) s.awaitValue( 5, TimeUnit.SECONDS );
        } catch (InterruptedException e) {
            e.printStackTrace();
        } catch (TimeoutException e) {
            e.printStackTrace();
        }
        return null;
    }

    public static  List getChildrenOf(ServiceName parentName, Class type) {
        List names = MSCBridgeActivator.REGISTRY.getServiceNames();

        List objects = new ArrayList<>();

        for (ServiceName name : names) {
            if ( parentName.isParentOf( name ) ) {
                objects.add( (T) MSCBridgeActivator.REGISTRY.getService( name ).getValue() );
            }
        }

        return objects;
    }
}