update 优化 sse工具类

This commit is contained in:
疯狂的狮子Li 2024-12-19 10:54:08 +08:00
parent 1a403361c9
commit ad85fa2016

View File

@ -26,7 +26,7 @@ public class SseMessageUtils {
* @param message 要发送的消息内容 * @param message 要发送的消息内容
*/ */
public static void sendMessage(Long userId, String message) { public static void sendMessage(Long userId, String message) {
if (SSE_ENABLE) { if (!isEnable()) {
return; return;
} }
MANAGER.sendMessage(userId, message); MANAGER.sendMessage(userId, message);
@ -38,7 +38,7 @@ public class SseMessageUtils {
* @param message 要发送的消息内容 * @param message 要发送的消息内容
*/ */
public static void sendMessage(String message) { public static void sendMessage(String message) {
if (SSE_ENABLE) { if (!isEnable()) {
return; return;
} }
MANAGER.sendMessage(message); MANAGER.sendMessage(message);
@ -50,7 +50,7 @@ public class SseMessageUtils {
* @param sseMessageDto 要发布的SSE消息对象 * @param sseMessageDto 要发布的SSE消息对象
*/ */
public static void publishMessage(SseMessageDto sseMessageDto) { public static void publishMessage(SseMessageDto sseMessageDto) {
if (SSE_ENABLE) { if (!isEnable()) {
return; return;
} }
MANAGER.publishMessage(sseMessageDto); MANAGER.publishMessage(sseMessageDto);
@ -62,10 +62,17 @@ public class SseMessageUtils {
* @param message 要发布的消息内容 * @param message 要发布的消息内容
*/ */
public static void publishAll(String message) { public static void publishAll(String message) {
if (SSE_ENABLE) { if (!isEnable()) {
return; return;
} }
MANAGER.publishAll(message); MANAGER.publishAll(message);
} }
/**
* 是否开启
*/
public static Boolean isEnable() {
return SSE_ENABLE;
}
} }