iSharkFly-Docs/java-tutorials
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Async refactor (#3570)

Browse Source
This commit is contained in:
Grzegorz Piwowarek 2018-02-02 15:40:03 +01:00 committed by GitHub
parent 55660e7702
commit 2169be4301
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 154 additions and 164 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

318
libraries/src/test/java/com/baeldung/asynchttpclient/AsyncHttpClientTestCase.java
View File

@ -1,15 +1,6 @@
package com.baeldung.asynchttpclient; package com.baeldung.asynchttpclient;
import static org.junit.Assert.assertEquals; import io.netty.handler.codec.http.HttpHeaders;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
import org.asynchttpclient.AsyncCompletionHandler; import org.asynchttpclient.AsyncCompletionHandler;
import org.asynchttpclient.AsyncHandler; import org.asynchttpclient.AsyncHandler;
import org.asynchttpclient.AsyncHttpClient; import org.asynchttpclient.AsyncHttpClient;
@ -27,193 +18,192 @@ import org.asynchttpclient.ws.WebSocketUpgradeHandler;
import org.junit.Before; import org.junit.Before;
import org.junit.Test; import org.junit.Test;
import io.netty.handler.codec.http.HttpHeaders; import java.util.concurrent.ExecutionException;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
public class AsyncHttpClientTestCase { public class AsyncHttpClientTestCase {
private static AsyncHttpClient HTTP_CLIENT; private static AsyncHttpClient HTTP_CLIENT;
@Before @Before
public void setup() { public void setup() {
AsyncHttpClientConfig clientConfig = Dsl.config().setConnectTimeout(15000).setRequestTimeout(15000).build();
HTTP_CLIENT = Dsl.asyncHttpClient(clientConfig);
}
AsyncHttpClientConfig clientConfig = Dsl.config().setConnectTimeout(15000).setRequestTimeout(15000).build(); @Test
HTTP_CLIENT = Dsl.asyncHttpClient(clientConfig); public void givenHttpClient_executeSyncGetRequest() {
}
@Test BoundRequestBuilder boundGetRequest = HTTP_CLIENT.prepareGet("http://www.baeldung.com");
public void givenHttpClient_executeSyncGetRequest() {
BoundRequestBuilder boundGetRequest = HTTP_CLIENT.prepareGet("http://www.baeldung.com"); Future<Response> responseFuture = boundGetRequest.execute();
try {
Response response = responseFuture.get(5000, TimeUnit.MILLISECONDS);
assertNotNull(response);
assertEquals(200, response.getStatusCode());
} catch (InterruptedException | ExecutionException | TimeoutException e) {
e.printStackTrace();
}
Future<Response> responseFuture = boundGetRequest.execute(); try {
try { Thread.sleep(5000);
Response response = responseFuture.get(5000, TimeUnit.MILLISECONDS); } catch (InterruptedException e) {
assertNotNull(response); e.printStackTrace();
assertEquals(200, response.getStatusCode()); }
}
} catch (InterruptedException e) { @Test
e.printStackTrace(); public void givenHttpClient_executeAsyncGetRequest() {
} catch (ExecutionException e) {
e.printStackTrace();
} catch (TimeoutException e) {
e.printStackTrace();
}
try { // execute an unbound GET request
Thread.sleep(5000); Request unboundGetRequest = Dsl.get("http://www.baeldung.com").build();
} catch (InterruptedException e) {
e.printStackTrace();
}
}
@Test HTTP_CLIENT.executeRequest(unboundGetRequest, new AsyncCompletionHandler<Integer>() {
public void givenHttpClient_executeAsyncGetRequest() { @Override
public Integer onCompleted(Response response) {
// execute an unbound GET request int resposeStatusCode = response.getStatusCode();
Request unboundGetRequest = Dsl.get("http://www.baeldung.com").build(); assertEquals(200, resposeStatusCode);
return resposeStatusCode;
}
});
HTTP_CLIENT.executeRequest(unboundGetRequest, new AsyncCompletionHandler<Integer>() { // execute a bound GET request
@Override BoundRequestBuilder boundGetRequest = HTTP_CLIENT.prepareGet("http://www.baeldung.com");
public Integer onCompleted(Response response) throws Exception {
int resposeStatusCode = response.getStatusCode(); boundGetRequest.execute(new AsyncCompletionHandler<Integer>() {
assertEquals(200, resposeStatusCode); @Override
return resposeStatusCode; public Integer onCompleted(Response response) {
} int resposeStatusCode = response.getStatusCode();
}); assertEquals(200, resposeStatusCode);
return resposeStatusCode;
}
});
// execute a bound GET request try {
BoundRequestBuilder boundGetRequest = HTTP_CLIENT.prepareGet("http://www.baeldung.com"); Thread.sleep(5000);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
boundGetRequest.execute(new AsyncCompletionHandler<Integer>() { @Test
@Override public void givenHttpClient_executeAsyncGetRequestWithAsyncHandler() {
public Integer onCompleted(Response response) throws Exception {
int resposeStatusCode = response.getStatusCode(); // execute an unbound GET request
assertEquals(200, resposeStatusCode); Request unboundGetRequest = Dsl.get("http://www.baeldung.com").build();
return resposeStatusCode;
}
});
try { HTTP_CLIENT.executeRequest(unboundGetRequest, new AsyncHandler<Integer>() {
Thread.sleep(5000);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
@Test int responseStatusCode = -1;
public void givenHttpClient_executeAsyncGetRequestWithAsyncHandler() {
// execute an unbound GET request @Override
Request unboundGetRequest = Dsl.get("http://www.baeldung.com").build(); public State onStatusReceived(HttpResponseStatus responseStatus) {
responseStatusCode = responseStatus.getStatusCode();
return State.CONTINUE;
}
HTTP_CLIENT.executeRequest(unboundGetRequest, new AsyncHandler<Integer>() { @Override
public State onHeadersReceived(HttpHeaders headers) {
return State.CONTINUE;
}
int responseStatusCode = -1; @Override
public State onBodyPartReceived(HttpResponseBodyPart bodyPart) {
return State.CONTINUE;
}
@Override @Override
public State onStatusReceived(HttpResponseStatus responseStatus) throws Exception { public void onThrowable(Throwable t) {
responseStatusCode = responseStatus.getStatusCode();
return State.CONTINUE;
}
@Override }
public State onHeadersReceived(HttpHeaders headers) throws Exception {
return State.CONTINUE;
}
@Override @Override
public State onBodyPartReceived(HttpResponseBodyPart bodyPart) throws Exception { public Integer onCompleted() {
return State.CONTINUE; assertEquals(200, responseStatusCode);
} return responseStatusCode;
}
});
@Override try {
public void onThrowable(Throwable t) { Thread.sleep(5000);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
} @Test
public void givenHttpClient_executeAsyncGetRequestWithListanableFuture() {
// execute an unbound GET request
Request unboundGetRequest = Dsl.get("http://www.baeldung.com").build();
@Override ListenableFuture<Response> listenableFuture = HTTP_CLIENT.executeRequest(unboundGetRequest);
public Integer onCompleted() throws Exception { listenableFuture.addListener(() -> {
assertEquals(200, responseStatusCode); Response response;
return responseStatusCode; try {
} response = listenableFuture.get(5000, TimeUnit.MILLISECONDS);
}); assertEquals(200, response.getStatusCode());
} catch (InterruptedException | ExecutionException | TimeoutException e) {
e.printStackTrace();
}
}, Executors.newCachedThreadPool());
try { try {
Thread.sleep(5000); Thread.sleep(5000);
} catch (InterruptedException e) { } catch (InterruptedException e) {
e.printStackTrace(); e.printStackTrace();
} }
} }
@Test @Test
public void givenHttpClient_executeAsyncGetRequestWithListanableFuture() { public void givenWebSocketClient_tryToConnect() {
// execute an unbound GET request
Request unboundGetRequest = Dsl.get("http://www.baeldung.com").build();
ListenableFuture<Response> listenableFuture = HTTP_CLIENT.executeRequest(unboundGetRequest); WebSocketUpgradeHandler.Builder upgradeHandlerBuilder = new WebSocketUpgradeHandler.Builder();
listenableFuture.addListener(() -> { WebSocketUpgradeHandler wsHandler = upgradeHandlerBuilder.addWebSocketListener(new WebSocketListener() {
Response response; @Override
try { public void onOpen(WebSocket websocket) {
response = listenableFuture.get(5000, TimeUnit.MILLISECONDS); // WebSocket connection opened
assertEquals(200, response.getStatusCode()); }
} catch (InterruptedException | ExecutionException | TimeoutException e) {
e.printStackTrace();
}
}, Executors.newCachedThreadPool()); @Override
public void onClose(WebSocket websocket, int code, String reason) {
// WebSocket connection closed
}
try { @Override
Thread.sleep(5000); public void onError(Throwable t) {
} catch (InterruptedException e) { // WebSocket connection error
e.printStackTrace(); assertTrue(t.getMessage().contains("Request timeout"));
} }
} }).build();
@Test WebSocket WEBSOCKET_CLIENT = null;
public void givenWebSocketClient_tryToConnect() { try {
WEBSOCKET_CLIENT = Dsl.asyncHttpClient()
.prepareGet("ws://localhost:5590/websocket")
.addHeader("header_name", "header_value")
.addQueryParam("key", "value")
.setRequestTimeout(5000)
.execute(wsHandler).get();
WebSocketUpgradeHandler.Builder upgradeHandlerBuilder = new WebSocketUpgradeHandler.Builder(); if (WEBSOCKET_CLIENT.isOpen()) {
WebSocketUpgradeHandler wsHandler = upgradeHandlerBuilder.addWebSocketListener(new WebSocketListener() { WEBSOCKET_CLIENT.sendPingFrame();
@Override WEBSOCKET_CLIENT.sendTextFrame("test message");
public void onOpen(WebSocket websocket) { WEBSOCKET_CLIENT.sendBinaryFrame(new byte[]{'t', 'e', 's', 't'});
// WebSocket connection opened }
} } catch (InterruptedException | ExecutionException e) {
e.printStackTrace();
@Override } finally {
public void onClose(WebSocket websocket, int code, String reason) { if (WEBSOCKET_CLIENT != null && WEBSOCKET_CLIENT.isOpen()) {
// WebSocket connection closed WEBSOCKET_CLIENT.sendCloseFrame(200, "OK");
} }
}
@Override }
public void onError(Throwable t) {
// WebSocket connection error
assertTrue(t.getMessage().contains("Request timeout"));
}
}).build();
WebSocket WEBSOCKET_CLIENT = null;
try {
WEBSOCKET_CLIENT = Dsl.asyncHttpClient()
.prepareGet("ws://localhost:5590/websocket")
.addHeader("header_name", "header_value")
.addQueryParam("key", "value")
.setRequestTimeout(5000)
.execute(wsHandler).get();
if (WEBSOCKET_CLIENT.isOpen()) {
WEBSOCKET_CLIENT.sendPingFrame();
WEBSOCKET_CLIENT.sendTextFrame("test message");
WEBSOCKET_CLIENT.sendBinaryFrame(new byte[] { 't', 'e', 's', 't' });
}
} catch (InterruptedException | ExecutionException e) {
e.printStackTrace();
} finally {
if (WEBSOCKET_CLIENT != null && WEBSOCKET_CLIENT.isOpen()) {
WEBSOCKET_CLIENT.sendCloseFrame(200, "OK");
}
}
}
} }