分享一个网上搜不到的「Redis」实现「聊天回合制」的方案

/**
 * 聊天回合制常量
 */
public final class ChatRoundConstants {
    /**
     * 聊天回合数key前缀
     */
    public static final String CHAT_NUM = "chat-num:";
    /**
     * 聊天对象key前缀
     */
    public static final String CHAT_WHO = "chat-who:";
    /**
     * redis-key过期时间
     */
    public static final Long EXPIRE_TIME = 48 * 3600L;
    /**
     * 聊天对象value值，医生-D，患者-P。
     */
    public static final String DOCTOR = "D";
    public static final String PATIENT = "P";
}

/**
 * 发起咨询成功
 */
public void consultSuccess() {
    // ....其他业务逻辑处理
    
    // 初始化聊天回合数
    initChatRoundNum(ConsultDTO consultDTO);
}

/**
 * 初始化聊天回合数
 * -- 过期时间48小时
 * @param consultDTO 咨询信息
 */
private void initChatRoundNum(ConsultDTO consultDTO) {
    // 初始6回合
    int chatNum = 6;
    
    // 获取系统配置的默认回合数，这里是伪代码根据自己需要编写。
    ParameterDTO parameterDTO = getConfigValue();
    if(!ObjectUtils.isEmpty(parameterDTO)) {
        chatNum = parameterDTO.getPvalue();
    }
    
    // 初始化到redis，key是chat-num:consultId
    redisService.set(ChatRoundConstants.CHAT_NUM + consultDTO.getId(), 
        chatNum, ChatRoundConstants.EXPIRE_TIME);
}

/**
 * 发消息
 */
public void sendMsg() {
    // ....其他业务逻辑
    
    // 更新聊天回合数
    handleChatRoundNum(consultDTO, consultDetailInfoDTO);
}


/**
 * 处理聊天回合数
 * @param consultDTO 咨询信息
 * @param consultDetailInfoDTO 聊天信息
 */
private void handleChatRoundNum(ConsultDTO consultDTO, 
                                ConsultDetailInfoDTO consultDetailInfoDTO) {
    
    // 获取redis保存的医生患者标识key
    String chatWhoKey = ChatRoundConstants.CHAT_WHO + consultDTO.getId();
    
    // 获取当前发消息的人对应的标识
    String current = ChatWhoEnum.getCodeById(consultDetailInfoDTO.getSource());
    
    // chat-who:consultId是否存在
    if(redisService.exists(chatWhoKey)) {
    
        String chatWhoValue = (String) redisService.get(chatWhoKey);
        
        // 判断当前发消息的人和chatWho的值是否相同，如果不同，更新chatWho为当前发消息的人。
        if(!Objects.equals(ChatWhoEnum.getIdByCode(chatWhoValue), 
        consultDetailInfoDTO.getSource())) {
        
            // 更新chatWho为当前发消息的人
            redisService.setRange(chatWhoKey, current, 0);
            
            // 判断当前发消息的人是否为D，是D的话才更新回合数。
            if(Objects.equals(ChatWhoEnum.DOCTOR.getId(), 
                                consultDetailInfoDTO.getSource())) {
            
                // 更新chatNum-1
                String chatNumKey = ChatRoundConstants.CHAT_NUM + consultDTO.getId();
                int chatNumValue = Integer.parseInt(
                                        (String) redisService.get(chatNumKey)
                                    );
                if(redisService.exists(chatNumKey) && chatNumValue > 0) {
                    redisService.decr(chatNumKey);
                }
                
            }
            
        }
    } else {
        // 不存在说明是第一条消息，创建这个key。
        redisService.set(chatWhoKey, current, ChatRoundConstants.EXPIRE_TIME);
    }
}

/**
 * 聊天对象来源的枚举类
 */
public enum ChatWhoEnum {
    // 来源 ：
    // 0 医生
    // 1 患者
    DOCTOR(0, "D", "医生"),
    PATIENT(1, "P", "患者");
    
    private final int id;
    private final String code;
    private final String label;
    
    ChatWhoEnum(final int id, final String code, final String label) {
        this.id = id;
        this.code = code;
        this.label = label;
    }
    
    public int getId() {
        return id;
    }
    public String getCode() {
        return code;
    }
    public String getLabel() {
        return label;
    }
    
    public static String getCodeById(int id) {
        for(ChatWhoEnum type: ChatWhoEnum.values()) {
            if(type.getId() == id) {
                return type.getCode();
            }
        }
        return null;
    }
    
    public static Integer getIdByCode(String code) {
        for(ChatWhoEnum type: ChatWhoEnum.values()) {
            if(code.equalsIgnoreCase(type.getCode())) {
                return type.getId();
            }
        }
        return null;
    }
}