前一陣子實(shí)現(xiàn)了我開源項(xiàng)目的單聊功能,在實(shí)現(xiàn)過程中遇到了需要將聊天記錄保存至數(shù)據(jù)庫的問題,在收到消息時(shí)肯定不能直接存數(shù)據(jù)庫,因?yàn)檫@樣在高并發(fā)的場景下,數(shù)據(jù)庫就炸了。
于是,我就想到了redis這個(gè)東西,第一次聽說它是在2年前,但是一直沒時(shí)間玩他,現(xiàn)在終于遇到了需要使用它的場景,在用的時(shí)候?qū)W它,本文就跟大家分享下我的實(shí)現(xiàn)思路以及過程,歡迎各位感興趣的開發(fā)者閱讀本文。
我的項(xiàng)目是基于SpringBoot2.x搭建的,電腦已經(jīng)安裝了redis,用的maven作為jar包管理工具,所以只需要在maven中添加需要的依賴包即可,如果你用的是其他管理工具,請自行查閱如何添加依賴。
!-- Redis --> dependency> groupId>org.springframework.boot/groupId> artifactId>spring-boot-starter-data-redis/artifactId> /dependency> !-- 定時(shí)任務(wù)調(diào)度 --> dependency> groupId>org.springframework.boot/groupId> artifactId>spring-boot-starter-quartz/artifactId> version>2.3.7.RELEASE/version> /dependency>
本文需要用到依賴:Redis 、quartz,在pom.xml文件的dependencies標(biāo)簽下添加下述代碼。
spring: # redis配置 redis: host: 127.0.0.1 # redis地址 port: 6379 # 端口號 password: # 密碼 timeout: 3000 # 連接超時(shí)時(shí)間,單位毫秒
在websocket的服務(wù)中,收到客戶端推送的消息后,我們對數(shù)據(jù)進(jìn)行解析,構(gòu)造聊天記錄實(shí)體類,將其保存至redis中,最后我們使用quartz設(shè)置定時(shí)任務(wù)將redis的數(shù)據(jù)定時(shí)寫入mysql中。
我們將上述思路進(jìn)行下整理:
實(shí)現(xiàn)思路很簡單,難在如何將實(shí)體類數(shù)據(jù)保存至redis,我們需要把redis這一塊配置好后,才能繼續(xù)實(shí)現(xiàn)我們的業(yè)務(wù)需求。
redis支持的數(shù)據(jù)結(jié)構(gòu)類型有:
redis的客戶端分為jedis 和 lettuce,在SpringBoot2.x中默認(rèn)客戶端是使用lettuce實(shí)現(xiàn)的,因此我們不用做過多配置,在使用的時(shí)候通過RedisTemplate.xxx來對redis進(jìn)行操作即可。
在RedisTemplate中,默認(rèn)是使用Java字符串序列化,將字符串存入redis后可讀性很差,因此,我們需要對他進(jìn)行自定義,使用Jackson 序列化,以 JSON 方式進(jìn)行存儲。
我們在項(xiàng)目的config包下,創(chuàng)建一個(gè)名為LettuceRedisConfig的Java文件,我們再此文件中配置其默認(rèn)序列化規(guī)則,它的代碼如下:
package com.lk.config; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.data.redis.connection.RedisConnectionFactory; import org.springframework.data.redis.core.RedisTemplate; import org.springframework.data.redis.serializer.GenericJackson2JsonRedisSerializer; import org.springframework.data.redis.serializer.StringRedisSerializer; // 自定義RedisTemplate設(shè)置序列化器, 方便轉(zhuǎn)換redis中的數(shù)據(jù)與實(shí)體類互轉(zhuǎn) @Configuration public class LettuceRedisConfig { /** * Redis 序列化配置 */ @Bean public RedisTemplateString, Object> redisTemplate(RedisConnectionFactory connectionFactory) { RedisTemplateString, Object> redisTemplate = new RedisTemplate>(); redisTemplate.setConnectionFactory(connectionFactory); // 使用GenericJackson2JsonRedisSerializer替換默認(rèn)序列化 GenericJackson2JsonRedisSerializer jackson2JsonRedisSerializer = new GenericJackson2JsonRedisSerializer(); // 設(shè)置 Key 和 Value 的序列化規(guī)則 redisTemplate.setKeySerializer(new StringRedisSerializer()); redisTemplate.setValueSerializer(jackson2JsonRedisSerializer); redisTemplate.setHashKeySerializer(new StringRedisSerializer()); redisTemplate.setHashValueSerializer(jackson2JsonRedisSerializer); // 初始化 RedisTemplate 序列化完成 redisTemplate.afterPropertiesSet(); return redisTemplate; } }
做完上述操作后,通過RedisTemplate存儲到redis中的數(shù)據(jù)就是json形式的了,接下來我們對其常用的操作封裝成工具類,方便我們在項(xiàng)目中使用。
在Utils包中創(chuàng)建一個(gè)名為RedisOperatingUtil,其代碼如下:
package com.lk.utils; import org.springframework.data.redis.connection.DataType; import org.springframework.data.redis.core.RedisTemplate; import org.springframework.stereotype.Component; import javax.annotation.Resource; import java.util.Arrays; import java.util.Collections; import java.util.List; import java.util.Map; import java.util.concurrent.TimeUnit; @Component // Redis操作工具類 public class RedisOperatingUtil { @Resource private RedisTemplateObject, Object> redisTemplate; /** * 指定 key 的過期時(shí)間 * * @param key 鍵 * @param time 時(shí)間(秒) */ public void setKeyTime(String key, long time) { redisTemplate.expire(key, time, TimeUnit.SECONDS); } /** * 根據(jù) key 獲取過期時(shí)間(-1 即為永不過期) * * @param key 鍵 * @return 過期時(shí)間 */ public Long getKeyTime(String key) { return redisTemplate.getExpire(key, TimeUnit.SECONDS); } /** * 判斷 key 是否存在 * * @param key 鍵 * @return 如果存在 key 則返回 true,否則返回 false */ public Boolean hasKey(String key) { return redisTemplate.hasKey(key); } /** * 刪除 key * * @param key 鍵 */ public Long delKey(String... key) { if (key == null || key.length 1) { return 0L; } return redisTemplate.delete(Arrays.asList(key)); } /** * 獲取 Key 的類型 * * @param key 鍵 */ public String keyType(String key) { DataType dataType = redisTemplate.type(key); assert dataType != null; return dataType.code(); } /** * 批量設(shè)置值 * * @param map 要插入的 key value 集合 */ public void barchSet(MapString, Object> map) { redisTemplate.opsForValue().multiSet(map); } /** * 批量獲取值 * * @param list 查詢的 Key 列表 * @return value 列表 */ public ListObject> batchGet(ListString> list) { return redisTemplate.opsForValue().multiGet(Collections.singleton(list)); } /** * 獲取指定對象類型key的值 * * @param key 鍵 * @return 值 */ public Object objectGetKey(String key) { return redisTemplate.opsForValue().get(key); } /** * 設(shè)置對象類型的數(shù)據(jù) * * @param key 鍵 * @param value 值 */ public void objectSetValue(String key, Object value) { redisTemplate.opsForValue().set(key, value); } /** * 向list的頭部插入一條數(shù)據(jù) * * @param key 鍵 * @param value 值 */ public Long listLeftPush(String key, Object value) { return redisTemplate.opsForList().leftPush(key, value); } /** * 向list的末尾插入一條數(shù)據(jù) * * @param key 鍵 * @param value 值 */ public Long listRightPush(String key, Object value) { return redisTemplate.opsForList().rightPush(key, value); } /** * 向list頭部添加list數(shù)據(jù) * * @param key 鍵 * @param value 值 */ public Long listLeftPushAll(String key, ListObject> value) { return redisTemplate.opsForList().leftPushAll(key, value); } /** * 向list末尾添加list數(shù)據(jù) * * @param key 鍵 * @param value 值 */ public Long listRightPushAll(String key, ListObject> value) { return redisTemplate.opsForList().rightPushAll(key, value); } /** * 通過索引設(shè)置list元素的值 * * @param key 鍵 * @param index 索引 * @param value 值 */ public void listIndexSet(String key, long index, Object value) { redisTemplate.opsForList().set(key, index, value); } /** * 獲取列表指定范圍內(nèi)的list元素,正數(shù)則表示正向查找,負(fù)數(shù)則倒敘查找 * * @param key 鍵 * @param start 開始 * @param end 結(jié)束 * @return boolean */ public Object listRange(String key, long start, long end) { return redisTemplate.opsForList().range(key, start, end); } /** * 從列表前端開始取出數(shù)據(jù) * * @param key 鍵 * @return 結(jié)果數(shù)組對象 */ public Object listPopLeftKey(String key) { return redisTemplate.opsForList().leftPop(key); } /** * 從列表末尾開始遍歷取出數(shù)據(jù) * * @param key 鍵 * @return 結(jié)果數(shù)組 */ public Object listPopRightKey(String key) { return redisTemplate.opsForList().rightPop(key); } /** * 獲取list長度 * * @param key 鍵 * @return 列表長度 */ public Long listLen(String key) { return redisTemplate.opsForList().size(key); } /** * 通過索引獲取list中的元素 * * @param key 鍵 * @param index 索引(index>=0時(shí),0 表頭,1 第二個(gè)元素,依次類推;index0時(shí),-1,表尾,-2倒數(shù)第二個(gè)元素,依次類推) * @return 列表中的元素 */ public Object listIndex(String key, long index) { return redisTemplate.opsForList().index(key, index); } /** * 移除list元素 * * @param key 鍵 * @param count 移除數(shù)量("負(fù)數(shù)"則從列表倒敘查找刪除 count 個(gè)對應(yīng)的值; "整數(shù)"則從列表正序查找刪除 count 個(gè)對應(yīng)的值;) * @param value 值 * @return 成功移除的個(gè)數(shù) */ public Long listRem(String key, long count, Object value) { return redisTemplate.opsForList().remove(key, count, value); } /** * 截取指定范圍內(nèi)的數(shù)據(jù), 移除不是范圍內(nèi)的數(shù)據(jù) * @param key 操作的key * @param start 截取開始位置 * @param end 截取激素位置 */ public void listTrim(String key, long start, long end) { redisTemplate.opsForList().trim(key, start, end); } }
做完上述操作后,最難弄的一關(guān)我們就已經(jīng)搞定了,接下來我們來對一會需要使用的方法進(jìn)行單元測試,確保其能夠正常運(yùn)行。
創(chuàng)建一個(gè)名為RedisTest的Java文件,注入需要用到的相關(guān)類。
@RunWith(SpringRunner.class) @SpringBootTest @Slf4j public class RedisTest { @Resource private RedisOperatingUtil redisOperatingUtil; @Resource private SubMessageMapper subMessageMapper; }
接下來,我們看下SubMessage實(shí)體類的代碼。
package com.lk.entity; import lombok.AllArgsConstructor; import lombok.Getter; import lombok.NoArgsConstructor; import lombok.Setter; @Getter @Setter @NoArgsConstructor @AllArgsConstructor // 聊天記錄-消息內(nèi)容 public class SubMessage { private Integer id; private String msgText; // 消息內(nèi)容 private String createTime; // 創(chuàng)建時(shí)間 private String userName; // 用戶名 private String userId; // 推送方用戶id private String avatarSrc; // 推送方頭像 private String msgId; // 接收方用戶id private Boolean status; // 消息狀態(tài) }
在單元測試類內(nèi)部加入下述代碼:
@Test public void testSerializableListRedisTemplate() { // 構(gòu)造聊天記錄實(shí)體類數(shù)據(jù) SubMessage subMessage = new SubMessage(); subMessage.setAvatarSrc("https://www.kaisir.cn/uploads/1ece3749801d4d45933ba8b31403c685touxiang.jpeg"); subMessage.setUserId("1090192"); subMessage.setUserName("神奇的程序員"); subMessage.setMsgText("你好"); subMessage.setMsgId("2901872"); subMessage.setCreateTime("2020-12-12 18:54:06"); subMessage.setStatus(false); // 將聊天記錄對象保存到redis中 redisOperatingUtil.listRightPush("subMessage", subMessage); // 獲取list中的數(shù)據(jù) Object resultObj = redisOperatingUtil.listRange("subMessage", 0, redisOperatingUtil.listLen("subMessage")); // 將Object安全的轉(zhuǎn)為List ListSubMessage> resultList = ObjectToOtherUtil.castList(resultObj, SubMessage.class); // 遍歷獲取到的結(jié)果 if (resultList != null) { for (SubMessage message : resultList) { System.out.println(message.getUserName()); } } }
在上述代碼中,我們從redis中取出的數(shù)據(jù)是Object類型的,我們要將它轉(zhuǎn)換為與之對應(yīng)的實(shí)體類,一開始我是用的類型強(qiáng)轉(zhuǎn),但是idea會報(bào)黃色警告,于是就寫了一個(gè)工具類用于將Object對象安全的轉(zhuǎn)換為與之對應(yīng)的類型,代碼如下:
package com.lk.utils; import java.util.ArrayList; import java.util.List; public class ObjectToOtherUtil { public static T> ListT> castList(Object obj, ClassT> clazz) { ListT> result = new ArrayList>(); if (obj instanceof List?>) { for (Object o : (List?>) obj) { result.add(clazz.cast(o)); } return result; } return null; } }
執(zhí)行后,我們看看redis是否有保存到我們寫入的數(shù)據(jù),如下所示,已經(jīng)成功保存。
我們再來看看,代碼的執(zhí)行結(jié)果,看看有沒有成功獲取到數(shù)據(jù),如下圖所示,也成功取到了。
注意:如果你的項(xiàng)目對websocket進(jìn)行了啟動配置,可能會導(dǎo)致單元測試失敗,報(bào)錯(cuò)java.lang.IllegalStateException: Failed to load ApplicationContext,解決方案就是注釋掉websocket配置文件中的@Configuration即可。
當(dāng)我們把redis中存儲的數(shù)據(jù)遷移到mysql后,需要刪除redis中的數(shù)據(jù),一開始我用的是它的delete方法,但是他的delete方法只能刪除與之匹配的值,不能選擇一個(gè)區(qū)間進(jìn)行刪除,于是就決定用它的pop方法進(jìn)行出棧操作。
我們來測試下工具類中的listPopLeftKey方法。
@Test public void testListPop() { long item = 0; // 獲取存儲在redis中聊天記錄的條數(shù) long messageListSize = redisOperatingUtil.listLen("subMessage"); for (int i = 0; i messageListSize; i++) { // 從頭向尾取出鏈表中的元素 SubMessage messageResult = (SubMessage) redisOperatingUtil.listPopLeftKey("subMessage"); log.info(messageResult.getMsgText()); item++; } log.info(item+"條數(shù)據(jù)已成功取出"); }
執(zhí)行結(jié)果如下所示,成功取出了redis中存儲的兩條數(shù)據(jù)。
接下來我們在redis中放入三條數(shù)據(jù)用于測試
我們測試下將redis中的數(shù)據(jù)取出,然后寫入數(shù)據(jù)庫,代碼如下:
// 測試聊天記錄轉(zhuǎn)移數(shù)據(jù)庫 @Test public void testRedisToMysqlTask() { // 獲取存儲在redis中聊天記錄的條數(shù) long messageListSize = redisOperatingUtil.listLen("subMessage"); // 寫入數(shù)據(jù)庫的數(shù)據(jù)總條數(shù) long resultCount = 0; for (int i = 0; i messageListSize; i++) { // 從頭到尾取出鏈表中的元素 SubMessage subMessage= (SubMessage) redisOperatingUtil.listPopLeftKey("subMessage"); // 向數(shù)據(jù)庫寫入數(shù)據(jù) int result = subMessageMapper.addMessageTextInfo(subMessage); if (result > 0) { // 寫入成功 resultCount++; } } log.info(resultCount+ "條聊天記錄,已寫入數(shù)據(jù)庫"); }
執(zhí)行結(jié)果如下,數(shù)據(jù)已成功寫入數(shù)據(jù)庫且redis中的數(shù)據(jù)也被刪除。
完成上述操作后,我們r(jià)edis那一塊的東西就搞定了,接下來就可以實(shí)現(xiàn)將客戶端的數(shù)據(jù)存到redis里了。
這里有個(gè)坑,因?yàn)閣ebsocket服務(wù)類中用到了@Component,會導(dǎo)致redis的工具類注入失敗,出現(xiàn)null的情況,解決這個(gè)問題需要將當(dāng)前類名聲明為靜態(tài)變量,然后在init中獲取賦值redis工具類,代碼如下:
// 解決redis操作工具類注入為null的問題 public static WebSocketServer webSocketServer; @PostConstruct public void init() { webSocketServer = this; webSocketServer.redisOperatingUtil = this.redisOperatingUtil; }
在websocket服務(wù)的@OnMessage注解中,收到客戶端發(fā)送的消息,我們將其保存到redis中,代碼如下:
/** * 收到客戶端消息后調(diào)用的方法 * * @param message 客戶端發(fā)送過來的消息 * // @param session 客戶端會話 */ @OnMessage public void onMessage(String message) { // 客戶端發(fā)送的消息 JSONObject jsReply = new JSONObject(message); // 添加在線人數(shù) jsReply.put("onlineUsers", getOnlineCount()); if (jsReply.has("buddyId")) { // 獲取推送方id String userId = jsReply.getString("userID"); // 獲取被推送方id String buddyId = jsReply.getString("buddyId"); // 非測試數(shù)據(jù)則推送消息 if (!buddyId.equals("121710f399b84322bdecc238199d6888")) { // 發(fā)送消息至推送方 this.sendInfo(jsReply.toString(), userId); } // 構(gòu)造聊天記錄實(shí)體類數(shù)據(jù) SubMessage subMessage = new SubMessage(); subMessage.setAvatarSrc(jsReply.getString("avatarSrc")); subMessage.setUserId(jsReply.getString("userID")); subMessage.setUserName(jsReply.getString("username")); subMessage.setMsgText(jsReply.getString("msg")); subMessage.setMsgId(jsReply.getString("msgId")); subMessage.setCreateTime(DateUtil.getThisTime()); subMessage.setStatus(false); // 將聊天記錄對象保存到redis中 webSocketServer.redisOperatingUtil.listRightPush("subMessage", subMessage); // 發(fā)送消息至被推送方 this.sendInfo(jsReply.toString(), buddyId); } }
做完上述操作后,收到客戶端發(fā)送的消息就會自動寫入redis。
接下來,我們使用quartz定時(shí)向mysql中寫入數(shù)據(jù),他執(zhí)行定時(shí)任務(wù)的步驟分為2步:
首先,創(chuàng)建quartzServer包,在其下創(chuàng)建RedisToMysqlTask.java文件,在此文件內(nèi)實(shí)現(xiàn)redis寫入mysql的代碼
package com.lk.quartzServer; import com.lk.dao.SubMessageMapper; import com.lk.entity.SubMessage; import com.lk.utils.RedisOperatingUtil; import lombok.extern.slf4j.Slf4j; import org.quartz.JobExecutionContext; import org.quartz.JobExecutionException; import org.springframework.scheduling.quartz.QuartzJobBean; import javax.annotation.Resource; // 將redis數(shù)據(jù)放進(jìn)mysql中 @Slf4j public class RedisToMysqlTask extends QuartzJobBean { @Resource private RedisOperatingUtil redisOperatingUtil; @Resource private SubMessageMapper subMessageMapper; @Override protected void executeInternal(JobExecutionContext jobExecutionContext) throws JobExecutionException { // 獲取存儲在redis中聊天記錄的條數(shù) long messageListSize = redisOperatingUtil.listLen("subMessage"); // 寫入數(shù)據(jù)庫的數(shù)據(jù)總條數(shù) long resultCount = 0; for (int i = 0; i messageListSize; i++) { // 從頭到尾取出鏈表中的元素 SubMessage subMessage= (SubMessage) redisOperatingUtil.listPopLeftKey("subMessage"); // 向數(shù)據(jù)庫寫入數(shù)據(jù) int result = subMessageMapper.addMessageTextInfo(subMessage); if (result > 0) { // 寫入成功 resultCount++; } } log.info(resultCount+ "條聊天記錄,已寫入數(shù)據(jù)庫"); } }
在config包下創(chuàng)建QuartzConfig.java文件,創(chuàng)建定時(shí)任務(wù)
package com.lk.config; import com.lk.quartzServer.RedisToMysqlTask; import org.quartz.*; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; /** * Quartz定時(shí)任務(wù)配置 */ @Configuration public class QuartzConfig { @Bean public JobDetail RedisToMysqlQuartz() { // 執(zhí)行定時(shí)任務(wù) return JobBuilder.newJob(RedisToMysqlTask.class).withIdentity("CallPayQuartzTask").storeDurably().build(); } @Bean public Trigger CallPayQuartzTaskTrigger() { //cron方式,從每月1號開始,每隔三天就執(zhí)行一次 return TriggerBuilder.newTrigger().forJob(RedisToMysqlQuartz()) .withIdentity("CallPayQuartzTask") .withSchedule(CronScheduleBuilder.cronSchedule("* * 4 1/3 * ?")) .build(); } }
這里我設(shè)置的定時(shí)任務(wù)是從每月1號開始,每隔三天就執(zhí)行一次,Quartz定時(shí)任務(wù)采用的是cron表達(dá)式,自己算這個(gè)比較麻煩,這里推薦一個(gè)在線網(wǎng)站,可以很容易的生成表達(dá)式:Cron表達(dá)式生成器
最后,配合Vue實(shí)現(xiàn)的瀏覽器端,跟大家展示下實(shí)現(xiàn)效果:
效果視頻:使用Vue實(shí)現(xiàn)單聊
項(xiàng)目瀏覽器端代碼地址:github/chat-system
項(xiàng)目在線體驗(yàn)地址:chat-system
到此這篇關(guān)于利用redis實(shí)現(xiàn)聊天記錄轉(zhuǎn)存功能的文章就介紹到這了,更多相關(guān)redis聊天記錄轉(zhuǎn)存內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
標(biāo)簽:邯鄲 本溪 吉安 鶴崗 景德鎮(zhèn) 大理 丹東 昭通
巨人網(wǎng)絡(luò)通訊聲明:本文標(biāo)題《利用redis實(shí)現(xiàn)聊天記錄轉(zhuǎn)存功能的全過程》,本文關(guān)鍵詞 利用,redis,實(shí)現(xiàn),聊天記錄,;如發(fā)現(xiàn)本文內(nèi)容存在版權(quán)問題,煩請?zhí)峁┫嚓P(guān)信息告之我們,我們將及時(shí)溝通與處理。本站內(nèi)容系統(tǒng)采集于網(wǎng)絡(luò),涉及言論、版權(quán)與本站無關(guān)。