使用 Socket 和 ServerSocket 进行 BIO 模型的演示,ServerSocket 代表的是**在服务器端的Socket**,而 Socket 代表的是**客户端的Socket**<br />![](https://cdn.nlark.com/yuque/0/2022/png/21405095/1641461703843-c8d31e4c-2481-459a-9015-bafdbffb92cf.png#clientId=u06a70c8a-6300-4&crop=0&crop=0&crop=1&crop=1&from=paste&height=359&id=u4c46567b&margin=%5Bobject%20Object%5D&originHeight=543&originWidth=764&originalType=url&ratio=1&rotation=0&showTitle=false&status=done&style=stroke&taskId=uc147f772-0804-4a38-a84c-53fb80b4e58&title=&width=505)<br />**BIO模型演示**
BIO步骤:
服务器创建 ServerSocket 对象的时候需要提供端口进行绑定(bind),实现绑定之后,ServerSocket 可以作为其他客户端与服务器进行通信时使用的端点,服务器端会一直监听的绑定端口,任何客户端应用只要把想要发送的服务器的消息发送服务器到绑定的端口,服务器就可以收到并进行处理
服务器完成绑定之后,会调用 accept 函数,该函数是一个阻塞式的调用,在没有客户端建立连接的时候,accept 函数会一直阻塞当前线程,直到客户端创建了 Socket 对象与服务器进程进行连接( 在生成客户端 Socket 时需要传入服务器所在的主机、监听端口,有这些信息才能实现与服务器 ServerSocket 的连接);一旦服务器的 accept 函数接收了客户端发起的连接将会返回一个 Socket 对象,该Socket对象可以和客户端的 Socket 进行通信
完成连接后,通过两边的端点使用 IO流 进行信息传输,两边信息交换完成后可以使用 close 函数关闭客户端的 Socket 对象与服务器端的 ServerSocket 对象
代码演示:
源码下载:
https://gitee.com/dmbjzorg/Netty
服务器端:
package study;
import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
/* 服务器端 */
public class MyServer {
private static final int port = 8888;
private static final String exit = "exit";
public static void main(String[] args) {
/* 创建ServerSocket对象并绑定端口 */
ServerSocket serverSocket = null;
try {
serverSocket = new ServerSocket(port);
System.out.println("服务器端创建serverSocket对象,开始监听"+port+"端口");
while (true){
Socket socket = serverSocket.accept(); //等待客户端连接,获取到连接后与客户端的Socket进行通信
System.out.println("服务器端收到端口为"+socket.getPort()+"客户端的Socket请求");
/* 从客户端Socket读取数据*/
InputStreamReader inputStream = new InputStreamReader(socket.getInputStream());
BufferedReader reader = new BufferedReader(inputStream);
OutputStreamWriter outputStream = new OutputStreamWriter(socket.getOutputStream());
BufferedWriter writer = new BufferedWriter(outputStream);
/* 修改收到的数据返还给服务器端 */
String lineMsg = null;
while ( (lineMsg = reader.readLine())!=null ) {
System.out.println("服务器端收到端口为"+socket.getPort()+"客户端的消息:"+lineMsg);
writer.write("服务器:"+lineMsg+"\n");
writer.flush();
/* 退出服务器端 */
if(exit.equals(lineMsg)){
System.out.println("端口为"+socket.getPort()+"的客户端断开连接");
break;
}
}
}
} catch (IOException e) {
e.printStackTrace();
}finally {
if(serverSocket!=null){
try {
serverSocket.close();
} catch (IOException e) {
e.printStackTrace();
}
System.out.println("服务端ServerSocket已关闭");
}
}
}
}
客户端:
package study;
import java.io.*;
import java.net.Socket;
/* 客户端请求 */
public class Browser {
private static final String address = "127.0.0.1";
private static final int port = 8888;
private static final String exit = "exit";
public static void main(String[] args) {
Socket socket = null;
BufferedWriter writer = null;
try {
socket = new Socket(address,port);
/* 创建IO流 */
BufferedReader reader = new BufferedReader(new InputStreamReader(socket.getInputStream()));
writer = new BufferedWriter(new OutputStreamWriter(socket.getOutputStream()));
/* 控制台输入信息 */
BufferedReader consoleReader = new BufferedReader(new InputStreamReader(System.in));
while (true){
String lineStr = consoleReader.readLine();
/* 发送消息给服务器 */
writer.write(lineStr+"\n");
writer.flush();
/* 读取服务器返回的消息 */
String msg = reader.readLine();
System.out.println("服务器返回的消息: "+msg);
/* 查看用户是否退出 */
if(exit.equals(lineStr)){
break;
}
}
} catch (IOException e) {
e.printStackTrace();
}finally {
/* 关闭 writer的时候会自动清空内部缓存区且关闭 Socket */
if(writer!=null){
try {
writer.close();
} catch (IOException e) {
e.printStackTrace();
}
System.out.println("客户端Socket已关闭");
}
}
}
}
测试:
运行 MyServer 先创建出 ServerSocket 对象,再运行 Browser 后 MyServer 获取到连接请求,使用IO流传输消息后收到指定信息进行退出操作
测试案例
聊天室案例:
用户加入到聊天室后可以看到后续每个成员的发言
BIO聊天室案例时序图
聊天室服务器端:
使用线程池对资源进行限制,由于核心数为10,因此包括服务器在内最多有9个用户可以在聊天室中进行正常交流,9位在线用户不会收到未加入到聊天室用户发送的消息
package chatroom;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
/* 聊天室服务器端 */
public class ChatServer {
private static int DEFAULT_PORT = 8888; //监听端口号
private static final String QUIT = "quit"; //用户输入quit时退出聊天
private ServerSocket serverSocket; //接收客户端的Socket
private ThreadPoolExecutor executorService; //线程池
private Map<Integer, Writer> connectedClients; //保存目前为止所有客户发送的消息,key:客户端的端口号 value:服务端向对应的客户端发送信息时的 writer
/* 初始化参数 */
public ChatServer() {
executorService = new ThreadPoolExecutor(5,10,200, TimeUnit.MILLISECONDS,new ArrayBlockingQueue<>(10));
connectedClients = new HashMap<>();
}
/* 新客户端用户加入到聊天室服务器 */
public synchronized void addClient(Socket socket) throws IOException {
if (socket!=null){
int port = socket.getPort();
BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(socket.getOutputStream()));
connectedClients.put(port,writer);
System.out.println("客户端["+port+"]已经连接到服务器");
}
}
/* 客户端用户退出聊天室服务器 */
public synchronized void removeClient(Socket socket) throws IOException {
if (socket!=null){
int port = socket.getPort();
//先判断当前客户端有没有在map中(之前有没有与客户端建立连接)
if (connectedClients.containsKey(port)){
//如果在,直接将writer流关闭
connectedClients.get(port).close();
}
connectedClients.remove(port); //将对应的值从map中去除
System.out.println("客户端["+port+"]断开连接");
}
}
/* 接收到某一客户发送的消息后将其发送给其他用户
* @param socket 发送消息的用户
* @param fwdMsg 客户端需要转发的消息
* */
public synchronized void forwardMessage(Socket socket,String fwdMsg){
connectedClients.forEach((k,v)->{
/* 如果该端口与发送消息的用户端口号不一致,则转发消息 */
if (!k.equals(socket.getPort())){
Writer writer = connectedClients.get(k);
//将消息转发给客户
try {
writer.write(fwdMsg);
writer.flush();
} catch (IOException e) {
e.printStackTrace();
}
}
});
}
/* 启动服务器端 */
public void start(){
try {
serverSocket = new ServerSocket(DEFAULT_PORT); //绑定监听端口
System.out.println("服务器已启动,监听端口:"+DEFAULT_PORT+"......");
while (true){
//等待是否有客户端需要尝试建立连接
Socket socket = serverSocket.accept();
//创建ChatHandler线程
executorService.execute(new ChatHandler(this,socket));
}
} catch (IOException e) {
e.printStackTrace();
} finally {
if (serverSocket!=null){
try {
serverSocket.close();
System.out.println("关闭ServerSocket");
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
/* 检查用户是否退出 */
public boolean readyToQuit(String msg){
return QUIT.equals(msg)?true:false;
}
public static void main(String[] args) {
ChatServer server = new ChatServer();
server.start();
}
}
聊天室建立对应关系方法:
package chatroom;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.Socket;
/* 用于建立客户端与用户一对一的交换(一个用户对应一个线程) */
public class ChatHandler implements Runnable{
private ChatServer server; //用操作存放在服务器端的map集合
private Socket socket; //建立连接的客户端socket
public ChatHandler(ChatServer server, Socket socket) {
this.server = server;
this.socket = socket;
}
@Override
public void run() {
try {
server.addClient(socket); //添加新上线用户
BufferedReader reader = new BufferedReader(new InputStreamReader(socket.getInputStream())); //读取用户发送来的消息
String msg = null;
while ( (msg = reader.readLine())!=null ){
String forMsg = "客户端["+socket.getPort()+"]"+msg+"\n";
System.out.println(forMsg);
/*
*将收到的消息转发给聊天室里在线的其他用户
*由于该方法读取的消息也是由readLine读取,而该方法要读取到\n才可以接受读取
*因此需要在msg后面加上\n,用于使读取接受
*/
server.forwardMessage(socket,forMsg);
/* 检查用户是否准备退出 */
if (server.readyToQuit(msg)){
break;
}
}
}catch (IOException e) {
e.printStackTrace();
}finally {
try {
//如果该用户退出聊天,也要将该用户的key、value从map中删除
server.removeClient(socket);
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
客户端:
package chatroom;
import java.io.*;
import java.net.Socket;
/* 用户端 */
public class ChatClient {
private final String DEFAULT_SERVER_HOST = "127.0.0.1";
private final int DEFAULT_SERVER_PORT = 8888;
private final String QUIT = "quit";
private Socket socket;
private BufferedReader reader;
private BufferedWriter writer;
/* 发送消息给服务器 */
public void send(String msg) throws IOException {
//判断该socket的输出流是否为开放的状态
if (!socket.isOutputShutdown()){
writer.write(msg+"\n");
writer.flush();
}
}
/* 从服务器接收消息 */
public String receive() throws IOException {
String msg = null;
if (!socket.isInputShutdown()){
msg = reader.readLine();
}
return msg;
}
/* 检查用户是否准备退出 */
public boolean readyToQuit(String msg){
return QUIT.equals(msg);
}
public void start(){
try {
//创建socket
socket = new Socket(DEFAULT_SERVER_HOST,DEFAULT_SERVER_PORT);
//创建用来发送和接收信息的io流
reader = new BufferedReader(new InputStreamReader(socket.getInputStream()));
writer = new BufferedWriter(new OutputStreamWriter(socket.getOutputStream()));
//处理用户的输入
new Thread(new UserInputHander(this)).start();
//读取服务器转发的各种信息
String msg = null;
while ((msg=receive())!=null){
System.out.println(msg);
}
} catch (IOException e) {
e.printStackTrace();
} finally {
if (writer!=null){
try {
System.out.println("关闭socket");
writer.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
public static void main(String[] args) {
ChatClient chatClient = new ChatClient();
chatClient.start();
}
}
客户端用户输入方法:
package chatroom;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
/* 处理用户在控制台上的输入 */
public class UserInputHander implements Runnable{
private ChatClient chatClient;
public UserInputHander(ChatClient chatClient) {
this.chatClient = chatClient;
}
@Override
public void run() {
try {
BufferedReader consoleReader = new BufferedReader(new InputStreamReader(System.in)); //等待用户输入信息
while (true){
String input = consoleReader.readLine();
chatClient.send(input); //向服务器发送消息
/* 检查用户是否准备退出 */
if (chatClient.readyToQuit(input)){
break;
}
}
} catch (IOException e) {
e.printStackTrace();
}
}
}
测试:
启动 ChatServer 后启动多个 ChatClient 实例,在不同的 ChatClient 实例窗口中模拟消息发送,每个窗口都可以收到其他用户的消息
聊天室测试图示