nio 代码实现简易多人聊天

Posted 2021-01-04 hf-china

这几天在学习nio相关知识。实现了一个简单的多人聊天程序。

服务端代码;

  1 import java.io.IOException;
  2 import java.net.InetSocketAddress;
  3 import java.nio.ByteBuffer;
  4 import java.nio.channels.*;
  5 import java.nio.charset.Charset;
  6 import java.util.*;
  7 
  8 /**
  9  * @ClassName CharRoomServer
 10  * @Description TODO
 11  * @Author hufeng8
 12  * @Date 2018/8/3 14:39
 13  * @Version 1.0
 14  */
 15 public class CharRoomServer implements Runnable{
 16 
 17     private ServerSocketChannel serverSocketChannel = null;
 18 
 19     private Selector selector = null;
 20 
 21     public static final int PORT_NUM = 1198;
 22 
 23     private boolean active = true;
 24 
 25     private Charset charset = Charset.forName("UTF-8");
 26 
 27     private List<String> users  = new ArrayList<String>();
 28 
 29     private ByteBuffer byteBuffer = ByteBuffer.allocate(2*1024);
 30 
 31     public static final String protocol = "#user#";
 32 
 33     public CharRoomServer() {
 34         this.init();
 35     }
 36 
 37     public void init() {
 38         try {
 39             selector = Selector.open();
 40             serverSocketChannel = ServerSocketChannel.open();
 41             serverSocketChannel.socket().bind(new InetSocketAddress(PORT_NUM));
 42             serverSocketChannel.configureBlocking(false);
 43             serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
 44         } catch (IOException e) {
 45             e.printStackTrace();
 46         }
 47     }
 48 
 49     public void run() {
 50         System.out.println("开始监听。。。。");
 51         while (active) {
 52             try {
 53                 //非阻塞接受连接
 54 //                int s = selector.selectNow();
 55 
 56                 //阻塞连接
 57                 int s = selector.select();
 58 
 59                 System.out.println("服务端接受到连接总数"+selector.keys().size());
 60             } catch (IOException e) {
 61                 e.printStackTrace();
 62             }
 63             System.out.println("服务端接受到的选择连接数"+selector.selectedKeys().size());
 64             Iterator<SelectionKey> keys = selector.selectedKeys().iterator();
 65             while (keys.hasNext()) {
 66                 SelectionKey k = keys.next();
 67                 keys.remove();
 68                 //处理逻辑
 69                 doHandler(selector, k);
 70             }
 71         }
 72     }
 73 
 74     private void doHandler(Selector selector, SelectionKey k) {
 75         try {
 76             //连接事件
 77             if (k.isConnectable()) {
 78                 System.out.println("Connectable 连接事件");
 79             } else if (k.isAcceptable()) {
 80                 ServerSocketChannel ser = (ServerSocketChannel) k.channel();
 81                 if (ser == serverSocketChannel) {
 82                     System.out.println("同一个连接");
 83                 }
 84                 SocketChannel socketChannel = ser.accept();
 85                 socketChannel.configureBlocking(false);
 86                 socketChannel.register(selector, SelectionKey.OP_READ);
 87                 socketChannel.write(charset.encode("please enter login name:"));
 88 
 89                 //设置k为接受事件，准备接受其它请求？
 90 
 91             } else if (k.isReadable()) {
 92                 //获取客户端连接
 93                 SocketChannel socketChannel = (SocketChannel) k.channel();
 94                 StringBuffer content = new StringBuffer();
 95                 int sum = 0;
 96                 try {
 97                     byteBuffer.clear();
 98                     while ((sum = socketChannel.read(byteBuffer)) > 0) {
 99                         byteBuffer.flip();
100                         content.append(charset.decode(byteBuffer));
101                     }
102                     System.out.println(sum);
103                     //判断客户端连接关闭
104                     if (sum == -1) {
105                         socketChannel.close();
106                         System.out.println("1--关闭连接");
107                     }
108 
109                     System.out.println("服务端：监听："+ content.toString());
110                 } catch (Exception e) {
111                     System.out.println("2--关闭连接");
112                     k.cancel();
113                     if (null != socketChannel) {
114                         socketChannel.close();
115                     }
116                 }
117                 if (content.length() > 0) {
118                     //按照协议切分内容
119                     String[] contents = content.toString().split(protocol);
120                     //登陆用户
121                     if (contents != null && contents.length == 1) {
122                         String user = contents[0];
123                         if (users.contains(user)) {
124                             socketChannel.write(charset.encode("登陆用户已存在！"));
125                         } else {
126                             users.add(user);
127                             //获取在线人数
128                             int i = onlineCount(selector);
129                             //广播登陆消息给当前房间所有人
130                             brokerMessage(selector, k, "欢迎"+user+"登陆，当前第"+i+"号");
131                         }
132                     } else if (contents != null && contents.length > 1) {
133                         String message = contents[0] + "say :" + contents[1];
134                         brokerMessage(selector, k, message);
135                     }
136                 }
137             } else if (k.isWritable()) {
138 
139             }
140         } catch (Exception e) {
141             e.printStackTrace();
142         }
143     }
144 
145     /**
146      * 广播消息
147      * @param content
148      */
149     private void brokerMessage(Selector selector, SelectionKey k, String content) {
150         for (SelectionKey key : selector.keys()) {
151             if (key.channel() instanceof SocketChannel && key != k) {
152                 try {
153                     SocketChannel sc = (SocketChannel) key.channel();
154                     sc.write(charset.encode(content));
155                 } catch (IOException e1) {
156                     e1.printStackTrace();
157                 }
158             }
159         }
160     }
161 
162     /**
163      * 统计在线人数
164      * @param selector
165      * @return
166      */
167     private int onlineCount(Selector selector) {
168 
169         return 0;
170     }
171 
172     public static void main(String[] args) {
173         System.out.println("开始启动服务");
174         new Thread(new CharRoomServer()).start();
175         System.out.println("服务启动");
176     }
177 }

View Code

客户端代码;

  1 import java.io.IOException;
  2 import java.net.InetSocketAddress;
  3 import java.nio.ByteBuffer;
  4 import java.nio.channels.SelectionKey;
  5 import java.nio.channels.Selector;
  6 import java.nio.channels.SocketChannel;
  7 import java.nio.charset.Charset;
  8 import java.util.Iterator;
  9 import java.util.Scanner;
 10 
 11 /**
 12  * @ClassName ChatRoomClient
 13  * @Description TODO
 14  * @Author hufeng8
 15  * @Date 2018/8/3 16:37
 16  * @Version 1.0
 17  */
 18 public class ChatRoomClient implements Runnable{
 19 
 20 
 21     private SocketChannel channel;
 22     private Selector selector;
 23     private boolean active = true;
 24     private Charset charset = Charset.forName("UTF-8");
 25 
 26     private String name = "";
 27 
 28     public ChatRoomClient() {
 29 
 30         try {
 31             selector = Selector.open();
 32             channel = SocketChannel.open(new InetSocketAddress("localhost",CharRoomServer.PORT_NUM));
 33             channel.configureBlocking(false);
 34             channel.register(selector, SelectionKey.OP_CONNECT|SelectionKey.OP_READ);
 35 
 36         } catch (IOException e) {
 37             e.printStackTrace();
 38         }
 39     }
 40 
 41     public void run() {
 42         System.out.println("客户端监听开始：");
 43         while (active) {
 44             try {
 45                 selector.select();
 46             } catch (IOException e) {
 47                 e.printStackTrace();
 48             }
 49 //            System.out.println(selector.selectedKeys().size());
 50             Iterator<SelectionKey> iterator = selector.selectedKeys().iterator();
 51             while (iterator.hasNext()) {
 52                 SelectionKey next = iterator.next();
 53                 iterator.remove();
 54                 if (channel == next.channel()) {
 55 //                    System.out.println("同一个对象");
 56                 }
 57 //                System.out.println("channel:"+channel+", sockchannel"+next.channel()+"连接事件："+next.isConnectable()+", "+next.isReadable()+", "+next.isWritable());
 58                 if (next.isConnectable()) {
 59                     try {
 60                         System.out.println("客户端连接事件");
 61                         SocketChannel sc = (SocketChannel) next.channel();
 62                         sc.configureBlocking(false);
 63                         sc.register(selector, SelectionKey.OP_READ);
 64                     } catch (IOException e) {
 65                         e.printStackTrace();
 66                     }
 67                 } else if (next.isWritable()) {
 68                     System.out.println("客户端写事件");
 69                 } else if (next.isReadable()) {
 70                     SocketChannel sc = (SocketChannel) next.channel();
 71                     if (channel == sc) {
 72                         System.out.println("同一个对象");
 73                     }
 74                     System.out.println("客户端读取事件");
 75                     ByteBuffer bf = ByteBuffer.allocate(2*1024);
 76                     int i = 0;
 77                     StringBuffer content = new StringBuffer();
 78                     try {
 79                         while ((i = sc.read(bf)) > 0) {
 80                             bf.flip();
 81                             content.append(charset.decode(bf));
 82                         }
 83                         System.out.println(content.toString());
 84 //                        next.interestOps(SelectionKey.OP_READ);
 85                     } catch (IOException e) {
 86                         e.printStackTrace();
 87                         next.cancel();
 88                         if (sc != null) {
 89                             try {
 90                                 sc.close();
 91                             } catch (IOException e1) {
 92                                 e1.printStackTrace();
 93                             }
 94                         }
 95                     }
 96                 }
 97             }
 98         }
 99     }
100 
101     public SocketChannel getChannel() {
102         return channel;
103     }
104 
105     public String getName() {
106         return name;
107     }
108 
109     public void setName(String name) {
110         this.name = name;
111     }
112 
113     public Charset getCharset() {
114         return charset;
115     }
116 
117     public void setCharset(Charset charset) {
118         this.charset = charset;
119     }
120 
121     public static void main(String[] args) {
122         ChatRoomClient chatRoomClient = new ChatRoomClient();
123         new Thread(chatRoomClient).start();
124         String name = chatRoomClient.getName();
125         SocketChannel sc = chatRoomClient.getChannel();
126         Charset charset = chatRoomClient.getCharset();
127         System.out.println("请输入：");
128         Scanner scan = new Scanner(System.in);
129         while (scan.hasNextLine()) {
130             String data = scan.next();
131             if ("".equals(data)) {
132                 continue;
133             } else if ("".equals(name)) {
134                 name = data;
135                 data = name + CharRoomServer.protocol;
136             } else {
137                 data = name + CharRoomServer.protocol + data;
138             }
139             try {
140                 sc.write(charset.encode(data));
141             } catch (IOException e) {
142 
143             }
144         }
145 
146     }
147 }

View Code