Java Code Examples for org.apache.cxf.BusFactory#getThreadDefaultBus()

private void increaseCounter(Exchange ex, MessageHandlingTimeRecorder mhtr) {
    Bus bus = ex.getBus();
    if (null == bus) {
        LOG.log(Level.INFO, "CAN_NOT_GET_BUS_FROM_EXCHANGE");
        bus = BusFactory.getThreadDefaultBus();
    }
    CounterRepository cr = bus.getExtension(CounterRepository.class);

    if (null == cr) {
        LOG.log(Level.WARNING, "NO_COUNTER_REPOSITORY");
        return;
    }
    ObjectName serviceCountername = this.getServiceCounterName(ex);
    cr.increaseCounter(serviceCountername, mhtr);

    ObjectName operationCounter = this.getOperationCounterName(ex, serviceCountername);
    cr.increaseCounter(operationCounter, mhtr);
}
 

@Test
public void testConfigureViaEPR() throws Exception {

    final Set<Class<?>> addressingClasses = new HashSet<>();
    addressingClasses.add(org.apache.cxf.ws.addressing.wsdl.ObjectFactory.class);
    addressingClasses.add(org.apache.cxf.ws.addressing.ObjectFactory.class);

    JAXBContext ctx = JAXBContextCache.getCachedContextAndSchemas(addressingClasses, null, null, null,
                                                                  true).getContext();
    Unmarshaller um = ctx.createUnmarshaller();
    InputStream inStream = getClass().getResourceAsStream("epr.xml");
    JAXBElement<?> el = (JAXBElement<?>)um.unmarshal(inStream);
    EndpointReferenceType ref = (EndpointReferenceType)el.getValue();

    Bus bus = BusFactory.getThreadDefaultBus();
    STSClient client = new STSClient(bus);
    client.configureViaEPR(ref, false);

    assertEquals("http://localhost:8080/jaxws-samples-wsse-policy-trust-sts/SecurityTokenService?wsdl",
                 client.getWsdlLocation());
    assertEquals(new QName("http://docs.oasis-open.org/ws-sx/ws-trust/200512/", "SecurityTokenService"),
                 client.getServiceQName());
    assertEquals(new QName("http://docs.oasis-open.org/ws-sx/ws-trust/200512/", "UT_Port"),
                 client.getEndpointQName());
}
 

public EndpointImpl(Bus bus, Service s, EndpointInfo ei) throws EndpointException {
    if (ei == null) {
        throw new NullPointerException("EndpointInfo can not be null!");
    }

    if (bus == null) {
        this.bus = BusFactory.getThreadDefaultBus();
    } else {
        this.bus = bus;
    }
    service = s;
    endpointInfo = ei;

    createBinding(endpointInfo.getBinding());

    inFaultObserver = new InFaultChainInitiatorObserver(bus);
    outFaultObserver = new OutFaultChainInitiatorObserver(bus);

    getInFaultInterceptors().add(new ClientFaultConverter());
    getOutInterceptors().add(new MessageSenderInterceptor());
    getOutFaultInterceptors().add(new MessageSenderInterceptor());
}
 

private void readBusProperties() {
    Bus b = BusFactory.getThreadDefaultBus(false);
    if (b != null) {
        String v = getBusProperty(b, CachedConstants.THRESHOLD_BUS_PROP, null);
        if (v != null && threshold == defaultThreshold) {
            threshold = Integer.parseInt(v);
        }
        v = getBusProperty(b, CachedConstants.MAX_SIZE_BUS_PROP, null);
        if (v != null) {
            maxSize = Integer.parseInt(v);
        }
        v = getBusProperty(b, CachedConstants.CIPHER_TRANSFORMATION_BUS_PROP, null);
        if (v != null) {
            cipherTransformation = v;
        }
        v = getBusProperty(b, CachedConstants.OUTPUT_DIRECTORY_BUS_PROP, null);
        if (v != null) {
            File f = new File(v);
            if (f.exists() && f.isDirectory()) {
                outputDir = f;
            }
        }
    }
}
 

@Test
public void testConfigureViaEPR() throws Exception {

    final Set<Class<?>> addressingClasses = new HashSet<Class<?>>();
    addressingClasses.add(org.apache.cxf.ws.addressing.wsdl.ObjectFactory.class);
    addressingClasses.add(org.apache.cxf.ws.addressing.ObjectFactory.class);

    JAXBContext ctx = JAXBContextCache.getCachedContextAndSchemas(addressingClasses, null, null, null,
                                                                  true).getContext();
    Unmarshaller um = ctx.createUnmarshaller();
    InputStream inStream = getClass().getResourceAsStream("epr.xml");
    JAXBElement<?> el = (JAXBElement<?>)um.unmarshal(inStream);
    EndpointReferenceType ref = (EndpointReferenceType)el.getValue();

    Bus bus = BusFactory.getThreadDefaultBus();
    STSClient client = new STSClient(bus);
    client.configureViaEPR(ref, false);

    assertEquals("http://localhost:8080/jaxws-samples-wsse-policy-trust-sts/SecurityTokenService?wsdl",
                 client.getWsdlLocation());
    assertEquals(new QName("http://docs.oasis-open.org/ws-sx/ws-trust/200512/", "SecurityTokenService"),
                 client.getServiceQName());
    assertEquals(new QName("http://docs.oasis-open.org/ws-sx/ws-trust/200512/", "UT_Port"),
                 client.getEndpointQName());
}
 

@Test
public void testConfigureViaEPR() throws Exception {

    final Set<Class<?>> addressingClasses = new HashSet<Class<?>>();
    addressingClasses.add(org.apache.cxf.ws.addressing.wsdl.ObjectFactory.class);
    addressingClasses.add(org.apache.cxf.ws.addressing.ObjectFactory.class);

    JAXBContext ctx = JAXBContextCache.getCachedContextAndSchemas(addressingClasses, null, null, null,
                                                                  true).getContext();
    Unmarshaller um = ctx.createUnmarshaller();
    InputStream inStream = getClass().getResourceAsStream("epr.xml");
    JAXBElement<?> el = (JAXBElement<?>)um.unmarshal(inStream);
    EndpointReferenceType ref = (EndpointReferenceType)el.getValue();

    Bus bus = BusFactory.getThreadDefaultBus();
    STSClient client = new STSClient(bus);
    client.configureViaEPR(ref, false);

    assertEquals("http://localhost:8080/jaxws-samples-wsse-policy-trust-sts/SecurityTokenService?wsdl",
                 client.getWsdlLocation());
    assertEquals(new QName("http://docs.oasis-open.org/ws-sx/ws-trust/200512/", "SecurityTokenService"),
                 client.getServiceQName());
    assertEquals(new QName("http://docs.oasis-open.org/ws-sx/ws-trust/200512/", "UT_Port"),
                 client.getEndpointQName());
}
 

/**
 * @return the bus
 */
public Bus getBus() {
    if (bus == null) {
        bus = BusFactory.getThreadDefaultBus();
    }

    return bus;
}
 

public static void updateWsdlExtensors(Bus bus,
                                       String wsdlLocation,
                                       String url,
                                       String encodedUrl) {
    try {
        if (encodedUrl == null) {
            encodedUrl = url;
        }
        if (bus == null) {
            bus = BusFactory.getThreadDefaultBus();
        }
        Definition def = bus.getExtension(WSDLManager.class)
            .getDefinition(wsdlLocation);
        Map<?, ?> map = def.getAllServices();
        for (Object o : map.values()) {
            Service service = (Service)o;
            Map<?, ?> ports = service.getPorts();
            adjustExtensibilityElements(service.getExtensibilityElements(), url, encodedUrl);

            for (Object p : ports.values()) {
                Port port = (Port)p;
                adjustExtensibilityElements(port.getExtensibilityElements(), url, encodedUrl);
                adjustExtensibilityElements(port.getBinding().getExtensibilityElements(), url, encodedUrl);
            }
        }
    } catch (Exception e) {
        e.printStackTrace();
    }
}
 

protected void run()  {
    Object implementor = new HelloImpl();
    String address = "http://localhost:" + PORT + "/wsa/responses";
    ep = new EndpointImpl(BusFactory.getThreadDefaultBus(),
                          implementor,
                          null,
                          getWsdl());
    ep.publish(address);
}
 

private ResourceManager getResourceManager() {
    Bus b = bus;
    if (b == null) {
        b = BusFactory.getThreadDefaultBus();
    }
    return b.getExtension(ResourceManager.class);
}
 

public static ServerProviderFactory createInstance(Bus bus) {
    if (bus == null) {
        bus = BusFactory.getThreadDefaultBus();
    }
    ServerProviderFactory factory = new ServerProviderFactory(bus);
    ProviderFactory.initFactory(factory);
    factory.setProviders(false, false,
                         new WebApplicationExceptionMapper(),
                         new NioMessageBodyWriter());
    factory.setBusProviders();
    return factory;
}
 

@Override
public Bus getBus() {
    if (cxfbus == null)
        cxfbus = BusFactory.getThreadDefaultBus();

    return cxfbus;
}
 

@Test
public void testSTSAsymmetricBinding() throws Exception {
    Bus bus = BusFactory.getThreadDefaultBus();
    STSClient stsClient = initStsClientAsymmeticBinding(bus);

    MessageImpl message = prepareMessage(bus, stsClient, SERVICE_ENDPOINT_ASSYMETRIC);
    STSTokenRetriever.TokenRequestParams params = new STSTokenRetriever.TokenRequestParams();

    SecurityToken token = STSTokenRetriever.getToken(message, params);
    validateSecurityToken(token);
}
 

private AddressBookProtos.AddressBookService connectExportedProtobufAddress(ExportDeclaration declaration) throws EndpointException, NoSuchMethodException, InvocationTargetException, IllegalAccessException, BinderException {
    ProtobufferExportDeclarationWrapper pojo = ProtobufferExportDeclarationWrapper.create(declaration);
    Bus cxfbus = BusFactory.getThreadDefaultBus();
    BindingFactoryManager mgr = cxfbus.getExtension(BindingFactoryManager.class);
    mgr.registerBindingFactory(ProtobufBindingFactory.PROTOBUF_BINDING_ID, new ProtobufBindingFactory(cxfbus));
    Class<?> bufferService = AddressBookProtos.AddressBookService.class;
    Class<?> bufferMessage = AddressBookProtos.AddressBookServiceMessage.class;
    Class<? extends Message> generic = bufferMessage.asSubclass(Message.class);
    RpcChannel channel = new SimpleRpcChannel(pojo.getAddress(), generic);
    Method method = bufferService.getMethod("newStub", RpcChannel.class);
    Object service = method.invoke(bufferService, channel);
    AddressBookProtos.AddressBookService addressBook = (AddressBookProtos.AddressBookService) service;
    return addressBook;
}
 

@Test
public void testCallAsyncWithFullWorkQueue() throws Exception {
    Bus bus = BusFactory.getThreadDefaultBus();
    WorkQueueManager workQueueManager = bus.getExtension(WorkQueueManager.class);
    AutomaticWorkQueueImpl automaticWorkQueue1 = (AutomaticWorkQueueImpl)workQueueManager.getAutomaticWorkQueue();
    updateAddressPort(g, PORT);

    Client client = ClientProxy.getClient(g);
    HTTPConduit http = (HTTPConduit) client.getConduit();

    HTTPClientPolicy httpClientPolicy = new HTTPClientPolicy();

    int asyncExecuteTimeout = 500;
    httpClientPolicy.setAsyncExecuteTimeout(asyncExecuteTimeout);

    http.setClient(httpClientPolicy);

    long repeat = automaticWorkQueue1.getHighWaterMark() + automaticWorkQueue1.getMaxSize() + 1;
    CountDownLatch initialThreadsLatch = new CountDownLatch(automaticWorkQueue1.getHighWaterMark());
    CountDownLatch doneLatch = new CountDownLatch((int) repeat);
    AtomicInteger threadCount = new AtomicInteger();

    for (long i = 0; i < repeat; i++) {
        g.greetMeLaterAsync(-50, res -> {

            try {
                int myCount = threadCount.getAndIncrement();

                if (myCount < automaticWorkQueue1.getHighWaterMark()) {
                    // Sleep long enough so that the workqueue will fill up and then
                    // handleResponseOnWorkqueue will fail for the calls from both
                    // responseReceived and consumeContent
                    Thread.sleep(3 * asyncExecuteTimeout);
                    initialThreadsLatch.countDown();
                } else {
                    Thread.sleep(50);
                }
                initialThreadsLatch.await();
                doneLatch.countDown();
            } catch (Exception e) {
                throw new RuntimeException(e);
            }
        });
    }
    doneLatch.await(30, TimeUnit.SECONDS);

    assertEquals("All responses should be handled eventually", 0, doneLatch.getCount());
}
 

public WSDLRefValidator(Definition wsdl, Document doc) {
    this(wsdl, doc, BusFactory.getThreadDefaultBus());
}
 

public Bus getBus() {
    if (bus == null) {
        bus = BusFactory.getThreadDefaultBus();
    }
    return bus;
}
 

private static Bus getBus(Bus bus) {
    return bus == null ? BusFactory.getThreadDefaultBus() : bus;
}
 

public Bus getBus() {
    Bus bus = super.getBus();
    return bus == null ? BusFactory.getThreadDefaultBus() : bus;
}
 

public JCacheOAuthDataProvider(boolean storeJwtTokenKeyOnly) {
    this(DEFAULT_CONFIG_URL, BusFactory.getThreadDefaultBus(true), storeJwtTokenKeyOnly);
}