实现一个基于nio的echo服务器

Somersames · Somersames · commit 8c45cc044169 · 2017-05-04T20:07:55.000+08:00
diff --git a/src/main/java/nio/myself/MyClient.java b/src/main/java/nio/myself/MyClient.java
@@ -0,0 +1,48 @@
+package nio.myself;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.net.InetSocketAddress;
+import java.nio.ByteBuffer;
+import java.nio.channels.SocketChannel;
+
+/**
+ * Created by szh on 2017/5/4.
+ */
+public class MyClient {
+    private void startClient() throws IOException {
+        SocketChannel socketChannel = SocketChannel.open();
+        socketChannel.connect(new InetSocketAddress("127.0.0.1", 9002));
+        ByteBuffer bytebuffer = ByteBuffer.allocateDirect(1024);
+//        String message = "测试";
+        BufferedReader br = new BufferedReader(new
+                InputStreamReader(System.in));
+//        try {
+        while (true) {
+
+            String message = br.readLine();
+
+            for (int i = 0; i < message.length(); i++) {
+                bytebuffer.put((byte) message.charAt(i));
+            }
+            bytebuffer.flip();
+
+            while (bytebuffer.hasRemaining()) {
+                socketChannel.write(bytebuffer);
+            }
+
+            bytebuffer.clear();
+            socketChannel.read(bytebuffer);
+            bytebuffer.flip();
+            while (bytebuffer.hasRemaining()){
+                System.out.print((char)bytebuffer.get()); //  System.out.print 不会导致换行
+            }
+            System.out.println();
+            bytebuffer.clear();
+        }
+    }
+    public static void main(String args[]) throws IOException {
+        new MyClient().startClient();
+    }
+}
diff --git a/src/main/java/nio/myself/MyServer.java b/src/main/java/nio/myself/MyServer.java
@@ -0,0 +1,46 @@
+package nio.myself;
+
+import java.io.IOException;
+import java.net.InetSocketAddress;
+import java.nio.ByteBuffer;
+import java.nio.channels.SelectionKey;
+import java.nio.channels.Selector;
+import java.nio.channels.ServerSocketChannel;
+import java.nio.channels.SocketChannel;
+import java.util.Set;
+
+/**
+ * Created by szh on 2017/5/3.
+ */
+public class MyServer {
+    public void startServer() throws IOException {
+        ServerSocketChannel serverSocketChannel= ServerSocketChannel.open();
+        serverSocketChannel.bind(new InetSocketAddress(9002));
+        Selector selector =Selector.open(); // 打开选择器
+        serverSocketChannel.configureBlocking(false);
+        serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT); // 将新进来的通道注册到选择器当中
+        while (true){
+            selector.select();
+            Set<SelectionKey> selectionKeys =selector.selectedKeys();
+            for(SelectionKey selectionKey : selectionKeys){
+                if(selectionKey.isAcceptable()){
+                    ServerSocketChannel ssc = (ServerSocketChannel) selectionKey.channel();
+                    SocketChannel socketChannel =ssc.accept(); // 返回SocketChannel ,非阻塞模式下会立即返回
+                    socketChannel.configureBlocking(false); // 非阻塞IO
+                    socketChannel.register(selector,selectionKey.OP_READ);
+                }else if(selectionKey.isReadable()){
+                    SocketChannel socketChannel = (SocketChannel) selectionKey.channel();
+                    ByteBuffer byteBuffer =ByteBuffer.allocate(1024);
+                    socketChannel.read(byteBuffer);
+                    byteBuffer.flip();
+                    socketChannel.write(byteBuffer);
+                }
+            }
+            selectionKeys.clear();
+        }
+    }
+    public static void  main(String args[]) throws IOException {
+        new MyServer().startServer();
+    }
+
+}
