Compare commits
No commits in common. '73c643ab5b1cc3faaf229fe0e2ac6361c75192df' and '05fed99163a67c4440a6456dfaeb84fd956b7d4d' have entirely different histories.
73c643ab5b
...
05fed99163
@ -1,68 +0,0 @@ |
||||
package com.alops.web.controller.message; |
||||
|
||||
import com.alops.common.annotation.Log; |
||||
import com.alops.common.core.domain.AjaxResult; |
||||
import com.alops.common.enums.BusinessType; |
||||
import com.alops.common.utils.SecurityUtils; |
||||
import com.alops.system.domain.InternalMessage; |
||||
import com.alops.system.domain.dto.AlertRuleAddDto; |
||||
import com.alops.system.domain.vo.AlertRuleVo; |
||||
import com.alops.system.service.IInternalMessageService; |
||||
import io.swagger.annotations.ApiOperation; |
||||
import org.springframework.beans.factory.annotation.Autowired; |
||||
import org.springframework.security.access.prepost.PreAuthorize; |
||||
import org.springframework.web.bind.annotation.*; |
||||
|
||||
import java.util.List; |
||||
|
||||
@RestController |
||||
@RequestMapping("/message") |
||||
public class messageController { |
||||
|
||||
@Autowired |
||||
public IInternalMessageService iInternalMessageService; |
||||
|
||||
//变已读
|
||||
@PreAuthorize("@ss.hasPermi('message:change')") |
||||
@ApiOperation("变已读") |
||||
@Log(title = "【变已读】") |
||||
@PostMapping("/change/{ids}") |
||||
public AjaxResult change(@PathVariable Long[] ids) { |
||||
return AjaxResult.success("修改成功", iInternalMessageService.markMessagesAsRead(ids)); |
||||
|
||||
} |
||||
|
||||
//根据用户id获取未读量
|
||||
@PreAuthorize("@ss.hasPermi('message:read')") |
||||
@ApiOperation("未读量") |
||||
@Log(title = "【未读量】") |
||||
@PostMapping("/read") |
||||
public AjaxResult read() { |
||||
Long userId = SecurityUtils.getLoginUser().getUserId(); |
||||
|
||||
// 调用Service
|
||||
int count = iInternalMessageService.getUnreadCountByUserId(userId); |
||||
|
||||
return AjaxResult.success("查询成功", count); |
||||
|
||||
} |
||||
|
||||
@PreAuthorize("@ss.hasPermi('message:view')") |
||||
@ApiOperation("获取用户消息列表") |
||||
@Log(title = "【消息列表】") |
||||
@GetMapping("/list") |
||||
public AjaxResult getMessageList() { |
||||
// 获取当前登录用户ID
|
||||
Long userId = SecurityUtils.getLoginUser().getUserId(); |
||||
|
||||
// 调用Service获取消息列表
|
||||
List<InternalMessage> messages = iInternalMessageService.getMessagesByUserId(userId); |
||||
|
||||
return AjaxResult.success(messages); |
||||
} |
||||
|
||||
|
||||
|
||||
|
||||
|
||||
} |
||||
@ -1,20 +0,0 @@ |
||||
package com.alops.system.config; |
||||
|
||||
import com.alops.system.hander.WebSocketServer; |
||||
import org.springframework.context.annotation.Configuration; |
||||
import org.springframework.web.socket.config.annotation.EnableWebSocket; |
||||
import org.springframework.web.socket.config.annotation.WebSocketConfigurer; |
||||
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry; |
||||
|
||||
//理解为WebSocket的controller
|
||||
@Configuration |
||||
@EnableWebSocket |
||||
public class WebSocketConfig implements WebSocketConfigurer { |
||||
@Override |
||||
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) { |
||||
|
||||
registry.addHandler(new WebSocketServer(), "/ws/onlineMessage") |
||||
.setAllowedOrigins("*"); |
||||
|
||||
} |
||||
} |
||||
@ -1,153 +0,0 @@ |
||||
package com.alops.system.hander; |
||||
|
||||
import com.fasterxml.jackson.databind.ObjectMapper; |
||||
import org.slf4j.Logger; |
||||
import org.slf4j.LoggerFactory; |
||||
import org.springframework.beans.factory.annotation.Autowired; |
||||
import org.springframework.stereotype.Component; |
||||
import org.springframework.util.MultiValueMap; |
||||
import org.springframework.web.socket.CloseStatus; |
||||
import org.springframework.web.socket.TextMessage; |
||||
import org.springframework.web.socket.WebSocketSession; |
||||
import org.springframework.web.socket.handler.TextWebSocketHandler; |
||||
import org.springframework.web.util.UriComponentsBuilder; |
||||
import com.fasterxml.jackson.databind.ObjectMapper; |
||||
|
||||
import java.util.Date; |
||||
import java.util.HashMap; |
||||
import java.util.Map; |
||||
import java.util.Set; |
||||
import java.util.concurrent.ConcurrentHashMap; |
||||
|
||||
@Component |
||||
public class WebSocketServer extends TextWebSocketHandler { |
||||
private static final Logger logger = LoggerFactory.getLogger(WebSocketServer.class); |
||||
// 存储在线用户 (Key: 用户ID, Value: WebSocketSession)
|
||||
public static final ConcurrentHashMap<Long, WebSocketSession> onlineUsers = new ConcurrentHashMap<>(); |
||||
|
||||
|
||||
private static ObjectMapper objectMapper; |
||||
|
||||
|
||||
@Autowired |
||||
public void setObjectMapper(ObjectMapper mapper) { |
||||
WebSocketServer.objectMapper = mapper; |
||||
} |
||||
|
||||
@Override |
||||
public void afterConnectionEstablished(WebSocketSession session) throws Exception { |
||||
Long userId = getUserIdFromSession(session); |
||||
if (userId != null) { |
||||
logger.info("WebSocket 连接成功，用户上线: " + userId); |
||||
onlineUsers.put(userId, session); |
||||
} else { |
||||
logger.warn("无效的用户连接，关闭连接: " + session.getId()); |
||||
session.close(); // 没有 userId 的连接直接关闭
|
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception { |
||||
Long userId = getUserIdFromSession(session); |
||||
if (userId != null) { |
||||
|
||||
logger.info("WebSocket 连接关闭，用户下线: " + userId + ", 关闭状态: " + status); |
||||
onlineUsers.remove(userId); |
||||
|
||||
} |
||||
} |
||||
|
||||
// 获取 WebSocketSession 中的 userId，并转换为 Long 类型
|
||||
public static Long getUserIdFromSession(WebSocketSession session) { |
||||
String query = session.getUri().getQuery(); |
||||
logger.info("WebSocket连接查询参数: {}", query); |
||||
|
||||
if (query != null) { |
||||
MultiValueMap<String, String> params = UriComponentsBuilder.newInstance() |
||||
.query(query) |
||||
.build() |
||||
.getQueryParams(); |
||||
String userIdStr = params.getFirst("userId"); |
||||
if (userIdStr != null) { |
||||
try { |
||||
return Long.parseLong(userIdStr); |
||||
} catch (NumberFormatException e) { |
||||
logger.error("WebSocket连接无效的用户ID: {}", userIdStr, e); |
||||
} |
||||
} else { |
||||
logger.warn("WebSocket连接没有 userId 参数"); |
||||
} |
||||
} else { |
||||
logger.warn("WebSocket连接没有查询参数"); |
||||
} |
||||
return null; |
||||
} |
||||
//
|
||||
// // 发送消息给指定用户
|
||||
// public void sendMessageToUser(Long userId, String content, String title) throws Exception {
|
||||
// // 获取对应用户的 WebSocket session
|
||||
// WebSocketSession session = onlineUsers.get(userId);
|
||||
//
|
||||
// // 检查 session 是否有效且连接未关闭
|
||||
// if (session != null && session.isOpen()) {
|
||||
// // 创建一个 Map 来构建消息内容
|
||||
// Map<String, Object> messageMap = new HashMap<>();
|
||||
// messageMap.put("from UserName", fromUserName); // 发送者姓名
|
||||
// messageMap.put("content", content); // 消息内容
|
||||
//
|
||||
// // 使用 Jackson ObjectMapper 将 Map 转换为 JSON 字符串
|
||||
// ObjectMapper objectMapper = new ObjectMapper();
|
||||
// String jsonMessage = objectMapper.writeValueAsString(messageMap);
|
||||
//
|
||||
// // 发送 JSON 格式的消息
|
||||
// session.sendMessage(new TextMessage(jsonMessage));
|
||||
// }
|
||||
// }
|
||||
|
||||
|
||||
/** |
||||
* 发送通知到指定用户 |
||||
* @param messageId 数据库中站内信的ID |
||||
* @param title 通知标题，例如 "设备异常预警" |
||||
* @param content 通知内容 |
||||
*/ |
||||
public static void sendAlertMessage(Long userId, Long messageId, String title, String content) { |
||||
WebSocketSession session = onlineUsers.get(userId); |
||||
try { |
||||
if (session != null && session.isOpen()) { |
||||
|
||||
// 构建消息内容
|
||||
Map<String, Object> messageMap = new HashMap<>(); |
||||
messageMap.put("messageId", messageId); // 对应数据库ID
|
||||
messageMap.put("title", title); // 弹窗标题
|
||||
messageMap.put("content", content); // 弹窗内容
|
||||
messageMap.put("timestamp", new Date()); // 发送时间
|
||||
|
||||
// 转换为 JSON
|
||||
String jsonMessage = objectMapper.writeValueAsString(messageMap); |
||||
|
||||
// 发送
|
||||
session.sendMessage(new TextMessage(jsonMessage)); |
||||
logger.info("成功向用户 [{}] 发送预警消息: {}", session.getId(), jsonMessage); |
||||
} |
||||
} catch (Exception e) { |
||||
logger.error("发送预警消息失败: messageId=" + messageId, e); |
||||
} |
||||
} |
||||
|
||||
|
||||
public void sendMessageToUser1(Long userId, String message) throws Exception { |
||||
WebSocketSession session = onlineUsers.get(userId); |
||||
if (session != null && session.isOpen()) { |
||||
session.sendMessage(new TextMessage(message)); |
||||
} |
||||
} |
||||
|
||||
// 获取所有在线用户 ID (Long 类型)
|
||||
public Set<Long> getOnlineUserIds() { |
||||
return onlineUsers.keySet(); |
||||
} |
||||
|
||||
|
||||
|
||||
} |
||||
@ -1,74 +0,0 @@ |
||||
package com.alops.system.service; |
||||
|
||||
import java.util.List; |
||||
|
||||
import com.alops.system.domain.InternalMessage; |
||||
|
||||
|
||||
/** |
||||
* 站内信Service接口 |
||||
* |
||||
* @author ruoyi |
||||
* @date 2025-10-23 |
||||
*/ |
||||
public interface IInternalMessageService |
||||
{ |
||||
|
||||
|
||||
/** |
||||
* 根据用户ID获取消息列表，按时间排序 |
||||
*/ |
||||
List<InternalMessage> getMessagesByUserId(Long userId); |
||||
/** |
||||
* 根据用户ID获取未读消息数量 |
||||
*/ |
||||
int getUnreadCountByUserId(Long userId); |
||||
public int markMessagesAsRead(Long[] ids); |
||||
/** |
||||
* 查询站内信 |
||||
* |
||||
* @param id 站内信主键 |
||||
* @return 站内信 |
||||
*/ |
||||
public InternalMessage selectInternalMessageById(Long id); |
||||
|
||||
/** |
||||
* 查询站内信列表 |
||||
* |
||||
* @param internalMessage 站内信 |
||||
* @return 站内信集合 |
||||
*/ |
||||
public List<InternalMessage> selectInternalMessageList(InternalMessage internalMessage); |
||||
|
||||
/** |
||||
* 新增站内信 |
||||
* |
||||
* @param internalMessage 站内信 |
||||
* @return 结果 |
||||
*/ |
||||
public int insertInternalMessage(InternalMessage internalMessage); |
||||
|
||||
/** |
||||
* 修改站内信 |
||||
* |
||||
* @param internalMessage 站内信 |
||||
* @return 结果 |
||||
*/ |
||||
public int updateInternalMessage(InternalMessage internalMessage); |
||||
|
||||
/** |
||||
* 批量删除站内信 |
||||
* |
||||
* @param ids 需要删除的站内信主键集合 |
||||
* @return 结果 |
||||
*/ |
||||
public int deleteInternalMessageByIds(Long[] ids); |
||||
|
||||
/** |
||||
* 删除站内信信息 |
||||
* |
||||
* @param id 站内信主键 |
||||
* @return 结果 |
||||
*/ |
||||
public int deleteInternalMessageById(Long id); |
||||
} |
||||
@ -1,123 +0,0 @@ |
||||
package com.alops.system.service.impl; |
||||
|
||||
import java.util.Collections; |
||||
import java.util.List; |
||||
|
||||
import com.alops.common.utils.DateUtils; |
||||
import com.alops.system.domain.InternalMessage; |
||||
import com.alops.system.mapper.InternalMessageMapper; |
||||
import com.alops.system.service.IInternalMessageService; |
||||
import org.springframework.beans.factory.annotation.Autowired; |
||||
import org.springframework.stereotype.Service; |
||||
import org.springframework.transaction.annotation.Transactional; |
||||
|
||||
|
||||
/** |
||||
* 站内信Service业务层处理 |
||||
* |
||||
* @author ruoyi |
||||
* @date 2025-10-23 |
||||
*/ |
||||
@Service |
||||
public class InternalMessageServiceImpl implements IInternalMessageService |
||||
{ |
||||
@Autowired |
||||
private InternalMessageMapper internalMessageMapper; |
||||
|
||||
@Override |
||||
public List<InternalMessage> getMessagesByUserId(Long userId) { |
||||
if (userId == null) { |
||||
return Collections.emptyList(); |
||||
} |
||||
return internalMessageMapper.selectMessagesByUserId(userId); |
||||
} |
||||
@Override |
||||
public int getUnreadCountByUserId(Long userId) { |
||||
if (userId == null) { |
||||
return 0; |
||||
} |
||||
return internalMessageMapper.countUnreadByUserId(userId); |
||||
} |
||||
/** |
||||
* 查询站内信 |
||||
* |
||||
* @param id 站内信主键 |
||||
* @return 站内信 |
||||
*/ |
||||
@Override |
||||
public InternalMessage selectInternalMessageById(Long id) |
||||
{ |
||||
return internalMessageMapper.selectInternalMessageById(id); |
||||
} |
||||
|
||||
@Transactional(rollbackFor = Exception.class) |
||||
public int markMessagesAsRead(Long[] ids) { |
||||
if (ids == null || ids.length == 0) { |
||||
return 0; |
||||
} |
||||
int updatedRows = internalMessageMapper.updateMessageStatusToRead(ids); |
||||
return updatedRows; |
||||
} |
||||
|
||||
|
||||
/** |
||||
* 查询站内信列表 |
||||
* |
||||
* @param internalMessage 站内信 |
||||
* @return 站内信 |
||||
*/ |
||||
@Override |
||||
public List<InternalMessage> selectInternalMessageList(InternalMessage internalMessage) |
||||
{ |
||||
return internalMessageMapper.selectInternalMessageList(internalMessage); |
||||
} |
||||
|
||||
/** |
||||
* 新增站内信 |
||||
* |
||||
* @param internalMessage 站内信 |
||||
* @return 结果 |
||||
*/ |
||||
@Override |
||||
public int insertInternalMessage(InternalMessage internalMessage) |
||||
{ |
||||
internalMessage.setCreateTime(DateUtils.getNowDate()); |
||||
return internalMessageMapper.insertInternalMessage(internalMessage); |
||||
} |
||||
|
||||
/** |
||||
* 修改站内信 |
||||
* |
||||
* @param internalMessage 站内信 |
||||
* @return 结果 |
||||
*/ |
||||
@Override |
||||
public int updateInternalMessage(InternalMessage internalMessage) |
||||
{ |
||||
return internalMessageMapper.updateInternalMessage(internalMessage); |
||||
} |
||||
|
||||
/** |
||||
* 批量删除站内信 |
||||
* |
||||
* @param ids 需要删除的站内信主键 |
||||
* @return 结果 |
||||
*/ |
||||
@Override |
||||
public int deleteInternalMessageByIds(Long[] ids) |
||||
{ |
||||
return internalMessageMapper.deleteInternalMessageByIds(ids); |
||||
} |
||||
|
||||
/** |
||||
* 删除站内信信息 |
||||
* |
||||
* @param id 站内信主键 |
||||
* @return 结果 |
||||
*/ |
||||
@Override |
||||
public int deleteInternalMessageById(Long id) |
||||
{ |
||||
return internalMessageMapper.deleteInternalMessageById(id); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue