一、前言

此项目作为练手，这篇文章作为总结。
包名，类名起的比较随意请见谅。

后端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()));
}