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

Fix AsyncEchoTest

This commit is contained in:
pivovarit 2016-12-04 09:02:10 +01:00
parent 450c250a0f
commit 3a00e57002
2 changed files with 20 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
core-java/src/test/java/com/baeldung/java/nio2/async/AsyncEchoClient.java
View File

@ -40,18 +40,22 @@ public class AsyncEchoClient {
}
}
public String sendMessage(String message) throws Exception {
public String sendMessage(String message) {
byte[] byteMsg = message.getBytes();
ByteBuffer buffer = ByteBuffer.wrap(byteMsg);
Future<Integer> writeResult = client.write(buffer);
//run some code
writeResult.get();
try {
writeResult.get();
} catch (Exception e) {
e.printStackTrace();
}
buffer.flip();
Future<Integer> readResult = client.read(buffer);
//run some code
readResult.get();
try {
readResult.get();
} catch (Exception e) {
e.printStackTrace();
String echo = new String(buffer.array()).trim();
buffer.clear();
return echo;

18
core-java/src/test/java/com/baeldung/java/nio2/async/AsyncEchoServer2.java
View File

@ -58,18 +58,19 @@ public class AsyncEchoServer2 {
@Override
public void completed(Integer result, Map<String, Object> attachment) {
String action = (String) attachment.get("action");
Map<String, Object> actionInfo = attachment;
String action = (String) actionInfo.get("action");
if ("read".equals(action)) {
ByteBuffer buffer = (ByteBuffer) attachment.get("buffer");
ByteBuffer buffer = (ByteBuffer) actionInfo.get("buffer");
buffer.flip();
attachment.put("action", "write");
clientChannel.write(buffer, attachment, this);
actionInfo.put("action", "write");
clientChannel.write(buffer, actionInfo, this);
buffer.clear();
} else if ("write".equals(action)) {
ByteBuffer buffer = ByteBuffer.allocate(32);
attachment.put("action", "read");
attachment.put("buffer", buffer);
clientChannel.read(buffer, attachment, this);
actionInfo.put("action", "read");
actionInfo.put("buffer", buffer);
clientChannel.read(buffer, actionInfo, this);
}
}
@ -81,6 +82,7 @@ public class AsyncEchoServer2 {
}
public static void main(String[] args) {
new AsyncEchoServer2();
}
@ -95,4 +97,4 @@ public class AsyncEchoServer2 {
return builder.start();
}
}
}