1. client包
1.1 package-info.java
/**
* 提供帮助实现客户端功能的类
*
* @author JuJunjian
* @version 1.0 2020-11-21
*/
package com.jujunjian.client;
1.2 Client.java
package com.jujunjian.client;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.net.Socket;
import java.net.SocketException;
import java.net.UnknownHostException;
/**
* 实现了客户端的功能
*
* @author JuJunjian
* @version 1.0 2020-11-21
*/
public class Client {
private static final String LOCALHOST = "127.0.0.1";
private static final String CHARSET = "UTF-8"; // 编码方式
private static final int TERMINAL_PORT = 8888; // 服务端的端口号
private Socket socket;
private InputStream inputStream;
private OutputStream oututStream;
/**
* 实例化一个客户端对象
*/
public Client() {
try {
socket = new Socket(LOCALHOST, TERMINAL_PORT);
inputStream = socket.getInputStream();
oututStream = socket.getOutputStream();
} catch (UnknownHostException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* 向服务终端申请账号的请求
*
* @param id 用户申请的账号,id为0代表申请一个随机的账号
*/
public void applyId(int id) {
byte[] byteId = toByte(id);
try {
oututStream.write(byteId);
oututStream.flush();
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* @return 返回申请账号的结果
*/
public int getApplyIdResult() {
int result = -1;
byte[] byteResult = new byte[4];
try {
inputStream.read(byteResult);
result = toInt(byteResult);
} catch (IOException e) {
e.printStackTrace();
}
return result;
}
/**
* 调用此方法以发送消息
*
* @param message 需发送的消息
*/
public void sendMessage(String message) {
try {
byte[] byteMessage = message.getBytes(CHARSET);
int length = byteMessage.length;
byte[] head = toByte(length); // 字节流前的报头,代表了消息字节流的长度,用于辅助接收端判断是否接收完毕
// 得到最终被传输的字节流
byte[] outByte = new byte[length + head.length];
System.arraycopy(head, 0, outByte, 0, head.length);
System.arraycopy(byteMessage, 0, outByte, head.length, byteMessage.length);
oututStream.write(outByte);
oututStream.flush();
} catch (UnsupportedEncodingException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* 用于接收新消息,每次调用只可接收一条新消息
*
* @return 返回接收到的信息(字符串类型)
*/
public String startReceiveMessage() {
String receiveMessage = "";
byte[] head = new byte[4]; // 消息的长度
try {
inputStream.read(head);
int messageLength = toInt(head);
byte[] byteMessage = new byte[messageLength];
inputStream.read(byteMessage);
receiveMessage = new String(byteMessage, CHARSET);
} catch (SocketException e) {
} catch (IOException e) {
e.printStackTrace();
}
return receiveMessage;
}
/**
* 调用此方法可用结束与服务终端的通讯
*/
public void endCommunication() {
try {
int result = -1; // -1 表示结束整个群聊
byte[] end = toByte(result);
oututStream.write(end);
oututStream.flush();
socket.close();
} catch (IOException e) {
e.printStackTrace();
}
}
// 将int类型的数据转换为4位byte类型的数据
private byte[] toByte(int intData) {
byte[] byteData = new byte[4];
byteData[0] = (byte) ((intData >> 24) & 0xFF);
byteData[1] = (byte) ((intData >> 16) & 0xFF);
byteData[2] = (byte) ((intData >> 8) & 0xFF);
byteData[3] = (byte) (intData & 0xFF);
return byteData;
}
// 将4位byte类型的数据转换为int类型的数据
private int toInt(byte[] byteData) {
int intData = 0;
for (int i = 0; i < byteData.length; i++) {
int shift = (3 - i) * 8;
intData += ((byteData[i] & 0xFF) << shift);
}
return intData;
}
}
2. server包
2.1 package-info.java
/**
* 提供了一个简单的服务端,用以接收、回馈和转发客户端的消息
*
* @author JuJunjian
* @version 1.0 2020-11-20
*/
package com.jujunjian.server;
2.2 UserSupervision.java
package com.jujunjian.server;
import java.io.IOException;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.net.Socket;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
/**
* 用于存储已注册用户的信息,同时提供向已注册用户发送消息、增加用户、删除用户以及其它一些辅助功能
*
* @author JuJunjian
* @version 1.0 2020-11-20
*/
public class UserSupervision {
private static final String CHARSET = "UTF-8"; // 传输数据的编码方式
private static final int SEND_THREADS_NUMBER = 10;
private static final int MAX_SIZE = 1000; // 最大聊天人数
private static final Integer MIN_ID = 9000;
private static final Integer MAX_ID = 9999;
private Map<Integer, OutputStream> userOutputStreamMap;
private Map<Integer, Socket> userSocketMap;
private ExecutorService sendMessagePool;
private int size; // 用于记录当前用户的数量
/**
* 实例化一个管理群聊用户的对象
*/
public UserSupervision() {
userOutputStreamMap = new HashMap<>();
userSocketMap = new HashMap<>();
sendMessagePool = Executors.newFixedThreadPool(SEND_THREADS_NUMBER);
size = 0;
}
/**
* 检查注册账号是否满足格式要求
*
* @param id 需检测的账号
* @return 账号符合要求返回true;否则返回false
*/
public boolean isRightId(Integer id) {
if (id >= MIN_ID && id <= MAX_ID) {
return true;
}
return false;
}
/**
* 检测群聊人数是否已满
*
* @return 聊人数是否已满返回true;否则返回false
*/
public boolean isFull() {