加入多线程
实现一个命令行界面的C/S聊天室应用,服务器端应该包含多个线程,每个Socket对应一个线程,该线程负责读取Socket对应输入流的数据(从客户端发送过来的数据),并将读到的数据向每个Socket输出流发送一次(将一个客户端发送的数据“广播”给其他客户端),因此需要在服务器端使用List来保存Socket。
首先是服务器端的实现代码,一共有两个类,一个是创建ServerSocket监听的主类,一个是负责处理每个Socket通信的线程类。
监听主类:
public class MyServer {
//定义保存所有Socket的ArrayList,并将其包装为线层安全的
public static List<Socket> socketList = Collections.synchronizedList(new ArrayList<>());
public static void main(String[] args) throws IOException{
ServerSocket ss = new ServerSocket(30001);
while (true){
//此行代码会阻塞,将一直等待别人的连接
Socket s = ss.accept();
socketList.add(s);
//每当客户端连接后启动一个ServerThread线程为该客户端服务
new Thread(new ServerThread(s)).start();
}
}
}
服务器端只负责接收客户端Socket的连接请求,每当客户端Socket连接到该ServerSocket之后,程序将对应Socket加入socketList集合中保存,并为该Socket启动一个线程,该线程负责处理该Socket所有的通信任务。
负责处理Socket通信的线程类:
public class ServerThread implements Runnable{
//定义当前线程所处理的Socket
Socket s = null;
//该线程所处理的Socket对应的输入流
BufferedReader br = null;
public ServerThread(Socket s) throws IOException{
this.s = s;
//初始化该Socket对应的输入流
br = new BufferedReader(new InputStreamReader(s.getInputStream()));
}
public void run(){
try{
String content = null;
//采用循环不断地从Socket中读取客户端发送过来的数据
while ((content = readFromClient()) != null){
//遍历socketList中的每个socket
//将读到的内容向每个Socket发送一次
for(Socket s : MyServer.socketList){
PrintStream ps = new PrintStream(s.getOutputStream());
ps.println(content);
}
}
}catch (IOException e){
e.printStackTrace();
}
}
//定义读取客户端数据的方法
private String readFromClient(){
try{
return br.readLine();
}
//如果捕获异常,则表明该Socket对应的客户端已经关闭
catch (IOException e){
//删除该Socket
MyServer.socketList.remove(s);
}
return null;
}
}
每个客户端包含两个线程,一个负责读取用户的键盘输入,并将用户输入的数据写入Socket对应的输出流中;一个负责读取Socket对应输入流中的数据(从服务端发送过来的数据),并将这些数据打印输出。其中负责读取用户键盘输入的线程由MyClient负责,也就是有程序的线程负责。
客户端主程序:
public class MyClient {
public static void main(String[] args) throws Exception{
Socket s = new Socket("127.0.0.1", 30001);
//客户端启动ClientThread线程不断地读取来自服务器的数据
new Thread(new ClientThread(s)).start();
//获取该Socket对应的输出流
PrintStream ps = new PrintStream(s.getOutputStream());
String line = null;
//不断地读取键盘输入
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
while ((line = br.readLine()) != null){
//将用户的键盘输入内容写入Socket对应的输出流
ps.println(line);
}
}
}
客户端线程:
public class ClientThread implements Runnable{
//该线程负责处理的Socket
private Socket s;
//该线程所处理的Socket对应的输入流
BufferedReader br = null;
public ClientThread(Socket s) throws IOException{
this.s = s;
br = new BufferedReader(new InputStreamReader(s.getInputStream()));
}
public void run(){
try{
String content = null;
//不断地读取Socket输入流中的内容,并将这些内容打印输出
while ((content = br.readLine()) != null){
System.out.println(content);
}
}catch (Exception e){
e.printStackTrace();
}
}
}