Java Socket编程（二）从单线程到多线程，实现聊天小程序

Java Socket编程（二）从单线程到多线程，实现聊天小程序

文章目录

单线程示例

实现功能

具体运行过程 代码

服务器端客户端封装的工具类 多线程示例思考

单线程示例 实现功能

服务端与客户端1对1聊天通信，需要先运行服务器端，后运行客户端，用户输入bye时，关闭socket，终止聊天。

具体运行过程

启动服务器端，创建ServerSocket，并阻塞于accept方法，等待客户端连接。启动客户端，连接服务器。服务器端接受连接，得到ServerSocket.accept( )返回的socket对象。服务器端和客户端分别创建各自监听消息的线程，之后可以通过socket的输入输出流进行信息交互。当某一方输入bye时，socket关闭，断开连接关闭相关资源，通信结束

代码 服务器端

//ServerDemo.java
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;

public class ServerDemo {
    public static void main(String[] args) {
        try {
            ServerSocket serverSocket = new ServerSocket(SocketUtils.PORT);
            Socket socket = serverSocket.accept();

            //开启接收消息线程
            new ReceiveMsg(socket).start();

            //主线程开始发送消息
            SocketUtils.sendMsg(socket, socket.getOutputStream());
            serverSocket.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

客户端

//ClientDemo.java
import java.io.IOException;
import java.net.InetAddress;
import java.net.Socket;

public class ClientDemo {
    public static void main(String[] args) {
        try {
            Socket socket = new Socket(InetAddress.getLocalHost(),SocketUtils.PORT);
            //接收消息线程开启
            new ReceiveMsg(socket).start();

            //主线程开始发送消息
            SocketUtils.sendMsg(socket, socket.getOutputStream());
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

封装的工具类

SocketUtils.java ：封装了发送用户控制台输入消息的方法，以及定义了端口常量。

import java.io.IOException;
import java.io.OutputStream;
import java.io.PrintWriter;
import java.net.Socket;
import java.util.Scanner;

public class SocketUtils {
    public static final int PORT = 8888;
    public static void sendMsg(Socket socket, OutputStream os){
        PrintWriter pw = new PrintWriter(os);
        Scanner sc = new Scanner(System.in);
        String msg;
        while (!socket.isClosed() && (msg = sc.next())!=null){
            // BufferedReader.readLine()方法直到遇到n或者r才会终止读取，是阻塞式方法
            //手动封装消息，增加n
            pw.write(msg+'n');
            pw.flush();
            if ("bye".equals(msg)){
                break;
            }
        }
        System.out.println("消息传输完毕，程序结束");
        try {
            os.close();
            socket.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

ReceiveMsg.java：创建新线程监听信息，并打印信息到控制台

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.Socket;
import java.net.SocketException;

public class ReceiveMsg extends Thread {
    InputStream inputStream;
    Socket socket;
    public ReceiveMsg(Socket socket) {
        this.socket = socket;
        try {
            this.inputStream = socket.getInputStream();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    @Override
    public void run() {
        System.out.println("线程被启动");
        //字节流转字符流
        InputStreamReader isr = new InputStreamReader(inputStream);
        //添加缓冲读数据
        BufferedReader br = new BufferedReader(isr);
        try {
            String msg;
            //当收到消息时把消息打印到控制台
            while ((msg=br.readLine())!=null) {
                System.out.println("msg:"+msg);
                if ("bye".equals(msg)){
                    System.out.println("接收消息socket关闭");
                    socket.close();
                    break;
                }
            }
        } catch (IOException e) {
            if (!(e instanceof SocketException)){
                e.printStackTrace();
            }
        }
    }
}

多线程示例

使server可以支持多个client连接并聊天

只需要增加一个线程类，稍微改动一下ServerDemo.java

//ServerDemo.java
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;

public class ServerDemo {
    public static void main(String[] args) {
        try {
            ServerSocket serverSocket = new ServerSocket(SocketUtils.PORT);
            while (true){
                Socket socket = serverSocket.accept();
                // 开启新线程 循环监听连接请求
                new ServerThread(socket).start();
            }
//            new ReceiveMsg(socket).start();
//            SocketUtils.sendMsg(socket, socket.getOutputStream());
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

增加的线程类ServerThread.java

import java.io.IOException;
import java.net.Socket;

public class ServerThread extends Thread{
    Socket socket;
    public ServerThread(Socket socket) {
        this.socket = socket;
    }
    @Override
    public void run() {
        super.run();
        //开启接收消息线程
        new ReceiveMsg(socket).start();
        //主线程开始发送消息
        try {
            SocketUtils.sendMsg(socket, socket.getOutputStream());
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

思考

不太明白的地方：

使用PrintWriter传输消息（例如从server向client传输消息），client的socket关闭之后，server发消息也不会抛出异常，不求甚解…

解决方法：server接收到bye后关闭socket，并在发送消息判断socket.isClosed()

但同时这又会带来一个问题，单线程进行聊天时，可以注意发送消息的这一段代码

while (!socket.isClosed() && (msg = sc.next())!=null){
	pw.write(msg+'n');
	pw.flush();
	if ("bye".equals(msg)){
	    break;
	}
}

当server启动之后，会阻塞于accept方法，client启动后，双方的代码都会阻塞在sc.next( )，socket.isClosed() 返回的是false。这时，任意一方输入bye关闭socket，另外一方还需要输入一行才能彻底结束。

可能可行的解决方法：通过注册–通知机制实现事件监听（观察者模式），达到实时监听socket是否关闭，还有待实现和验证…

此外，还存在一个问题，当使用多线程时，会造成聊天混乱的情况，两个连接client的线程并发运行，会导致server需要交替输入。

参考资料：

Scanner 类抛出java.util.NoSuchElementException

java 线程监听事件_java中的事件监听是怎样实现随时监听的，是通过线程吗

java-用java.net.Socket和java.net.ServerSocket实现简单的聊天程序