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

前言

为什么说网上搜不到,因为关于聊天回合制的方案作者本人快把百度搜秃噜了也没找到,好在最终是公司一个关系不错的大佬帮提供了点思路,最终作者将其完整实现了出来。

分享出来大家可以收藏,万一你哪天也碰到这样的需求,可不就节省大把时间了吗。

场景

先说下我这边的场景,读过我文章的同好都知道,我是做互联网医疗行业的,我们的项目中是包含聊天功能的,我们服务的对象主要是医院的医生,患者在网上找医生问诊时,往往会出现不停问的情况。

医生目前唯一的做法是自己结束这个咨询,或等待系统自动结束,这就带来了一个问题,不管是系统结束还是医生手动结束,患者都喜欢投诉和打差评,导致医生不敢擅自结束,问烦了又不好不回复,不回复也要被投诉。

最终聊天回合制这个需求就摆出来了,主动告诉患者我们的聊天是有回合的,所以你要一次问清楚,回合数满了我们不会再回复,如果患者硬要投诉,医生也可以说,这是做这个产品的公司自己设定的。

结下来就是,我们要把锅端好。

实际上,聊天回合制的诞生,基本上都和这个场景的诉求类似,为了减少用户频繁且无休止的咨询。

思路

结合redis能够很好的实现聊天回合制,当然也可以直接通过数据库来实现,但显然redis操作更简单性能更优越。

总体思路如下:

1)、redis中存储两个key,一个是表示对象,声明为chat-who:consultId,value为对象标识,比如这里就是医生和患者,医生用D标识,患者用P标识;另一个key是表示回合数,声明为chat-num:consultId,value就是当前回合数。这里的consultId是动态的,表示这个咨询的id,可以根据自己的业务来定;

2)、这两个key的过期时间我们都定为2天,具体过期时间要根据自己业务规则来适配;

3)、我们在特定的位置进行初始化,只要是进入聊天之前都可以,比如这里的场景,就是患者发起咨询成功后才开始聊天,我们就在发起成功后的方法中初始化聊天回合数为默认值6个回合,这个默认值还可以做成配置的形式进行动态读取的;

4)、我们在发消息的方法中做一个判断,获取redis中的chat-who:consultId,看是否存在,存在就往下执行,不存在就说明发的是第一条消息,那就创建chat-who:consultId这个key到redis中,value为当前发消息人D或者P;

5)、承接4,如果chat-who存在,我们继续将当前发消息的对象和redis的chat-who存储的对象值进行比较,如果一样,则跳过不管,如果不一样,更新chat-who的值为当前发消息的人。同时,我们判断当前发消息的人是不是医生也就是D,是D的话才更新回合数,执行-1操作,这样做的目的是把医生作为回合数更新的维度,维度只能有一个,这样才能保证回合数更新最准确。

实现

接下来,我使用伪代码把整个思路写出来。

1、定义redis-key

/**
 * 聊天回合制常量
 */
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";
}

2、初始化聊天回合数

在聊天之前初始化,这里我们项目的场景是患者发起咨询成功后,就在这个成功后的方法中初始化。

/**
 * 发起咨询成功
 */
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);
}

3、更新回合数

这里是核心逻辑,主要分为两步:初始化chat-who:consultId,更新chat-num:consultId。

/**
 * 发消息
 */
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;
    }
}

总结

其实写起来很简单,思路也不难,但忽然间让你来实现这个小功能的话还是挺费劲的,理不清楚就会一直卡在里面,理清楚了瞬间就念头通达。

这个功能目前已经上线,并且运行稳定没有任何问题,感兴趣的可以收藏起来,如果有一天做聊天相关业务的话,说不定就会遇到类似的需求。


本人原创文章纯手打,觉得有一滴滴帮助就请点个推荐吧~

本人持续分享实际工作经验和主流技术,喜欢的话可以关注下哦~