Java多线程实现多人聊天室功能

package works;

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

public class ServerChat {

 //定义Map集合用于存储用户的Socket以及用户的名字   key:Socket    Value:用户名
 public final static Map<Socket,String> socketsMaps = Collections.synchronizedMap(new HashMap<Socket,String>());
 public static void main(String[] args) {
  // TODO Auto-generated method stub
  try {
   // 创建服务端套接字
   ServerSocket serverSocket = new ServerSocket(9999); 
System.out.println("------服务端暴露-------");
   while (true) {
    // 监听客户端套接字，若有客户端连接，则代码不会往下执行，否则会堵塞在此处。
    Socket socket = serverSocket.accept();

    // 开启线程，用于读取客户端发送的信息，并转发给每一个客户端
    new ThreadServer(socket).start();
   }
  } catch (Exception e) {
   // TODO: handle exception
   e.printStackTrace();
  }
 }
}

class ThreadServer extends Thread {
 private Socket socket;
 ThreadServer(){};
 ThreadServer(Socket socket)
 {
  this.socket = socket;
 }
 @Override
 public void run() {
  try {
   while(true)
   {
    DataInputStream dataInputStream = new DataInputStream(socket.getInputStream());
    String data = dataInputStream.readUTF();
    
    if(data.startsWith("①②③④")&&data.endsWith("①②③④"))
    {
     //发送过来的是用户名
     //将Socket以及用户名字都存放在Map集合中
     ServerChat.socketsMaps.put(socket, data.replace("①②③④",""));
     //获取所有的key(Socket)，将所有用户的名字发送至客户端
     Set<Socket> sockets = ServerChat.socketsMaps.keySet();
     //获取所有的用户的名字，将这些名字拼装成一个字符串
     Collection<String> names = ServerChat.socketsMaps.values();
     StringBuffer sbf = new StringBuffer();
     for(String userName :names)
     {
      sbf.append(userName).append(",");
     }
     System.out.println("sbf:"+sbf.toString());
     for(Socket soc:sockets)
     {
      DataOutputStream dataOutputStream = new DataOutputStream(soc.getOutputStream());
      dataOutputStream.writeUTF("①②③④"+sbf.toString()+"①②③④");
      dataOutputStream.flush();
     }
    }
    else{
     //发送过来的是聊天信息
     //获取所有的key(Socket)，将所有用户的名字发送至客户端
     Set<Socket> sockets = ServerChat.socketsMaps.keySet();
     //?⒘奶煨畔⒐悴コ鋈
     for(Socket soc:sockets)
     {
      DataOutputStream dataOutputStream = new DataOutputStream(soc.getOutputStream());
      dataOutputStream.writeUTF("[ "+ServerChat.socketsMaps.get(socket)+" ]说："+data);
      dataOutputStream.flush();
     }
     
     
    }
    
   }
  } catch (IOException e) {
   // TODO Auto-generated catch block
   e.printStackTrace();
  }
 }
}

package works;

import java.awt.BorderLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.net.Socket;

import javax.swing.Box;
import javax.swing.BoxLayout;
import javax.swing.DefaultListModel;
import javax.swing.JButton;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JList;
import javax.swing.JOptionPane;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTextArea;

public class ClientChat {

 private JFrame mainWin = new JFrame("聊天窗口");

 // 消息展示框
 private JTextArea displayTa = new JTextArea(14, 40);
 // 在线用户名称展示框
 private DefaultListModel<String> userListModel = new DefaultListModel<>();
 private JList<String> userList = new JList<>(userListModel);
 // 消息发送框
 private JTextArea inputTF = new JTextArea(4, 40);
 // 消息按钮
 private JButton sendBn = new JButton("发送");
 // 用户记录当前聊天用户名
 private String curUser;

 public static void main(String[] args) {
  new ClientChat().init();
 }

 private void init() {
  try {
   // 通过弹出对话框获取用户输入的用户名
   String userName = JOptionPane.showInputDialog(mainWin, "请输入您的用户名：");
   // 把用户输入的用户名，赋给curUser
   curUser = userName;
   mainWin.setTitle(curUser + "的聊天窗口");

   // 创建套接字
   Socket socket = new Socket("192.168.193.1", 9999);
   // 向服务器声明
   DataOutputStream dataOutputStream = new DataOutputStream(socket.getOutputStream());
   //发送用户名到服务端
   dataOutputStream.writeUTF("①②③④"+userName+"①②③④");
   dataOutputStream.flush();

   // 开启线程，用于读取服务器发送的信息
   new ThreadClient(socket, this).start();

   JPanel bottomPanel = new JPanel();

   // 将消息框和按钮添加到窗口的底端
   mainWin.add(bottomPanel, BorderLayout.SOUTH);
   bottomPanel.add(inputTF);
   bottomPanel.add(sendBn);

   ActionListener listener = new ActionListener() {
    @Override
    public void actionPerformed(ActionEvent e) {
     // 获取用户发送的消息
     String message = inputTF.getText();
     sendSms(message,socket);
    }
   };
   // 给发送消息按钮绑定点击事件监听器
   sendBn.addActionListener(listener);

   JPanel centerPanel = new JPanel();

   // 将展示消息区centerPanel添加到窗口的中间
   mainWin.add(centerPanel);
   // 让展示消息区可以滚动
   centerPanel.add(new JScrollPane(displayTa));
   displayTa.setEditable(false);
   // 用户列表和是否私聊放到窗口的最右边
   Box rightBox = new Box(BoxLayout.Y_AXIS);
   userList.setFixedCellWidth(60);
   userList.setVisibleRowCount(13);
   rightBox.add(new JLabel("用户列表："));
   rightBox.add(new JScrollPane(userList));

   centerPanel.add(rightBox);

   // 关闭窗口退出当前程序
   mainWin.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
   mainWin.pack(); // swing加上这句就可以拥有关闭窗口的功能
   mainWin.setVisible(true);

  } catch (Exception e) {
   // TODO: handle exception
   e.printStackTrace();
  }
 }
  //点击发送后将消息发送到服务器
  protected void sendSms(String sms, Socket socket) {
   try {
    //发送聊天消息到服务端
    DataOutputStream dataOutputStream = new DataOutputStream(socket.getOutputStream());
    dataOutputStream.writeUTF(sms);
    dataOutputStream.flush(); 
     
   } catch (Exception e) {
    // TODO: handle exception
    e.printStackTrace();
   }
  }

  public DefaultListModel<String> getUserListModel() {
   return userListModel;
  }

  public JTextArea getDisplayTa() {
   return displayTa;
  }
  public JTextArea getInputTF()
  {
   return inputTF;
  }
}

// 定义线程类，用来读取服务器发送的信息
class ThreadClient extends Thread {
 private Socket socket;
 private ClientChat clientChat;

 ThreadClient() {
 }

 ThreadClient(Socket socket, ClientChat clientChat) {
  this.socket = socket;
  this.clientChat = clientChat;
 }

 @Override
 public void run() {

  try {
   while (true) {
    DataInputStream DataInputStream = new DataInputStream(socket.getInputStream());
    String message = DataInputStream.readUTF();

    if(message.startsWith("①②③④")&&message.endsWith("①②③④"))
    {
     //说明信息是用户名
     String[] names = message.replace("①②③④","").split(",");
     // 将用户列表先清空
     clientChat.getUserListModel().clear();
     for (int i = 0; i < names.length; ++i) {
      clientChat.getUserListModel().addElement(names[i]);
     }
    }
    else
    {
     //说明是聊天信息，将聊天信息放在displayTa中
     clientChat.getInputTF().setText("");
     clientChat.getDisplayTa().append(message+"\t\n");
    }

    
   }
  } catch (IOException e) {
   // TODO Auto-generated catch block
    e.printStackTrace();
   }
  }
}