说说Java网络编程

 1 /*
 2 1. 功能实现：这个类的作用就像Acceptor。它有两个比较关键的全局变量，一个就是存储在线用户信息的Map，一个就是线程池。
 3 这个类会监听端口，接收客户端的请求，然后为客户端分配工作线程。
 4 还会提供一些常用的工具方法给每个工作线程调用，比如：发送消息、添加在线用户等。
 5 */
 6 
 7 import java.io.*;
 8 import java.net.*;
 9 import java.util.Map;
10 import java.util.concurrent.*;
11 
12 public class ChatServer {
13     private int DEFAULT_PORT = 8888;
14     /**
15      * 创建一个Map存储在线用户的信息。这个map可以统计在线用户、针对这些用户可以转发其他用户发送的消息
16      * 因为会有多个线程操作这个map，所以为了安全起见用ConcurrentHashMap
17      * 在这里key就是客户端的端口号，但在实际中肯定不会用端口号区分用户，如果是web的话一般用session。
18      * value是IO的Writer，用以存储客户端发送的消息
19      */
20     private Map<Integer, Writer> map = new ConcurrentHashMap<>();
21     /**
22      * 创建线程池，线程上限为10个，如果第11个客户端请求进来，服务器会接收但是不会去分配线程处理它。
23      * 前10个客户端的聊天记录，它看不见。当有一个客户端下线时，这第11个客户端就会被分配线程，服务器显示在线
24      * 大家可以把10再设置小一点，测试看看
25      * */
26     private ExecutorService executorService = Executors.newFixedThreadPool(10);
27     //客户端连接时往map添加客户端
28     public void addClient(Socket socket) throws IOException {
29         if (socket != null) {
30             BufferedWriter writer = new BufferedWriter(
31                     new OutputStreamWriter(socket.getOutputStream())
32             );
33             map.put(socket.getPort(), writer);
34             System.out.println("Client["+socket.getPort()+"]:Online");
35         }
36     }
37 
38     //断开连接时map里移除客户端
39     public void removeClient(Socket socket) throws Exception {
40         if (socket != null) {
41             if (map.containsKey(socket.getPort())) {
42                 map.get(socket.getPort()).close();
43                 map.remove(socket.getPort());
44             }
45             System.out.println("Client[" + socket.getPort() + "]Offline");
46         }
47     }
48 
49     //转发客户端消息，这个方法就是把消息发送给在线的其他的所有客户端
50     public void sendMessage(Socket socket, String msg) throws IOException {
51         //遍历在线客户端
52         for (Integer port : map.keySet()) {
53             //发送给在线的其他客户端
54             if (port != socket.getPort()) {
55                 Writer writer = map.get(port);
56                 writer.write(msg);
57                 writer.flush();
58             }
59         }
60     }
61 
62     //接收客户端请求，并分配Handler去处理请求
63     public void start() {
64         try (ServerSocket serverSocket = new ServerSocket(DEFAULT_PORT)) {
65             System.out.println("Server Start,The Port is:"+DEFAULT_PORT);
66             while (true){
67                 //等待客户端连接
68                 Socket socket=serverSocket.accept();
69                 //为客户端分配一个ChatHandler线程
70                 executorService.execute(new ChatHandler(this, socket));
71             }
72         } catch (IOException e) {
73             e.printStackTrace();
74         }
75     }
76 
77     public static void main(String[] args) {
78         ChatServer server=new ChatServer();
79         server.start();
80     }
81 }