diff --git a/core-java/src/test/java/com/baeldung/java/nio2/async/AsyncEchoClient.java b/core-java/src/test/java/com/baeldung/java/nio2/async/AsyncEchoClient.java index eaaf3dea06..576d1f7a3f 100644 --- a/core-java/src/test/java/com/baeldung/java/nio2/async/AsyncEchoClient.java +++ b/core-java/src/test/java/com/baeldung/java/nio2/async/AsyncEchoClient.java @@ -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 writeResult = client.write(buffer); - //run some code - writeResult.get(); + try { + writeResult.get(); + } catch (Exception e) { + e.printStackTrace(); + } buffer.flip(); Future 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; diff --git a/core-java/src/test/java/com/baeldung/java/nio2/async/AsyncEchoServer2.java b/core-java/src/test/java/com/baeldung/java/nio2/async/AsyncEchoServer2.java index a7432da464..03ce233ce9 100644 --- a/core-java/src/test/java/com/baeldung/java/nio2/async/AsyncEchoServer2.java +++ b/core-java/src/test/java/com/baeldung/java/nio2/async/AsyncEchoServer2.java @@ -58,18 +58,19 @@ public class AsyncEchoServer2 { @Override public void completed(Integer result, Map attachment) { - String action = (String) attachment.get("action"); + Map 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(); } -} \ No newline at end of file +}