gitee/snowy
1
0
Fork 0
mirror of https://gitee.com/xiaonuobase/snowy.git synced 2026-03-22 10:47:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

【重构】站内信消息SSE彻底更改为ws方案

Browse Source
This commit is contained in:
俞宝山
2025-12-24 18:08:58 +08:00
parent cc5bf91e89
commit 315eee3794
19 changed files with 382 additions and 916 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
snowy-admin-web/package.json
View File

@@ -24,7 +24,6 @@
"@chenfengyuan/vue-qrcode": "2.0.0", "@chenfengyuan/vue-qrcode": "2.0.0",
"@highlightjs/vue-plugin": "2.1.0", "@highlightjs/vue-plugin": "2.1.0",
"@kangc/v-md-editor": "2.3.18", "@kangc/v-md-editor": "2.3.18",
"@microsoft/fetch-event-source": "2.0.1",
"@wangeditor/editor": "5.1.23", "@wangeditor/editor": "5.1.23",
"@wangeditor/editor-for-vue": "5.1.12", "@wangeditor/editor-for-vue": "5.1.12",
"@vue-office/docx": "1.6.2", "@vue-office/docx": "1.6.2",
@@ -37,7 +36,6 @@
"echarts": "5.5.1", "echarts": "5.5.1",
"echarts-stat": "1.2.0", "echarts-stat": "1.2.0",
"enquire.js": "2.1.6", "enquire.js": "2.1.6",
"event-source-polyfill": "1.0.31",
"fuse.js": "7.0.0", "fuse.js": "7.0.0",
"highlight.js": "11.10.0", "highlight.js": "11.10.0",
"hotkeys-js": "3.13.7", "hotkeys-js": "3.13.7",

265
snowy-admin-web/src/layout/components/message.vue
View File

@@ -1,55 +1,84 @@
<template> <template>
<div class="msg panel-item" @click="showMsg"> <div class="msg panel-item" @click="showMsg">
<a-badge :count="unreadMessageNum" class="badge"> <a-badge :count="unreadMessageNum" class="badge">
<comment-outlined /> <bell-outlined class="msg-icon-trigger" />
</a-badge> </a-badge>
<a-drawer v-model:open="msgVisible" title="新消息" placement="right" :width="500"> <a-drawer
<a-list :data-source="messageList" size="small" class="mb-3" :loading="miniMessageLoading"> v-model:open="msgVisible"
<template #renderItem="{ item }"> title="站内消息"
<a-list-item> placement="right"
<a-list-item-meta :description="item.createTime"> :width="500"
<template #title> :body-style="{ padding: '0' }"
<a @click="messageDetail(item)">{{ item.subject }}</a> :footer-style="{ textAlign: 'right', borderTop: '1px solid #f0f0f0' }"
</template> >
</a-list-item-meta> <div class="msg-list-container">
</a-list-item> <a-list :data-source="messageList" :loading="miniMessageLoading" item-layout="horizontal">
</template> <template #renderItem="{ item }">
</a-list> <a-list-item class="msg-item" :class="{ unread: !item.read }" @click="messageDetail(item)">
<a-space class="xn-fdr"> <a-list-item-meta>
<a-button v-if="unreadMessageNum > 0" @click="markRead">全部设为已读</a-button> <template #avatar>
<a-button type="primary" @click="leaveFor('/usercenter')">消息中心</a-button> <a-avatar
</a-space> shape="square"
class="msg-avatar"
:style="{ backgroundColor: item.read ? '#d9d9d9' : '#1890ff' }"
>
<template #icon><BellOutlined /></template>
</a-avatar>
</template>
<template #title>
<div class="msg-title-row">
<span class="msg-title text-ellipsis">{{ item.subject }}</span>
<span class="msg-time">{{ item.createTime }}</span>
</div>
</template>
<template #description>
<div class="msg-desc text-ellipsis">{{ item.content || '暂无摘要' }}</div>
</template>
</a-list-item-meta>
</a-list-item>
</template>
<template #empty>
<a-empty description="暂无新消息" style="margin-top: 50px" />
</template>
</a-list>
</div>
<template #footer>
<a-space>
<a-button v-if="unreadMessageNum > 0" @click="markRead">全部已读</a-button>
<a-button type="primary" @click="leaveFor('/usercenter')">消息中心</a-button>
</a-space>
</template>
</a-drawer> </a-drawer>
<xn-form-container title="详情" :width="700" :open="visible" :destroy-on-close="true" @close="onClose">
<a-form ref="formRef" :model="formData" layout="vertical"> <xn-form-container title="消息详情" :width="700" :open="visible" :destroy-on-close="true" @close="onClose">
<a-form-item label="主题：" name="subject"> <a-descriptions bordered :column="1" size="middle" class="mb-4">
<span>{{ formData.subject }}</span> <a-descriptions-item label="主题">{{ formData.subject }}</a-descriptions-item>
</a-form-item> <a-descriptions-item label="发送时间">{{ formData.createTime }}</a-descriptions-item>
<a-form-item label="发送时间：" name="createTime"> <a-descriptions-item label="内容">
<span>{{ formData.createTime }}</span> <div class="msg-content-detail">{{ formData.content }}</div>
</a-form-item> </a-descriptions-item>
<a-form-item label="内容：" name="content"> </a-descriptions>
<span>{{ formData.content }}</span>
</a-form-item> <div class="receive-info-title">查收情况</div>
<a-form-item label="查收情况：" name="receiveInfoList"> <s-table
<s-table ref="tableRef"
ref="tableRef" :columns="columns"
:columns="columns" :data="loadData"
:data="loadData" :alert="false"
:alert="false" :showPagination="false"
:showPagination="false" bordered
bordered :row-key="(record) => record.id"
:row-key="(record) => record.id" compSize="small"
> >
<template #bodyCell="{ column, record }"> <template #bodyCell="{ column, record }">
<template v-if="column.dataIndex === 'read'"> <template v-if="column.dataIndex === 'read'">
<span v-if="record.read" class="xn-color-d9d9d9">已读</span> <a-tag :color="record.read ? 'green' : 'red'">
<span v-else class="xn-color-ff4d4f">未读</span> {{ record.read ? '已读' : '未读' }}
</template> </a-tag>
</template> </template>
</s-table> </template>
</a-form-item> </s-table>
</a-form>
</xn-form-container> </xn-form-container>
</div> </div>
</template> </template>
@@ -57,11 +86,11 @@
<script setup name="devUserMessage"> <script setup name="devUserMessage">
import indexApi from '@/api/sys/indexApi' import indexApi from '@/api/sys/indexApi'
import { message } from 'ant-design-vue' import { message } from 'ant-design-vue'
import { BellOutlined } from '@ant-design/icons-vue'
import router from '@/router' import router from '@/router'
import { onMounted } from 'vue' import { onMounted } from 'vue'
import sysConfig from '@/config' import sysConfig from '@/config'
import { convertUrl } from '@/utils/apiAdaptive' import { convertUrl } from '@/utils/apiAdaptive'
import { EventSourcePolyfill } from 'event-source-polyfill'
import tool from '@/utils/tool' import tool from '@/utils/tool'
const miniMessageLoading = ref(false) const miniMessageLoading = ref(false)
@@ -70,40 +99,38 @@
const unreadMessageNum = ref(0) const unreadMessageNum = ref(0)
onMounted(() => { onMounted(() => {
createSseConnect() createWebSocketConnect()
}) })
// 创建sse连接 // 创建ws连接
const createSseConnect = () => { const createWebSocketConnect = () => {
if (window.EventSource) { if ('WebSocket' in window) {
let clientId = tool.data.get('CLIENTID') ? tool.data.get('CLIENTID') : '' let url = sysConfig.API_URL + convertUrl('/dev/message/ws') + '?token=' + tool.data.get('TOKEN')
let url = sysConfig.API_URL + convertUrl('/dev/message/createSseConnect?clientId=') + clientId url = url.replace('https://', 'wss://').replace('http://', 'ws://')
// heartbeatTimeout:心跳超时监测 30s const socket = new WebSocket(url)
let source = new EventSourcePolyfill(url, {
headers: { token: tool.data.get('TOKEN') },
heartbeatTimeout: 300000
})
// 监听打开事件 // 监听打开事件
source.addEventListener('open', (e) => {}) socket.onopen = () => {
// console.log('WebSocket连接已打开')
}
// 监听消息事件 // 监听消息事件
source.addEventListener('message', (e) => { socket.onmessage = (event) => {
const result = JSON.parse(e.data) const result = JSON.parse(event.data)
const code = result.code const code = result.code
const data = result.data const data = result.data
if (code === 200) { if (code === 200) {
unreadMessageNum.value = data unreadMessageNum.value = data
} else if (code === 0) {
// 初次建立连接，客户端id储存本地
tool.data.set('CLIENTID', data)
} }
}) }
// 监听错误事件 // 监听错误事件
source.addEventListener('error', (e) => { socket.onerror = () => {
console.error('发生错误，消息实时获取已断开与服务器的连接') console.error('WebSocket连接发生错误')
source.close() }
}) // 监听关闭事件
socket.onclose = () => {
// console.log('WebSocket连接已关闭')
}
} else { } else {
message.warning('该浏览器不支持消息功能') message.warning('该浏览器不支持WebSocket消息功能')
} }
} }
@@ -182,7 +209,101 @@
} }
</script> </script>
<style scoped> <style scoped lang="less">
.msg {
height: 100%;
display: flex;
align-items: center;
justify-content: center;
padding: 0 10px;
cursor: pointer;
transition: all 0.3s;
}
.msg:hover {
background: rgba(0, 0, 0, 0.025);
}
.msg-icon-trigger {
font-size: 16px;
}
.msg-list-container {
max-height: calc(100vh - 150px);
overflow-y: auto;
}
.msg-item {
cursor: pointer;
padding: 12px 16px;
transition: background-color 0.3s;
border-bottom: 1px solid #f0f0f0;
&:hover {
background-color: #fafafa;
}
&.unread {
background-color: #f0f9ff;
&:hover {
background-color: #e6f7ff;
}
.msg-title {
font-weight: 600;
color: #1890ff;
}
}
}
.msg-avatar {
margin-top: 4px;
}
.msg-title-row {
display: flex;
justify-content: space-between;
align-items: center;
margin-bottom: 4px;
}
.msg-title {
flex: 1;
color: #333;
font-size: 14px;
margin-right: 8px;
}
.msg-time {
color: #999;
font-size: 12px;
white-space: nowrap;
}
.msg-desc {
color: #666;
font-size: 13px;
}
.text-ellipsis {
overflow: hidden;
white-space: nowrap;
text-overflow: ellipsis;
}
.msg-content-detail {
white-space: pre-wrap;
line-height: 1.6;
max-height: 400px;
overflow-y: auto;
}
.receive-info-title {
font-size: 15px;
font-weight: 600;
margin: 20px 0 10px;
padding-left: 10px;
border-left: 3px solid #1890ff;
line-height: 1;
}
:deep(.ant-badge-count) { :deep(.ant-badge-count) {
padding: 0px; padding: 0px;
min-width: 15px; min-width: 15px;

4
snowy-admin-web/src/layout/components/userbar.vue
View File

@@ -5,7 +5,7 @@
<div v-if="!isMobile" class="screen panel-item hidden-sm-and-down" @click="fullscreen"> <div v-if="!isMobile" class="screen panel-item hidden-sm-and-down" @click="fullscreen">
<fullscreen-outlined /> <fullscreen-outlined />
</div> </div>
<!-- <dev-user-message />--> <dev-user-message />
<a-dropdown class="user panel-item"> <a-dropdown class="user panel-item">
<div class="user-avatar"> <div class="user-avatar">
<a-avatar :src="userInfo ? userInfo.avatar : undefined" /> <a-avatar :src="userInfo ? userInfo.avatar : undefined" />
@@ -130,7 +130,7 @@
router.replace({ path: '/login' }) router.replace({ path: '/login' })
nextTick(() => { nextTick(() => {
// 清理缓存内的个人信息 // 清理缓存内的个人信息
store.userInfo = undefined store.setUserInfo(undefined)
}) })
}) })
.catch(() => { .catch(() => {

31
snowy-common/src/main/java/vip/xiaonuo/common/sse/CommonSseParam.java
View File

@@ -1,31 +0,0 @@
/*
* Copyright [2022] [https://www.xiaonuo.vip]
*
* Snowy采用APACHE LICENSE 2.0开源协议，您在使用过程中，需要注意以下几点：
*
* 1.请不要删除和修改根目录下的LICENSE文件。
* 2.请不要删除和修改Snowy源码头部的版权声明。
* 3.本项目代码可免费商业使用，商业使用请保留源码和相关描述文件的项目出处，作者声明等。
* 4.分发源码时候，请注明软件出处 https://www.xiaonuo.vip
* 5.不可二次分发开源参与同类竞品，如有想法可联系团队xiaonuobase@qq.com商议合作。
* 6.若您的项目无法满足以上几点，需要更多功能代码，获取Snowy商业授权许可，请在官网购买授权，地址为 https://www.xiaonuo.vip
*/
package vip.xiaonuo.common.sse;
import lombok.Getter;
import lombok.Setter;
/**
* 通用SSE参数
*
* @author diantu
* @date 2023/7/10
*/
@Getter
@Setter
public class CommonSseParam {
private String clientId;
private String loginId;
}

69
snowy-plugin-api/snowy-plugin-dev-api/src/main/java/vip/xiaonuo/dev/api/DevSseApi.java
View File

@@ -1,69 +0,0 @@
/*
* Copyright [2022] [https://www.xiaonuo.vip]
*
* Snowy采用APACHE LICENSE 2.0开源协议，您在使用过程中，需要注意以下几点：
*
* 1.请不要删除和修改根目录下的LICENSE文件。
* 2.请不要删除和修改Snowy源码头部的版权声明。
* 3.本项目代码可免费商业使用，商业使用请保留源码和相关描述文件的项目出处，作者声明等。
* 4.分发源码时候，请注明软件出处 https://www.xiaonuo.vip
* 5.不可二次分发开源参与同类竞品，如有想法可联系团队xiaonuobase@qq.com商议合作。
* 6.若您的项目无法满足以上几点，需要更多功能代码，获取Snowy商业授权许可，请在官网购买授权，地址为 https://www.xiaonuo.vip
*/
package vip.xiaonuo.dev.api;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
import vip.xiaonuo.common.sse.CommonSseParam;
import java.util.function.Consumer;
/**
* SSE API接口
*
* @author diantu
* @date 2023/7/5
**/
public interface DevSseApi {
/**
* 创建SSE连接
*
* @param clientId 客户端id,不传则自动生成
* @param setHeartBeat 是否设置心跳定时任务,默认为false（true:设置 false:不设置）
* @param defaultHeartbeat 是否使用默认心跳任务
* @param consumer 自定义心跳任务,需要自定义实现Consumer接口中的accept方法（setHeartBeat必须为true,defaultHeartbeat为false才有意义）
* @return 初次建立连接会推送客户端id,状态码为0
* @author diantu
* @date 2023/7/5
**/
public SseEmitter createSseConnect(String clientId, Boolean setHeartBeat, Boolean defaultHeartbeat, Consumer<CommonSseParam> consumer);
/**
* 关闭连接
*
* @param clientId 客户端id
* @author diantu
* @date 2023/7/5
**/
public void closeSseConnect(String clientId);
/**
* 推送消息到所有客户端
*
* @param msg 推送消息
* @author diantu
* @date 2023/7/5
**/
public void sendMessageToAllClient(String msg);
/**
* 根据clientId发送消息给某一客户端
*
* @param clientId 客户端id
* @param msg 推送消息
* @author diantu
* @date 2023/7/5
**/
public void sendMessageToOneClient(String clientId, String msg);
}

6
snowy-plugin/snowy-plugin-dev/pom.xml
View File

@@ -33,5 +33,11 @@
<groupId>vip.xiaonuo</groupId> <groupId>vip.xiaonuo</groupId>
<artifactId>snowy-plugin-sys-api</artifactId> <artifactId>snowy-plugin-sys-api</artifactId>
</dependency> </dependency>
<!-- websocket -->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
</dependencies> </dependencies>
</project> </project>

25
snowy-plugin/snowy-plugin-dev/src/main/java/vip/xiaonuo/dev/core/config/WebSocketConfig.java Normal file
View File

@@ -0,0 +1,25 @@
package vip.xiaonuo.dev.core.config;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;
/**
* WebSocket配置
*
* @author yubaoshan
* @date 2025/12/24 18:01
*/
@Configuration
@EnableScheduling
public class WebSocketConfig {
/**
* 注入ServerEndpointExporter，这个bean会自动注册使用了@ServerEndpoint注解声明的Websocket endpoint
*/
@Bean
public ServerEndpointExporter serverEndpointExporter() {
return new ServerEndpointExporter();
}
}

3
snowy-plugin/snowy-plugin-dev/src/main/java/vip/xiaonuo/dev/modular/message/provider/DevMessageProvider.java
View File

@@ -25,6 +25,7 @@ import vip.xiaonuo.dev.modular.message.param.DevMessageIdParam;
import vip.xiaonuo.dev.modular.message.param.DevMessageListParam; import vip.xiaonuo.dev.modular.message.param.DevMessageListParam;
import vip.xiaonuo.dev.modular.message.param.DevMessageSendParam; import vip.xiaonuo.dev.modular.message.param.DevMessageSendParam;
import vip.xiaonuo.dev.modular.message.service.DevMessageService; import vip.xiaonuo.dev.modular.message.service.DevMessageService;
import vip.xiaonuo.dev.modular.message.websocket.DevMessageWebSocket;
import vip.xiaonuo.dev.modular.relation.entity.DevRelation; import vip.xiaonuo.dev.modular.relation.entity.DevRelation;
import vip.xiaonuo.dev.modular.relation.enums.DevRelationCategoryEnum; import vip.xiaonuo.dev.modular.relation.enums.DevRelationCategoryEnum;
import vip.xiaonuo.dev.modular.relation.service.DevRelationService; import vip.xiaonuo.dev.modular.relation.service.DevRelationService;
@@ -105,5 +106,7 @@ public class DevMessageProvider implements DevMessageApi {
.eq(DevRelation::getTargetId, StpUtil.getLoginIdAsString()) .eq(DevRelation::getTargetId, StpUtil.getLoginIdAsString())
.eq(DevRelation::getCategory, DevRelationCategoryEnum.MSG_TO_USER.getValue()) .eq(DevRelation::getCategory, DevRelationCategoryEnum.MSG_TO_USER.getValue())
.set(DevRelation::getExtJson, myMessageExtJson)); .set(DevRelation::getExtJson, myMessageExtJson));
// 发送WebSocket消息
DevMessageWebSocket.sendMessage(StpUtil.getLoginIdAsString());
} }
} }

12
snowy-plugin/snowy-plugin-dev/src/main/java/vip/xiaonuo/dev/modular/message/service/impl/DevMessageServiceImpl.java
View File

@@ -41,6 +41,7 @@ import vip.xiaonuo.dev.modular.message.param.DevMessagePageParam;
import vip.xiaonuo.dev.modular.message.param.DevMessageSendParam; import vip.xiaonuo.dev.modular.message.param.DevMessageSendParam;
import vip.xiaonuo.dev.modular.message.result.DevMessageResult; import vip.xiaonuo.dev.modular.message.result.DevMessageResult;
import vip.xiaonuo.dev.modular.message.service.DevMessageService; import vip.xiaonuo.dev.modular.message.service.DevMessageService;
import vip.xiaonuo.dev.modular.message.websocket.DevMessageWebSocket;
import vip.xiaonuo.dev.modular.relation.entity.DevRelation; import vip.xiaonuo.dev.modular.relation.entity.DevRelation;
import vip.xiaonuo.dev.modular.relation.enums.DevRelationCategoryEnum; import vip.xiaonuo.dev.modular.relation.enums.DevRelationCategoryEnum;
import vip.xiaonuo.dev.modular.relation.service.DevRelationService; import vip.xiaonuo.dev.modular.relation.service.DevRelationService;
@@ -81,6 +82,8 @@ public class DevMessageServiceImpl extends ServiceImpl<DevMessageMapper, DevMess
.set("read", false))).collect(Collectors.toList()); .set("read", false))).collect(Collectors.toList());
devRelationService.saveRelationBatchWithAppend(devMessage.getId(), receiverIdList, devRelationService.saveRelationBatchWithAppend(devMessage.getId(), receiverIdList,
DevRelationCategoryEnum.MSG_TO_USER.getValue(), extJsonList); DevRelationCategoryEnum.MSG_TO_USER.getValue(), extJsonList);
// 发送WebSocket消息
DevMessageWebSocket.sendMessage(receiverIdList);
} }
} }
@@ -155,9 +158,10 @@ public class DevMessageServiceImpl extends ServiceImpl<DevMessageMapper, DevMess
@Override @Override
public Long unreadCount(String loginId){ public Long unreadCount(String loginId){
return devRelationService.getRelationListByTargetIdAndCategory(loginId, return devRelationService.count(new LambdaQueryWrapper<DevRelation>()
DevRelationCategoryEnum.MSG_TO_USER.getValue()).stream().filter(devRelation -> JSONUtil .eq(DevRelation::getTargetId, loginId)
.parseObj(devRelation.getExtJson()).getBool("read").equals(false)).count(); .eq(DevRelation::getCategory, DevRelationCategoryEnum.MSG_TO_USER.getValue())
.like(DevRelation::getExtJson, "\"read\":false"));
} }
@Transactional(rollbackFor = Exception.class) @Transactional(rollbackFor = Exception.class)
@@ -188,6 +192,8 @@ public class DevMessageServiceImpl extends ServiceImpl<DevMessageMapper, DevMess
devRelationService.update(new LambdaUpdateWrapper<DevRelation>() devRelationService.update(new LambdaUpdateWrapper<DevRelation>()
.eq(DevRelation::getObjectId, devMessage.getId()).eq(DevRelation::getTargetId, StpUtil.getLoginIdAsString()) .eq(DevRelation::getObjectId, devMessage.getId()).eq(DevRelation::getTargetId, StpUtil.getLoginIdAsString())
.eq(DevRelation::getCategory, DevRelationCategoryEnum.MSG_TO_USER.getValue()).set(DevRelation::getExtJson, myMessageExtJson)); .eq(DevRelation::getCategory, DevRelationCategoryEnum.MSG_TO_USER.getValue()).set(DevRelation::getExtJson, myMessageExtJson));
// 发送WebSocket消息
DevMessageWebSocket.sendMessage(StpUtil.getLoginIdAsString());
} }
List<DevMessageResult.DevReceiveInfo> receiveInfoList = devRelationService.getRelationListByObjectIdAndCategory(devMessage.getId(), List<DevMessageResult.DevReceiveInfo> receiveInfoList = devRelationService.getRelationListByObjectIdAndCategory(devMessage.getId(),
DevRelationCategoryEnum.MSG_TO_USER.getValue()).stream().map(devRelation -> { DevRelationCategoryEnum.MSG_TO_USER.getValue()).stream().map(devRelation -> {

144
snowy-plugin/snowy-plugin-dev/src/main/java/vip/xiaonuo/dev/modular/message/websocket/DevMessageWebSocket.java Normal file
View File

@@ -0,0 +1,144 @@
package vip.xiaonuo.dev.modular.message.websocket;
import cn.dev33.satoken.stp.StpUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.http.HttpUtil;
import cn.hutool.extra.spring.SpringUtil;
import cn.hutool.json.JSONUtil;
import jakarta.websocket.*;
import jakarta.websocket.server.ServerEndpoint;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import vip.xiaonuo.dev.modular.message.service.DevMessageService;
import java.io.IOException;
import java.nio.charset.Charset;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
/**
* 站内信WebSocket
*
* @author yubaoshan
* @date 2025/12/24 18:01
*/
@Component
@Slf4j
@ServerEndpoint("/dev/message/ws")
public class DevMessageWebSocket {
/**
* WebSocket Session池，key: sessionId, value: session
*/
private static final ConcurrentHashMap<String, Session> SESSION_POOL = new ConcurrentHashMap<>();
/**
* SessionUser映射，key: sessionId, value: userId
*/
private static final ConcurrentHashMap<String, String> SESSION_USER_MAP = new ConcurrentHashMap<>();
@OnOpen
public void onOpen(Session session) {
try {
String queryString = session.getQueryString();
if (ObjectUtil.isEmpty(queryString)) {
session.close();
return;
}
Map<String, String> paramMap = HttpUtil.decodeParamMap(queryString, Charset.defaultCharset());
String token = paramMap.get("token");
if (ObjectUtil.isEmpty(token)) {
session.close();
return;
}
// 验证token
Object loginId = StpUtil.getLoginIdByToken(token);
if (ObjectUtil.isEmpty(loginId)) {
session.close();
return;
}
String userId = loginId.toString();
SESSION_POOL.put(session.getId(), session);
SESSION_USER_MAP.put(session.getId(), userId);
// 发送初始未读数量
sendUnreadCount(session, userId);
} catch (Exception e) {
log.error("WebSocket onOpen error", e);
}
}
@OnClose
public void onClose(Session session) {
SESSION_POOL.remove(session.getId());
SESSION_USER_MAP.remove(session.getId());
}
@OnError
public void onError(Session session, Throwable error) {
log.error("WebSocket onError", error);
onClose(session);
}
@OnMessage
public void onMessage(String message, Session session) {
// 收到消息，暂不处理
}
/**
* 发送未读消息数量给指定用户
*
* @param userId 用户ID
*/
public static void sendMessage(String userId) {
SESSION_USER_MAP.forEach((sessionId, sessionUserId) -> {
if (sessionUserId.equals(userId)) {
Session session = SESSION_POOL.get(sessionId);
if (session != null && session.isOpen()) {
sendUnreadCount(session, userId);
}
}
});
}
/**
* 发送未读消息数量给指定用户列表
*
* @author yubaoshan
* @date 2025/12/24 22:20
* @param userIds 用户ID列表
*/
public static void sendMessage(List<String> userIds) {
Set<String> userIdSet = new HashSet<>(userIds);
SESSION_USER_MAP.forEach((sessionId, sessionUserId) -> {
if (userIdSet.contains(sessionUserId)) {
Session session = SESSION_POOL.get(sessionId);
if (session != null && session.isOpen()) {
sendUnreadCount(session, sessionUserId);
}
}
});
}
/**
* 发送未读消息数量
*/
private static void sendUnreadCount(Session session, String userId) {
try {
if (session.isOpen()) {
DevMessageService devMessageService = SpringUtil.getBean(DevMessageService.class);
Long count = devMessageService.unreadCount(userId);
Map<String, Object> result = Map.of("code", 200, "data", count);
session.getBasicRemote().sendText(JSONUtil.toJsonStr(result));
}
} catch (IOException e) {
log.error("Send message error", e);
}
}
}

98
snowy-plugin/snowy-plugin-dev/src/main/java/vip/xiaonuo/dev/modular/sse/controller/DevSseEmitterController.java
View File

@@ -1,98 +0,0 @@
/*
* Copyright [2022] [https://www.xiaonuo.vip]
*
* Snowy采用APACHE LICENSE 2.0开源协议，您在使用过程中，需要注意以下几点：
*
* 1.请不要删除和修改根目录下的LICENSE文件。
* 2.请不要删除和修改Snowy源码头部的版权声明。
* 3.本项目代码可免费商业使用，商业使用请保留源码和相关描述文件的项目出处，作者声明等。
* 4.分发源码时候，请注明软件出处 https://www.xiaonuo.vip
* 5.不可二次分发开源参与同类竞品，如有想法可联系团队xiaonuobase@qq.com商议合作。
* 6.若您的项目无法满足以上几点，需要更多功能代码，获取Snowy商业授权许可，请在官网购买授权，地址为 https://www.xiaonuo.vip
*/
package vip.xiaonuo.dev.modular.sse.controller;
import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
import com.github.xiaoymin.knife4j.annotations.ApiSupport;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.annotation.Resource;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
import vip.xiaonuo.common.sse.CommonSseParam;
import vip.xiaonuo.dev.modular.sse.service.DevSseEmitterService;
import java.util.function.Consumer;
/**
* SSE通信控制器
*
* @author diantu
* @date 2023/7/3
**/
@Tag(name = "SSE通信控制器")
@ApiSupport(author = "SNOWY_TEAM", order = 10)
@RestController
@Validated
public class DevSseEmitterController {
@Resource
private DevSseEmitterService devSseEmitterService;
/**
* 创建sse连接
*
* @author diantu
* @date 2023/7/3
**/
@ApiOperationSupport(order = 1)
@Operation(summary = "创建sse连接")
@GetMapping("/dev/sse/createConnect")
public SseEmitter createConnect(String clientId,
@RequestParam(required = false)Boolean setHeartBeat,
@RequestParam(required = false)Boolean defaultHeartbeat,
@RequestParam(required = false)Consumer<CommonSseParam> consumer){
return devSseEmitterService.createSseConnect(clientId,setHeartBeat,defaultHeartbeat,consumer);
}
/**
* 关闭sse连接
*
* @author diantu
* @date 2023/7/3
**/
@ApiOperationSupport(order = 2)
@Operation(summary = "关闭sse连接")
@GetMapping("/dev/sse/closeSseConnect")
public void closeSseConnect(String clientId){
devSseEmitterService.closeSseConnect(clientId);
}
/**
* 推送消息到所有客户端
*
* @author diantu
* @date 2023/7/3
**/
@ApiOperationSupport(order = 3)
@Operation(summary = "推送消息到所有客户端")
@PostMapping("/dev/sse/broadcast")
public void sendMessageToAllClient(@RequestBody(required = false) String msg){
devSseEmitterService.sendMessageToAllClient(msg);
}
/**
* 根据clientId发送消息给某一客户端
*
* @author diantu
* @date 2023/7/3
**/
@ApiOperationSupport(order = 4)
@Operation(summary = "根据clientId发送消息给某一客户端")
@PostMapping("/dev/sse/sendMessage")
public void sendMessageToOneClient(String clientId,String msg){
devSseEmitterService.sendMessageToOneClient(clientId,msg);
}
}

47
snowy-plugin/snowy-plugin-dev/src/main/java/vip/xiaonuo/dev/modular/sse/enums/DevSseEmitterParameterEnum.java
View File

@@ -1,47 +0,0 @@
/*
* Copyright [2022] [https://www.xiaonuo.vip]
*
* Snowy采用APACHE LICENSE 2.0开源协议，您在使用过程中，需要注意以下几点：
*
* 1.请不要删除和修改根目录下的LICENSE文件。
* 2.请不要删除和修改Snowy源码头部的版权声明。
* 3.本项目代码可免费商业使用，商业使用请保留源码和相关描述文件的项目出处，作者声明等。
* 4.分发源码时候，请注明软件出处 https://www.xiaonuo.vip
* 5.不可二次分发开源参与同类竞品，如有想法可联系团队xiaonuobase@qq.com商议合作。
* 6.若您的项目无法满足以上几点，需要更多功能代码，获取Snowy商业授权许可，请在官网购买授权，地址为 https://www.xiaonuo.vip
*/
package vip.xiaonuo.dev.modular.sse.enums;
import lombok.Getter;
/**
* SSE通信参数枚举
*
* @author diantu
* @date 2023/7/17
**/
@Getter
public enum DevSseEmitterParameterEnum {
/**
* 通信
*/
EMITTER("EMITTER"),
/**
* 心跳
*/
FUTURE("FUTURE"),
/**
* 用户ID
*/
LOGINID("LOGINID");
private final String value;
DevSseEmitterParameterEnum(String value) {
this.value = value;
}
}

79
snowy-plugin/snowy-plugin-dev/src/main/java/vip/xiaonuo/dev/modular/sse/provider/DevSseProvider.java
View File

@@ -1,79 +0,0 @@
/*
* Copyright [2022] [https://www.xiaonuo.vip]
*
* Snowy采用APACHE LICENSE 2.0开源协议，您在使用过程中，需要注意以下几点：
*
* 1.请不要删除和修改根目录下的LICENSE文件。
* 2.请不要删除和修改Snowy源码头部的版权声明。
* 3.本项目代码可免费商业使用，商业使用请保留源码和相关描述文件的项目出处，作者声明等。
* 4.分发源码时候，请注明软件出处 https://www.xiaonuo.vip
* 5.不可二次分发开源参与同类竞品，如有想法可联系团队xiaonuobase@qq.com商议合作。
* 6.若您的项目无法满足以上几点，需要更多功能代码，获取Snowy商业授权许可，请在官网购买授权，地址为 https://www.xiaonuo.vip
*/
package vip.xiaonuo.dev.modular.sse.provider;
import jakarta.annotation.Resource;
import org.springframework.stereotype.Service;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
import vip.xiaonuo.common.sse.CommonSseParam;
import vip.xiaonuo.dev.api.DevSseApi;
import vip.xiaonuo.dev.modular.sse.service.DevSseEmitterService;
import java.util.function.Consumer;
/**
* SSE API接口提供者
*
* @author diantu
* @date 2023/7/5
**/
@Service
public class DevSseProvider implements DevSseApi {
@Resource
private DevSseEmitterService devSseEmitterService;
/**
* 创建SSE连接
*
* @author diantu
* @date 2023/7/5
**/
@Override
public SseEmitter createSseConnect(String clientId, Boolean setHeartBeat, Boolean defaultHeartbeat, Consumer<CommonSseParam> consumer) {
return devSseEmitterService.createSseConnect(clientId,setHeartBeat,defaultHeartbeat,consumer);
}
/**
* 关闭连接
*
* @author diantu
* @date 2023/7/5
**/
@Override
public void closeSseConnect(String clientId) {
devSseEmitterService.closeSseConnect(clientId);
}
/**
* 推送消息到所有客户端
*
* @author diantu
* @date 2023/7/5
**/
@Override
public void sendMessageToAllClient(String msg) {
devSseEmitterService.sendMessageToAllClient(msg);
}
/**
* 根据clientId发送消息给某一客户端
*
* @author diantu
* @date 2023/7/5
**/
@Override
public void sendMessageToOneClient(String clientId, String msg) {
devSseEmitterService.sendMessageToOneClient(clientId,msg);
}
}

59
snowy-plugin/snowy-plugin-dev/src/main/java/vip/xiaonuo/dev/modular/sse/service/DevSseEmitterService.java
View File

@@ -1,59 +0,0 @@
/*
* Copyright [2022] [https://www.xiaonuo.vip]
*
* Snowy采用APACHE LICENSE 2.0开源协议，您在使用过程中，需要注意以下几点：
*
* 1.请不要删除和修改根目录下的LICENSE文件。
* 2.请不要删除和修改Snowy源码头部的版权声明。
* 3.本项目代码可免费商业使用，商业使用请保留源码和相关描述文件的项目出处，作者声明等。
* 4.分发源码时候，请注明软件出处 https://www.xiaonuo.vip
* 5.不可二次分发开源参与同类竞品，如有想法可联系团队xiaonuobase@qq.com商议合作。
* 6.若您的项目无法满足以上几点，需要更多功能代码，获取Snowy商业授权许可，请在官网购买授权，地址为 https://www.xiaonuo.vip
*/
package vip.xiaonuo.dev.modular.sse.service;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
import vip.xiaonuo.common.sse.CommonSseParam;
import java.util.function.Consumer;
/**
* SSE通信Service接口
*
* @author diantu
* @date 2023/7/3
**/
public interface DevSseEmitterService {
/**
* 创建连接
*
* @author diantu
* @date 2023/7/3
**/
public SseEmitter createSseConnect(String clientId,Boolean setHeartBeat,Boolean defaultHeartbeat,Consumer<CommonSseParam> consumer);
/**
* 关闭连接
*
* @author diantu
* @date 2023/7/3
**/
public void closeSseConnect(String clientId);
/**
* 推送消息到所有客户端
*
* @author diantu
* @date 2023/7/3
**/
public void sendMessageToAllClient(String msg);
/**
* 根据clientId发送消息给某一客户端
*
* @author diantu
* @date 2023/7/3
**/
public void sendMessageToOneClient(String clientId, String msg);
}

135
snowy-plugin/snowy-plugin-dev/src/main/java/vip/xiaonuo/dev/modular/sse/service/impl/DevSseEmitterServiceImpl.java
View File

@@ -1,135 +0,0 @@
/*
* Copyright [2022] [https://www.xiaonuo.vip]
*
* Snowy采用APACHE LICENSE 2.0开源协议，您在使用过程中，需要注意以下几点：
*
* 1.请不要删除和修改根目录下的LICENSE文件。
* 2.请不要删除和修改Snowy源码头部的版权声明。
* 3.本项目代码可免费商业使用，商业使用请保留源码和相关描述文件的项目出处，作者声明等。
* 4.分发源码时候，请注明软件出处 https://www.xiaonuo.vip
* 5.不可二次分发开源参与同类竞品，如有想法可联系团队xiaonuobase@qq.com商议合作。
* 6.若您的项目无法满足以上几点，需要更多功能代码，获取Snowy商业授权许可，请在官网购买授权，地址为 https://www.xiaonuo.vip
*/
package vip.xiaonuo.dev.modular.sse.service.impl;
import cn.dev33.satoken.stp.StpUtil;
import cn.hutool.core.util.IdUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
import vip.xiaonuo.common.pojo.CommonResult;
import vip.xiaonuo.common.sse.CommonSseParam;
import vip.xiaonuo.dev.modular.sse.service.DevSseEmitterService;
import vip.xiaonuo.dev.modular.sse.util.DevSseCacheUtil;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.TimeUnit;
import java.util.function.Consumer;
/**
* SSE通信Service接口实现类
*
* @author diantu
* @date 2023/7/3
**/
@Slf4j
@Service
public class DevSseEmitterServiceImpl implements DevSseEmitterService {
/**
* 心跳线程池
*/
private static final ScheduledExecutorService heartbeatExecutors = Executors.newScheduledThreadPool(10);
/**
* 创建连接
*
* @author diantu
* @date 2023/7/3
**/
@Override
public SseEmitter createSseConnect(String clientId, Boolean setHeartBeat, Boolean defaultHeartbeat, Consumer<CommonSseParam> consumer) {
// 设置超时时间，0表示不过期。默认30秒，超过时间未完成会抛出异常：AsyncRequestTimeoutException
SseEmitter sseEmitter = new SseEmitter(0L);
String loginId = StpUtil.getLoginIdAsString();
// 判断连接是否有效
if (DevSseCacheUtil.connectionValidity(clientId,loginId)) {
return DevSseCacheUtil.getSseEmitterByClientId(clientId);
}else{
DevSseCacheUtil.removeConnection(clientId);
}
clientId = IdUtil.simpleUUID();
String finalClientId = clientId;
// 增加心跳
final ScheduledFuture<?> future;
// 是否自定义心跳任务
if (setHeartBeat!=null&&setHeartBeat) {
//是否使用默认心跳任务
if(defaultHeartbeat!=null&&defaultHeartbeat){
//默认心跳任务
future = heartbeatExecutors.scheduleAtFixedRate(() ->
DevSseCacheUtil.sendMessageToOneClient(finalClientId,finalClientId+"-"+loginId),
2, 10, TimeUnit.SECONDS);
}else{
//自定义心跳任务
CommonSseParam commonSseParam = new CommonSseParam();
commonSseParam.setClientId(clientId);
commonSseParam.setLoginId(loginId);
future = heartbeatExecutors.scheduleAtFixedRate(() -> consumer.accept(commonSseParam),
2, 10, TimeUnit.SECONDS);
}
// 增加连接
DevSseCacheUtil.addConnection(clientId, loginId, sseEmitter, future);
} else {
// 增加连接
DevSseCacheUtil.addConnection(clientId, loginId, sseEmitter, null);
}
// 长链接完成后回调(即关闭连接时调用)
sseEmitter.onCompletion(DevSseCacheUtil.completionCallBack(clientId));
// 连接超时回调
sseEmitter.onTimeout(DevSseCacheUtil.timeoutCallBack(clientId));
// 推送消息异常回调
sseEmitter.onError(DevSseCacheUtil.errorCallBack(clientId));
// 初次建立连接,推送客户端id
CommonResult<String> message = new CommonResult<>(0,"",clientId);
DevSseCacheUtil.sendMessageToClientByClientId(clientId,message);
return sseEmitter;
}
/**
* 关闭连接
*
* @author diantu
* @date 2023/7/3
**/
@Override
public void closeSseConnect(String clientId){
DevSseCacheUtil.removeConnection(clientId);
}
/**
* 推送消息到所有客户端
*
* @author diantu
* @date 2023/7/3
**/
@Override
public void sendMessageToAllClient(String msg) {
DevSseCacheUtil.sendMessageToAllClient(msg);
}
/**
* 根据clientId发送消息给某一客户端
*
* @author diantu
* @date 2023/7/3
**/
@Override
public void sendMessageToOneClient(String clientId, String msg) {
DevSseCacheUtil.sendMessageToOneClient(clientId,msg);
}
}

281
snowy-plugin/snowy-plugin-dev/src/main/java/vip/xiaonuo/dev/modular/sse/util/DevSseCacheUtil.java
View File

@@ -1,281 +0,0 @@
/*
* Copyright [2022] [https://www.xiaonuo.vip]
*
* Snowy采用APACHE LICENSE 2.0开源协议，您在使用过程中，需要注意以下几点：
*
* 1.请不要删除和修改根目录下的LICENSE文件。
* 2.请不要删除和修改Snowy源码头部的版权声明。
* 3.本项目代码可免费商业使用，商业使用请保留源码和相关描述文件的项目出处，作者声明等。
* 4.分发源码时候，请注明软件出处 https://www.xiaonuo.vip
* 5.不可二次分发开源参与同类竞品，如有想法可联系团队xiaonuobase@qq.com商议合作。
* 6.若您的项目无法满足以上几点，需要更多功能代码，获取Snowy商业授权许可，请在官网购买授权，地址为 https://www.xiaonuo.vip
*/
package vip.xiaonuo.dev.modular.sse.util;
import cn.hutool.core.util.StrUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.MediaType;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
import vip.xiaonuo.common.exception.CommonException;
import vip.xiaonuo.common.pojo.CommonResult;
import vip.xiaonuo.dev.modular.sse.enums.DevSseEmitterParameterEnum;
import java.util.Map;
import java.util.Objects;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ScheduledFuture;
import java.util.function.Consumer;
/**
* SseEmitter工具类
*
* @author diantu
* @date 2023/7/3
**/
@Slf4j
public class DevSseCacheUtil {
/**
* 创建一个容器来存储所有的 SseEmitter(使用ConcurrentHashMap是因为它是线程安全的)。
*/
public static Map<String, Map<String, Object>> sseCache = new ConcurrentHashMap<>();
/**
* 根据客户端id获取连接对象
*
* @author diantu
* @date 2023/7/3
**/
public static SseEmitter getSseEmitterByClientId(String clientId) {
Map<String, Object> map = sseCache.get(clientId);
if (map == null || map.isEmpty()) {
return null;
}
return (SseEmitter) map.get(DevSseEmitterParameterEnum.EMITTER.getValue());
}
/**
* 根据客户端id获取心跳
*
* @author diantu
* @date 2023/7/18
**/
public static ScheduledFuture<?> getSseFutureByClientId(String clientId) {
Map<String, Object> map = sseCache.get(clientId);
if (map == null || map.isEmpty()) {
return null;
}
return (ScheduledFuture<?>) map.get(DevSseEmitterParameterEnum.FUTURE.getValue());
}
/**
* 根据客户端id获取用户id
*
* @author diantu
* @date 2023/7/18
**/
public static ScheduledFuture<?> getLoginIdByClientId(String clientId) {
Map<String, Object> map = sseCache.get(clientId);
if (map == null || map.isEmpty()) {
return null;
}
return (ScheduledFuture<?>) map.get(DevSseEmitterParameterEnum.LOGINID.getValue());
}
/**
* 根据用户id获取客户端id
*
* @author diantu
* @date 2023/7/18
**/
public static String getClientIdByLoginId(String loginId) {
if (existSseCache()) {
for (Map.Entry<String, Map<String, Object>> entry : sseCache.entrySet()) {
Map<String, Object> map = sseCache.get(entry.getKey());
String lId = (String) map.get(DevSseEmitterParameterEnum.LOGINID.getValue());
if (loginId.equals(lId)) {
return entry.getKey();
}
}
}
return null;
}
/**
* 判断容器是否存在连接
*
* @author diantu
* @date 2023/7/3
**/
public static boolean existSseCache() {
return !sseCache.isEmpty();
}
/**
* 判断连接是否有效
*
* @author diantu
* @date 2023/7/3
**/
public static boolean connectionValidity(String clientId, String loginId) {
if (sseCache.get(clientId) == null) {
return false;
}
return Objects.equals(loginId, sseCache.get(clientId).get(DevSseEmitterParameterEnum.LOGINID.getValue()));
}
/**
* 增加连接
*
* @author diantu
* @date 2023/7/3
**/
public static void addConnection(String clientId, String loginId, SseEmitter emitter, ScheduledFuture<?> future) {
final SseEmitter oldEmitter = getSseEmitterByClientId(clientId);
if (oldEmitter != null) {
throw new CommonException("连接已存在:{}", clientId);
}
Map<String, Object> map = new ConcurrentHashMap<>();
map.put(DevSseEmitterParameterEnum.EMITTER.getValue(), emitter);
if (future != null) {
map.put(DevSseEmitterParameterEnum.FUTURE.getValue(), future);
}
map.put(DevSseEmitterParameterEnum.LOGINID.getValue(), loginId);
sseCache.put(clientId, map);
}
/**
* 移除连接
*
* @author diantu
* @date 2023/7/3
**/
public static void removeConnection(String clientId) {
SseEmitter emitter = getSseEmitterByClientId(clientId);
if (emitter != null) {
cancelScheduledFuture(clientId);
}
sseCache.remove(clientId);
log.info("移除连接:{}", clientId);
}
/**
* 中断心跳发送任务
*
* @author diantu
* @date 2023/7/3
*/
public static void cancelScheduledFuture(String clientId) {
ScheduledFuture<?> future = getSseFutureByClientId(clientId);
if (future != null) {
future.cancel(true);
}
}
/**
* 长链接完成后回调
*
* @author diantu
* @date 2023/7/3
**/
public static Runnable completionCallBack(String clientId) {
return () -> {
log.info("结束连接:{}", clientId);
removeConnection(clientId);
cancelScheduledFuture(clientId);
};
}
/**
* 连接超时回调
*
* @author diantu
* @date 2023/7/3
**/
public static Runnable timeoutCallBack(String clientId) {
return () -> {
log.info("连接超时:{}", clientId);
removeConnection(clientId);
cancelScheduledFuture(clientId);
};
}
/**
* 推送消息异常时回调
*
* @author diantu
* @date 2023/7/3
**/
public static Consumer<Throwable> errorCallBack(String clientId) {
return throwable -> {
log.info("推送消息异常:{}", clientId);
removeConnection(clientId);
cancelScheduledFuture(clientId);
};
}
/**
* 推送消息到所有客户端
*
* @author diantu
* @date 2023/7/3
**/
public static void sendMessageToAllClient(String msg) {
if (!existSseCache()) {
return;
}
// 判断发送的消息是否为空
if (StrUtil.isEmpty(msg)) {
log.info("群发消息为空");
return;
}
CommonResult<String> message = new CommonResult<>(CommonResult.CODE_SUCCESS, "", msg);
for (Map.Entry<String, Map<String, Object>> entry : sseCache.entrySet()) {
sendMessageToClientByClientId(entry.getKey(), message);
}
}
/**
* 根据clientId发送消息给某一客户端
*
* @author diantu
* @date 2023/7/3
**/
public static void sendMessageToOneClient(String clientId, String msg) {
if (StrUtil.isEmpty(clientId)) {
log.info("客户端ID为空");
return;
}
if (StrUtil.isEmpty(msg)) {
log.info("向客户端{}推送消息为空", clientId);
return;
}
CommonResult<String> message = new CommonResult<>(CommonResult.CODE_SUCCESS, "", msg);
sendMessageToClientByClientId(clientId, message);
}
/**
* 推送消息到客户端
*
* @author diantu
* @date 2023/7/3
**/
public static void sendMessageToClientByClientId(String clientId, CommonResult<String> message) {
Map<String, Object> map = sseCache.get(clientId);
if (map == null || map.isEmpty()) {
log.error("推送消息失败:客户端{}未创建长链接,失败消息:{}", clientId, message.toString());
return;
}
SseEmitter.SseEventBuilder sendData = SseEmitter.event().data(message, MediaType.APPLICATION_JSON);
SseEmitter sseEmitter = getSseEmitterByClientId(clientId);
try {
Objects.requireNonNull(sseEmitter).send(sendData);
} catch (Exception e) {
log.error("推送消息失败,报错异常:", e);
removeConnection(clientId);
}
}
}

13
snowy-plugin/snowy-plugin-sys/src/main/java/vip/xiaonuo/sys/modular/index/controller/SysIndexController.java
View File

@@ -158,19 +158,6 @@ public class SysIndexController {
return CommonResult.data(sysIndexService.opLogList()); return CommonResult.data(sysIndexService.opLogList());
} }
/**
* 创建sse连接
*
* @author diantu
* @date 2023/7/10
**/
@ApiOperationSupport(order = 9)
@Operation(summary = "创建sse连接")
@GetMapping("/dev/message/createSseConnect")
public SseEmitter createSseConnect(String clientId){
return sysIndexService.createSseConnect(clientId);
}
/** /**
* 获取基础系统业务数据 * 获取基础系统业务数据
* *

8
snowy-plugin/snowy-plugin-sys/src/main/java/vip/xiaonuo/sys/modular/index/service/SysIndexService.java
View File

@@ -90,14 +90,6 @@ public interface SysIndexService {
*/ */
List<SysIndexOpLogListResult> opLogList(); List<SysIndexOpLogListResult> opLogList();
/**
* 创建连接
*
* @author diantu
* @date 2023/7/10
**/
SseEmitter createSseConnect(String clientId);
/** /**
* 获取基础系统业务数据 * 获取基础系统业务数据
* *

17
snowy-plugin/snowy-plugin-sys/src/main/java/vip/xiaonuo/sys/modular/index/service/impl/SysIndexServiceImpl.java
View File

@@ -21,15 +21,12 @@ import cn.hutool.json.JSONUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import jakarta.annotation.Resource; import jakarta.annotation.Resource;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
import vip.xiaonuo.auth.api.AuthApi; import vip.xiaonuo.auth.api.AuthApi;
import vip.xiaonuo.auth.core.pojo.SaBaseLoginUser; import vip.xiaonuo.auth.core.pojo.SaBaseLoginUser;
import vip.xiaonuo.auth.core.util.StpLoginUserUtil; import vip.xiaonuo.auth.core.util.StpLoginUserUtil;
import vip.xiaonuo.common.sse.CommonSseParam;
import vip.xiaonuo.dev.api.DevApi; import vip.xiaonuo.dev.api.DevApi;
import vip.xiaonuo.dev.api.DevLogApi; import vip.xiaonuo.dev.api.DevLogApi;
import vip.xiaonuo.dev.api.DevMessageApi; import vip.xiaonuo.dev.api.DevMessageApi;
import vip.xiaonuo.dev.api.DevSseApi;
import vip.xiaonuo.sys.modular.index.param.*; import vip.xiaonuo.sys.modular.index.param.*;
import vip.xiaonuo.sys.modular.index.result.*; import vip.xiaonuo.sys.modular.index.result.*;
import vip.xiaonuo.sys.modular.index.service.SysIndexService; import vip.xiaonuo.sys.modular.index.service.SysIndexService;
@@ -63,9 +60,6 @@ public class SysIndexServiceImpl implements SysIndexService {
@Resource @Resource
private DevLogApi devLogApi; private DevLogApi devLogApi;
@Resource
private DevSseApi devSseApi;
@Resource @Resource
private SysUserService sysUserService; private SysUserService sysUserService;
@@ -143,17 +137,6 @@ public class SysIndexServiceImpl implements SysIndexService {
.map(jsonObject -> JSONUtil.toBean(jsonObject, SysIndexOpLogListResult.class)).collect(Collectors.toList()); .map(jsonObject -> JSONUtil.toBean(jsonObject, SysIndexOpLogListResult.class)).collect(Collectors.toList());
} }
@Override
public SseEmitter createSseConnect(String clientId){
Consumer<CommonSseParam> consumer = m -> {
//获取用户未读消息
long unreadMessageNum = devMessageApi.unreadCount(m.getLoginId());
//发送消息
devSseApi.sendMessageToOneClient(m.getClientId(), String.valueOf(unreadMessageNum));
};
return devSseApi.createSseConnect(clientId,true,false,consumer);
}
@Override @Override
public SysBizDataCountResult getBizDataCount() { public SysBizDataCountResult getBizDataCount() {
SysBizDataCountResult result = new SysBizDataCountResult(); SysBizDataCountResult result = new SysBizDataCountResult();