JAVA NIO Demo

发表于

JAVA NIO Demo

Nio 服务端

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
package com.zsr.test.nio;
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.Iterator;
/**
 * nio 服务端 Created by david.zhang
 */
public class NIOServer {
  // 通道管理器
  private Selector selector;
  /**
   * 获得一个ServerSocket通道，并对该通道做一些初始化的工作
   * 
   * @param port 绑定的端口号
   * @throws IOException
   */
  public void initServer(int port) throws IOException {
    // 获得一个ServerSocket通道
    ServerSocketChannel serverChannel = ServerSocketChannel.open();
    // 设置通道为非阻塞
    serverChannel.configureBlocking(false);
    // 将该通道对应的ServerSocket绑定到指定端口
    serverChannel.socket().bind(new InetSocketAddress(port));
    // 获得一个通道管理器
    this.selector = Selector.open();
    // 将通道管理器和该通道绑定，并为该通道注册SelectionKey.OP_ACCEPT事件,注册该事件后，
    // 当该事件到达时，selector.select()会返回，如果该事件没到达selector.select()会一直阻塞。
    serverChannel.register(selector, SelectionKey.OP_ACCEPT);
  }
  /**
   * 采用轮询的方式监听selector上是否有需要处理的事件，如果有，则进行处理
   * 
   * @throws IOException
   */
  public void listen() throws IOException {
    System.out.println("nio server listen start.");
    // 轮询访问selector
    while (true) {
      // 当注册的事件到达时，方法返回；否则,该方法会一直阻塞
      selector.select();
      // 获得selector中选中的项的迭代器，选中的项为注册的事件
      Iterator<SelectionKey> it = selector.selectedKeys().iterator();
      while (it.hasNext()) {
        SelectionKey key = it.next();
        // 删除已选的key,以防重复处理
        it.remove();
        // 客户端请求连接事件
        if (key.isAcceptable()) {
          ServerSocketChannel server = (ServerSocketChannel) key.channel();
          // 获得和客户端连接的通道
          SocketChannel channel = server.accept();
          // 设置成非阻塞
          channel.configureBlocking(false);
          // 在这里可以给客户端发送信息
          channel.write(ByteBuffer.wrap(new String("服务器收到客户端连接请求.").getBytes()));
          // 在和客户端连接成功之后，为了可以接收到客户端的信息，需要给通道设置读的权限。
          channel.register(selector, SelectionKey.OP_READ);
          // 获得了可读事件
        } else if (key.isReadable()) {
          read(key);
        }
      }
    }
  }
  /**
   * 处理读取客户端发来的信息 的事件
   * 
   * @param key
   * @throws IOException
   */
  public void read(SelectionKey key) throws IOException {
    // 服务器可读取消息:得到事件发生的Socket通道
    SocketChannel channel = (SocketChannel) key.channel();
    // 创建读取的缓冲区
    ByteBuffer buffer = ByteBuffer.allocate(1024);
    channel.read(buffer);
    byte[] data = buffer.array();
    String msg = new String(data);
    System.out.println("服务器端收到客户端信息:" + msg);
    // 将消息回送给客户端
    ByteBuffer outBuffer = ByteBuffer.wrap(new String("服务器端发送客户端消息: world.").getBytes());
    channel.write(outBuffer);
  }
  // 服务器端测试
  public static void main(String[] args) throws IOException {
    NIOServer nioServer = new NIOServer();
    nioServer.initServer(8000);
    nioServer.listen();
  }
}

Nio 客户端

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
package com.zsr.test.nio;
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.SocketChannel;
import java.util.Iterator;
/**
 * nio 客户端 Created by david.zhang
 */
public class NIOClient {
  // 通道管理器
  private Selector selector;
  /**
   * 获得一个Socket通道，并对该通道做一些初始化的工作
   * 
   * @param ip 连接的服务器的ip
   * @param port 连接的服务器的端口号
   * @throws IOException
   */
  public void initClient(String ip, int port) throws IOException {
    // 获得一个Socket通道
    SocketChannel channel = SocketChannel.open();
    // 设置通道为非阻塞
    channel.configureBlocking(false);
    // 获得一个通道管理器
    this.selector = Selector.open();
    // 客户端连接服务器,其实方法执行并没有实现连接，需要在listen（）方法中调
    // 用channel.finishConnect();才能完成连接
    channel.connect(new InetSocketAddress(ip, port));
    // 将通道管理器和该通道绑定，并为该通道注册SelectionKey.OP_CONNECT事件。
    channel.register(selector, SelectionKey.OP_CONNECT);
  }
  /**
   * 采用轮询的方式监听selector上是否有需要处理的事件，如果有，则进行处理
   * 
   * @throws IOException
   */
  public void listen() throws IOException {
    System.out.println("nio client liesten start.");
    // 轮询访问selector
    while (true) {
      // 当注册的事件到达时，方法返回；否则,该方法会一直阻塞
      selector.select();
      // 获得selector中选中的项的迭代器，选中的项为注册的事件
      Iterator<SelectionKey> it = selector.selectedKeys().iterator();
      while (it.hasNext()) {
        SelectionKey key = it.next();
        // 删除已选的key,以防重复处理
        it.remove();
        // 连接事件发生
        if (key.isConnectable()) {
          SocketChannel channel = (SocketChannel) key.channel();
          // 如果正在连接，则完成连接
          if (channel.isConnectionPending()) {
            channel.finishConnect();
          }
          // 设置非阻塞
          channel.configureBlocking(false);
          // 向服务器端发送信息
          channel.write(ByteBuffer.wrap(new String("客户端请求连接服务器端.").getBytes()));
          // 在和服务端连接成功之后，为了可以接收到服务端的信息，需要给通道设置读的权限。
          channel.register(selector, SelectionKey.OP_READ);
          // 获得了可读的事件
        } else if (key.isReadable()) {
          read(key);
        }
      }
    }
  }
  /**
   * 处理读取服务器端发来的信息 的事件
   * 
   * @param key
   * @throws IOException
   */
  public void read(SelectionKey key) throws IOException {
    // 服务器可读取消息:得到事件发生的Socket通道
    SocketChannel channel = (SocketChannel) key.channel();
    // 创建读取的缓冲区
    ByteBuffer buffer = ByteBuffer.allocate(1024);
    channel.read(buffer);
    byte[] data = buffer.array();
    String msg = new String(data);
    System.out.println("客户端收到服务器信息:" + msg);
    ByteBuffer outBuffer = ByteBuffer.wrap(new String("客户端发送服务器消息: hello").getBytes());
    channel.write(outBuffer);
  }
  // 客户端测试
  public static void main(String[] args) throws IOException {
    NIOClient nioClient = new NIOClient();
    nioClient.initClient("127.0.0.1", 8000);
    nioClient.listen();
  }
}

热评文章