分享一個網上搜不到的「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;
}
}
總結
其實寫起來很簡單，思路也不難，但忽然間讓你來實現這個小功能的話還是挺費勁的，理不清楚就會一直卡在里面，理清楚了瞬間就念頭通達。
這個功能目前已經上線，并且運行穩定沒有任何問題，感興趣的可以收藏起來，如果有一天做聊天相關業務的話，說不定就會遇到類似的需求。
