Java Examples for org.springframework.util.concurrent.SuccessCallback

@Test
public void testAsyncGet2() throws Exception {
    ListenableFuture<String> listenableFuture = asyncService.asyncGet2();
    SuccessCallback<String> successCallback = new SuccessCallback<String>() {

        @Override
        public void onSuccess(String str) {
            System.out.println("�첽�ص��ɹ���, return : " + str);
        }
    };
    FailureCallback failureCallback = new FailureCallback() {

        @Override
        public void onFailure(Throwable throwable) {
            System.out.println("�첽�ص�ʧ����, exception message : " + throwable.getMessage());
        }
    };
    listenableFuture.addCallback(successCallback, failureCallback);
    Assert.assertEquals("123", listenableFuture.get());
}

@Override
public void addCallback(SuccessCallback<? super V> successCallback, FailureCallback failureCallback) {
    try {
        if (this.executionException != null) {
            Throwable cause = this.executionException.getCause();
            failureCallback.onFailure(cause != null ? cause : this.executionException);
        } else {
            successCallback.onSuccess(this.value);
        }
    } catch (Throwable ex) {
    }
}

@Test
public void setConfigurationOK() throws Exception {
    // Mock the task scheduler and capture the runnable
    ArgumentCaptor<Runnable> runnableArg = ArgumentCaptor.forClass(Runnable.class);
    when(taskScheduler.scheduleWithFixedDelay(runnableArg.capture(), anyLong())).thenReturn(Mockito.mock(ScheduledFuture.class));
    // Mock the response to the subsribeContext
    ArgumentCaptor<SuccessCallback> successArg = ArgumentCaptor.forClass(SuccessCallback.class);
    ListenableFuture<SubscribeContextResponse> responseFuture = Mockito.mock(ListenableFuture.class);
    doNothing().when(responseFuture).addCallback(successArg.capture(), any());
    Configuration configuration = getBasicConf();
    subscriptionManager.setConfiguration(configuration);
    // Capture the arg of subscription and return the mocked future
    ArgumentCaptor<String> urlProviderArg = ArgumentCaptor.forClass(String.class);
    ArgumentCaptor<SubscribeContext> subscribeContextArg = ArgumentCaptor.forClass(SubscribeContext.class);
    when(ngsiClient.subscribeContext(urlProviderArg.capture(), eq(null), subscribeContextArg.capture())).thenReturn(responseFuture);
    // Execute scheduled runnable
    runnableArg.getValue().run();
    // Return the SubscribeContextResponse
    callSuccessCallback(successArg);
    SubscribeContext subscribeContext = subscribeContextArg.getValue();
    assertEquals("S.*", subscribeContext.getEntityIdList().get(0).getId());
    assertEquals("TempSensor", subscribeContext.getEntityIdList().get(0).getType());
    assertEquals(true, subscribeContext.getEntityIdList().get(0).getIsPattern());
    assertEquals("temp", subscribeContext.getAttributeList().get(0));
    assertEquals("PT1H", subscribeContext.getDuration());
    assertEquals("http://iotAgent", urlProviderArg.getValue());
    Set<Provider> providers = configuration.getEventTypeIns().get(0).getProviders();
    for (Provider provider : providers) {
        assertEquals("12345678", provider.getSubscriptionId());
        assertNotNull(provider.getSubscriptionDate());
    }
}

@Test
public void run() throws InterruptedException {
    threadPoolTaskExecutor.setCorePoolSize(8);
    threadPoolTaskExecutor.setMaxPoolSize(10);
    threadPoolTaskExecutor.setQueueCapacity(100);
    threadPoolTaskExecutor.setAllowCoreThreadTimeOut(true);
    threadPoolTaskExecutor.setKeepAliveSeconds(30);
    threadPoolTaskExecutor.setThreadGroupName("elb-check-group");
    threadPoolTaskExecutor.setThreadNamePrefix("elb-check-");
    threadPoolTaskExecutor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
    threadPoolTaskExecutor.setDaemon(true);
    threadPoolTaskExecutor.setWaitForTasksToCompleteOnShutdown(true);
    threadPoolTaskExecutor.afterPropertiesSet();
    try {
        httpclient = HttpClientBuilder.create().disableAuthCaching().disableAutomaticRetries().disableConnectionState().disableCookieManagement().disableRedirectHandling().setDefaultRequestConfig(config).setHostnameVerifier(new AllowAllHostnameVerifier()).setSslcontext(new SSLContextBuilder().loadTrustMaterial(null, ( arrayX509Certificate,  value) -> true).build()).build();
    } catch (NoSuchAlgorithmExceptionKeyManagementException | KeyStoreException |  e) {
        log.error(e.getMessage(), e);
    }
    final List<String> addresses = newArrayList("www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com", "www.google.de", "www.google.it", "www.google.com");
    for (final String address : addresses) {
        for (final Integer allowedPort : allowedPorts) {
            final HttpCall httpCall = new HttpCall(httpclient, address, allowedPort);
            final ListenableFuture<Void> listenableFuture = threadPoolTaskExecutor.submitListenable(httpCall);
            listenableFuture.addCallback(new SuccessCallback<Void>() {

                @Override
                public void onSuccess(final Void result) {
                    log.info("address: {} and port: {}", address, allowedPort);
                }
            }, new FailureCallback() {

                @Override
                public void onFailure(final Throwable ex) {
                    log.warn(ex.getMessage(), ex);
                }
            });
            log.info("getActiveCount: {}", threadPoolTaskExecutor.getActiveCount());
            log.info("### - Thread: {}", Thread.currentThread().getId());
        }
    }
//TODO: important use this to let the test run all thread!
//TimeUnit.MINUTES.sleep(5);
}

/* (non-Javadoc)
	 * @see org.springframework.util.concurrent.ListenableFuture#addCallback(org.springframework.util.concurrent.SuccessCallback, org.springframework.util.concurrent.FailureCallback)
	 */
@Override
public void addCallback(SuccessCallback<? super T> successCallback, FailureCallback failureCallback) {
    future.addCallback(successCallback, failureCallback);
}

@Override
public void addCallback(SuccessCallback<? super T> successCallback, FailureCallback failureCallback) {
    getWrappedFuture().addCallbacks(successCallback::onSuccess, failureCallback::onFailure);
}