org.apache.cxf.transport.http.DestinationRegistry Java Examples

@Test
public void testUseCustomAtmoosphereInterceptor() throws Exception {
    Bus bus = new ExtensionManagerBus();
    bus.setProperty("atmosphere.interceptors", new CustomInterceptor1());
    DestinationRegistry registry = new HTTPTransportFactory().getRegistry();
    EndpointInfo endpoint = new EndpointInfo();
    endpoint.setAddress(ENDPOINT_ADDRESS);
    endpoint.setName(ENDPOINT_NAME);

    AtmosphereWebSocketServletDestination dest =
        new AtmosphereWebSocketServletDestination(bus, registry, endpoint, ENDPOINT_ADDRESS);

    List<AtmosphereInterceptor> ais = dest.getAtmosphereFramework().interceptors();
    int added = 0;
    for (AtmosphereInterceptor a : ais) {
        if (CustomInterceptor1.class.equals(a.getClass())) {
            added++;
            break;
        }
    }
    assertEquals(1, added);
}
 

@Test
public void testUseCXFDefaultAtmoosphereInterceptor() throws Exception {
    Bus bus = new ExtensionManagerBus();
    DestinationRegistry registry = new HTTPTransportFactory().getRegistry();
    EndpointInfo endpoint = new EndpointInfo();
    endpoint.setAddress(ENDPOINT_ADDRESS);
    endpoint.setName(ENDPOINT_NAME);

    AtmosphereWebSocketServletDestination dest =
        new AtmosphereWebSocketServletDestination(bus, registry, endpoint, ENDPOINT_ADDRESS);

    List<AtmosphereInterceptor> ais = dest.getAtmosphereFramework().interceptors();
    int added = 0;
    for (AtmosphereInterceptor a : ais) {
        if (DefaultProtocolInterceptor.class.equals(a.getClass())) {
            added++;
            break;
        }
    }
    assertEquals(1, added);
}
 

@Test
public void testUseCustomAtmoosphereInterceptors() throws Exception {
    Bus bus = new ExtensionManagerBus();
    bus.setProperty("atmosphere.interceptors", Arrays.asList(new CustomInterceptor1(), new CustomInterceptor2()));
    DestinationRegistry registry = new HTTPTransportFactory().getRegistry();
    EndpointInfo endpoint = new EndpointInfo();
    endpoint.setAddress(ENDPOINT_ADDRESS);
    endpoint.setName(ENDPOINT_NAME);

    AtmosphereWebSocketServletDestination dest =
        new AtmosphereWebSocketServletDestination(bus, registry, endpoint, ENDPOINT_ADDRESS);

    List<AtmosphereInterceptor> ais = dest.getAtmosphereFramework().interceptors();
    int added = 0;
    for (AtmosphereInterceptor a : ais) {
        if (CustomInterceptor1.class.equals(a.getClass())) {
            added++;
        } else if (CustomInterceptor2.class.equals(a.getClass())) {
            added++;
            break;
        }
    }
    assertEquals(2, added);
}
 

@Test
public void testRegisteration() throws Exception {
    Bus bus = new ExtensionManagerBus();
    DestinationRegistry registry = new HTTPTransportFactory().getRegistry();
    EndpointInfo endpoint = new EndpointInfo();
    endpoint.setAddress(ENDPOINT_ADDRESS);
    endpoint.setName(ENDPOINT_NAME);

    TestAtmosphereWebSocketServletDestination dest =
        new TestAtmosphereWebSocketServletDestination(bus, registry, endpoint, ENDPOINT_ADDRESS);

    dest.activate();

    assertNotNull(registry.getDestinationForPath(ENDPOINT_ADDRESS));

    dest.deactivate();

    assertNull(registry.getDestinationForPath(ENDPOINT_ADDRESS));
}
 

@Test
public void testUseCXFDefaultAtmoosphereInterceptor() throws Exception {
    Bus bus = new ExtensionManagerBus();
    DestinationRegistry registry = new HTTPTransportFactory().getRegistry();
    EndpointInfo endpoint = new EndpointInfo();
    endpoint.setAddress(ENDPOINT_ADDRESS);
    endpoint.setName(ENDPOINT_NAME);

    AtmosphereWebSocketServletDestination dest =
        new AtmosphereWebSocketServletDestination(bus, registry, endpoint, ENDPOINT_ADDRESS);

    List<AtmosphereInterceptor> ais = dest.getAtmosphereFramework().interceptors();
    int added = 0;
    for (AtmosphereInterceptor a : ais) {
        if (DefaultProtocolInterceptor.class.equals(a.getClass())) {
            added++;
            break;
        }
    }
    assertEquals(1, added);
}
 

@Test
public void testUseCustomAtmoosphereInterceptor() throws Exception {
    Bus bus = new ExtensionManagerBus();
    bus.setProperty("atmosphere.interceptors", new CustomInterceptor1());
    DestinationRegistry registry = new HTTPTransportFactory().getRegistry();
    EndpointInfo endpoint = new EndpointInfo();
    endpoint.setAddress(ENDPOINT_ADDRESS);
    endpoint.setName(ENDPOINT_NAME);

    AtmosphereWebSocketServletDestination dest =
        new AtmosphereWebSocketServletDestination(bus, registry, endpoint, ENDPOINT_ADDRESS);

    List<AtmosphereInterceptor> ais = dest.getAtmosphereFramework().interceptors();
    int added = 0;
    for (AtmosphereInterceptor a : ais) {
        if (CustomInterceptor1.class.equals(a.getClass())) {
            added++;
            break;
        }
    }
    assertEquals(1, added);
}
 

@Test
public void testUseCustomAtmoosphereInterceptors() throws Exception {
    Bus bus = new ExtensionManagerBus();
    bus.setProperty("atmosphere.interceptors", Arrays.asList(new CustomInterceptor1(), new CustomInterceptor2()));
    DestinationRegistry registry = new HTTPTransportFactory().getRegistry();
    EndpointInfo endpoint = new EndpointInfo();
    endpoint.setAddress(ENDPOINT_ADDRESS);
    endpoint.setName(ENDPOINT_NAME);

    AtmosphereWebSocketServletDestination dest =
        new AtmosphereWebSocketServletDestination(bus, registry, endpoint, ENDPOINT_ADDRESS);

    List<AtmosphereInterceptor> ais = dest.getAtmosphereFramework().interceptors();
    int added = 0;
    for (AtmosphereInterceptor a : ais) {
        if (CustomInterceptor1.class.equals(a.getClass())) {
            added++;
        } else if (CustomInterceptor2.class.equals(a.getClass())) {
            added++;
            break;
        }
    }
    assertEquals(2, added);
}
 

public AtmosphereWebSocketUndertowDestination(Bus bus, DestinationRegistry registry, EndpointInfo ei,
                                              UndertowHTTPServerEngineFactory serverEngineFactory)
                                                  throws IOException {
    super(bus, registry, ei, serverEngineFactory);
    framework = new AtmosphereFramework(false, true);
    framework.setUseNativeImplementation(false);
    framework.addInitParameter(ApplicationConfig.PROPERTY_NATIVE_COMETSUPPORT, "true");
    framework.addInitParameter(ApplicationConfig.PROPERTY_SESSION_SUPPORT, "true");
    framework.addInitParameter(ApplicationConfig.WEBSOCKET_SUPPORT, "true");
    framework.addInitParameter(ApplicationConfig.WEBSOCKET_PROTOCOL_EXECUTION, "true");
    // workaround for atmosphere's jsr356 initialization requiring servletConfig
    framework.addInitParameter(ApplicationConfig.WEBSOCKET_SUPPRESS_JSR356, "true");
    AtmosphereUtils.addInterceptors(framework, bus);
    framework.addAtmosphereHandler("/", new DestinationHandler());
    framework.init();
    executor = bus.getExtension(WorkQueueManager.class).getAutomaticWorkQueue();
}
 

public AtmosphereWebSocketJettyDestination(Bus bus, DestinationRegistry registry, EndpointInfo ei,
                                 JettyHTTPServerEngineFactory serverEngineFactory)
    throws IOException {
    super(bus, registry, ei, 
          serverEngineFactory == null ? null : new URL(getNonWSAddress(ei)),
          serverEngineFactory);

    framework = new AtmosphereFramework(false, true);
    framework.setUseNativeImplementation(false);
    addInitParameter(framework, initParams, ApplicationConfig.PROPERTY_NATIVE_COMETSUPPORT, "true");
    addInitParameter(framework, initParams, ApplicationConfig.PROPERTY_SESSION_SUPPORT, "true");
    addInitParameter(framework, initParams, ApplicationConfig.WEBSOCKET_SUPPORT, "true");
    addInitParameter(framework, initParams, ApplicationConfig.WEBSOCKET_PROTOCOL_EXECUTION, "true");
    // workaround for atmosphere's jsr356 initialization requiring servletConfig
    addInitParameter(framework, initParams, ApplicationConfig.WEBSOCKET_SUPPRESS_JSR356, "true");
    AtmosphereUtils.addInterceptors(framework, bus);
    framework.addAtmosphereHandler("/", new DestinationHandler());        
}
 

@Test
public void testRegisteration() throws Exception {
    Bus bus = new ExtensionManagerBus();
    DestinationRegistry registry = new HTTPTransportFactory().getRegistry();
    EndpointInfo endpoint = new EndpointInfo();
    endpoint.setAddress(ENDPOINT_ADDRESS);
    endpoint.setName(ENDPOINT_NAME);
    JettyHTTPServerEngine engine = EasyMock.createMock(JettyHTTPServerEngine.class);

    control.replay();

    TestJettyWebSocketDestination dest = new TestJettyWebSocketDestination(bus, registry, endpoint, null, engine);

    dest.activate();

    assertNotNull(registry.getDestinationForPath(ENDPOINT_ADDRESS));

    dest.deactivate();

    assertNull(registry.getDestinationForPath(ENDPOINT_ADDRESS));
}
 

public ServletController(DestinationRegistry destinationRegistry,
                            ServletConfig config,
                            HttpServlet serviceListGenerator) {
    this.servletConfig = config;
    this.destinationRegistry = destinationRegistry;
    this.serviceListGenerator = serviceListGenerator;
    init();
}
 

private static DestinationRegistry getDestinationRegistry(Bus bus) {
    DestinationFactoryManager dfm = bus.getExtension(DestinationFactoryManager.class);
    try {
        DestinationFactory df = dfm
            .getDestinationFactory("http://cxf.apache.org/transports/http/configuration");
        if (df instanceof HTTPTransportFactory) {
            HTTPTransportFactory transportFactory = (HTTPTransportFactory)df;
            return transportFactory.getRegistry();
        }
    } catch (Exception e) {
        // why are we throwing a busexception if the DF isn't found?
    }
    return null;
}
 

@Override
public void init(final ServletConfig sc) throws ServletException {
    super.init(sc);

    // just logging the endpoints
    final LogFacade log = new LogFacade(CxfCdiAutoSetup.class.getName());
    final String transportId = sc.getInitParameter(TRANSPORT_ID);
    final DestinationRegistry registry = getDestinationRegistryFromBusOrDefault(transportId);
    prefixes = registry.getDestinations().stream()
            .filter(ServletDestination.class::isInstance)
            .map(ServletDestination.class::cast)
            .map(getServletDestinationPath(sc, log))
            .filter(Objects::nonNull)
            .toArray(String[]::new);
}
 

private static Constructor<?> probeConstructor(String name) {
    try {
        Class<?> clz = Class.forName(name, true, WebSocketDestinationFactory.class.getClassLoader());
        return clz.getConstructor(Bus.class, DestinationRegistry.class, EndpointInfo.class,
                                  JettyHTTPServerEngineFactory.class);
    } catch (Throwable t) {
        return null;
    }
}
 

private static Constructor<?> probeUndertowConstructor(String name) {
    try {
        Class<?> clz = Class.forName(name, true, WebSocketDestinationFactory.class.getClassLoader());
        return clz.getConstructor(Bus.class, DestinationRegistry.class, EndpointInfo.class,
                                  UndertowHTTPServerEngineFactory.class);
    } catch (Throwable t) {
        return null;
    }
}
 

public ServiceListGeneratorServlet(DestinationRegistry destinationRegistry, Bus bus) {
    this.destinationRegistry = destinationRegistry;
    this.bus = bus;
    if (this.bus == null) {
        this.bus = BusFactory.getDefaultBus(false);
    }

    this.title = "CXF - Service list";
}
 

public ServletDestination(Bus b,
                          DestinationRegistry registry,
                          EndpointInfo ei,
                          String path,
                          boolean dp)
    throws IOException {
    // would add the default port to the address
    super(b, registry, ei, path, dp);
}
 

public DestinationRegistry getRegistry() {
    try {
        final HTTPTransportFactory transportFactory = HTTPTransportFactory.class
                .cast(bus
                        .getExtension(DestinationFactoryManager.class)
                        .getDestinationFactory("http://cxf.apache.org/transports/http" + "/configuration"));
        return transportFactory.getRegistry();
    } catch (final BusException e) {
        throw new IllegalStateException(e);
    }
}
 

protected DestinationRegistry getDestinationRegistryFromBusOrDefault(final String transportId) {
    DestinationFactoryManager dfm = bus.getExtension(DestinationFactoryManager.class);
    try {
        String peferredTransportId = transportId;

        // Check if the preferred transport is set on a bus level (f.e., from any
        // extension or customization).
        if (StringUtils.isEmpty(peferredTransportId) && getBus() != null) {
            peferredTransportId = (String)getBus().getProperty(AbstractTransportFactory.PREFERRED_TRANSPORT_ID);
        }

        if (StringUtils.isEmpty(peferredTransportId)) {
            final Set<String> candidates = dfm.getRegisteredDestinationFactoryNames();

            // If the default transport is present, fall back to it and don't even
            // consider other candidates
            if (!candidates.contains(DEFAULT_TRANSPORT_ID)) {
                peferredTransportId = candidates
                    .stream()
                    .filter(name -> name.endsWith("/configuration"))
                    .findAny()
                    .orElse(DEFAULT_TRANSPORT_ID);
            }
        }

        DestinationFactory df = StringUtils.isEmpty(peferredTransportId)
            ? dfm.getDestinationFactory(DEFAULT_TRANSPORT_ID)
                : dfm.getDestinationFactory(peferredTransportId);
        if (df instanceof HTTPTransportFactory) {
            HTTPTransportFactory transportFactory = (HTTPTransportFactory)df;
            return transportFactory.getRegistry();
        }
    } catch (BusException e) {
        // why are we throwing a busexception if the DF isn't found?
    }
    return null;
}
 

public void start(final BundleContext context) throws Exception {

        ConfigAdminHttpConduitConfigurer conduitConfigurer = new ConfigAdminHttpConduitConfigurer();

        registerService(context, ManagedServiceFactory.class, conduitConfigurer,
                        ConfigAdminHttpConduitConfigurer.FACTORY_PID);
        registerService(context, HTTPConduitConfigurer.class, conduitConfigurer,
                        "org.apache.cxf.http.conduit-configurer");

        if (PropertyUtils.isTrue(context.getProperty(DISABLE_DEFAULT_HTTP_TRANSPORT))) {
            //TODO: Review if it also makes sense to support "http.transport.disable"
            //      directly in the CXF_CONFIG_SCOPE properties file
            return;
        }

        DestinationRegistry destinationRegistry = new DestinationRegistryImpl();
        HTTPTransportFactory transportFactory = new HTTPTransportFactory(destinationRegistry);

        HttpServiceTrackerCust customizer = new HttpServiceTrackerCust(destinationRegistry, context);
        httpServiceTracker = new ServiceTracker<>(context, HttpService.class, customizer);
        httpServiceTracker.open();

        context.registerService(DestinationRegistry.class.getName(), destinationRegistry, null);
        context.registerService(HTTPTransportFactory.class.getName(), transportFactory, null);

        BlueprintNameSpaceHandlerFactory factory = new BlueprintNameSpaceHandlerFactory() {

            @Override
            public Object createNamespaceHandler() {
                return new HttpBPHandler();
            }
        };
        NamespaceHandlerRegisterer.register(context, factory,
                                            "http://cxf.apache.org/transports/http/configuration");
    }
 

@Before
public void setUp() {
    req = EasyMock.createMock(HttpServletRequest.class);
    res = EasyMock.createMock(HttpServletResponse.class);
    registry = EasyMock.createMock(DestinationRegistry.class);
    serviceListGenerator = EasyMock.createMock(HttpServlet.class);
}
 

/**
 * Constructor
 *
 * @param bus  the associated Bus
 * @param registry the associated destinationRegistry
 * @param ei the endpoint info of the destination
 * @param serverEngineFactory the serverEngineFactory which could be used to create ServerEngine
 * @throws java.io.IOException
 */
public JettyHTTPDestination(
        Bus bus,
        DestinationRegistry registry,
        EndpointInfo ei,
        JettyHTTPServerEngineFactory serverEngineFactory
) throws IOException {
    this(bus, registry, ei, 
         serverEngineFactory == null ? null : new URL(getAddressValue(ei, true).getAddress()),
         serverEngineFactory);
}
 

protected JettyHTTPDestination(Bus bus,
                               DestinationRegistry registry,
                               EndpointInfo ei,
                               URL nurl,
                               JettyHTTPServerEngineFactory serverEngineFactory)
    throws IOException {
    //Add the default port if the address is missing it
    super(bus, registry, ei, getAddressValue(ei, true).getAddress(), true);
    this.serverEngineFactory = serverEngineFactory;
    this.nurl = nurl;
    loader = bus.getExtension(ClassLoader.class);
}
 

EasyMockJettyHTTPDestination(Bus bus,
                                    DestinationRegistry registry,
                                    EndpointInfo endpointInfo,
                                    JettyHTTPServerEngineFactory serverEngineFactory,
                                    JettyHTTPServerEngine easyMockEngine) throws IOException {
    super(bus, registry, endpointInfo, serverEngineFactory);
    engine = easyMockEngine;
}
 

/**
 * Constructor
 *
 * @param bus  the associated Bus
 * @param registry the associated destinationRegistry
 * @param ei the endpoint info of the destination
 * @param serverEngineFactory the serverEngineFactory which could be used to create ServerEngine
 * @throws java.io.IOException
 */
public UndertowHTTPDestination(Bus bus,
                               DestinationRegistry registry,
                               EndpointInfo ei,
                               UndertowHTTPServerEngineFactory serverEngineFactory)
    throws IOException {
    //Add the default port if the address is missing it
    super(bus, registry, ei, getAddressValue(ei, true).getAddress(), true);
    this.serverEngineFactory = serverEngineFactory;
    loader = bus.getExtension(ClassLoader.class);
}
 

EasyMockUndertowHTTPDestination(Bus bus,
                                    DestinationRegistry registry,
                                    EndpointInfo endpointInfo,
                                    UndertowHTTPServerEngineFactory serverEngineFactory,
                                    UndertowHTTPServerEngine easyMockEngine) throws IOException {
    super(bus, registry, endpointInfo, serverEngineFactory);
    engine = easyMockEngine;
}
 

public NettyHttpTransportFactory(DestinationRegistry registry) {
    super(DEFAULT_NAMESPACES);
    if (registry == null) {
        registry = new DestinationRegistryImpl();
    }
    this.registry = registry;
}
 

EasyMockJettyHTTPDestination(Bus bus,
                             DestinationRegistry registry,
                             EndpointInfo endpointInfo,
                             NettyHttpServerEngineFactory serverEngineFactory,
                             NettyHttpServerEngine easyMockEngine) throws IOException {
    super(bus, registry, endpointInfo, serverEngineFactory);
    engine = easyMockEngine;
}
 

private static DestinationRegistry getDestinationRegistryFromBus(Bus bus) {
	DestinationFactoryManager dfm = bus.getExtension(DestinationFactoryManager.class);
	try {
		DestinationFactory df = dfm.getDestinationFactory("http://cxf.apache.org/transports/http/configuration");
		if (df instanceof HTTPTransportFactory) {
			HTTPTransportFactory transportFactory = (HTTPTransportFactory) df;
			return transportFactory.getRegistry();
		}
	} catch (BusException e) {
		// why are we throwing a busexception if the DF isn't found?
	}
	return null;
}
 

/**
 * Constructor
 *
 * @param bus the associated Bus
 * @param registry the associated destinationRegistry
 * @param ei the endpoint info of the destination
 * @throws java.io.IOException
 */
public UndertowHTTPDestination(Bus bus, DestinationRegistry registry, EndpointInfo ei) throws IOException {
    //Add the default port if the address is missing it
    super(bus, registry, ei, getAddressValue(ei, true).getAddress(), true);
    this.serverEngineFactory = bus.getExtension(HttpServerEngineFactory.class);
    if (serverEngineFactory != null) {
        nurl = new URL(getAddress(endpointInfo));
    }
    loader = bus.getExtension(ClassLoader.class);
}