Compare commits
40 Commits
feature-Ki
...
f6769c8c69
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
f6769c8c69 | ||
|
|
5e1f3b7e8a | ||
|
|
e30a0dba97 | ||
|
|
74d47d586f | ||
|
|
0bb69075f7 | ||
|
|
1d179da910 | ||
|
|
c0969406b3 | ||
|
|
2a9acb946e | ||
|
|
fb2b7823b8 | ||
|
|
951cd7249d | ||
|
|
b12128fad6 | ||
|
|
212668ae1c | ||
|
|
a648ecad2a | ||
|
|
b8ec4a434d | ||
|
|
3bc00334ea | ||
| eb2b0cb999 | |||
| 30db5ad706 | |||
|
|
5151ccfacc | ||
|
|
ab2bde0517 | ||
|
|
c9cb410819 | ||
|
|
07a6b86d7d | ||
|
|
1cf9ba0aa9 | ||
|
|
6a68fefbd4 | ||
|
|
da66c3b048 | ||
|
|
a9730a8414 | ||
|
|
0da6cde202 | ||
|
|
59838e1f5b | ||
|
|
7601fbb3c0 | ||
|
|
170ce83fe1 | ||
|
|
aa16c30d57 | ||
|
|
8fdbad8bd5 | ||
|
|
a9598ba795 | ||
|
|
439b6789fa | ||
|
|
9c7afc4470 | ||
|
|
e92cd31c27 | ||
|
|
bd12c599b0 | ||
|
|
b5a15a3f01 | ||
|
|
039443dc0f | ||
| ae0cca5437 | |||
| c329eaef79 |
12
pom.xml
12
pom.xml
@@ -88,6 +88,12 @@
|
|||||||
<artifactId>spring-boot-starter-aop</artifactId>
|
<artifactId>spring-boot-starter-aop</artifactId>
|
||||||
</dependency>
|
</dependency>
|
||||||
|
|
||||||
|
<!-- WebSocket 起步依赖 -->
|
||||||
|
<dependency>
|
||||||
|
<groupId>org.springframework.boot</groupId>
|
||||||
|
<artifactId>spring-boot-starter-websocket</artifactId>
|
||||||
|
</dependency>
|
||||||
|
|
||||||
<!-- 阿里云相关依赖 -->
|
<!-- 阿里云相关依赖 -->
|
||||||
<dependency>
|
<dependency>
|
||||||
<groupId>com.aliyun.oss</groupId>
|
<groupId>com.aliyun.oss</groupId>
|
||||||
@@ -165,6 +171,12 @@
|
|||||||
<artifactId>spring-boot-starter-data-redis</artifactId>
|
<artifactId>spring-boot-starter-data-redis</artifactId>
|
||||||
</dependency>
|
</dependency>
|
||||||
|
|
||||||
|
<dependency>
|
||||||
|
<groupId>com.github.pagehelper</groupId>
|
||||||
|
<artifactId>pagehelper-spring-boot-starter</artifactId>
|
||||||
|
<version>1.4.7</version>
|
||||||
|
</dependency>
|
||||||
|
|
||||||
</dependencies>
|
</dependencies>
|
||||||
<dependencyManagement>
|
<dependencyManagement>
|
||||||
<dependencies>
|
<dependencies>
|
||||||
|
|||||||
@@ -1,16 +0,0 @@
|
|||||||
package com.bao.dating.common;
|
|
||||||
|
|
||||||
|
|
||||||
import lombok.Data;
|
|
||||||
|
|
||||||
import java.time.LocalDateTime;
|
|
||||||
/**
|
|
||||||
* 动态表
|
|
||||||
*/
|
|
||||||
@Data
|
|
||||||
public class Post {
|
|
||||||
private Long post_id; // 动态ID
|
|
||||||
private String content; // 动态内容
|
|
||||||
private Long user_id; // 发布人ID
|
|
||||||
private LocalDateTime created_at; // 创建时间
|
|
||||||
}
|
|
||||||
28
src/main/java/com/bao/dating/common/result/PageResult.java
Normal file
28
src/main/java/com/bao/dating/common/result/PageResult.java
Normal file
@@ -0,0 +1,28 @@
|
|||||||
|
package com.bao.dating.common.result;
|
||||||
|
|
||||||
|
import lombok.Data;
|
||||||
|
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
@Data
|
||||||
|
public class PageResult<T> {
|
||||||
|
/** 总记录数 */
|
||||||
|
private Long total;
|
||||||
|
/** 当前页码 */
|
||||||
|
private Integer page;
|
||||||
|
/** 每页大小 */
|
||||||
|
private Integer size;
|
||||||
|
/** 总页数 */
|
||||||
|
private Integer pages;
|
||||||
|
/** 当前页数据 */
|
||||||
|
private List<T> records;
|
||||||
|
|
||||||
|
public PageResult(Long total, Integer page, Integer size, List<T> records) {
|
||||||
|
this.total = total;
|
||||||
|
this.page = page;
|
||||||
|
this.size = size;
|
||||||
|
this.records = records;
|
||||||
|
// 计算总页数
|
||||||
|
this.pages = (int) Math.ceil((double) total / size);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -14,9 +14,9 @@ import org.springframework.data.redis.serializer.StringRedisSerializer;
|
|||||||
@Configuration
|
@Configuration
|
||||||
public class RedisConfig {
|
public class RedisConfig {
|
||||||
@Bean
|
@Bean
|
||||||
public RedisTemplate<String, Object> redisTemplate(RedisConnectionFactory redisConnectionFactory) {
|
public RedisTemplate redisTemplate(RedisConnectionFactory redisConnectionFactory) {
|
||||||
// 创建RedisTemplate对象
|
// 创建RedisTemplate对象
|
||||||
RedisTemplate<String, Object> redisTemplate = new RedisTemplate<>();
|
RedisTemplate redisTemplate = new RedisTemplate<>();
|
||||||
// 设置redis的连接工厂对象
|
// 设置redis的连接工厂对象
|
||||||
redisTemplate.setConnectionFactory(redisConnectionFactory);
|
redisTemplate.setConnectionFactory(redisConnectionFactory);
|
||||||
|
|
||||||
|
|||||||
35
src/main/java/com/bao/dating/config/WebSocketConfig.java
Normal file
35
src/main/java/com/bao/dating/config/WebSocketConfig.java
Normal file
@@ -0,0 +1,35 @@
|
|||||||
|
package com.bao.dating.config;
|
||||||
|
|
||||||
|
|
||||||
|
import com.bao.dating.controller.websocket.ChatWebSocketHandler;
|
||||||
|
import com.bao.dating.interceptor.WsAuthInterceptor;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
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 配置类
|
||||||
|
* @author lenovo
|
||||||
|
*/
|
||||||
|
@Configuration
|
||||||
|
@EnableWebSocket
|
||||||
|
public class WebSocketConfig implements WebSocketConfigurer {
|
||||||
|
@Autowired
|
||||||
|
private ChatWebSocketHandler chatWebSocketHandler;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private WsAuthInterceptor wsAuthInterceptor;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 注册 WebSocket 处理器
|
||||||
|
* @param registry
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
|
||||||
|
registry.addHandler(chatWebSocketHandler, "/ws/chat")
|
||||||
|
.addInterceptors(wsAuthInterceptor)
|
||||||
|
.setAllowedOrigins("*");
|
||||||
|
}
|
||||||
|
}
|
||||||
60
src/main/java/com/bao/dating/controller/ChatController.java
Normal file
60
src/main/java/com/bao/dating/controller/ChatController.java
Normal file
@@ -0,0 +1,60 @@
|
|||||||
|
package com.bao.dating.controller;
|
||||||
|
|
||||||
|
import com.bao.dating.common.Result;
|
||||||
|
import com.bao.dating.common.ResultCode;
|
||||||
|
import com.bao.dating.common.result.PageResult;
|
||||||
|
import com.bao.dating.context.UserContext;
|
||||||
|
import com.bao.dating.pojo.dto.ChatCursorPageDTO;
|
||||||
|
import com.bao.dating.pojo.dto.ChatHistoryQueryDTO;
|
||||||
|
import com.bao.dating.pojo.vo.ChatRecordsVO;
|
||||||
|
import com.bao.dating.pojo.vo.ChatSessionsVO;
|
||||||
|
import com.bao.dating.service.ChatService;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
import org.springframework.web.bind.annotation.*;
|
||||||
|
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 聊天控制器
|
||||||
|
* @author lenovo
|
||||||
|
*/
|
||||||
|
@Slf4j
|
||||||
|
@RestController
|
||||||
|
@RequestMapping("/chat")
|
||||||
|
public class ChatController {
|
||||||
|
@Autowired
|
||||||
|
private ChatService chatService;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取聊天记录
|
||||||
|
* @param targetUserId 目标用户ID
|
||||||
|
* @param pageDTO 分页参数
|
||||||
|
* @return 聊天记录列表
|
||||||
|
*/
|
||||||
|
@GetMapping("/history/{targetUserId}")
|
||||||
|
public Result<List<ChatRecordsVO>> getChatHistory(
|
||||||
|
@PathVariable Long targetUserId,
|
||||||
|
ChatCursorPageDTO pageDTO){
|
||||||
|
Long currentUserId = UserContext.getUserId();
|
||||||
|
ChatHistoryQueryDTO queryDTO = new ChatHistoryQueryDTO();
|
||||||
|
queryDTO.setCurrentUserId(currentUserId);
|
||||||
|
queryDTO.setTargetUserId(targetUserId);
|
||||||
|
queryDTO.setCursor(pageDTO.getCursor());
|
||||||
|
queryDTO.setSize(pageDTO.getSize());
|
||||||
|
chatService.markChatMessagesAsRead(currentUserId, targetUserId);
|
||||||
|
List<ChatRecordsVO> history = chatService.getChatHistory(queryDTO);
|
||||||
|
return Result.success(ResultCode.SUCCESS, "获取聊天记录成功", history);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取会话列表
|
||||||
|
* @return 会话列表
|
||||||
|
*/
|
||||||
|
@GetMapping("/sessions")
|
||||||
|
public Result<List<ChatSessionsVO>> getSessionList() {
|
||||||
|
Long currentUserId = UserContext.getUserId();
|
||||||
|
List<ChatSessionsVO> list = chatService.getSessionList(currentUserId);
|
||||||
|
return Result.success(ResultCode.SUCCESS, "获取会话列表成功", list);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -4,12 +4,11 @@ import com.bao.dating.anno.Log;
|
|||||||
import com.bao.dating.common.Result;
|
import com.bao.dating.common.Result;
|
||||||
import com.bao.dating.common.ResultCode;
|
import com.bao.dating.common.ResultCode;
|
||||||
import com.bao.dating.context.UserContext;
|
import com.bao.dating.context.UserContext;
|
||||||
import com.bao.dating.pojo.dto.UserInfoUpdateDTO;
|
import com.bao.dating.pojo.dto.UserInfoDTO;
|
||||||
import com.bao.dating.pojo.dto.UserLoginDTO;
|
import com.bao.dating.pojo.dto.UserLoginDTO;
|
||||||
import com.bao.dating.pojo.vo.UserInfoVO;
|
import com.bao.dating.pojo.vo.UserInfoVO;
|
||||||
import com.bao.dating.pojo.vo.UserLoginVO;
|
import com.bao.dating.pojo.vo.UserLoginVO;
|
||||||
import com.bao.dating.service.UserService;
|
import com.bao.dating.service.UserService;
|
||||||
import io.jsonwebtoken.Jwt;
|
|
||||||
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.bind.annotation.*;
|
||||||
import org.springframework.web.multipart.MultipartFile;
|
import org.springframework.web.multipart.MultipartFile;
|
||||||
@@ -91,7 +90,7 @@ public class UserController {
|
|||||||
*/
|
*/
|
||||||
@Log
|
@Log
|
||||||
@PostMapping("/info/update")
|
@PostMapping("/info/update")
|
||||||
public Result<UserInfoVO> userInfoUpdate(@RequestBody UserInfoUpdateDTO userInfoUpdateDTO) {
|
public Result<UserInfoVO> userInfoUpdate(@RequestBody UserInfoDTO userInfoUpdateDTO) {
|
||||||
Long userId = UserContext.getUserId();
|
Long userId = UserContext.getUserId();
|
||||||
userInfoUpdateDTO.setUserId(userId);
|
userInfoUpdateDTO.setUserId(userId);
|
||||||
UserInfoVO userInfoVO =userService.updateUserInfo(userInfoUpdateDTO);
|
UserInfoVO userInfoVO =userService.updateUserInfo(userInfoUpdateDTO);
|
||||||
|
|||||||
@@ -0,0 +1,107 @@
|
|||||||
|
package com.bao.dating.controller.websocket;
|
||||||
|
|
||||||
|
import com.bao.dating.message.WsMessage;
|
||||||
|
import com.bao.dating.pojo.dto.ChatRecordSendDTO;
|
||||||
|
import com.bao.dating.pojo.vo.ChatRecordsVO;
|
||||||
|
import com.bao.dating.service.ChatService;
|
||||||
|
import com.bao.dating.session.WsSessionManager;
|
||||||
|
import com.fasterxml.jackson.core.type.TypeReference;
|
||||||
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
import org.springframework.stereotype.Component;
|
||||||
|
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;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* WebSocket处理器类
|
||||||
|
* @author lenovo
|
||||||
|
*/
|
||||||
|
@Slf4j
|
||||||
|
@Component
|
||||||
|
public class ChatWebSocketHandler extends TextWebSocketHandler {
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private ObjectMapper objectMapper;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private ChatService chatService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private WsSessionManager sessionManager;
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 用户建立连接(上线)
|
||||||
|
* @param session
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
public void afterConnectionEstablished(WebSocketSession session) {
|
||||||
|
Long userId = (Long) session.getAttributes().get("userId");
|
||||||
|
sessionManager.addSession(userId, session);
|
||||||
|
log.info("用户 " + userId + " 已上线");
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 接收并处理消息
|
||||||
|
* @param session
|
||||||
|
* @param message
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception{
|
||||||
|
// 获取当前用户ID
|
||||||
|
Long senderUserId = (Long) session.getAttributes().get("userId");
|
||||||
|
if (senderUserId == null) {
|
||||||
|
log.error("WebSocket session 中未找到 userId");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// 解析消息
|
||||||
|
WsMessage<ChatRecordSendDTO> wsMessage =
|
||||||
|
objectMapper.readValue(message.getPayload(),
|
||||||
|
new TypeReference<WsMessage<ChatRecordSendDTO>>(){});
|
||||||
|
|
||||||
|
// 处理私聊消息
|
||||||
|
if ("chat".equals(wsMessage.getType())) {
|
||||||
|
handlePrivateChat(senderUserId, wsMessage.getData());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 私聊处理
|
||||||
|
*/
|
||||||
|
private void handlePrivateChat(Long senderUserId, ChatRecordSendDTO dto) throws Exception {
|
||||||
|
|
||||||
|
// 1. 消息入库 + 会话更新
|
||||||
|
ChatRecordsVO chatRecordsVO = chatService.createSession(senderUserId, dto);
|
||||||
|
|
||||||
|
// 2. 推送给接收方
|
||||||
|
WebSocketSession receiverSession =
|
||||||
|
sessionManager.getSession(dto.getReceiverUserId());
|
||||||
|
|
||||||
|
if (receiverSession != null && receiverSession.isOpen()) {
|
||||||
|
WsMessage<ChatRecordsVO> pushMsg = new WsMessage<>();
|
||||||
|
pushMsg.setType("chat");
|
||||||
|
pushMsg.setData(chatRecordsVO);
|
||||||
|
|
||||||
|
receiverSession.sendMessage(
|
||||||
|
new TextMessage(objectMapper.writeValueAsString(pushMsg))
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 用户断开连接(下线)
|
||||||
|
* @param session
|
||||||
|
* @param status
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
public void afterConnectionClosed(WebSocketSession session, CloseStatus status){
|
||||||
|
// 下线处理
|
||||||
|
Long userId = (Long) session.getAttributes().get("userId");
|
||||||
|
sessionManager.removeSession(userId);
|
||||||
|
log.info("用户 " + userId + " 已下线");
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -23,7 +23,7 @@ import org.springframework.web.servlet.HandlerInterceptor;
|
|||||||
public class TokenInterceptor implements HandlerInterceptor {
|
public class TokenInterceptor implements HandlerInterceptor {
|
||||||
|
|
||||||
@Autowired
|
@Autowired
|
||||||
private RedisTemplate<String, Object> redisTemplate;
|
private RedisTemplate redisTemplate;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 在请求处理之前进行拦截
|
* 在请求处理之前进行拦截
|
||||||
@@ -68,7 +68,7 @@ public class TokenInterceptor implements HandlerInterceptor {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 解析 token
|
// 解析 token
|
||||||
String userId = JwtUtil.getSubjectFromToken(token);
|
Long userId = Long.valueOf(JwtUtil.getSubjectFromToken(token));
|
||||||
|
|
||||||
// 从Redis获取存储的token进行比对
|
// 从Redis获取存储的token进行比对
|
||||||
Object redisTokenObj = redisTemplate.opsForValue().get("login:token:" + userId);
|
Object redisTokenObj = redisTemplate.opsForValue().get("login:token:" + userId);
|
||||||
@@ -85,7 +85,7 @@ public class TokenInterceptor implements HandlerInterceptor {
|
|||||||
|
|
||||||
log.info("用户: {}", userId);
|
log.info("用户: {}", userId);
|
||||||
// 保存 userId 到 ThreadLocal
|
// 保存 userId 到 ThreadLocal
|
||||||
UserContext.setUserId(Long.valueOf(userId));
|
UserContext.setUserId(userId);
|
||||||
return true;
|
return true;
|
||||||
} catch (Exception e) {
|
} catch (Exception e) {
|
||||||
log.error("Token 校验失败: {}", e.getMessage());
|
log.error("Token 校验失败: {}", e.getMessage());
|
||||||
|
|||||||
111
src/main/java/com/bao/dating/interceptor/WsAuthInterceptor.java
Normal file
111
src/main/java/com/bao/dating/interceptor/WsAuthInterceptor.java
Normal file
@@ -0,0 +1,111 @@
|
|||||||
|
package com.bao.dating.interceptor;
|
||||||
|
|
||||||
|
import com.bao.dating.util.JwtUtil;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.apache.commons.lang3.StringUtils;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
import org.springframework.data.redis.core.RedisTemplate;
|
||||||
|
import org.springframework.http.server.ServerHttpRequest;
|
||||||
|
import org.springframework.http.server.ServerHttpResponse;
|
||||||
|
import org.springframework.http.server.ServletServerHttpRequest;
|
||||||
|
import org.springframework.stereotype.Component;
|
||||||
|
import org.springframework.web.socket.WebSocketHandler;
|
||||||
|
import org.springframework.web.socket.server.HandshakeInterceptor;
|
||||||
|
|
||||||
|
import javax.servlet.http.HttpServletRequest;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* WebSocket 认证拦截器
|
||||||
|
*/
|
||||||
|
@Slf4j
|
||||||
|
@Component
|
||||||
|
public class WsAuthInterceptor implements HandshakeInterceptor {
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private RedisTemplate<String, Object> redisTemplate;
|
||||||
|
/**
|
||||||
|
* 拦截WebSocket连接请求
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception {
|
||||||
|
log.info("开始WebSocket握手认证");
|
||||||
|
|
||||||
|
// 获取请求参数
|
||||||
|
if (!(request instanceof ServletServerHttpRequest)) {
|
||||||
|
log.error("WebSocket握手失败:非HTTP请求");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取HttpServletRequest对象
|
||||||
|
HttpServletRequest servletRequest = ((ServletServerHttpRequest) request).getServletRequest();
|
||||||
|
|
||||||
|
// 从URL参数中获取token
|
||||||
|
String token = servletRequest.getParameter("token");
|
||||||
|
|
||||||
|
if (StringUtils.isBlank(token)) {
|
||||||
|
log.error("WebSocket握手失败:令牌丢失");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
// 验证 token 是否有效(包括是否过期)
|
||||||
|
if (!JwtUtil.validateToken(token)) {
|
||||||
|
log.error("Token无效或已过期: {}", token);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查 token 是否在黑名单中
|
||||||
|
String blacklistKey = "jwt:blacklist:" + token;
|
||||||
|
Object blacklistToken = redisTemplate.opsForValue().get(blacklistKey);
|
||||||
|
if (blacklistToken != null) {
|
||||||
|
log.error("Token已在黑名单中: {}", token);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 验证token并获取用户ID
|
||||||
|
String userIdStr = JwtUtil.getSubjectFromToken(token);
|
||||||
|
|
||||||
|
if (StringUtils.isBlank(userIdStr) || !userIdStr.matches("\\d+")) {
|
||||||
|
log.error("无效的用户ID格式: {}", userIdStr);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
Long userId = Long.valueOf(userIdStr);
|
||||||
|
|
||||||
|
// 从Redis获取存储的token进行比对
|
||||||
|
String redisTokenKey = "login:token:" + userId;
|
||||||
|
Object redisTokenObj = redisTemplate.opsForValue().get(redisTokenKey);
|
||||||
|
String redisToken = redisTokenObj != null ? redisTokenObj.toString() : null;
|
||||||
|
log.info("Redis中存储的token: {}", redisToken != null ? "存在" : "不存在");
|
||||||
|
|
||||||
|
// 验证Redis中的token是否存在且匹配
|
||||||
|
if (redisToken == null || !redisToken.equals(token)) {
|
||||||
|
log.error("登录已失效 - Redis中token不存在或不匹配");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
log.info("WebSocket认证成功,用户ID: {}", userId);
|
||||||
|
// 将用户ID保存到attributes中
|
||||||
|
attributes.put("userId", userId);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
catch (NumberFormatException e) {
|
||||||
|
log.error("用户ID格式转换异常: {}", e.getMessage());
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
catch (Exception e) {
|
||||||
|
log.error("WebSocket握手失败:{}", e.getMessage(), e);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Exception exception) {
|
||||||
|
if (exception != null) {
|
||||||
|
log.error("WebSocket握手后出现异常:", exception);
|
||||||
|
} else {
|
||||||
|
log.info("WebSocket握手完成");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
35
src/main/java/com/bao/dating/mapper/ChatRecordsMapper.java
Normal file
35
src/main/java/com/bao/dating/mapper/ChatRecordsMapper.java
Normal file
@@ -0,0 +1,35 @@
|
|||||||
|
package com.bao.dating.mapper;
|
||||||
|
|
||||||
|
|
||||||
|
import com.bao.dating.pojo.dto.ChatHistoryQueryDTO;
|
||||||
|
import com.bao.dating.pojo.dto.ChatMarkReadDTO;
|
||||||
|
import com.bao.dating.pojo.entity.ChatRecords;
|
||||||
|
import org.apache.ibatis.annotations.Mapper;
|
||||||
|
import org.apache.ibatis.annotations.Param;
|
||||||
|
|
||||||
|
import java.time.LocalDateTime;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
@Mapper
|
||||||
|
public interface ChatRecordsMapper {
|
||||||
|
/**
|
||||||
|
* 插入聊天记录
|
||||||
|
* @param chatRecords 聊天记录
|
||||||
|
* @return 影响行数
|
||||||
|
*/
|
||||||
|
int insert(ChatRecords chatRecords);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 根据时间游标查询聊天记录
|
||||||
|
* @param queryDTO 查询参数
|
||||||
|
* @return 聊天记录列表
|
||||||
|
*/
|
||||||
|
List<ChatRecords> selectChatHistoryByCursor(ChatHistoryQueryDTO queryDTO);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 标记聊天记录为已读
|
||||||
|
* @param markReadDTO 标记参数
|
||||||
|
* @return 影响行数
|
||||||
|
*/
|
||||||
|
int markMessagesAsRead(ChatMarkReadDTO markReadDTO);
|
||||||
|
}
|
||||||
45
src/main/java/com/bao/dating/mapper/ChatSessionsMapper.java
Normal file
45
src/main/java/com/bao/dating/mapper/ChatSessionsMapper.java
Normal file
@@ -0,0 +1,45 @@
|
|||||||
|
package com.bao.dating.mapper;
|
||||||
|
|
||||||
|
|
||||||
|
import com.bao.dating.pojo.entity.ChatRecords;
|
||||||
|
import com.bao.dating.pojo.entity.ChatSessions;
|
||||||
|
import org.apache.ibatis.annotations.Mapper;
|
||||||
|
import org.apache.ibatis.annotations.Param;
|
||||||
|
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
@Mapper
|
||||||
|
public interface ChatSessionsMapper {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 如果发送方存在会话则更新,不存在则创建
|
||||||
|
* @param chatSessions 会话
|
||||||
|
* @return 影响行数
|
||||||
|
*/
|
||||||
|
int upsertSessionForSender(ChatSessions chatSessions);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 如果接收方存在会话则更新,不存在则创建
|
||||||
|
* @param chatSessions 会话
|
||||||
|
* @return 影响行数
|
||||||
|
*/
|
||||||
|
int upsertSessionForReceiver(ChatSessions chatSessions);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 清空会话的未读数
|
||||||
|
* @param userId 用户ID
|
||||||
|
* @param targetUserId 目标用户ID
|
||||||
|
* @return 影响行数
|
||||||
|
*/
|
||||||
|
int clearUnreadCount(
|
||||||
|
@Param("userId") Long userId,
|
||||||
|
@Param("targetUserId") Long targetUserId
|
||||||
|
);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 查询当前用户的会话列表
|
||||||
|
* @param userId 当前用户ID
|
||||||
|
* @return 会话列表
|
||||||
|
*/
|
||||||
|
List<ChatSessions> selectSessionsByUserId(@Param("userId") Long userId);
|
||||||
|
}
|
||||||
@@ -1,6 +1,6 @@
|
|||||||
package com.bao.dating.mapper;
|
package com.bao.dating.mapper;
|
||||||
|
|
||||||
import com.bao.dating.pojo.dto.UserInfoUpdateDTO;
|
import com.bao.dating.pojo.dto.UserInfoDTO;
|
||||||
import com.bao.dating.pojo.entity.User;
|
import com.bao.dating.pojo.entity.User;
|
||||||
import org.apache.ibatis.annotations.Mapper;
|
import org.apache.ibatis.annotations.Mapper;
|
||||||
|
|
||||||
@@ -31,6 +31,6 @@ public interface UserMapper {
|
|||||||
* 更新用户信息
|
* 更新用户信息
|
||||||
* @param userInfoUpdateDTO 用户信息更新参数
|
* @param userInfoUpdateDTO 用户信息更新参数
|
||||||
*/
|
*/
|
||||||
void updateUserInfoByUserId(UserInfoUpdateDTO userInfoUpdateDTO);
|
void updateUserInfoByUserId(UserInfoDTO userInfoUpdateDTO);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|||||||
17
src/main/java/com/bao/dating/message/WsMessage.java
Normal file
17
src/main/java/com/bao/dating/message/WsMessage.java
Normal file
@@ -0,0 +1,17 @@
|
|||||||
|
package com.bao.dating.message;
|
||||||
|
|
||||||
|
import lombok.Data;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* WebSocket 消息
|
||||||
|
* @author KilLze
|
||||||
|
*/
|
||||||
|
@Data
|
||||||
|
public class WsMessage<T> {
|
||||||
|
|
||||||
|
/** 消息类型:chat / read / system */
|
||||||
|
private String type;
|
||||||
|
|
||||||
|
/** 消息体 */
|
||||||
|
private T data;
|
||||||
|
}
|
||||||
25
src/main/java/com/bao/dating/pojo/dto/ChatCursorPageDTO.java
Normal file
25
src/main/java/com/bao/dating/pojo/dto/ChatCursorPageDTO.java
Normal file
@@ -0,0 +1,25 @@
|
|||||||
|
package com.bao.dating.pojo.dto;
|
||||||
|
|
||||||
|
import lombok.Data;
|
||||||
|
import org.springframework.format.annotation.DateTimeFormat;
|
||||||
|
|
||||||
|
import java.time.LocalDateTime;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 聊天记录分页参数
|
||||||
|
* @author lenovo
|
||||||
|
*/
|
||||||
|
@Data
|
||||||
|
public class ChatCursorPageDTO {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 时间游标:最后一条消息的发送时间
|
||||||
|
*/
|
||||||
|
@DateTimeFormat(iso = DateTimeFormat.ISO.DATE_TIME)
|
||||||
|
private LocalDateTime cursor;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 拉取条数
|
||||||
|
*/
|
||||||
|
private Integer size;
|
||||||
|
}
|
||||||
@@ -0,0 +1,17 @@
|
|||||||
|
package com.bao.dating.pojo.dto;
|
||||||
|
|
||||||
|
import lombok.Data;
|
||||||
|
|
||||||
|
import java.time.LocalDateTime;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 聊天记录查询参数
|
||||||
|
* @author KilLze
|
||||||
|
*/
|
||||||
|
@Data
|
||||||
|
public class ChatHistoryQueryDTO {
|
||||||
|
private Long currentUserId;
|
||||||
|
private Long targetUserId;
|
||||||
|
private LocalDateTime cursor;
|
||||||
|
private Integer size;
|
||||||
|
}
|
||||||
16
src/main/java/com/bao/dating/pojo/dto/ChatMarkReadDTO.java
Normal file
16
src/main/java/com/bao/dating/pojo/dto/ChatMarkReadDTO.java
Normal file
@@ -0,0 +1,16 @@
|
|||||||
|
package com.bao.dating.pojo.dto;
|
||||||
|
|
||||||
|
import lombok.Data;
|
||||||
|
|
||||||
|
import java.time.LocalDateTime;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 聊天记录已读参数
|
||||||
|
* @author lenovo
|
||||||
|
*/
|
||||||
|
@Data
|
||||||
|
public class ChatMarkReadDTO {
|
||||||
|
private Long senderUserId;
|
||||||
|
private Long receiverUserId;
|
||||||
|
private LocalDateTime readTime;
|
||||||
|
}
|
||||||
17
src/main/java/com/bao/dating/pojo/dto/ChatRecordSendDTO.java
Normal file
17
src/main/java/com/bao/dating/pojo/dto/ChatRecordSendDTO.java
Normal file
@@ -0,0 +1,17 @@
|
|||||||
|
package com.bao.dating.pojo.dto;
|
||||||
|
|
||||||
|
import lombok.Data;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 聊天记录发送数据传输对象
|
||||||
|
* @author KilLze
|
||||||
|
*/
|
||||||
|
@Data
|
||||||
|
public class ChatRecordSendDTO {
|
||||||
|
/** 接收者用户ID */
|
||||||
|
private Long receiverUserId;
|
||||||
|
/** 消息内容 */
|
||||||
|
private String messageContent;
|
||||||
|
/** 消息类型 (1-文本消息,2-文件消息) */
|
||||||
|
private Integer messageType;
|
||||||
|
}
|
||||||
@@ -8,11 +8,11 @@ import java.time.LocalDateTime;
|
|||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 用户信息更新数据传输对象
|
* 用户信息数据传输对象
|
||||||
* @author KilLze
|
* @author KilLze
|
||||||
*/
|
*/
|
||||||
@Data
|
@Data
|
||||||
public class UserInfoUpdateDTO implements Serializable {
|
public class UserInfoDTO implements Serializable {
|
||||||
private Long userId;
|
private Long userId;
|
||||||
private String userName;
|
private String userName;
|
||||||
private String nickname;
|
private String nickname;
|
||||||
35
src/main/java/com/bao/dating/pojo/entity/ChatRecords.java
Normal file
35
src/main/java/com/bao/dating/pojo/entity/ChatRecords.java
Normal file
@@ -0,0 +1,35 @@
|
|||||||
|
package com.bao.dating.pojo.entity;
|
||||||
|
|
||||||
|
import lombok.Data;
|
||||||
|
|
||||||
|
import java.time.LocalDateTime;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 聊天记录表
|
||||||
|
* @author lenovo
|
||||||
|
*/
|
||||||
|
@Data
|
||||||
|
public class ChatRecords {
|
||||||
|
/** 聊天记录ID */
|
||||||
|
private Long chatId;
|
||||||
|
/** 发送者用户ID */
|
||||||
|
private Long senderUserId;
|
||||||
|
/** 接收者用户ID */
|
||||||
|
private Long receiverUserId;
|
||||||
|
/** 消息内容 */
|
||||||
|
private String messageContent;
|
||||||
|
/** 消息类型 (1-文本消息,2-文件消息) */
|
||||||
|
private Integer messageType;
|
||||||
|
/** 阅读状态 (0-未读,1-已读) */
|
||||||
|
private Integer readStatus;
|
||||||
|
/** 阅读时间 */
|
||||||
|
private LocalDateTime readTime;
|
||||||
|
/** 发送时间 */
|
||||||
|
private LocalDateTime sendTime;
|
||||||
|
/** 消息状态 (1-正常,2-已撤回,3-已删除) */
|
||||||
|
private Integer messageStatus;
|
||||||
|
/** 创建时间 */
|
||||||
|
private LocalDateTime createdAt;
|
||||||
|
/** 更新时间 */
|
||||||
|
private LocalDateTime updatedAt;
|
||||||
|
}
|
||||||
40
src/main/java/com/bao/dating/pojo/entity/ChatSessions.java
Normal file
40
src/main/java/com/bao/dating/pojo/entity/ChatSessions.java
Normal file
@@ -0,0 +1,40 @@
|
|||||||
|
package com.bao.dating.pojo.entity;
|
||||||
|
|
||||||
|
import lombok.Data;
|
||||||
|
|
||||||
|
import java.time.LocalDateTime;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 会话表
|
||||||
|
* @author KilLze
|
||||||
|
*/
|
||||||
|
@Data
|
||||||
|
public class ChatSessions {
|
||||||
|
/** 会话ID */
|
||||||
|
private Long sessionId;
|
||||||
|
/** 所属用户ID */
|
||||||
|
private Long userId;
|
||||||
|
/** 目标用户ID */
|
||||||
|
private Long targetUserId;
|
||||||
|
/** 会话名称 */
|
||||||
|
private String sessionName;
|
||||||
|
/** 最后一条消息ID (关联chat_records.chat_id) */
|
||||||
|
private Long lastMessageId;
|
||||||
|
/** 最后一条消息内容 */
|
||||||
|
private String lastMessageContent;
|
||||||
|
/** 最后一条消息时间 */
|
||||||
|
private LocalDateTime lastMessageTime;
|
||||||
|
/** 未读消息数量 */
|
||||||
|
private Integer unreadCount;
|
||||||
|
/** 会话状态 (1-正常,2-已隐藏,3-已删除) */
|
||||||
|
private Integer sessionStatus;
|
||||||
|
/** 置顶状态 (0-未置顶,1-已置顶) */
|
||||||
|
private Integer topStatus;
|
||||||
|
/** 免打扰状态 (0-正常提醒,1-免打扰) */
|
||||||
|
private Integer muteStatus;
|
||||||
|
/** 创建时间 */
|
||||||
|
private LocalDateTime createdAt;
|
||||||
|
/** 更新时间 */
|
||||||
|
private LocalDateTime updatedAt;
|
||||||
|
|
||||||
|
}
|
||||||
29
src/main/java/com/bao/dating/pojo/vo/ChatRecordsVO.java
Normal file
29
src/main/java/com/bao/dating/pojo/vo/ChatRecordsVO.java
Normal file
@@ -0,0 +1,29 @@
|
|||||||
|
package com.bao.dating.pojo.vo;
|
||||||
|
|
||||||
|
import lombok.Data;
|
||||||
|
|
||||||
|
import java.time.LocalDateTime;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 聊天记录返回数据
|
||||||
|
* @author KilLze
|
||||||
|
*/
|
||||||
|
@Data
|
||||||
|
public class ChatRecordsVO {
|
||||||
|
/** 聊天记录ID */
|
||||||
|
private Long chatId;
|
||||||
|
/** 发送者用户ID */
|
||||||
|
private Long senderUserId;
|
||||||
|
/** 接收者用户ID */
|
||||||
|
private Long receiverUserId;
|
||||||
|
/** 消息内容 */
|
||||||
|
private String messageContent;
|
||||||
|
/** 消息类型 (1-文本消息,2-文件消息) */
|
||||||
|
private Integer messageType;
|
||||||
|
/** 阅读状态 (0-未读,1-已读) */
|
||||||
|
private Integer readStatus;
|
||||||
|
/** 发送时间 */
|
||||||
|
private LocalDateTime sendTime;
|
||||||
|
/** 消息状态 (1-正常,2-已撤回,3-已删除) */
|
||||||
|
private Integer messageStatus;
|
||||||
|
}
|
||||||
32
src/main/java/com/bao/dating/pojo/vo/ChatSessionsVO.java
Normal file
32
src/main/java/com/bao/dating/pojo/vo/ChatSessionsVO.java
Normal file
@@ -0,0 +1,32 @@
|
|||||||
|
package com.bao.dating.pojo.vo;
|
||||||
|
|
||||||
|
import lombok.Data;
|
||||||
|
|
||||||
|
import java.time.LocalDateTime;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 会话信息
|
||||||
|
* @author KilLze
|
||||||
|
*/
|
||||||
|
@Data
|
||||||
|
public class ChatSessionsVO {
|
||||||
|
|
||||||
|
/** 会话ID */
|
||||||
|
private Long sessionId;
|
||||||
|
/** 对方用户ID */
|
||||||
|
private Long targetUserId;
|
||||||
|
/** 会话名称 */
|
||||||
|
private String sessionName;
|
||||||
|
/** 对方用户头像 */
|
||||||
|
private String avatarUrl;
|
||||||
|
/** 最后一条消息内容 */
|
||||||
|
private String lastMessageContent;
|
||||||
|
/** 最后一条消息时间 */
|
||||||
|
private LocalDateTime lastMessageTime;
|
||||||
|
/** 未读消息数量 */
|
||||||
|
private Integer unreadCount;
|
||||||
|
/** 置顶状态 */
|
||||||
|
private Integer topStatus;
|
||||||
|
/** 免打扰状态 */
|
||||||
|
private Integer muteStatus;
|
||||||
|
}
|
||||||
46
src/main/java/com/bao/dating/service/ChatService.java
Normal file
46
src/main/java/com/bao/dating/service/ChatService.java
Normal file
@@ -0,0 +1,46 @@
|
|||||||
|
package com.bao.dating.service;
|
||||||
|
|
||||||
|
import com.bao.dating.common.result.PageResult;
|
||||||
|
import com.bao.dating.pojo.dto.ChatHistoryQueryDTO;
|
||||||
|
import com.bao.dating.pojo.dto.ChatRecordSendDTO;
|
||||||
|
import com.bao.dating.pojo.vo.ChatRecordsVO;
|
||||||
|
import com.bao.dating.pojo.vo.ChatSessionsVO;
|
||||||
|
|
||||||
|
import java.time.LocalDateTime;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 聊天服务
|
||||||
|
* @author lenovo
|
||||||
|
*/
|
||||||
|
public interface ChatService {
|
||||||
|
/**
|
||||||
|
* 消息入库,如果会话不存在则创建会话
|
||||||
|
* @param senderUserId 发送方用户ID
|
||||||
|
* @param dto 发送参数
|
||||||
|
* @return 聊天记录VO
|
||||||
|
*/
|
||||||
|
ChatRecordsVO createSession(Long senderUserId, ChatRecordSendDTO dto);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取聊天记录
|
||||||
|
* @param dto 查询参数
|
||||||
|
* @return 聊天记录列表
|
||||||
|
*/
|
||||||
|
List<ChatRecordsVO> getChatHistory(ChatHistoryQueryDTO dto);
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 标记聊天记录为已读
|
||||||
|
* @param currentUserId 当前用户ID
|
||||||
|
* @param targetUserId 目标用户ID
|
||||||
|
*/
|
||||||
|
void markChatMessagesAsRead(Long currentUserId, Long targetUserId);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取会话列表
|
||||||
|
* @param currentUserId 当前用户ID
|
||||||
|
* @return 会话列表
|
||||||
|
*/
|
||||||
|
List<ChatSessionsVO> getSessionList(Long currentUserId);
|
||||||
|
}
|
||||||
@@ -1,6 +1,6 @@
|
|||||||
package com.bao.dating.service;
|
package com.bao.dating.service;
|
||||||
|
|
||||||
import com.bao.dating.pojo.dto.UserInfoUpdateDTO;
|
import com.bao.dating.pojo.dto.UserInfoDTO;
|
||||||
import com.bao.dating.pojo.dto.UserLoginDTO;
|
import com.bao.dating.pojo.dto.UserLoginDTO;
|
||||||
import com.bao.dating.pojo.vo.UserInfoVO;
|
import com.bao.dating.pojo.vo.UserInfoVO;
|
||||||
import com.bao.dating.pojo.vo.UserLoginVO;
|
import com.bao.dating.pojo.vo.UserLoginVO;
|
||||||
@@ -51,5 +51,12 @@ public interface UserService {
|
|||||||
* @param userInfoUpdateDTO 用户信息
|
* @param userInfoUpdateDTO 用户信息
|
||||||
* @return 更新后的用户信息
|
* @return 更新后的用户信息
|
||||||
*/
|
*/
|
||||||
UserInfoVO updateUserInfo(UserInfoUpdateDTO userInfoUpdateDTO);
|
UserInfoVO updateUserInfo(UserInfoDTO userInfoUpdateDTO);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 根据用户ID查询用户信息
|
||||||
|
* @param userId 用户ID
|
||||||
|
* @return 用户
|
||||||
|
*/
|
||||||
|
UserInfoDTO getUserInfoById(Long userId);
|
||||||
}
|
}
|
||||||
|
|||||||
182
src/main/java/com/bao/dating/service/impl/ChatServiceImpl.java
Normal file
182
src/main/java/com/bao/dating/service/impl/ChatServiceImpl.java
Normal file
@@ -0,0 +1,182 @@
|
|||||||
|
package com.bao.dating.service.impl;
|
||||||
|
|
||||||
|
import com.bao.dating.mapper.ChatRecordsMapper;
|
||||||
|
import com.bao.dating.mapper.ChatSessionsMapper;
|
||||||
|
import com.bao.dating.pojo.dto.ChatHistoryQueryDTO;
|
||||||
|
import com.bao.dating.pojo.dto.ChatMarkReadDTO;
|
||||||
|
import com.bao.dating.pojo.dto.ChatRecordSendDTO;
|
||||||
|
import com.bao.dating.pojo.dto.UserInfoDTO;
|
||||||
|
import com.bao.dating.pojo.entity.ChatRecords;
|
||||||
|
import com.bao.dating.pojo.entity.ChatSessions;
|
||||||
|
import com.bao.dating.pojo.vo.ChatRecordsVO;
|
||||||
|
import com.bao.dating.pojo.vo.ChatSessionsVO;
|
||||||
|
import com.bao.dating.service.ChatService;
|
||||||
|
import com.bao.dating.service.UserService;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.springframework.beans.BeanUtils;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
import org.springframework.stereotype.Service;
|
||||||
|
import org.springframework.transaction.annotation.Transactional;
|
||||||
|
|
||||||
|
import java.time.LocalDateTime;
|
||||||
|
import java.util.Collections;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.stream.Collectors;
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 聊天服务实现类
|
||||||
|
* @author lenovo
|
||||||
|
*/
|
||||||
|
@Slf4j
|
||||||
|
@Service
|
||||||
|
public class ChatServiceImpl implements ChatService {
|
||||||
|
@Autowired
|
||||||
|
private ChatRecordsMapper chatRecordsMapper;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private ChatSessionsMapper chatSessionsMapper;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private UserService userService;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 消息入库,如果会话不存在则创建会话
|
||||||
|
* @param senderUserId 发送者用户ID
|
||||||
|
* @param dto 消息
|
||||||
|
* @return 聊天记录
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
@Transactional(rollbackFor = Exception.class)
|
||||||
|
public ChatRecordsVO createSession(Long senderUserId, ChatRecordSendDTO dto) {
|
||||||
|
ChatRecords record = new ChatRecords();
|
||||||
|
record.setSenderUserId(senderUserId);
|
||||||
|
record.setReceiverUserId(dto.getReceiverUserId());
|
||||||
|
record.setMessageContent(dto.getMessageContent());
|
||||||
|
record.setMessageType(dto.getMessageType());
|
||||||
|
record.setReadStatus(0);
|
||||||
|
record.setMessageStatus(1);
|
||||||
|
record.setSendTime(LocalDateTime.now());
|
||||||
|
|
||||||
|
// 插入消息记录
|
||||||
|
chatRecordsMapper.insert(record);
|
||||||
|
|
||||||
|
// 创建发送方会话
|
||||||
|
ChatSessions sessions = new ChatSessions();
|
||||||
|
sessions.setUserId(senderUserId);
|
||||||
|
sessions.setTargetUserId(dto.getReceiverUserId());
|
||||||
|
// 获取接收方昵称作为发送方会话名称
|
||||||
|
try {
|
||||||
|
UserInfoDTO receiverNicknameInfo = userService.getUserInfoById(dto.getReceiverUserId());
|
||||||
|
if (receiverNicknameInfo != null && receiverNicknameInfo.getNickname() != null) {
|
||||||
|
sessions.setSessionName(receiverNicknameInfo.getNickname());
|
||||||
|
} else {
|
||||||
|
log.warn("接收用户ID获取用户信息时出现错误: {} , 使用默认id", dto.getReceiverUserId());
|
||||||
|
sessions.setSessionName("用户" + dto.getReceiverUserId());
|
||||||
|
}
|
||||||
|
} catch (Exception e) {
|
||||||
|
log.error("接收用户ID获取用户信息时出现错误: {} , 使用默认id", dto.getReceiverUserId(), e);
|
||||||
|
sessions.setSessionName("用户" + dto.getReceiverUserId());
|
||||||
|
}
|
||||||
|
sessions.setLastMessageId(record.getChatId());
|
||||||
|
sessions.setLastMessageContent(record.getMessageContent());
|
||||||
|
sessions.setLastMessageTime(record.getSendTime());
|
||||||
|
|
||||||
|
chatSessionsMapper.upsertSessionForSender(sessions);
|
||||||
|
|
||||||
|
// 创建接收方会话
|
||||||
|
sessions.setUserId(dto.getReceiverUserId());
|
||||||
|
sessions.setTargetUserId(senderUserId);
|
||||||
|
try {
|
||||||
|
UserInfoDTO senderNicknameInfo = userService.getUserInfoById(senderUserId);
|
||||||
|
if (senderNicknameInfo != null && senderNicknameInfo.getNickname() != null) {
|
||||||
|
sessions.setSessionName(senderNicknameInfo.getNickname());
|
||||||
|
} else {
|
||||||
|
log.warn("接收用户ID获取用户信息时出现错误: {} , 使用默认id", senderUserId);
|
||||||
|
sessions.setSessionName("用户" + senderUserId);
|
||||||
|
}
|
||||||
|
} catch (Exception e) {
|
||||||
|
log.error("接收用户ID获取用户信息时出现错误: {} , 使用默认id", senderUserId, e);
|
||||||
|
sessions.setSessionName("用户" + senderUserId);
|
||||||
|
}
|
||||||
|
|
||||||
|
chatSessionsMapper.upsertSessionForReceiver(sessions);
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
// 3. 返回 VO
|
||||||
|
ChatRecordsVO vo = new ChatRecordsVO();
|
||||||
|
BeanUtils.copyProperties(record, vo);
|
||||||
|
return vo;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取聊天记录
|
||||||
|
* @return 聊天记录列表
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
public List<ChatRecordsVO> getChatHistory(ChatHistoryQueryDTO dto) {
|
||||||
|
|
||||||
|
Integer size = dto.getSize();
|
||||||
|
if (size == null || size < 1 || size > 100) {
|
||||||
|
dto.setSize(50);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 查询聊天记录
|
||||||
|
List<ChatRecords> records = chatRecordsMapper.selectChatHistoryByCursor(dto);
|
||||||
|
|
||||||
|
// 倒序
|
||||||
|
Collections.reverse(records);
|
||||||
|
|
||||||
|
return records.stream().map(record -> {
|
||||||
|
ChatRecordsVO vo = new ChatRecordsVO();
|
||||||
|
BeanUtils.copyProperties(record, vo);
|
||||||
|
return vo;
|
||||||
|
}).collect(Collectors.toList());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
@Transactional(rollbackFor = Exception.class)
|
||||||
|
public void markChatMessagesAsRead(Long currentUserId, Long targetUserId) {
|
||||||
|
// 更新 chat_records:把对方发给我的未读消息设为已读
|
||||||
|
ChatMarkReadDTO markReadDTO = new ChatMarkReadDTO();
|
||||||
|
// 因为读的是对方发给我的消息所以要把填入的参数反过来
|
||||||
|
markReadDTO.setSenderUserId(targetUserId);
|
||||||
|
markReadDTO.setReceiverUserId(currentUserId);
|
||||||
|
markReadDTO.setReadTime(LocalDateTime.now());
|
||||||
|
|
||||||
|
chatRecordsMapper.markMessagesAsRead(markReadDTO);
|
||||||
|
|
||||||
|
// 更新 chat_sessions:清空当前会话的未读数
|
||||||
|
chatSessionsMapper.clearUnreadCount(
|
||||||
|
currentUserId,
|
||||||
|
targetUserId
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取会话列表
|
||||||
|
* @return 会话列表
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
public List<ChatSessionsVO> getSessionList(Long currentUserId) {
|
||||||
|
List<ChatSessions> sessions = chatSessionsMapper.selectSessionsByUserId(currentUserId);
|
||||||
|
|
||||||
|
return sessions.stream().map(session -> {
|
||||||
|
ChatSessionsVO vo = new ChatSessionsVO();
|
||||||
|
BeanUtils.copyProperties(session, vo);
|
||||||
|
|
||||||
|
UserInfoDTO targetUser = userService.getUserInfoById(session.getTargetUserId());
|
||||||
|
if (targetUser != null){
|
||||||
|
vo.setSessionName(targetUser.getNickname());
|
||||||
|
vo.setAvatarUrl(targetUser.getAvatarUrl());
|
||||||
|
}else {
|
||||||
|
vo.setSessionName("用户" + session.getTargetUserId());
|
||||||
|
vo.setAvatarUrl(null);
|
||||||
|
}
|
||||||
|
return vo;
|
||||||
|
}).collect(Collectors.toList());
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
@@ -5,10 +5,9 @@ import com.bao.dating.common.aliyun.GreenImageScan;
|
|||||||
import com.bao.dating.common.aliyun.GreenTextScan;
|
import com.bao.dating.common.aliyun.GreenTextScan;
|
||||||
import com.bao.dating.common.result.AliOssResult;
|
import com.bao.dating.common.result.AliOssResult;
|
||||||
import com.bao.dating.common.result.GreenAuditResult;
|
import com.bao.dating.common.result.GreenAuditResult;
|
||||||
import com.bao.dating.config.RedisConfig;
|
|
||||||
import com.bao.dating.context.UserContext;
|
import com.bao.dating.context.UserContext;
|
||||||
import com.bao.dating.mapper.UserMapper;
|
import com.bao.dating.mapper.UserMapper;
|
||||||
import com.bao.dating.pojo.dto.UserInfoUpdateDTO;
|
import com.bao.dating.pojo.dto.UserInfoDTO;
|
||||||
import com.bao.dating.pojo.dto.UserLoginDTO;
|
import com.bao.dating.pojo.dto.UserLoginDTO;
|
||||||
import com.bao.dating.pojo.entity.User;
|
import com.bao.dating.pojo.entity.User;
|
||||||
import com.bao.dating.pojo.vo.UserInfoVO;
|
import com.bao.dating.pojo.vo.UserInfoVO;
|
||||||
@@ -231,7 +230,7 @@ public class UserServiceImpl implements UserService {
|
|||||||
* @param userInfoUpdateDTO 用户信息更新参数
|
* @param userInfoUpdateDTO 用户信息更新参数
|
||||||
*/
|
*/
|
||||||
@Override
|
@Override
|
||||||
public UserInfoVO updateUserInfo(UserInfoUpdateDTO userInfoUpdateDTO) {
|
public UserInfoVO updateUserInfo(UserInfoDTO userInfoUpdateDTO) {
|
||||||
Long userId = userInfoUpdateDTO.getUserId();
|
Long userId = userInfoUpdateDTO.getUserId();
|
||||||
User user = userMapper.selectByUserId(userId);
|
User user = userMapper.selectByUserId(userId);
|
||||||
if (user == null) {
|
if (user == null) {
|
||||||
@@ -311,4 +310,23 @@ public class UserServiceImpl implements UserService {
|
|||||||
BeanUtils.copyProperties(updatedUser, userInfoVO);
|
BeanUtils.copyProperties(updatedUser, userInfoVO);
|
||||||
return userInfoVO;
|
return userInfoVO;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 根据用户ID获取用户昵称和头像
|
||||||
|
*
|
||||||
|
* @param userId 用户ID
|
||||||
|
* @return 用户昵称
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
public UserInfoDTO getUserInfoById(Long userId) {
|
||||||
|
// 查询数据库获取昵称
|
||||||
|
User user = userMapper.selectByUserId(userId);
|
||||||
|
if (user == null) {
|
||||||
|
throw new RuntimeException("没有此用户");
|
||||||
|
}
|
||||||
|
UserInfoDTO dto = new UserInfoDTO();
|
||||||
|
dto.setUserId(user.getUserId());
|
||||||
|
dto.setNickname(user.getNickname());
|
||||||
|
return dto;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
51
src/main/java/com/bao/dating/session/WsSessionManager.java
Normal file
51
src/main/java/com/bao/dating/session/WsSessionManager.java
Normal file
@@ -0,0 +1,51 @@
|
|||||||
|
package com.bao.dating.session;
|
||||||
|
|
||||||
|
import org.springframework.stereotype.Component;
|
||||||
|
import org.springframework.web.socket.WebSocketSession;
|
||||||
|
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.concurrent.ConcurrentHashMap;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* WebSocketSession 管理类
|
||||||
|
* @author KilLze
|
||||||
|
*/
|
||||||
|
@Component
|
||||||
|
public class WsSessionManager {
|
||||||
|
|
||||||
|
private final Map<Long, WebSocketSession> SESSION_MAP = new ConcurrentHashMap<>();
|
||||||
|
/**
|
||||||
|
* 添加 WebSocketSession
|
||||||
|
* @param userId 用户ID
|
||||||
|
* @param session WebSocketSession
|
||||||
|
*/
|
||||||
|
public void addSession(Long userId, WebSocketSession session) {
|
||||||
|
SESSION_MAP.put(userId, session);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 移除 WebSocketSession
|
||||||
|
* @param userId 用户ID
|
||||||
|
*/
|
||||||
|
public void removeSession(Long userId) {
|
||||||
|
SESSION_MAP.remove(userId);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取 WebSocketSession
|
||||||
|
* @param userId 用户ID
|
||||||
|
* @return WebSocketSession
|
||||||
|
*/
|
||||||
|
public WebSocketSession getSession(Long userId) {
|
||||||
|
return SESSION_MAP.get(userId);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 判断用户是否在线
|
||||||
|
* @param userId 用户ID
|
||||||
|
* @return true-在线,false-离线
|
||||||
|
*/
|
||||||
|
public boolean isOnline(Long userId) {
|
||||||
|
return SESSION_MAP.containsKey(userId);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -88,6 +88,3 @@ public class MD5Util {
|
|||||||
return sb.toString();
|
return sb.toString();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,67 @@
|
|||||||
|
<?xml version="1.0" encoding="UTF-8" ?>
|
||||||
|
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
|
||||||
|
"http://mybatis.org/dtd/mybatis-3-mapper.dtd">
|
||||||
|
<mapper namespace="com.bao.dating.mapper.ChatRecordsMapper">
|
||||||
|
|
||||||
|
<!-- 插入聊天记录 -->
|
||||||
|
<insert id="insert" useGeneratedKeys="true" keyProperty="chatId">
|
||||||
|
INSERT INTO chat_records
|
||||||
|
(
|
||||||
|
sender_user_id,
|
||||||
|
receiver_user_id,
|
||||||
|
message_content,
|
||||||
|
message_type,
|
||||||
|
read_status,
|
||||||
|
send_time,
|
||||||
|
message_status,
|
||||||
|
created_at,
|
||||||
|
updated_at
|
||||||
|
)
|
||||||
|
VALUES
|
||||||
|
(
|
||||||
|
#{senderUserId},
|
||||||
|
#{receiverUserId},
|
||||||
|
#{messageContent},
|
||||||
|
#{messageType},
|
||||||
|
#{readStatus},
|
||||||
|
#{sendTime},
|
||||||
|
#{messageStatus},
|
||||||
|
NOW(),
|
||||||
|
NOW()
|
||||||
|
)
|
||||||
|
</insert>
|
||||||
|
|
||||||
|
<!-- 根据两个用户ID查询聊天记录 (按发送时间倒序) - PageHelper 会自动处理分页 -->
|
||||||
|
<select id="selectChatHistoryByCursor" resultType="com.bao.dating.pojo.entity.ChatRecords">
|
||||||
|
SELECT
|
||||||
|
chat_id, sender_user_id, receiver_user_id, message_content, message_type,
|
||||||
|
read_status, read_time, send_time, message_status, created_at, updated_at
|
||||||
|
FROM chat_records
|
||||||
|
WHERE
|
||||||
|
message_status = 1
|
||||||
|
AND (
|
||||||
|
(sender_user_id = #{currentUserId} AND receiver_user_id = #{targetUserId})
|
||||||
|
OR
|
||||||
|
(sender_user_id = #{targetUserId} AND receiver_user_id = #{currentUserId})
|
||||||
|
)
|
||||||
|
<if test="cursor != null">
|
||||||
|
AND send_time < #{cursor}
|
||||||
|
</if>
|
||||||
|
ORDER BY send_time DESC
|
||||||
|
LIMIT #{size}
|
||||||
|
</select>
|
||||||
|
|
||||||
|
<!-- 标记消息为已读 -->
|
||||||
|
<update id="markMessagesAsRead">
|
||||||
|
UPDATE chat_records
|
||||||
|
SET
|
||||||
|
read_status = 1,
|
||||||
|
read_time = #{readTime},
|
||||||
|
updated_at = NOW()
|
||||||
|
WHERE
|
||||||
|
sender_user_id = #{senderUserId}
|
||||||
|
AND receiver_user_id = #{receiverUserId}
|
||||||
|
AND read_status = 0
|
||||||
|
AND message_status = 1
|
||||||
|
</update>
|
||||||
|
</mapper>
|
||||||
@@ -0,0 +1,63 @@
|
|||||||
|
<?xml version="1.0" encoding="UTF-8" ?>
|
||||||
|
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
|
||||||
|
"http://mybatis.org/dtd/mybatis-3-mapper.dtd">
|
||||||
|
<mapper namespace="com.bao.dating.mapper.ChatSessionsMapper">
|
||||||
|
|
||||||
|
<!-- 插入或更新发送方会话 -->
|
||||||
|
<insert id="upsertSessionForSender" parameterType="com.bao.dating.pojo.entity.ChatSessions">
|
||||||
|
INSERT INTO chat_sessions
|
||||||
|
(user_id, target_user_id, session_name, last_message_id, last_message_content, last_message_time,
|
||||||
|
unread_count, session_status, top_status, mute_status, created_at, updated_at)
|
||||||
|
VALUES
|
||||||
|
(
|
||||||
|
#{userId}, #{targetUserId}, #{sessionName}, #{lastMessageId}, #{lastMessageContent}, #{lastMessageTime},
|
||||||
|
0, 1, 0, 0, NOW(), NOW()
|
||||||
|
)
|
||||||
|
ON DUPLICATE KEY UPDATE
|
||||||
|
last_message_id = VALUES(last_message_id),
|
||||||
|
last_message_content = VALUES(last_message_content),
|
||||||
|
last_message_time = VALUES(last_message_time),
|
||||||
|
unread_count = 0,
|
||||||
|
updated_at = NOW();
|
||||||
|
</insert>
|
||||||
|
|
||||||
|
<!-- 插入或更新接收方会话 -->
|
||||||
|
<insert id="upsertSessionForReceiver" parameterType="com.bao.dating.pojo.entity.ChatSessions">
|
||||||
|
INSERT INTO chat_sessions
|
||||||
|
(user_id, target_user_id, session_name, last_message_id, last_message_content, last_message_time,
|
||||||
|
unread_count, session_status, top_status, mute_status, created_at, updated_at)
|
||||||
|
VALUES
|
||||||
|
(
|
||||||
|
#{userId}, #{targetUserId}, #{sessionName}, #{lastMessageId}, #{lastMessageContent}, #{lastMessageTime},
|
||||||
|
1, 1, 0, 0, NOW(), NOW()
|
||||||
|
)
|
||||||
|
ON DUPLICATE KEY UPDATE
|
||||||
|
last_message_id = VALUES(last_message_id),
|
||||||
|
last_message_content = VALUES(last_message_content),
|
||||||
|
last_message_time = VALUES(last_message_time),
|
||||||
|
unread_count = unread_count + 1,
|
||||||
|
updated_at = NOW();
|
||||||
|
</insert>
|
||||||
|
|
||||||
|
<!-- 清空未读消息数 -->
|
||||||
|
<update id="clearUnreadCount">
|
||||||
|
UPDATE chat_sessions
|
||||||
|
SET
|
||||||
|
unread_count = 0,
|
||||||
|
updated_at = NOW()
|
||||||
|
WHERE
|
||||||
|
user_id = #{userId}
|
||||||
|
AND target_user_id = #{targetUserId}
|
||||||
|
AND session_status = 1
|
||||||
|
</update>
|
||||||
|
|
||||||
|
<!-- 查询会话列表 -->
|
||||||
|
<select id="selectSessionsByUserId" resultType="com.bao.dating.pojo.entity.ChatSessions">
|
||||||
|
SELECT session_id, target_user_id, session_name, last_message_content,
|
||||||
|
last_message_time, unread_count, top_status, mute_status
|
||||||
|
FROM chat_sessions
|
||||||
|
WHERE user_id = #{userId}
|
||||||
|
AND session_status = 1
|
||||||
|
ORDER BY top_status DESC, last_message_time DESC
|
||||||
|
</select>
|
||||||
|
</mapper>
|
||||||
Reference in New Issue
Block a user