Compare commits
53 Commits
60df001385
...
feature-We
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
cf6f9b8b7c | ||
|
|
0a17eb8deb | ||
|
|
448ce1d3d6 | ||
|
|
a4e66b39d1 | ||
|
|
0a1425c7ab | ||
|
|
21c0a94a5d | ||
|
|
b6953cb8d0 | ||
|
|
f6769c8c69 | ||
|
|
5e1f3b7e8a | ||
|
|
e30a0dba97 | ||
|
|
74d47d586f | ||
|
|
0bb69075f7 | ||
|
|
1d179da910 | ||
|
|
c0969406b3 | ||
|
|
2a9acb946e | ||
|
|
fb2b7823b8 | ||
|
|
951cd7249d | ||
|
|
b12128fad6 | ||
|
|
212668ae1c | ||
|
|
a648ecad2a | ||
|
|
b8ec4a434d | ||
|
|
3bc00334ea | ||
|
|
a6259875f2 | ||
| 34cad7457b | |||
| eb2b0cb999 | |||
| 61c4c9d442 | |||
| 30db5ad706 | |||
|
|
717c0a0507 | ||
|
|
413bafa275 | ||
|
|
2ce8116126 | ||
|
|
5151ccfacc | ||
|
|
ab2bde0517 | ||
|
|
c9cb410819 | ||
|
|
07a6b86d7d | ||
|
|
1cf9ba0aa9 | ||
|
|
6a68fefbd4 | ||
|
|
da66c3b048 | ||
|
|
a9730a8414 | ||
|
|
0da6cde202 | ||
|
|
59838e1f5b | ||
|
|
7601fbb3c0 | ||
|
|
170ce83fe1 | ||
|
|
aa16c30d57 | ||
|
|
8fdbad8bd5 | ||
|
|
a9598ba795 | ||
|
|
439b6789fa | ||
|
|
9c7afc4470 | ||
|
|
e92cd31c27 | ||
|
|
bd12c599b0 | ||
|
|
b5a15a3f01 | ||
|
|
039443dc0f | ||
| ae0cca5437 | |||
| c329eaef79 |
18
pom.xml
18
pom.xml
@@ -88,6 +88,12 @@
|
||||
<artifactId>spring-boot-starter-aop</artifactId>
|
||||
</dependency>
|
||||
|
||||
<!-- WebSocket 起步依赖 -->
|
||||
<dependency>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-starter-websocket</artifactId>
|
||||
</dependency>
|
||||
|
||||
<!-- 阿里云相关依赖 -->
|
||||
<dependency>
|
||||
<groupId>com.aliyun.oss</groupId>
|
||||
@@ -159,6 +165,18 @@
|
||||
<artifactId>spring-boot-starter-mail</artifactId>
|
||||
</dependency>
|
||||
|
||||
<!-- Redis 依赖 -->
|
||||
<dependency>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-starter-data-redis</artifactId>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>com.github.pagehelper</groupId>
|
||||
<artifactId>pagehelper-spring-boot-starter</artifactId>
|
||||
<version>1.4.7</version>
|
||||
</dependency>
|
||||
|
||||
</dependencies>
|
||||
<dependencyManagement>
|
||||
<dependencies>
|
||||
|
||||
@@ -4,11 +4,32 @@ import org.mybatis.spring.annotation.MapperScan;
|
||||
import org.springframework.boot.SpringApplication;
|
||||
import org.springframework.boot.autoconfigure.SpringBootApplication;
|
||||
|
||||
import java.io.BufferedReader;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.io.InputStreamReader;
|
||||
|
||||
@MapperScan("com.bao.dating.mapper")
|
||||
@SpringBootApplication
|
||||
public class DatingApplication {
|
||||
public static void main(String[] args) {
|
||||
SpringApplication.run(DatingApplication.class, args);
|
||||
// 读取并打印 ciallo.txt 文件内容
|
||||
printCialloFile();
|
||||
}
|
||||
|
||||
}
|
||||
/**
|
||||
* 读取并打印 ciallo.txt 文件内容
|
||||
*/
|
||||
private static void printCialloFile() {
|
||||
try (InputStream inputStream = DatingApplication.class.getClassLoader().getResourceAsStream("ciallo.txt");
|
||||
BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream))) {
|
||||
String line;
|
||||
while ((line = reader.readLine()) != null) {
|
||||
System.out.println(line);
|
||||
}
|
||||
} catch (IOException e) {
|
||||
System.err.println("读取 ciallo.txt 文件时发生错误: " + e.getMessage());
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -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; // 创建时间
|
||||
}
|
||||
@@ -19,6 +19,14 @@ public enum ResultCode {
|
||||
FORBIDDEN(403, "无权限"),
|
||||
/** 数据不存在 */
|
||||
NOT_FOUND(404, "数据不存在"),
|
||||
/** 验证码已发送 */
|
||||
CODE_SENT(200, "验证码已发送"),
|
||||
/** 验证码错误 */
|
||||
CODE_ERROR(400, "验证码错误"),
|
||||
/** 验证码已过期 */
|
||||
CODE_EXPIRED(400, "验证码已过期"),
|
||||
/** 验证码发送失败 */
|
||||
CODE_SEND_FAIL(500, "验证码发送失败"),
|
||||
/** 系统异常 */
|
||||
SYSTEM_ERROR(500, "系统异常"),
|
||||
/** 操作失败 */
|
||||
|
||||
@@ -149,3 +149,4 @@ public class SmsUtil {
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
@@ -0,0 +1,15 @@
|
||||
package com.bao.dating.common.result;
|
||||
|
||||
import com.bao.dating.pojo.vo.ChatRecordsVO;
|
||||
import lombok.Data;
|
||||
|
||||
/**
|
||||
* 聊天发送结果
|
||||
* @author KilLze
|
||||
*/
|
||||
@Data
|
||||
public class ChatSendResult {
|
||||
private boolean success;
|
||||
private String message;
|
||||
private ChatRecordsVO record;
|
||||
}
|
||||
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,15 +14,19 @@ import org.springframework.data.redis.serializer.StringRedisSerializer;
|
||||
@Configuration
|
||||
public class RedisConfig {
|
||||
@Bean
|
||||
public RedisTemplate<String, Object> redisTemplate(RedisConnectionFactory redisConnectionFactory) {
|
||||
public RedisTemplate redisTemplate(RedisConnectionFactory redisConnectionFactory) {
|
||||
// 创建RedisTemplate对象
|
||||
RedisTemplate<String, Object> redisTemplate = new RedisTemplate<>();
|
||||
RedisTemplate redisTemplate = new RedisTemplate<>();
|
||||
// 设置redis的连接工厂对象
|
||||
redisTemplate.setConnectionFactory(redisConnectionFactory);
|
||||
|
||||
// key 使用 String
|
||||
// 设置redis key的序列化器
|
||||
redisTemplate.setKeySerializer(new StringRedisSerializer());
|
||||
// 设置value的序列化器
|
||||
redisTemplate.setValueSerializer(new GenericJackson2JsonRedisSerializer());
|
||||
// 设置hash类型的key和value的序列化器
|
||||
redisTemplate.setHashKeySerializer(new StringRedisSerializer());
|
||||
redisTemplate.setHashValueSerializer(new GenericJackson2JsonRedisSerializer());
|
||||
|
||||
return redisTemplate;
|
||||
}
|
||||
|
||||
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("*");
|
||||
}
|
||||
}
|
||||
104
src/main/java/com/bao/dating/controller/ChatController.java
Normal file
104
src/main/java/com/bao/dating/controller/ChatController.java
Normal file
@@ -0,0 +1,104 @@
|
||||
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.*;
|
||||
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 org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* 聊天控制器
|
||||
* @author lenovo
|
||||
*/
|
||||
@Slf4j
|
||||
@RestController
|
||||
@RequestMapping("/chat")
|
||||
public class ChatController {
|
||||
@Autowired
|
||||
private ChatService chatService;
|
||||
|
||||
/**
|
||||
* 上传文件
|
||||
* @param file 文件
|
||||
* @return 文件URL
|
||||
*/
|
||||
@PostMapping("/upload")
|
||||
public Result<String> uploadVideo(@RequestParam MultipartFile file) {
|
||||
String url = chatService.uploadChat(file);
|
||||
return Result.success(ResultCode.SUCCESS, "文件上传成功", url);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取聊天记录
|
||||
* @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);
|
||||
}
|
||||
|
||||
/**
|
||||
* 更新会话状态
|
||||
* @param dto 会话状态
|
||||
* @return 无
|
||||
*/
|
||||
@PostMapping("/session/status")
|
||||
public Result<Void> updateSessionStatus(@RequestBody ChatSessionStatusDTO dto) {
|
||||
chatService.updateSessionStatus(UserContext.getUserId(), dto);
|
||||
return Result.success(ResultCode.SUCCESS, null);
|
||||
}
|
||||
|
||||
/**
|
||||
* 置顶会话
|
||||
* @param dto 置顶状态
|
||||
* @return 无
|
||||
*/
|
||||
@PostMapping("/session/top")
|
||||
public Result<Void> updateTopStatus(@RequestBody ChatSessionTopDTO dto) {
|
||||
chatService.updateTopStatus(UserContext.getUserId(), dto);
|
||||
return Result.success(ResultCode.SUCCESS, null);
|
||||
}
|
||||
|
||||
/**
|
||||
* 静音会话
|
||||
* @param dto 静音状态
|
||||
* @return 无
|
||||
*/
|
||||
@PostMapping("/session/mute")
|
||||
public Result<Void> updateMuteStatus(@RequestBody ChatSessionMuteDTO dto) {
|
||||
chatService.updateMuteStatus(UserContext.getUserId(), dto);
|
||||
return Result.success(ResultCode.SUCCESS, null);
|
||||
}
|
||||
}
|
||||
@@ -31,7 +31,6 @@ public class PostController {
|
||||
* @param files 媒体文件数组
|
||||
* @return 上传后的文件URL列表
|
||||
*/
|
||||
@Log
|
||||
@PostMapping(value = "/upload", consumes = "multipart/form-data")
|
||||
public Result<List<String>> uploadMedia(@RequestParam("files") MultipartFile[] files) {
|
||||
List<String> fileUrls = postService.uploadMedia(files);
|
||||
@@ -43,7 +42,6 @@ public class PostController {
|
||||
* @param postDTO 动态信息
|
||||
* @return 发布的动态对象
|
||||
*/
|
||||
@Log
|
||||
@PostMapping( "/createPost")
|
||||
public Result<Post> createPostJson(@RequestBody PostRequestDTO postDTO) {
|
||||
// 调用 Service 层处理发布动态业务逻辑
|
||||
@@ -57,11 +55,10 @@ public class PostController {
|
||||
* @param postIds 动态ID
|
||||
* @return 删除结果
|
||||
*/
|
||||
@Log
|
||||
@PostMapping("/deletePost")
|
||||
public Result<String> deleteById(@RequestBody List<Long> postIds){
|
||||
int deletedCount = postService.deletePostById(postIds);
|
||||
return Result.success(ResultCode.SUCCESS_DELETE, deletedCount > 0 ? "成功删除" : "删除失败,该动态不存在", null);
|
||||
return Result.success(ResultCode.SUCCESS_DELETE, "成功删除" + deletedCount + "条动态", null);
|
||||
}
|
||||
|
||||
/**
|
||||
@@ -81,7 +78,6 @@ public class PostController {
|
||||
* @param postRequestDTO 动态信息
|
||||
* @return 更新后的动态对象
|
||||
*/
|
||||
@Log
|
||||
@PostMapping("/{postId}/updatePost")
|
||||
public Result<PostEditVO> updatePost(@PathVariable Long postId, @RequestBody PostRequestDTO postRequestDTO) {
|
||||
PostEditVO result = postService.updatePost(postId, postRequestDTO);
|
||||
|
||||
@@ -4,12 +4,11 @@ import com.bao.dating.anno.Log;
|
||||
import com.bao.dating.common.Result;
|
||||
import com.bao.dating.common.ResultCode;
|
||||
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.vo.UserInfoVO;
|
||||
import com.bao.dating.pojo.vo.UserLoginVO;
|
||||
import com.bao.dating.service.UserService;
|
||||
import io.jsonwebtoken.Jwt;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
import org.springframework.web.multipart.MultipartFile;
|
||||
@@ -65,7 +64,6 @@ public class UserController {
|
||||
* @param file 头像文件
|
||||
* @return 上传后的文件URL列表
|
||||
*/
|
||||
@Log
|
||||
@PostMapping(value = "/info/uploadAvatar", consumes = "multipart/form-data")
|
||||
public Result<String> uploadAvatar(@RequestParam("file") MultipartFile file) {
|
||||
String fileUrl = userService.uploadAvatar(file);
|
||||
@@ -77,7 +75,6 @@ public class UserController {
|
||||
* @param file 背景文件
|
||||
* @return 上传后的文件URL列表
|
||||
*/
|
||||
@Log
|
||||
@PostMapping(value = "/info/uploadBackground", consumes = "multipart/form-data")
|
||||
public Result<String> uploadBackground(@RequestParam("file") MultipartFile file) {
|
||||
String fileUrl = userService.uploadBackground(file);
|
||||
@@ -89,9 +86,8 @@ public class UserController {
|
||||
* @param userInfoUpdateDTO 用户信息更新参数
|
||||
* @return 更新后的用户信息
|
||||
*/
|
||||
@Log
|
||||
@PostMapping("/info/update")
|
||||
public Result<UserInfoVO> userInfoUpdate(@RequestBody UserInfoUpdateDTO userInfoUpdateDTO) {
|
||||
public Result<UserInfoVO> userInfoUpdate(@RequestBody UserInfoDTO userInfoUpdateDTO) {
|
||||
Long userId = UserContext.getUserId();
|
||||
userInfoUpdateDTO.setUserId(userId);
|
||||
UserInfoVO userInfoVO =userService.updateUserInfo(userInfoUpdateDTO);
|
||||
|
||||
@@ -0,0 +1,90 @@
|
||||
package com.bao.dating.controller;
|
||||
|
||||
import com.bao.dating.common.Result;
|
||||
import com.bao.dating.common.ResultCode;
|
||||
import com.bao.dating.service.VerificationCodeService;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.util.StringUtils;
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
|
||||
/**
|
||||
* 验证码控制器
|
||||
* @author KilLze
|
||||
*/
|
||||
@Slf4j
|
||||
@RestController
|
||||
@RequestMapping("/api/verification")
|
||||
public class VerificationCodeController {
|
||||
|
||||
@Autowired
|
||||
private VerificationCodeService verificationCodeService;
|
||||
|
||||
/**
|
||||
* 发送邮箱验证码
|
||||
* @param email 邮箱地址
|
||||
* @return 结果
|
||||
*/
|
||||
@PostMapping("/send-email-code")
|
||||
public Result<String> sendEmailCode(@RequestParam String email) {
|
||||
// 参数校验
|
||||
if (!StringUtils.hasText(email)) {
|
||||
return Result.error(ResultCode.PARAM_ERROR, "邮箱地址不能为空");
|
||||
}
|
||||
|
||||
// 简单的邮箱格式校验
|
||||
if (!isValidEmail(email)) {
|
||||
return Result.error(ResultCode.PARAM_ERROR, "邮箱格式不正确");
|
||||
}
|
||||
|
||||
// 发送验证码
|
||||
boolean success = verificationCodeService.sendEmailCode(email);
|
||||
|
||||
if (success) {
|
||||
return Result.success(ResultCode.CODE_SENT, "验证码已发送到您的邮箱,请查收");
|
||||
} else {
|
||||
return Result.error(ResultCode.CODE_SEND_FAIL, "验证码发送失败,请稍后重试");
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证邮箱验证码
|
||||
* @param email 邮箱地址
|
||||
* @param code 验证码
|
||||
* @return 结果
|
||||
*/
|
||||
@PostMapping("/verify-email-code")
|
||||
public Result<String> verifyEmailCode(@RequestParam String email, @RequestParam String code) {
|
||||
// 参数校验
|
||||
if (!StringUtils.hasText(email)) {
|
||||
return Result.error(ResultCode.PARAM_ERROR, "邮箱地址不能为空");
|
||||
}
|
||||
|
||||
if (!StringUtils.hasText(code)) {
|
||||
return Result.error(ResultCode.PARAM_ERROR, "验证码不能为空");
|
||||
}
|
||||
|
||||
// 验证验证码
|
||||
boolean success = verificationCodeService.verifyEmailCode(email, code);
|
||||
|
||||
if (success) {
|
||||
return Result.success(ResultCode.SUCCESS, "验证码验证成功");
|
||||
} else {
|
||||
return Result.error(ResultCode.CODE_ERROR, "验证码错误或已过期");
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 简单的邮箱格式校验
|
||||
* @param email 邮箱地址
|
||||
* @return 是否有效
|
||||
*/
|
||||
private boolean isValidEmail(String email) {
|
||||
if (email == null || email.trim().isEmpty()) {
|
||||
return false;
|
||||
}
|
||||
// 简单的邮箱格式校验:包含@和.
|
||||
return email.contains("@") && email.contains(".") && email.length() > 5;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,165 @@
|
||||
package com.bao.dating.controller.websocket;
|
||||
|
||||
import com.bao.dating.message.WsMessage;
|
||||
import com.bao.dating.pojo.dto.ChatRecallDTO;
|
||||
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.JsonNode;
|
||||
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 WebSocketSession
|
||||
*/
|
||||
@Override
|
||||
public void afterConnectionEstablished(WebSocketSession session) {
|
||||
Long userId = (Long) session.getAttributes().get("userId");
|
||||
sessionManager.addSession(userId, session);
|
||||
log.info("用户 " + userId + " 已上线");
|
||||
}
|
||||
|
||||
/**
|
||||
* 接收并处理消息
|
||||
* @param session WebSocketSession
|
||||
* @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;
|
||||
}
|
||||
|
||||
JsonNode node = objectMapper.readTree(message.getPayload());
|
||||
String type = node.get("type").asText();
|
||||
// 根据消息类型解析消息
|
||||
WsMessage wsMessage = objectMapper.readValue(message.getPayload(), WsMessage.class);
|
||||
|
||||
// 先获取消息类型,再根据类型进行相应处理和转换
|
||||
if ("chat".equals(type)) {
|
||||
// 处理私聊消息
|
||||
WsMessage<ChatRecordSendDTO> chatWsMessage =
|
||||
objectMapper.convertValue(node, new TypeReference<WsMessage<ChatRecordSendDTO>>(){});
|
||||
handlePrivateChat(session, senderUserId, chatWsMessage.getData());
|
||||
} else if ("recall".equals(type)) {
|
||||
// 处理撤回消息
|
||||
WsMessage<ChatRecallDTO> recallWsMessage =
|
||||
objectMapper.convertValue(node, new TypeReference<WsMessage<ChatRecallDTO>>(){});
|
||||
handleRecallMessage(session, senderUserId, recallWsMessage.getData());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 私聊处理
|
||||
*/
|
||||
private void handlePrivateChat(WebSocketSession session, Long senderUserId, ChatRecordSendDTO dto) throws Exception {
|
||||
|
||||
// 1. 消息入库 + 会话更新
|
||||
ChatRecordsVO chatRecordsVO = chatService.createSession(senderUserId, dto);
|
||||
if (chatRecordsVO == null){
|
||||
WsMessage<String> errorMsg = new WsMessage<>();
|
||||
errorMsg.setType("error");
|
||||
errorMsg.setData("会话已删除,无法发送消息");
|
||||
// 返回错误信息
|
||||
session.sendMessage(new TextMessage(objectMapper.writeValueAsString(errorMsg)));
|
||||
return;
|
||||
}
|
||||
// 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))
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 消息撤回处理
|
||||
*/
|
||||
private void handleRecallMessage(WebSocketSession session, Long senderUserId, Object data) throws Exception {
|
||||
|
||||
// 转 DTO
|
||||
ChatRecallDTO dto = objectMapper.convertValue(data, ChatRecallDTO.class);
|
||||
|
||||
// 撤回逻辑
|
||||
boolean success = chatService.recallMessage(senderUserId, dto.getChatId());
|
||||
// 如果返回false,说明消息撤回失败
|
||||
if (!success) {
|
||||
WsMessage<String> errorMsg = new WsMessage<>();
|
||||
errorMsg.setType("error");
|
||||
errorMsg.setData("撤回失败,消息可能无法撤回或不存在");
|
||||
// 返回错误信息
|
||||
session.sendMessage(new TextMessage(objectMapper.writeValueAsString(errorMsg)));
|
||||
return;
|
||||
}
|
||||
|
||||
// 创建撤回通知消息
|
||||
WsMessage<ChatRecallDTO> pushMsg = new WsMessage<>();
|
||||
pushMsg.setType("recall");
|
||||
pushMsg.setData(dto);
|
||||
|
||||
// 通知自己
|
||||
if (session.isOpen()) {
|
||||
session.sendMessage(new TextMessage(objectMapper.writeValueAsString(pushMsg)));
|
||||
}
|
||||
|
||||
// 通知对方
|
||||
WebSocketSession receiverSession =
|
||||
sessionManager.getSession(dto.getReceiverUserId());
|
||||
|
||||
if (receiverSession != null && receiverSession.isOpen()) {
|
||||
receiverSession.sendMessage(new TextMessage(objectMapper.writeValueAsString(pushMsg))
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 用户断开连接(下线)
|
||||
* @param session WebSocketSession
|
||||
* @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 {
|
||||
|
||||
@Autowired
|
||||
private RedisTemplate<String, Object> redisTemplate;
|
||||
private RedisTemplate redisTemplate;
|
||||
|
||||
/**
|
||||
* 在请求处理之前进行拦截
|
||||
@@ -68,11 +68,10 @@ public class TokenInterceptor implements HandlerInterceptor {
|
||||
}
|
||||
|
||||
// 解析 token
|
||||
String userId = JwtUtil.getSubjectFromToken(token);
|
||||
Long userId = Long.valueOf(JwtUtil.getSubjectFromToken(token));
|
||||
|
||||
// 从Redis获取存储的token进行比对
|
||||
Object redisTokenObj = redisTemplate.opsForValue()
|
||||
.get("login:token:" + userId);
|
||||
Object redisTokenObj = redisTemplate.opsForValue().get("login:token:" + userId);
|
||||
String redisToken = redisTokenObj != null ? redisTokenObj.toString() : null;
|
||||
|
||||
// 验证Redis中的token是否存在且匹配
|
||||
@@ -86,7 +85,7 @@ public class TokenInterceptor implements HandlerInterceptor {
|
||||
|
||||
log.info("用户: {}", userId);
|
||||
// 保存 userId 到 ThreadLocal
|
||||
UserContext.setUserId(Long.valueOf(userId));
|
||||
UserContext.setUserId(userId);
|
||||
return true;
|
||||
} catch (Exception e) {
|
||||
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握手完成");
|
||||
}
|
||||
}
|
||||
}
|
||||
51
src/main/java/com/bao/dating/mapper/ChatRecordsMapper.java
Normal file
51
src/main/java/com/bao/dating/mapper/ChatRecordsMapper.java
Normal file
@@ -0,0 +1,51 @@
|
||||
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);
|
||||
|
||||
/**
|
||||
* 根据ID查询聊天记录
|
||||
* @param chatId 聊天记录ID
|
||||
* @return 聊天记录
|
||||
*/
|
||||
ChatRecords selectById(@Param("chatId") Long chatId);
|
||||
|
||||
/**
|
||||
* 撤回聊天记录
|
||||
* @param chatId 聊天记录ID
|
||||
* @param senderUserId 发送者ID
|
||||
* @return 影响行数
|
||||
*/
|
||||
int recallMessage(@Param("chatId") Long chatId,
|
||||
@Param("senderUserId") Long senderUserId);
|
||||
}
|
||||
88
src/main/java/com/bao/dating/mapper/ChatSessionsMapper.java
Normal file
88
src/main/java/com/bao/dating/mapper/ChatSessionsMapper.java
Normal file
@@ -0,0 +1,88 @@
|
||||
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 org.apache.ibatis.annotations.Select;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
@Mapper
|
||||
public interface ChatSessionsMapper {
|
||||
|
||||
/**
|
||||
* 如果发送方存在会话则更新,不存在则创建
|
||||
* @param chatSessions 会话
|
||||
* @return 影响行数
|
||||
*/
|
||||
int upsertSessionForSender(ChatSessions chatSessions);
|
||||
|
||||
/**
|
||||
* 获取会话
|
||||
* @param userId 用户ID
|
||||
* @param targetUserId 目标用户ID
|
||||
* @return 会话
|
||||
*/
|
||||
@Select("SELECT * FROM chat_sessions WHERE user_id = #{userId} AND target_user_id = #{targetUserId} LIMIT 1")
|
||||
ChatSessions getSession(@Param("userId") Long userId, @Param("targetUserId") Long targetUserId);
|
||||
|
||||
/**
|
||||
* 如果接收方存在会话则更新,不存在则创建
|
||||
* @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);
|
||||
|
||||
/**
|
||||
* 更新会话状态
|
||||
* @param userId 用户ID
|
||||
* @param targetUserId 目标用户ID
|
||||
* @param sessionStatus 会话状态
|
||||
* @return 影响行数
|
||||
*/
|
||||
int updateSessionStatus(@Param("userId") Long userId,
|
||||
@Param("targetUserId") Long targetUserId,
|
||||
@Param("sessionStatus") Integer sessionStatus);
|
||||
|
||||
/**
|
||||
* 更新会话置顶状态
|
||||
* @param userId 用户ID
|
||||
* @param targetUserId 目标用户ID
|
||||
* @param topStatus 置顶状态
|
||||
* @return 影响行数
|
||||
*/
|
||||
int updateTopStatus(@Param("userId") Long userId,
|
||||
@Param("targetUserId") Long targetUserId,
|
||||
@Param("topStatus") Integer topStatus);
|
||||
|
||||
/**
|
||||
* 更新会话免打扰状态
|
||||
* @param userId 用户ID
|
||||
* @param targetUserId 目标用户ID
|
||||
* @param muteStatus 免打扰状态
|
||||
* @return 影响行数
|
||||
*/
|
||||
int updateMuteStatus(@Param("userId") Long userId,
|
||||
@Param("targetUserId") Long targetUserId,
|
||||
@Param("muteStatus") Integer muteStatus);
|
||||
}
|
||||
@@ -18,4 +18,11 @@ public interface CommentsMapper {
|
||||
// 根据动态ID查询评论列表
|
||||
@Select("SELECT * FROM comments WHERE post_id = #{post_id} ORDER BY created_at DESC")
|
||||
List<Comments> getCommentsByPostId(@Param("post_id") Long post_id);
|
||||
|
||||
/**
|
||||
* 根据动态ID批量删除评论
|
||||
* @param postIds
|
||||
* @return
|
||||
*/
|
||||
int deleteCommentsByPostIds(@Param("postIds") List<Long> postIds);
|
||||
}
|
||||
|
||||
@@ -12,4 +12,12 @@ public interface PostFavoriteMapper {
|
||||
List<Long> selectUserIDByPostID(@Param("postId") Long postId);
|
||||
int addPostFavorite(PostFavorite postFavorite);
|
||||
int deletePostFavorite(@Param("postId") Long postId);
|
||||
|
||||
/**
|
||||
* 批量删除收藏
|
||||
* @param postIds
|
||||
* @return
|
||||
*/
|
||||
int deleteFavoritesByPostIds(@Param("postIds") List<Long> postIds);
|
||||
|
||||
}
|
||||
|
||||
@@ -4,6 +4,8 @@ import com.bao.dating.pojo.entity.PostLike;
|
||||
import org.apache.ibatis.annotations.Mapper;
|
||||
import org.apache.ibatis.annotations.Param;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
@Mapper
|
||||
public interface PostLikeMapper {
|
||||
/**
|
||||
@@ -31,4 +33,12 @@ public interface PostLikeMapper {
|
||||
* @return
|
||||
*/
|
||||
int deleteByPostIdAndUserId(@Param("postId") Long postId, @Param("userId") Long userId);
|
||||
|
||||
/**
|
||||
* 批量删除点赞记录
|
||||
*
|
||||
* @param postIds
|
||||
* @return
|
||||
*/
|
||||
int deleteLikesByPostIds(@Param("postIds") List<Long> postIds);
|
||||
}
|
||||
|
||||
@@ -21,11 +21,11 @@ public interface PostMapper {
|
||||
void insert(Post post);
|
||||
|
||||
/**
|
||||
* 根据ID删除动态
|
||||
* 根据ID修改动态状态
|
||||
*
|
||||
* @param postIds 动态ID
|
||||
*/
|
||||
int deletePostByIds(List<Long> postIds);
|
||||
int updatePublicById(@Param("postIds") List<Long> postIds, @Param("userId") Long userId);
|
||||
|
||||
/**
|
||||
* 根据ID查询动态
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
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 org.apache.ibatis.annotations.Mapper;
|
||||
|
||||
@@ -31,6 +31,6 @@ public interface UserMapper {
|
||||
* 更新用户信息
|
||||
* @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;
|
||||
}
|
||||
13
src/main/java/com/bao/dating/pojo/dto/ChatRecallDTO.java
Normal file
13
src/main/java/com/bao/dating/pojo/dto/ChatRecallDTO.java
Normal file
@@ -0,0 +1,13 @@
|
||||
package com.bao.dating.pojo.dto;
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
/**
|
||||
* 聊天记录撤回参数
|
||||
* @author lenovo
|
||||
*/
|
||||
@Data
|
||||
public class ChatRecallDTO {
|
||||
private Long chatId;
|
||||
private Long receiverUserId;
|
||||
}
|
||||
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;
|
||||
}
|
||||
@@ -0,0 +1,13 @@
|
||||
package com.bao.dating.pojo.dto;
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
/**
|
||||
* 聊天会话静音参数
|
||||
* @author lenovo
|
||||
*/
|
||||
@Data
|
||||
public class ChatSessionMuteDTO {
|
||||
private Long targetUserId;
|
||||
private Integer muteStatus;
|
||||
}
|
||||
@@ -0,0 +1,13 @@
|
||||
package com.bao.dating.pojo.dto;
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
/**
|
||||
* 会话状态参数
|
||||
* @author lenovo
|
||||
*/
|
||||
@Data
|
||||
public class ChatSessionStatusDTO {
|
||||
private Long targetUserId;
|
||||
private Integer sessionStatus;
|
||||
}
|
||||
13
src/main/java/com/bao/dating/pojo/dto/ChatSessionTopDTO.java
Normal file
13
src/main/java/com/bao/dating/pojo/dto/ChatSessionTopDTO.java
Normal file
@@ -0,0 +1,13 @@
|
||||
package com.bao.dating.pojo.dto;
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
/**
|
||||
* 会话置顶参数
|
||||
* @author lenovo
|
||||
*/
|
||||
@Data
|
||||
public class ChatSessionTopDTO {
|
||||
private Long targetUserId;
|
||||
private Integer topStatus;
|
||||
}
|
||||
@@ -8,11 +8,11 @@ import java.time.LocalDateTime;
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* 用户信息更新数据传输对象
|
||||
* 用户信息数据传输对象
|
||||
* @author KilLze
|
||||
*/
|
||||
@Data
|
||||
public class UserInfoUpdateDTO implements Serializable {
|
||||
public class UserInfoDTO implements Serializable {
|
||||
private Long userId;
|
||||
private String userName;
|
||||
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;
|
||||
}
|
||||
82
src/main/java/com/bao/dating/service/ChatService.java
Normal file
82
src/main/java/com/bao/dating/service/ChatService.java
Normal file
@@ -0,0 +1,82 @@
|
||||
package com.bao.dating.service;
|
||||
|
||||
import com.bao.dating.common.result.PageResult;
|
||||
import com.bao.dating.pojo.dto.*;
|
||||
import com.bao.dating.pojo.vo.ChatRecordsVO;
|
||||
import com.bao.dating.pojo.vo.ChatSessionsVO;
|
||||
import org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
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 file 文件
|
||||
* @return 文件URL列表
|
||||
*/
|
||||
String uploadChat(MultipartFile file);
|
||||
|
||||
/**
|
||||
* 获取聊天记录
|
||||
* @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);
|
||||
|
||||
/**
|
||||
* 更新会话状态
|
||||
* @param userId 用户ID
|
||||
* @param dto 更新参数
|
||||
*/
|
||||
void updateSessionStatus(Long userId, ChatSessionStatusDTO dto);
|
||||
|
||||
/**
|
||||
* 置顶会话
|
||||
* @param userId 用户ID
|
||||
* @param dto 置顶参数
|
||||
*/
|
||||
void updateTopStatus(Long userId, ChatSessionTopDTO dto);
|
||||
|
||||
/**
|
||||
* 免打扰会话
|
||||
* @param userId 用户ID
|
||||
* @param dto 免打扰参数
|
||||
*/
|
||||
void updateMuteStatus(Long userId, ChatSessionMuteDTO dto);
|
||||
|
||||
/**
|
||||
* 撤回消息
|
||||
* @param senderUserId 发送方ID
|
||||
* @param chatId 聊天记录ID
|
||||
* @return 撤回结果
|
||||
*/
|
||||
boolean recallMessage(Long senderUserId, Long chatId);
|
||||
}
|
||||
@@ -27,7 +27,7 @@ public interface PostService {
|
||||
Post createPost(PostRequestDTO postRequestDTO);
|
||||
|
||||
/**
|
||||
* 批量删除动态
|
||||
* 批量删除动态(将动态状态改为已删除)
|
||||
* @param postIds 动态ID
|
||||
* @return 删除的动态对象
|
||||
*/
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
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.vo.UserInfoVO;
|
||||
import com.bao.dating.pojo.vo.UserLoginVO;
|
||||
@@ -51,5 +51,12 @@ public interface UserService {
|
||||
* @param userInfoUpdateDTO 用户信息
|
||||
* @return 更新后的用户信息
|
||||
*/
|
||||
UserInfoVO updateUserInfo(UserInfoUpdateDTO userInfoUpdateDTO);
|
||||
UserInfoVO updateUserInfo(UserInfoDTO userInfoUpdateDTO);
|
||||
|
||||
/**
|
||||
* 根据用户ID查询用户信息
|
||||
* @param userId 用户ID
|
||||
* @return 用户
|
||||
*/
|
||||
UserInfoDTO getUserInfoById(Long userId);
|
||||
}
|
||||
|
||||
@@ -0,0 +1,31 @@
|
||||
package com.bao.dating.service;
|
||||
|
||||
/**
|
||||
* 验证码服务接口
|
||||
* @author KilLze
|
||||
*/
|
||||
public interface VerificationCodeService {
|
||||
|
||||
/**
|
||||
* 发送邮箱验证码
|
||||
* @param email 邮箱地址
|
||||
* @return 是否发送成功
|
||||
*/
|
||||
boolean sendEmailCode(String email);
|
||||
|
||||
/**
|
||||
* 验证邮箱验证码
|
||||
* @param email 邮箱地址
|
||||
* @param code 验证码
|
||||
* @return 是否验证成功
|
||||
*/
|
||||
boolean verifyEmailCode(String email, String code);
|
||||
|
||||
/**
|
||||
* 生成验证码
|
||||
* @param length 验证码长度(默认6位)
|
||||
* @return 验证码字符串
|
||||
*/
|
||||
String generateCode(int length);
|
||||
}
|
||||
|
||||
325
src/main/java/com/bao/dating/service/impl/ChatServiceImpl.java
Normal file
325
src/main/java/com/bao/dating/service/impl/ChatServiceImpl.java
Normal file
@@ -0,0 +1,325 @@
|
||||
package com.bao.dating.service.impl;
|
||||
|
||||
import com.bao.dating.common.aliyun.AliOssUtil;
|
||||
import com.bao.dating.common.result.AliOssResult;
|
||||
import com.bao.dating.context.UserContext;
|
||||
import com.bao.dating.mapper.ChatRecordsMapper;
|
||||
import com.bao.dating.mapper.ChatSessionsMapper;
|
||||
import com.bao.dating.pojo.dto.*;
|
||||
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 com.bao.dating.util.FileUtil;
|
||||
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 org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.time.Duration;
|
||||
import java.time.LocalDate;
|
||||
import java.time.LocalDateTime;
|
||||
import java.time.format.DateTimeFormatter;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
import java.util.UUID;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
|
||||
/**
|
||||
* 聊天服务实现类
|
||||
* @author lenovo
|
||||
*/
|
||||
@Slf4j
|
||||
@Service
|
||||
public class ChatServiceImpl implements ChatService {
|
||||
@Autowired
|
||||
private ChatRecordsMapper chatRecordsMapper;
|
||||
|
||||
@Autowired
|
||||
private ChatSessionsMapper chatSessionsMapper;
|
||||
|
||||
@Autowired
|
||||
private AliOssUtil ossUtil;
|
||||
|
||||
@Autowired
|
||||
private UserService userService;
|
||||
|
||||
/**
|
||||
* 消息入库,如果会话不存在则创建会话
|
||||
* @param senderUserId 发送者用户ID
|
||||
* @param dto 消息
|
||||
* @return 聊天记录
|
||||
*/
|
||||
@Override
|
||||
@Transactional(rollbackFor = Exception.class)
|
||||
public ChatRecordsVO createSession(Long senderUserId, ChatRecordSendDTO dto) {
|
||||
|
||||
ChatSessions session = chatSessionsMapper.getSession(senderUserId, dto.getReceiverUserId());
|
||||
if (session != null && session.getSessionStatus() == 3) {
|
||||
log.warn("会话已删除,无法发送消息");
|
||||
return null;
|
||||
}
|
||||
|
||||
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 上传后的文件URL
|
||||
*/
|
||||
@Override
|
||||
public String uploadChat(MultipartFile file) {
|
||||
// 参数校验
|
||||
if (file == null || file.isEmpty()) {
|
||||
throw new RuntimeException("文件不存在");
|
||||
}
|
||||
|
||||
String originalFilename = file.getOriginalFilename();
|
||||
if (originalFilename == null) {
|
||||
throw new RuntimeException("文件名非法");
|
||||
}
|
||||
|
||||
String fileType = FileUtil.getFileType(originalFilename);
|
||||
// 仅支持图片和视频文件上传
|
||||
if (!AliOssResult.IMAGE.equals(fileType) && !AliOssResult.VIDEO.equals(fileType)) {
|
||||
throw new RuntimeException("仅支持图片和视频文件上传");
|
||||
}
|
||||
|
||||
//生成 OSS 路径
|
||||
String extension = FileUtil.getFileExtension(originalFilename);
|
||||
String fileName = UUID.randomUUID().toString().replace("-", "") + "." + extension;
|
||||
Long userId = UserContext.getUserId();
|
||||
String objectKey = "chat/" + userId + "/" + fileName;
|
||||
|
||||
try {
|
||||
byte[] fileBytes = file.getBytes();
|
||||
String ossUrl = ossUtil.upload(fileBytes, objectKey);
|
||||
|
||||
if (ossUrl == null || ossUrl.isEmpty()) {
|
||||
throw new RuntimeException("图片上传失败");
|
||||
}
|
||||
|
||||
return ossUrl;
|
||||
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException("上传图片失败", e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取聊天记录
|
||||
* @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());
|
||||
}
|
||||
|
||||
/**
|
||||
* 更新会话状态
|
||||
* @param userId 用户ID
|
||||
* @param dto 会话状态
|
||||
*/
|
||||
@Override
|
||||
public void updateSessionStatus(Long userId, ChatSessionStatusDTO dto) {
|
||||
chatSessionsMapper.updateSessionStatus(
|
||||
userId,
|
||||
dto.getTargetUserId(),
|
||||
dto.getSessionStatus()
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* 置顶会话
|
||||
* @param userId 用户ID
|
||||
* @param dto 置顶状态
|
||||
*/
|
||||
@Override
|
||||
public void updateTopStatus(Long userId, ChatSessionTopDTO dto) {
|
||||
chatSessionsMapper.updateTopStatus(
|
||||
userId,
|
||||
dto.getTargetUserId(),
|
||||
dto.getTopStatus()
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* 免打扰会话
|
||||
* @param userId 用户ID
|
||||
* @param dto 免打扰状态
|
||||
*/
|
||||
@Override
|
||||
public void updateMuteStatus(Long userId, ChatSessionMuteDTO dto) {
|
||||
chatSessionsMapper.updateMuteStatus(
|
||||
userId,
|
||||
dto.getTargetUserId(),
|
||||
dto.getMuteStatus()
|
||||
);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 撤回消息
|
||||
* @param senderUserId 发送者用户ID
|
||||
* @param chatId 聊天记录ID
|
||||
* @return 是否成功
|
||||
*/
|
||||
@Override
|
||||
@Transactional(rollbackFor = Exception.class)
|
||||
public boolean recallMessage(Long senderUserId, Long chatId) {
|
||||
|
||||
// 查询聊天记录
|
||||
ChatRecords record = chatRecordsMapper.selectById(chatId);
|
||||
// 消息不存在
|
||||
if (record == null) {
|
||||
log.info("消息不存在,chatId: {}", chatId);
|
||||
return false;
|
||||
}
|
||||
|
||||
// 只能撤回自己发的
|
||||
if (!record.getSenderUserId().equals(senderUserId)) {
|
||||
log.info("不能撤回别人发的消息,chatId: {},当前用户: {},消息发送者: {}", chatId, senderUserId, record.getSenderUserId());
|
||||
return false;
|
||||
}
|
||||
|
||||
// 已撤回或已删除
|
||||
if (record.getMessageStatus() != 1) {
|
||||
log.info("消息已撤回或已删除,chatId: {},当前状态: {}", chatId, record.getMessageStatus());
|
||||
return false;
|
||||
}
|
||||
|
||||
// 时间限制(2 分钟)
|
||||
Duration duration = Duration.between(record.getSendTime(), LocalDateTime.now());
|
||||
if (duration.toMinutes() > 2) {
|
||||
log.info("消息已超过 2 分钟,不能撤回,chatId: {},发送时间: {}", chatId, record.getSendTime());
|
||||
return false;
|
||||
}
|
||||
|
||||
return chatRecordsMapper.recallMessage(chatId, senderUserId) > 0;
|
||||
}
|
||||
}
|
||||
@@ -4,17 +4,22 @@ import com.bao.dating.common.aliyun.GreenImageScan;
|
||||
import com.bao.dating.common.aliyun.GreenTextScan;
|
||||
import com.bao.dating.common.result.GreenAuditResult;
|
||||
import com.bao.dating.context.UserContext;
|
||||
import com.bao.dating.mapper.CommentsMapper;
|
||||
import com.bao.dating.mapper.PostFavoriteMapper;
|
||||
import com.bao.dating.mapper.PostLikeMapper;
|
||||
import com.bao.dating.mapper.PostMapper;
|
||||
import com.bao.dating.pojo.dto.PostRequestDTO;
|
||||
import com.bao.dating.pojo.entity.Post;
|
||||
import com.bao.dating.pojo.vo.PostEditVO;
|
||||
import com.bao.dating.service.PostService;
|
||||
import com.bao.dating.common.aliyun.AliOssUtil;
|
||||
import com.bao.dating.service.UserService;
|
||||
import com.bao.dating.util.FileUtil;
|
||||
import org.springframework.beans.BeanUtils;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.springframework.transaction.annotation.Transactional;
|
||||
import org.springframework.util.CollectionUtils;
|
||||
import org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
import java.io.IOException;
|
||||
@@ -43,6 +48,15 @@ public class PostServiceImpl implements PostService {
|
||||
@Autowired
|
||||
private PostMapper postMapper;
|
||||
|
||||
@Autowired
|
||||
private PostLikeMapper postLikeMapper;
|
||||
|
||||
@Autowired
|
||||
private PostFavoriteMapper postFavoriteMapper;
|
||||
|
||||
@Autowired
|
||||
private CommentsMapper commentsMapper;
|
||||
|
||||
/**
|
||||
* 上传媒体文件
|
||||
* @param files 媒体文件数组
|
||||
@@ -177,7 +191,7 @@ public class PostServiceImpl implements PostService {
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量删除动态
|
||||
* 批量删除动态(将动态状态改为已删除)
|
||||
*
|
||||
* @param postIds 动态ID
|
||||
* @return 删除的动态对象
|
||||
@@ -188,19 +202,21 @@ public class PostServiceImpl implements PostService {
|
||||
// 判断用户权限
|
||||
Long userId = UserContext.getUserId();
|
||||
|
||||
// 遍历所有要删除的帖子ID,验证权限
|
||||
for (Long postId : postIds) {
|
||||
Post post = postMapper.selectById(postId);
|
||||
if (post == null) {
|
||||
throw new RuntimeException("动态不存在");
|
||||
}
|
||||
// 验证用户权限
|
||||
if (post.getUserId() == null || !post.getUserId().equals(userId)) {
|
||||
throw new RuntimeException("无权限删除此动态");
|
||||
}
|
||||
if (CollectionUtils.isEmpty(postIds)) {
|
||||
return 0;
|
||||
}
|
||||
// 批量删除动态
|
||||
return postMapper.deletePostByIds(postIds);
|
||||
int affected = postMapper.updatePublicById(postIds, userId);
|
||||
|
||||
if (affected == 0) {
|
||||
throw new RuntimeException("未删除任何动态,可能无权限或动态不存在");
|
||||
}
|
||||
|
||||
// 删除动态下的评论、点赞、收藏
|
||||
commentsMapper.deleteCommentsByPostIds(postIds);
|
||||
postLikeMapper.deleteLikesByPostIds(postIds);
|
||||
postFavoriteMapper.deleteFavoritesByPostIds(postIds);
|
||||
|
||||
return affected;
|
||||
}
|
||||
|
||||
/**
|
||||
|
||||
@@ -5,10 +5,9 @@ import com.bao.dating.common.aliyun.GreenImageScan;
|
||||
import com.bao.dating.common.aliyun.GreenTextScan;
|
||||
import com.bao.dating.common.result.AliOssResult;
|
||||
import com.bao.dating.common.result.GreenAuditResult;
|
||||
import com.bao.dating.config.RedisConfig;
|
||||
import com.bao.dating.context.UserContext;
|
||||
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.entity.User;
|
||||
import com.bao.dating.pojo.vo.UserInfoVO;
|
||||
@@ -231,7 +230,7 @@ public class UserServiceImpl implements UserService {
|
||||
* @param userInfoUpdateDTO 用户信息更新参数
|
||||
*/
|
||||
@Override
|
||||
public UserInfoVO updateUserInfo(UserInfoUpdateDTO userInfoUpdateDTO) {
|
||||
public UserInfoVO updateUserInfo(UserInfoDTO userInfoUpdateDTO) {
|
||||
Long userId = userInfoUpdateDTO.getUserId();
|
||||
User user = userMapper.selectByUserId(userId);
|
||||
if (user == null) {
|
||||
@@ -311,4 +310,23 @@ public class UserServiceImpl implements UserService {
|
||||
BeanUtils.copyProperties(updatedUser, 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;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,122 @@
|
||||
package com.bao.dating.service.impl;
|
||||
|
||||
import com.bao.dating.service.VerificationCodeService;
|
||||
import com.bao.dating.util.EmailUtil;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.data.redis.core.StringRedisTemplate;
|
||||
import org.springframework.stereotype.Service;
|
||||
|
||||
import java.util.Random;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
/**
|
||||
* 验证码服务实现类
|
||||
* @author KilLze
|
||||
*/
|
||||
@Slf4j
|
||||
@Service
|
||||
public class VerificationCodeServiceImpl implements VerificationCodeService {
|
||||
|
||||
@Autowired
|
||||
private EmailUtil emailUtil;
|
||||
|
||||
@Autowired
|
||||
private StringRedisTemplate redisTemplate;
|
||||
|
||||
/**
|
||||
* Redis中验证码的key前缀
|
||||
*/
|
||||
private static final String CODE_KEY_PREFIX = "email:code:";
|
||||
|
||||
/**
|
||||
* 验证码过期时间(分钟)
|
||||
*/
|
||||
private static final long CODE_EXPIRE_MINUTES = 10;
|
||||
|
||||
/**
|
||||
* 验证码长度
|
||||
*/
|
||||
private static final int CODE_LENGTH = 6;
|
||||
|
||||
/**
|
||||
* 发送邮箱验证码
|
||||
* @param email 邮箱地址
|
||||
* @return 是否发送成功
|
||||
*/
|
||||
@Override
|
||||
public boolean sendEmailCode(String email) {
|
||||
try {
|
||||
// 生成验证码
|
||||
String code = generateCode(CODE_LENGTH);
|
||||
|
||||
// 存储到Redis,设置过期时间
|
||||
String key = CODE_KEY_PREFIX + email;
|
||||
redisTemplate.opsForValue().set(key, code, CODE_EXPIRE_MINUTES, TimeUnit.MINUTES);
|
||||
|
||||
// 发送邮件
|
||||
boolean sendResult = emailUtil.sendVerificationCode(email, code);
|
||||
|
||||
if (sendResult) {
|
||||
log.info("邮箱验证码发送成功,邮箱:{},验证码:{}", email, code);
|
||||
return true;
|
||||
} else {
|
||||
// 如果发送失败,删除Redis中的验证码
|
||||
redisTemplate.delete(key);
|
||||
log.error("邮箱验证码发送失败,邮箱:{}", email);
|
||||
return false;
|
||||
}
|
||||
} catch (Exception e) {
|
||||
log.error("发送邮箱验证码异常,邮箱:{},异常信息:{}", email, e.getMessage(), e);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证邮箱验证码
|
||||
* @param email 邮箱地址
|
||||
* @param code 验证码
|
||||
* @return 是否验证成功
|
||||
*/
|
||||
@Override
|
||||
public boolean verifyEmailCode(String email, String code) {
|
||||
try {
|
||||
String key = CODE_KEY_PREFIX + email;
|
||||
String storedCode = redisTemplate.opsForValue().get(key);
|
||||
|
||||
if (storedCode == null) {
|
||||
log.warn("验证码不存在或已过期,邮箱:{}", email);
|
||||
return false;
|
||||
}
|
||||
|
||||
if (storedCode.equals(code)) {
|
||||
// 验证成功后,删除验证码(防止重复使用)
|
||||
redisTemplate.delete(key);
|
||||
log.info("邮箱验证码验证成功,邮箱:{}", email);
|
||||
return true;
|
||||
} else {
|
||||
log.warn("邮箱验证码错误,邮箱:{},输入的验证码:{},正确的验证码:{}", email, code, storedCode);
|
||||
return false;
|
||||
}
|
||||
} catch (Exception e) {
|
||||
log.error("验证邮箱验证码异常,邮箱:{},异常信息:{}", email, e.getMessage(), e);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成验证码
|
||||
* @param length 验证码长度
|
||||
* @return 验证码字符串
|
||||
*/
|
||||
@Override
|
||||
public String generateCode(int length) {
|
||||
Random random = new Random();
|
||||
StringBuilder code = new StringBuilder();
|
||||
for (int i = 0; i < length; i++) {
|
||||
code.append(random.nextInt(10)); // 生成0-9的随机数字
|
||||
}
|
||||
return code.toString();
|
||||
}
|
||||
}
|
||||
|
||||
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);
|
||||
}
|
||||
}
|
||||
@@ -9,7 +9,7 @@ import java.security.NoSuchAlgorithmException;
|
||||
* @author KilLze
|
||||
*/
|
||||
public class MD5Util {
|
||||
|
||||
|
||||
/**
|
||||
* 对字符串进行MD5加密
|
||||
* @param input 待加密的字符串
|
||||
@@ -19,7 +19,7 @@ public class MD5Util {
|
||||
if (input == null || input.isEmpty()) {
|
||||
return null;
|
||||
}
|
||||
|
||||
|
||||
try {
|
||||
MessageDigest md = MessageDigest.getInstance("MD5");
|
||||
byte[] digest = md.digest(input.getBytes());
|
||||
@@ -28,7 +28,7 @@ public class MD5Util {
|
||||
throw new RuntimeException("MD5算法不可用", e);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 对字符串进行MD5加密(带盐值)
|
||||
* @param input 待加密的字符串
|
||||
@@ -44,7 +44,7 @@ public class MD5Util {
|
||||
}
|
||||
return encrypt(input + salt);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 验证字符串与MD5值是否匹配
|
||||
* @param input 原始字符串
|
||||
@@ -57,7 +57,7 @@ public class MD5Util {
|
||||
}
|
||||
return encrypt(input).equalsIgnoreCase(md5Hash);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 验证字符串与MD5值是否匹配(带盐值)
|
||||
* @param input 原始字符串
|
||||
@@ -74,7 +74,7 @@ public class MD5Util {
|
||||
}
|
||||
return encryptWithSalt(input, salt).equalsIgnoreCase(md5Hash);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 将字节数组转换为十六进制字符串
|
||||
* @param bytes 字节数组
|
||||
@@ -88,6 +88,3 @@ public class MD5Util {
|
||||
return sb.toString();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
@@ -13,8 +13,8 @@ spring:
|
||||
password: JoyeeServe2025
|
||||
driver-class-name: com.mysql.cj.jdbc.Driver
|
||||
redis:
|
||||
host: 127.0.0.1
|
||||
port: 6379
|
||||
host: 110.42.41.177
|
||||
port: 6389
|
||||
password: ""
|
||||
database: 0
|
||||
timeout: 10000
|
||||
|
||||
9
src/main/resources/ciallo.txt
Normal file
9
src/main/resources/ciallo.txt
Normal file
@@ -0,0 +1,9 @@
|
||||
|
||||
▄▄▄▄ ██ ▄▄▄▄ ▄▄▄▄ ▄▄ ▄▄
|
||||
██▀▀▀▀█ ▀▀ ▀▀██ ▀▀██ ██ ▄▄ ██
|
||||
██▀ ████ ▄█████▄ ██ ██ ▄████▄ ▄▄▄ ▄█▀ ▄█▀ ██ █▄ ▄▄▄█ ▀█▄ ▄▄▄▄
|
||||
██ ██ ▀ ▄▄▄██ ██ ██ ██▀ ▀██ ▀ ▀▀▄▄ ▄ ██ ▄█▀ ██ ██ ██ ▄▄█▀▀▀ ██ █▀▀ ▀█▄ █▄
|
||||
██▄ ██ ▄██▀▀▀██ ██ ██ ██ ██ ▀▀▀ ██ ▄█▄▄▄▄▄ ▀▀ ██ ██ ██ ▀▀█▄▄▄ ██ █▀ █ ▀▀████▀
|
||||
██▄▄▄▄█ ▄▄▄██▄▄▄ ██▄▄▄███ ██▄▄▄ ██▄▄▄ ▀██▄▄██▀ ▀█▄ ▀▀▀▀▀▀▀▀ ██▄██▄██ ▀▀▀█ ▄█▀ █▀▀█
|
||||
▀▀▀▀ ▀▀▀▀▀▀▀▀ ▀▀▀▀ ▀▀ ▀▀▀▀ ▀▀▀▀ ▀▀▀▀ ██ ▀▀▀ ▀▀▀ ██
|
||||
▀▀ ▀▀
|
||||
@@ -0,0 +1,92 @@
|
||||
<?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>
|
||||
|
||||
<!-- 根据ID查询聊天记录 -->
|
||||
<select id="selectById" resultType="com.bao.dating.pojo.entity.ChatRecords">
|
||||
SELECT
|
||||
chat_id,
|
||||
sender_user_id,
|
||||
receiver_user_id,
|
||||
send_time,
|
||||
message_status
|
||||
FROM chat_records
|
||||
WHERE chat_id = #{chatId}
|
||||
</select>
|
||||
|
||||
<!-- 撤回消息 -->
|
||||
<update id="recallMessage">
|
||||
UPDATE chat_records
|
||||
SET
|
||||
message_status = 2,
|
||||
updated_at = NOW()
|
||||
WHERE
|
||||
chat_id = #{chatId}
|
||||
AND sender_user_id = #{senderUserId}
|
||||
AND message_status = 1
|
||||
</update>
|
||||
|
||||
</mapper>
|
||||
@@ -0,0 +1,91 @@
|
||||
<?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 in (1,2)
|
||||
ORDER BY top_status DESC, last_message_time DESC
|
||||
</select>
|
||||
|
||||
<!-- 更新会话状态 -->
|
||||
<update id="updateSessionStatus">
|
||||
UPDATE chat_sessions
|
||||
SET session_status = #{sessionStatus},
|
||||
updated_at = NOW()
|
||||
WHERE user_id = #{userId}
|
||||
AND target_user_id = #{targetUserId}
|
||||
</update>
|
||||
|
||||
<!-- 置顶会话 -->
|
||||
<update id="updateTopStatus">
|
||||
UPDATE chat_sessions
|
||||
SET top_status = #{topStatus},
|
||||
updated_at = NOW()
|
||||
WHERE user_id = #{userId}
|
||||
AND target_user_id = #{targetUserId}
|
||||
</update>
|
||||
|
||||
<!-- 静音会话 -->
|
||||
<update id="updateMuteStatus">
|
||||
UPDATE chat_sessions
|
||||
SET mute_status = #{muteStatus},
|
||||
updated_at = NOW()
|
||||
WHERE user_id = #{userId}
|
||||
AND target_user_id = #{targetUserId}
|
||||
</update>
|
||||
|
||||
</mapper>
|
||||
15
src/main/resources/com/bao/dating/mapper/CommentsMapper.xml
Normal file
15
src/main/resources/com/bao/dating/mapper/CommentsMapper.xml
Normal file
@@ -0,0 +1,15 @@
|
||||
<?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.CommentsMapper">
|
||||
|
||||
<!-- 批量删除动态下的所有评论 -->
|
||||
<delete id="deleteCommentsByPostIds">
|
||||
DELETE FROM comments
|
||||
WHERE post_id IN
|
||||
<foreach collection="postIds" item="postId" open="(" close=")" separator=",">
|
||||
#{postId}
|
||||
</foreach>
|
||||
</delete>
|
||||
|
||||
</mapper>
|
||||
@@ -14,4 +14,14 @@
|
||||
<select id="selectUserIDByPostID" resultType="java.lang.Long">
|
||||
SELECT user_id FROM post_favorite WHERE post_id = #{postId}
|
||||
</select>
|
||||
|
||||
<!--批量删除动态收藏-->
|
||||
<delete id="deleteFavoritesByPostIds">
|
||||
DELETE FROM post_favorite
|
||||
WHERE post_id IN
|
||||
<foreach collection="postIds" item="postId" open="(" close=")" separator=",">
|
||||
#{postId}
|
||||
</foreach>
|
||||
</delete>
|
||||
|
||||
</mapper>
|
||||
@@ -14,4 +14,14 @@
|
||||
<delete id="deleteByPostIdAndUserId">
|
||||
delete from dating.post_like where post_id = #{postId} and user_id = #{userId}
|
||||
</delete>
|
||||
|
||||
<!--批量删除点赞记录-->
|
||||
<delete id="deleteLikesByPostIds">
|
||||
DELETE FROM post_like
|
||||
WHERE post_id IN
|
||||
<foreach collection="postIds" item="postId" open="(" close=")" separator=",">
|
||||
#{postId}
|
||||
</foreach>
|
||||
</delete>
|
||||
|
||||
</mapper>
|
||||
@@ -28,25 +28,19 @@
|
||||
#{isPublic}, 0, 0, #{createdAt}, #{updatedAt})
|
||||
</insert>
|
||||
|
||||
<!--动态删除-->
|
||||
<delete id="deletePostByIds">
|
||||
DELETE FROM post WHERE post_id IN
|
||||
<!--修改动态状态-->
|
||||
<update id="updatePublicById">
|
||||
UPDATE post
|
||||
<set>
|
||||
is_public = 3,
|
||||
updated_at = NOW()
|
||||
</set>
|
||||
WHERE post_id IN
|
||||
<foreach item="postId" index="index" collection="postIds" separator="," open="(" close=")">
|
||||
#{postId}
|
||||
</foreach>
|
||||
</delete>
|
||||
<!--删除收藏记录-->
|
||||
<delete id="1">
|
||||
DELETE FROM post_favorite WHERE post_id = #{postId}
|
||||
</delete>
|
||||
<!--删除点赞记录-->
|
||||
<delete id="2">
|
||||
DELETE FROM post_like WHERE post_id = #{postId}
|
||||
</delete>
|
||||
<!--动态评论删除-->
|
||||
<delete id="3">
|
||||
DELETE FROM comments WHERE post_id = #{postId}
|
||||
</delete>
|
||||
AND user_id = #{userId}
|
||||
</update>
|
||||
|
||||
<!--动态查询-->
|
||||
<resultMap id="PostResultMap" type="com.bao.dating.pojo.entity.Post">
|
||||
|
||||
304
src/test/java/com/bao/dating/common/aliyun/OssUploadTest.java
Normal file
304
src/test/java/com/bao/dating/common/aliyun/OssUploadTest.java
Normal file
@@ -0,0 +1,304 @@
|
||||
package com.bao.dating.common.aliyun;
|
||||
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.boot.test.context.SpringBootTest;
|
||||
|
||||
import java.io.ByteArrayOutputStream;
|
||||
import java.io.File;
|
||||
import java.io.FileInputStream;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.net.URL;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.Paths;
|
||||
import java.time.LocalDateTime;
|
||||
import java.time.format.DateTimeFormatter;
|
||||
import java.util.UUID;
|
||||
|
||||
/**
|
||||
* OSS图片上传测试类
|
||||
* @author KilLze
|
||||
*/
|
||||
@SpringBootTest
|
||||
public class OssUploadTest {
|
||||
|
||||
@Autowired
|
||||
private AliOssUtil aliOssUtil;
|
||||
|
||||
/**
|
||||
* 测试上传本地图片文件
|
||||
*/
|
||||
@Test
|
||||
public void testUploadLocalImage() {
|
||||
try {
|
||||
// 本地图片文件路径(请修改为实际存在的图片路径)
|
||||
String localImagePath = "D:/image/image.jpg"; // 请修改为实际路径
|
||||
|
||||
File imageFile = new File(localImagePath);
|
||||
if (!imageFile.exists()) {
|
||||
System.out.println("图片文件不存在: " + localImagePath);
|
||||
System.out.println("请修改localImagePath为实际存在的图片路径");
|
||||
return;
|
||||
}
|
||||
|
||||
// 读取文件字节
|
||||
byte[] imageBytes = Files.readAllBytes(Paths.get(localImagePath));
|
||||
|
||||
// 生成OSS对象名称(使用时间戳和UUID确保唯一性)
|
||||
String objectName = generateObjectName("test", "jpg");
|
||||
|
||||
// 上传到OSS
|
||||
String fileUrl = aliOssUtil.upload(imageBytes, objectName);
|
||||
|
||||
System.out.println("========== 本地图片上传测试 ==========");
|
||||
System.out.println("本地文件路径: " + localImagePath);
|
||||
System.out.println("文件大小: " + imageBytes.length + " 字节");
|
||||
System.out.println("OSS对象名称: " + objectName);
|
||||
System.out.println("文件访问URL: " + fileUrl);
|
||||
System.out.println("上传结果: 成功");
|
||||
System.out.println("=====================================");
|
||||
|
||||
} catch (IOException e) {
|
||||
System.out.println("上传失败: " + e.getMessage());
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试从网络URL下载图片并上传到OSS
|
||||
*/
|
||||
@Test
|
||||
public void testUploadImageFromUrl() {
|
||||
try {
|
||||
// 网络图片URL(请修改为实际的图片URL)
|
||||
String imageUrl = "https://example.com/image.jpg"; // 请修改为实际URL
|
||||
|
||||
System.out.println("========== 网络图片上传测试 ==========");
|
||||
System.out.println("开始下载图片: " + imageUrl);
|
||||
|
||||
// 从URL下载图片
|
||||
URL url = new URL(imageUrl);
|
||||
try (InputStream inputStream = url.openStream()) {
|
||||
byte[] imageBytes = readAllBytes(inputStream);
|
||||
|
||||
// 从URL中提取文件扩展名
|
||||
String extension = getFileExtensionFromUrl(imageUrl);
|
||||
if (extension.isEmpty()) {
|
||||
extension = "jpg"; // 默认扩展名
|
||||
}
|
||||
|
||||
// 生成OSS对象名称
|
||||
String objectName = generateObjectName("download", extension);
|
||||
|
||||
// 上传到OSS
|
||||
String fileUrl = aliOssUtil.upload(imageBytes, objectName);
|
||||
|
||||
System.out.println("图片下载成功");
|
||||
System.out.println("文件大小: " + imageBytes.length + " 字节");
|
||||
System.out.println("OSS对象名称: " + objectName);
|
||||
System.out.println("文件访问URL: " + fileUrl);
|
||||
System.out.println("上传结果: 成功");
|
||||
}
|
||||
|
||||
System.out.println("=====================================");
|
||||
|
||||
} catch (Exception e) {
|
||||
System.out.println("上传失败: " + e.getMessage());
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试上传不同格式的图片
|
||||
*/
|
||||
@Test
|
||||
public void testUploadDifferentImageFormats() {
|
||||
String[] imagePaths = {
|
||||
"D:/test/image1.jpg", // 请修改为实际路径
|
||||
"D:/test/image2.png", // 请修改为实际路径
|
||||
"D:/test/image3.gif" // 请修改为实际路径
|
||||
};
|
||||
|
||||
System.out.println("========== 多格式图片上传测试 ==========");
|
||||
|
||||
for (String imagePath : imagePaths) {
|
||||
try {
|
||||
File imageFile = new File(imagePath);
|
||||
if (!imageFile.exists()) {
|
||||
System.out.println("跳过不存在的文件: " + imagePath);
|
||||
continue;
|
||||
}
|
||||
|
||||
// 读取文件
|
||||
byte[] imageBytes = Files.readAllBytes(Paths.get(imagePath));
|
||||
|
||||
// 获取文件扩展名
|
||||
String extension = getFileExtension(imagePath);
|
||||
|
||||
// 生成OSS对象名称
|
||||
String objectName = generateObjectName("format-test", extension);
|
||||
|
||||
// 上传
|
||||
String fileUrl = aliOssUtil.upload(imageBytes, objectName);
|
||||
|
||||
System.out.println("\n文件: " + imagePath);
|
||||
System.out.println("格式: " + extension);
|
||||
System.out.println("大小: " + imageBytes.length + " 字节");
|
||||
System.out.println("URL: " + fileUrl);
|
||||
System.out.println("状态: 成功");
|
||||
|
||||
} catch (Exception e) {
|
||||
System.out.println("\n文件: " + imagePath);
|
||||
System.out.println("状态: 失败 - " + e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
System.out.println("\n=====================================");
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试上传图片到指定目录
|
||||
*/
|
||||
@Test
|
||||
public void testUploadImageToDirectory() {
|
||||
try {
|
||||
// 本地图片文件路径
|
||||
String localImagePath = "D:/test/image.jpg"; // 请修改为实际路径
|
||||
|
||||
File imageFile = new File(localImagePath);
|
||||
if (!imageFile.exists()) {
|
||||
System.out.println("图片文件不存在: " + localImagePath);
|
||||
return;
|
||||
}
|
||||
|
||||
// 读取文件
|
||||
byte[] imageBytes = Files.readAllBytes(Paths.get(localImagePath));
|
||||
|
||||
// 上传到指定目录(例如:images/2025/12/24/xxx.jpg)
|
||||
String directory = "images/" + LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy/MM/dd"));
|
||||
String fileName = UUID.randomUUID().toString() + ".jpg";
|
||||
String objectName = directory + "/" + fileName;
|
||||
|
||||
// 上传
|
||||
String fileUrl = aliOssUtil.upload(imageBytes, objectName);
|
||||
|
||||
System.out.println("========== 目录上传测试 ==========");
|
||||
System.out.println("目录: " + directory);
|
||||
System.out.println("文件名: " + fileName);
|
||||
System.out.println("完整路径: " + objectName);
|
||||
System.out.println("文件访问URL: " + fileUrl);
|
||||
System.out.println("上传结果: 成功");
|
||||
System.out.println("=================================");
|
||||
|
||||
} catch (Exception e) {
|
||||
System.out.println("上传失败: " + e.getMessage());
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试使用FileInputStream上传
|
||||
*/
|
||||
@Test
|
||||
public void testUploadWithFileInputStream() {
|
||||
try {
|
||||
// 本地图片文件路径
|
||||
String localImagePath = "D:/test/image.jpg"; // 请修改为实际路径
|
||||
|
||||
File imageFile = new File(localImagePath);
|
||||
if (!imageFile.exists()) {
|
||||
System.out.println("图片文件不存在: " + localImagePath);
|
||||
return;
|
||||
}
|
||||
|
||||
// 使用FileInputStream读取文件
|
||||
try (FileInputStream fis = new FileInputStream(imageFile)) {
|
||||
byte[] imageBytes = new byte[(int) imageFile.length()];
|
||||
fis.read(imageBytes);
|
||||
|
||||
// 生成OSS对象名称
|
||||
String objectName = generateObjectName("stream", "jpg");
|
||||
|
||||
// 上传
|
||||
String fileUrl = aliOssUtil.upload(imageBytes, objectName);
|
||||
|
||||
System.out.println("========== FileInputStream上传测试 ==========");
|
||||
System.out.println("文件路径: " + localImagePath);
|
||||
System.out.println("文件大小: " + imageBytes.length + " 字节");
|
||||
System.out.println("OSS对象名称: " + objectName);
|
||||
System.out.println("文件访问URL: " + fileUrl);
|
||||
System.out.println("上传结果: 成功");
|
||||
System.out.println("===========================================");
|
||||
}
|
||||
|
||||
} catch (Exception e) {
|
||||
System.out.println("上传失败: " + e.getMessage());
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成OSS对象名称
|
||||
* @param prefix 前缀
|
||||
* @param extension 文件扩展名
|
||||
* @return 对象名称
|
||||
*/
|
||||
private String generateObjectName(String prefix, String extension) {
|
||||
// 格式: prefix/yyyyMMdd/HHmmss-uuid.extension
|
||||
String dateTime = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyyMMdd/HHmmss"));
|
||||
String uuid = UUID.randomUUID().toString().substring(0, 8);
|
||||
return String.format("%s/%s-%s.%s", prefix, dateTime, uuid, extension);
|
||||
}
|
||||
|
||||
/**
|
||||
* 从文件路径获取扩展名
|
||||
* @param filePath 文件路径
|
||||
* @return 扩展名
|
||||
*/
|
||||
private String getFileExtension(String filePath) {
|
||||
int lastDot = filePath.lastIndexOf('.');
|
||||
if (lastDot > 0 && lastDot < filePath.length() - 1) {
|
||||
return filePath.substring(lastDot + 1).toLowerCase();
|
||||
}
|
||||
return "jpg"; // 默认扩展名
|
||||
}
|
||||
|
||||
/**
|
||||
* 从URL获取文件扩展名
|
||||
* @param url URL地址
|
||||
* @return 扩展名
|
||||
*/
|
||||
private String getFileExtensionFromUrl(String url) {
|
||||
try {
|
||||
// 移除查询参数
|
||||
String path = url.split("\\?")[0];
|
||||
int lastDot = path.lastIndexOf('.');
|
||||
int lastSlash = path.lastIndexOf('/');
|
||||
|
||||
if (lastDot > lastSlash && lastDot < path.length() - 1) {
|
||||
return path.substring(lastDot + 1).toLowerCase();
|
||||
}
|
||||
} catch (Exception e) {
|
||||
// 忽略异常
|
||||
}
|
||||
return "jpg"; // 默认扩展名
|
||||
}
|
||||
|
||||
/**
|
||||
* 从InputStream读取所有字节(兼容方法)
|
||||
* @param inputStream 输入流
|
||||
* @return 字节数组
|
||||
* @throws IOException IO异常
|
||||
*/
|
||||
private byte[] readAllBytes(InputStream inputStream) throws IOException {
|
||||
ByteArrayOutputStream buffer = new ByteArrayOutputStream();
|
||||
byte[] data = new byte[8192]; // 8KB缓冲区
|
||||
int nRead;
|
||||
while ((nRead = inputStream.read(data, 0, data.length)) != -1) {
|
||||
buffer.write(data, 0, nRead);
|
||||
}
|
||||
return buffer.toByteArray();
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,91 @@
|
||||
package com.bao.dating.service;
|
||||
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.boot.test.context.SpringBootTest;
|
||||
|
||||
/**
|
||||
* 验证码服务测试类
|
||||
* @author KilLze
|
||||
*/
|
||||
@SpringBootTest
|
||||
public class VerificationCodeServiceTest {
|
||||
|
||||
@Autowired
|
||||
private VerificationCodeService verificationCodeService;
|
||||
|
||||
/**
|
||||
* 测试发送邮箱验证码
|
||||
*/
|
||||
@Test
|
||||
public void testSendEmailCode() {
|
||||
String email = "test@example.com"; // 请修改为实际邮箱地址
|
||||
|
||||
System.out.println("========== 发送邮箱验证码测试 ==========");
|
||||
System.out.println("邮箱地址: " + email);
|
||||
|
||||
boolean result = verificationCodeService.sendEmailCode(email);
|
||||
|
||||
System.out.println("发送结果: " + (result ? "成功" : "失败"));
|
||||
System.out.println("=====================================");
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试验证邮箱验证码
|
||||
*/
|
||||
@Test
|
||||
public void testVerifyEmailCode() {
|
||||
String email = "test@example.com"; // 请修改为实际邮箱地址
|
||||
String code = "123456"; // 请修改为实际收到的验证码
|
||||
|
||||
System.out.println("========== 验证邮箱验证码测试 ==========");
|
||||
System.out.println("邮箱地址: " + email);
|
||||
System.out.println("验证码: " + code);
|
||||
|
||||
boolean result = verificationCodeService.verifyEmailCode(email, code);
|
||||
|
||||
System.out.println("验证结果: " + (result ? "成功" : "失败"));
|
||||
System.out.println("=====================================");
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试完整流程:发送验证码 -> 验证验证码
|
||||
*/
|
||||
@Test
|
||||
public void testCompleteFlow() {
|
||||
String email = "test@example.com"; // 请修改为实际邮箱地址
|
||||
|
||||
System.out.println("========== 完整流程测试 ==========");
|
||||
System.out.println("邮箱地址: " + email);
|
||||
|
||||
// 1. 发送验证码
|
||||
System.out.println("\n1. 发送验证码...");
|
||||
boolean sendResult = verificationCodeService.sendEmailCode(email);
|
||||
System.out.println("发送结果: " + (sendResult ? "成功" : "失败"));
|
||||
|
||||
if (sendResult) {
|
||||
// 2. 等待用户输入验证码(这里模拟,实际应该从控制台或API获取)
|
||||
System.out.println("\n2. 请查看邮箱获取验证码,然后手动测试验证功能");
|
||||
System.out.println(" 使用 testVerifyEmailCode() 方法进行验证");
|
||||
}
|
||||
|
||||
System.out.println("\n=====================================");
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试生成验证码
|
||||
*/
|
||||
@Test
|
||||
public void testGenerateCode() {
|
||||
System.out.println("========== 生成验证码测试 ==========");
|
||||
|
||||
// 测试不同长度的验证码
|
||||
for (int length = 4; length <= 8; length++) {
|
||||
String code = verificationCodeService.generateCode(length);
|
||||
System.out.println(length + "位验证码: " + code);
|
||||
}
|
||||
|
||||
System.out.println("=====================================");
|
||||
}
|
||||
}
|
||||
|
||||
@@ -75,3 +75,4 @@ public class EmailAndSmsTest {
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user