一、前言
此项目作为练手,这篇文章作为总结。
包名,类名起的比较随意请见谅。
后端95%的代码来自DavidDingXu
前端样式
前端代码写的不太好,我也不想改了,也就这样了。
在看这篇文章之前,请大致看一下java-socket长连接demo体验这篇文章
项目具体的实现就不介绍了,全在上面这篇文章里。
记录一下学习的过程。
二、注意点及实现方案
1 用户聊天记录保存和获取
1.1 问题说明
项目用没有用到数据库,所以用户消息缓存在内存中。假设:
A 发送给 B 一条消息 (A -> B)
B 接收到 A 一条消息 (B <- A)
B.Socket无法推送给前端HTML消息
Socket Client和Server Socket如何保存记录
1.2 解决思路
1.2.1 消息的保存
Server:消息类型如果是3(FunctionCodeEnum.MESSAGE),添加两条记录。比如:
Map<String, List<UserMessage>> map = new HashMap<>(16);
map.put("A",// List UserMessage(SEND,B,data));
map.put("B",// List UserMessage(RECEIVE,A,data));
A Client:A客户端需要在自己的集合中添加一条发送的消息。
String userId = A
List<UserMessage> list;
list.add(// UserMessage(SEND,B,data))
B Client:B客户端(接收方)接收到从Server发来的消息,保存集合。
// A Client -> Server -> B Client
String userId = B
List<UserMessage> list;
list.add(// UserMessage(RECEIVE,A,data))
当Client断开连接时候,保存在客户端的聊天记录会丢失。当用户再次登录时,可从Server中获取之前聊天记录。(此功能目前未实现)
1.2.2 消息的获取
因为后端无法主动推送给前端,解决办法有点LOW,JS 定时请求接口获取用户聊天记录。(目前没有想到更好的办法)
1.3 具体代码实现
1.3.1 用户记录实体类
public class UserMessage {
/**
* 1 接收 2 发送
* */
private Integer state;
private String userId;
private String message;
private Date date;
}
1.3.2 Server端保存记录
/**
* 存储 用户聊天记录
*/
ConcurrentHashMap<String,List<UserMessage>> userMessageMap = new ConcurrentHashMap<>();
当用户登录时,初始化集合。
// 初始化用户记录
userMessageMap.put(userId, Collections.synchronizedList(new ArrayList<>()));
服务端接收消息的类型是MESSAGE
// 消息
// 用户登录后一定存在 不需要在判断key存不存在
// A -> B A - B,message,date,2
userMessageMap.get(messageDto.getUserId()).add(
new UserMessage(messageDto.getSendUserId(), messageDto.getMessage(), new Date(), MessageCodeEnum.SEND.getValue()));
// B <- A B - A,message,date,1
userMessageMap.get(messageDto.getSendUserId()).add(
new UserMessage(messageDto.getUserId(), messageDto.getMessage(), new Date(), MessageCodeEnum.RECEIVE.getValue()));
// A.socket( uId = A, sId = B) -> message -> B.socket( uId = B, sId = A)
// message userId和sendUserId 要倒过来
String swap = messageDto.getUserId();
messageDto.setUserId(messageDto.getSendUserId());
messageDto.setSendUserId(swap);
connection.println(JSONObject.toJSONString(messageDto));
1.3.3 Client端保存记录
封装Connection,用户消息,心跳线程
public class SocketClient {
private Connection connection;
/**
* 保存消息
* */
private List<UserMessage> messageList;
private ScheduledExecutorService clientHeartExecutor;
}
缓存所有Client信息
ConcurrentMap<String, SocketClient> existSocketClientMap = new ConcurrentHashMap<>();
发送消息时直接添加到messageList中。
public void sendMessage(String userId, String sendUserId, String message) {
existSocketClientMap.get(userId).getMessageList().add(
new UserMessage(sendUserId, message, new Date(), MessageCodeEnum.SEND.getValue()));
SocketMessageDto messageDto = new SocketMessageDto();
messageDto.setFunctionCode(FunctionCodeEnum.MESSAGE.getValue());
messageDto.setUserId(userId);
messageDto.setSendUserId(sendUserId);
messageDto.setMessage(message);
existSocketClientMap.get(userId).getConnection().println(messageDto);
}
接收消息时直接添加到messageList中。
if (functionCode.equals(FunctionCodeEnum.MESSAGE.getValue())) {
// 消息
existSocketClientMap.get(socketMessage.getUserId()).getMessageList().add(
new UserMessage(socketMessage.getSendUserId(), socketMessage.getMessage(), new Date(), MessageCodeEnum.RECEIVE.getValue()));
}
未解决问题:
- 前端广播群发(没啥意思,不想写)
- 前端接收消息必须点一下。比如:
A -> B B页面必须在当前在线的其他用户点击A用户,才可以接收消息。并且没有任何提现。(没想到什么好的办法) - A用户登陆,并没有点退出,直接关闭页面。(导致Socket一直存在)。
- 心跳机制(我的项目里并没有加)
- A登陆后,无法继承上次登陆后记录。(登陆退出都没解决,还有登陆记录的问题!!!)
。。。。。。。
自我总结:
总之,这个项目确实学到了不少东西,比如handler处理(我的项目里并没有用)。熟悉了Socket,还动了脑子。总之这个练习还是蛮有意义的???
发现大佬项目有一些自己感觉不合理的地方,总之还是大佬强。希望自己以后继续努力!!!也成为一名大佬。
最后捞一下自己jiHongYuan
大佬githubDavidDingXu
大佬知乎DavidDingXu