myplayer/myplayer-backend
1
0
Fork 0
Code Issues 11 Pull Requests Actions Packages Projects Releases Wiki Activity

feat: introduce middleware rabbitmq and redis

Browse Source
This commit is contained in:
merlin
2025-12-11 10:37:03 +08:00
parent a1a23bec7a
commit 44133d3667
22 changed files with 404 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
pom.xml
View File

@@ -38,9 +38,14 @@
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
<dependency>
<groupId>org.springframework.kafka</groupId>
<artifactId>spring-kafka</artifactId>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-amqp</artifactId>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>

16
src/main/java/xin/merlin/myplayerbackend/config/RabbitMQConfig.java Normal file
View File

@@ -0,0 +1,16 @@
package xin.merlin.myplayerbackend.config;
import org.springframework.amqp.core.Queue;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
@Configuration
public class RabbitMQConfig {
public static final String WS_MESSAGE_QUEUE = "ws.message";
@Bean
public Queue wsMessageQueue() {
return new Queue(WS_MESSAGE_QUEUE, true); // 持久化队列
}
}

24
src/main/java/xin/merlin/myplayerbackend/config/RedisConfig.java Normal file
View File

@@ -0,0 +1,24 @@
package xin.merlin.myplayerbackend.config;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.serializer.GenericJackson2JsonRedisSerializer;
import org.springframework.data.redis.serializer.StringRedisSerializer;
@Configuration
public class RedisConfig {
@Bean
public RedisTemplate<String, Object> redisTemplate(RedisConnectionFactory factory) {
RedisTemplate<String, Object> template = new RedisTemplate<>();
template.setConnectionFactory(factory);
template.setKeySerializer(new StringRedisSerializer());
template.setValueSerializer(new GenericJackson2JsonRedisSerializer());
return template;
}
}

4
src/main/java/xin/merlin/myplayerbackend/config/security/WebsocketInterceptor.java
View File

@@ -26,6 +26,10 @@ public class WebsocketInterceptor implements HandshakeInterceptor {
String token = request.getHeaders().getFirst("Authorization");
if (token != null && token.startsWith("Bearer ")) {
token = token.substring(7);
if (jwtUtil.isTokenExpired(token)){
log.info("token expired");
return false;
}
} else {
return false;
}

29
src/main/java/xin/merlin/myplayerbackend/controller/FriendController.java
View File

@@ -5,14 +5,14 @@ import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.*;
import xin.merlin.myplayerbackend.entity.Friends;
import xin.merlin.myplayerbackend.entity.UserInfo;
import xin.merlin.myplayerbackend.service.impl.FriendsServiceImpl;
import xin.merlin.myplayerbackend.service.impl.InvitingServiceImpl;
import xin.merlin.myplayerbackend.utils.JwtUtil;
import xin.merlin.myplayerbackend.utils.result.Response;
import xin.merlin.myplayerbackend.utils.result.ResultCode;
import static com.baomidou.mybatisplus.extension.ddl.DdlScriptErrorHandler.PrintlnLogErrorHandler.log;
import java.util.List;
@Slf4j
@RequestMapping("/friend")
@@ -54,14 +54,27 @@ public class FriendController {
@PostMapping("/nickname")
Response nickname(@RequestHeader("Authorization")String token, @RequestBody Friends friends) {
token = token.substring(7);
Integer id = jwtUtil.getId(token);
if (!id.equals(friends.getId())) return Response.success(ResultCode.SERVER_ERROR);
friendsService.saveOrUpdate(friends);
return Response.success(ResultCode.SUCCESS);
try {
token = token.substring(7);
Integer id = jwtUtil.getId(token);
if (!id.equals(friends.getId())) return Response.success(ResultCode.SERVER_ERROR);
friendsService.saveOrUpdate(friends);
return Response.success(ResultCode.SUCCESS);
} catch (Exception e) {
log.error(e.getMessage());
return Response.fail(ResultCode.SERVER_ERROR);
}
}
@PostMapping("/status")
Response status(@RequestBody List<Integer> friends) {
try {
return Response.success(ResultCode.SUCCESS,friendsService.status(friends));
} catch (Exception e) {
log.error(e.getMessage());
return Response.fail(ResultCode.SERVER_ERROR);
}
}
}

5
src/main/java/xin/merlin/myplayerbackend/controller/GroupController.java
View File

@@ -6,7 +6,6 @@ import lombok.RequiredArgsConstructor;
import org.springframework.web.bind.annotation.*;
import xin.merlin.myplayerbackend.entity.GroupInfo;
import xin.merlin.myplayerbackend.entity.Groups;
import xin.merlin.myplayerbackend.entity.Playrooms;
import xin.merlin.myplayerbackend.entity.UserInfo;
import xin.merlin.myplayerbackend.service.impl.GroupServiceImpl;
import xin.merlin.myplayerbackend.service.impl.GroupsServiceImpl;
@@ -56,7 +55,7 @@ public class GroupController {
}
@PostMapping("/search")
Response searchGroup(@RequestHeader("Authorization")String token, @RequestBody GroupInfo groupInfo){
Response searchGroup(@RequestBody GroupInfo groupInfo){
// TODO：视情况开放api参数currentPage和 pageSize
try {
Integer currentPage = 1;
@@ -113,7 +112,7 @@ public class GroupController {
token = token.substring(7);
Integer id = jwtUtil.getId(token);
if(!isAdmin(id,g_id)) return Response.success(ResultCode.ACCOUNT_PERMISSION_DENY);
if(!groupService.leaveGroup(id,g_id)) return Response.success(ResultCode.GROUP_USER_NOT_EXISTED);
if(!groupService.leaveGroup(userInfo.getId(),g_id)) return Response.success(ResultCode.GROUP_USER_NOT_EXISTED);
return Response.success(ResultCode.SUCCESS);
} catch (Exception e) {
log.error(e.getMessage());

5
src/main/java/xin/merlin/myplayerbackend/controller/InvitingController.java
View File

@@ -5,6 +5,7 @@ import org.springframework.web.bind.annotation.*;
import xin.merlin.myplayerbackend.entity.Inviting;
import xin.merlin.myplayerbackend.entity.UserInfo;
import xin.merlin.myplayerbackend.service.impl.InvitingServiceImpl;
import xin.merlin.myplayerbackend.service.impl.PlayroomsServiceImpl;
import xin.merlin.myplayerbackend.utils.JwtUtil;
import xin.merlin.myplayerbackend.utils.result.Response;
import xin.merlin.myplayerbackend.utils.result.ResultCode;
@@ -20,11 +21,13 @@ public class InvitingController {
private final InvitingServiceImpl invitingService;
private final PlayroomsServiceImpl playroomsService;
private final JwtUtil jwtUtil;
// playroom鉴权
private Boolean isAdmin(Integer id,Integer r_id){
return invitingService.playroomIsAdmin(id,r_id)==0;
return playroomsService.playroomIsAdmin(id,r_id)==0;
}
@PostMapping("/friends")

45
src/main/java/xin/merlin/myplayerbackend/service/OnlineStatusService.java Normal file
View File

@@ -0,0 +1,45 @@
package xin.merlin.myplayerbackend.service;
import jakarta.annotation.PreDestroy;
import lombok.RequiredArgsConstructor;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Service;
import java.util.Set;
@Service
@RequiredArgsConstructor
public class OnlineStatusService {
private static final String ONLINE_USERS_KEY = "online:users";
private final StringRedisTemplate redis;
/** 用户上线 */
public void online(Integer uid) {
redis.opsForSet().add(ONLINE_USERS_KEY, uid.toString());
}
/** 用户下线 */
public void offline(Integer uid) {
redis.opsForSet().remove(ONLINE_USERS_KEY, uid.toString());
}
/** 是否在线 */
public boolean isOnline(Integer uid) {
return Boolean.TRUE.equals(redis.opsForSet().isMember(ONLINE_USERS_KEY, uid.toString()));
}
/** 获取所有在线用户 */
public Set<String> getOnlineUsers() {
return redis.opsForSet().members(ONLINE_USERS_KEY);
}
@PreDestroy
public void cleanup() {
// 清空在线状态
redis.delete(ONLINE_USERS_KEY);
}
}

22
src/main/java/xin/merlin/myplayerbackend/service/impl/FriendsServiceImpl.java
View File

@@ -11,9 +11,12 @@ import xin.merlin.myplayerbackend.entity.UserInfo;
import xin.merlin.myplayerbackend.entity.http.Friend;
import xin.merlin.myplayerbackend.mapper.FriendsMapper;
import xin.merlin.myplayerbackend.mapper.UserMapper;
import xin.merlin.myplayerbackend.service.OnlineStatusService;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@Service
@RequiredArgsConstructor
@@ -23,6 +26,8 @@ public class FriendsServiceImpl extends ServiceImpl<FriendsMapper, Friends> {
private final UserMapper userMapper;
private final OnlineStatusService onlineStatusService;
public List<Friend> getFriends(Integer id, Integer size, Integer page) {
List<Friends> friends = friendsMapper.selectList(new Page<>(page,size),Wrappers.<Friends>lambdaQuery().eq(Friends::getId, id));
@@ -44,4 +49,21 @@ public class FriendsServiceImpl extends ServiceImpl<FriendsMapper, Friends> {
throw new RuntimeException(e);
}
}
public Map<Integer,Integer> status(List<Integer> friends){
try {
Map<Integer,Integer> map = new HashMap<>();
for (Integer f : friends) {
if(onlineStatusService.isOnline(f)){
map.put(f,1);
}else {
map.put(f,0);
}
}
return map;
} catch (Exception e) {
throw new RuntimeException(e);
}
}
}

1
src/main/java/xin/merlin/myplayerbackend/service/impl/GroupServiceImpl.java
View File

@@ -8,7 +8,6 @@ import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import xin.merlin.myplayerbackend.entity.*;
import xin.merlin.myplayerbackend.entity.http.GroupDetails;
import xin.merlin.myplayerbackend.entity.http.PlayroomDetails;
import xin.merlin.myplayerbackend.mapper.GroupMapper;
import xin.merlin.myplayerbackend.mapper.GroupsMapper;
import xin.merlin.myplayerbackend.mapper.UserMapper;

20
src/main/java/xin/merlin/myplayerbackend/service/impl/InvitingServiceImpl.java
View File

@@ -94,16 +94,16 @@ public class InvitingServiceImpl extends ServiceImpl<InvitingMapper, Inviting> {
}
public Integer playroomIsAdmin(Integer id, Integer r_id) {
try {
Playrooms playrooms = playroomsMapper.selectOne(Wrappers.<Playrooms>lambdaQuery().eq(Playrooms::getId,id).eq(Playrooms::getR_id,r_id));
if (playrooms == null) return 1;
else return playrooms.getRole();
} catch (Exception e) {
log.error(e.getMessage());
throw new RuntimeException(e);
}
}
// public Integer playroomIsAdmin(Integer id, Integer r_id) {
// try {
// Playrooms playrooms = playroomsMapper.selectOne(Wrappers.<Playrooms>lambdaQuery().eq(Playrooms::getId,id).eq(Playrooms::getR_id,r_id));
// if (playrooms == null) return 1;
// else return playrooms.getRole();
// } catch (Exception e) {
// log.error(e.getMessage());
// throw new RuntimeException(e);
// }
// }
public Boolean handlePlayroomInviting(Inviting inviting) {
try {

1
src/main/java/xin/merlin/myplayerbackend/service/impl/PlayroomServiceImpl.java
View File

@@ -4,7 +4,6 @@ import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.RequiredArgsConstructor;
import org.apache.catalina.User;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import xin.merlin.myplayerbackend.entity.PlayroomInfo;

58
src/main/java/xin/merlin/myplayerbackend/utils/websocket/CustomWebSocketHandler.java
View File

@@ -2,13 +2,19 @@ package xin.merlin.myplayerbackend.utils.websocket;
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONObject;
import jakarta.annotation.PreDestroy;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.context.event.ContextClosedEvent;
import org.springframework.context.event.EventListener;
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;
import xin.merlin.myplayerbackend.config.RabbitMQConfig;
import xin.merlin.myplayerbackend.service.OnlineStatusService;
@Slf4j
@@ -18,9 +24,16 @@ public class CustomWebSocketHandler extends TextWebSocketHandler {
private final WebSocketSessionManager webSocketSessionManager;
private final RabbitTemplate rabbitTemplate;
private final OnlineStatusService onlineStatusService;
@Override
public void afterConnectionEstablished(WebSocketSession session) throws Exception {
Integer userId = (Integer) session.getAttributes().get("userId");
onlineStatusService.online(userId);
webSocketSessionManager.addSession(userId, session);
log.info("用户 {} 已连接", userId);
}
@@ -28,29 +41,42 @@ public class CustomWebSocketHandler extends TextWebSocketHandler {
@Override
protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
String payload = message.getPayload();
JSONObject msg = JSON.parseObject(payload);
String type = msg.getString("type");
switch (type) {
case "chat":
Integer toUser = msg.getInteger("toUser");
webSocketSessionManager.sendToUser(toUser, payload);
break;
rabbitTemplate.convertAndSend(RabbitMQConfig.WS_MESSAGE_QUEUE, payload);
case "broadcast":
webSocketSessionManager.broadcast(payload);
break;
case "signal":
webSocketSessionManager.sendToUser(msg.getInteger("toUser"), payload);
break;
}
// JSONObject msg = JSON.parseObject(payload);
// String type = msg.getString("type");
// switch (type) {
// case "chat":
// Integer toUser = msg.getInteger("toUser");
// webSocketSessionManager.sendToUser(toUser, payload);
// break;
//
// case "broadcast":
// webSocketSessionManager.broadcast(payload);
// break;
//
// case "signal":
// webSocketSessionManager.sendToUser(msg.getInteger("toUser"), payload);
// break;
//
// }
}
@Override
public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
Integer userId = (Integer) session.getAttributes().get("userId");
onlineStatusService.offline(userId);
webSocketSessionManager.removeSession(userId);
log.info("用户 {} 已断开", userId);
}
@EventListener
public void onShutdown(ContextClosedEvent event) {
log.info("Shutting down... closing websocket sessions");
webSocketSessionManager.closeAll();
}
}

42
src/main/java/xin/merlin/myplayerbackend/utils/websocket/WebSocketMessageConsumer.java Normal file
View File

@@ -0,0 +1,42 @@
package xin.merlin.myplayerbackend.utils.websocket;
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONObject;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.stereotype.Component;
import xin.merlin.myplayerbackend.config.RabbitMQConfig;
import xin.merlin.myplayerbackend.utils.websocket.command.CommandDispatcher;
@Slf4j
@Component
@RequiredArgsConstructor
public class WebSocketMessageConsumer {
private final WebSocketSessionManager sessionManager;
private final CommandDispatcher commandDispatcher;
@RabbitListener(queues = RabbitMQConfig.WS_MESSAGE_QUEUE)
public void onMessage(String json) {
try {
JSONObject msg = JSON.parseObject(json);
String to = msg.getString("to");
if (to == null || to.isEmpty()) {
// 这是广播 / 系统消息
sessionManager.broadcast(json);
} else {
// 单发消息
// sessionManager.sendToUser(Integer.valueOf(to), json);
commandDispatcher.dispatch(msg);
}
} catch (Exception e) {
log.info(e.getMessage());
}
}
}

8
src/main/java/xin/merlin/myplayerbackend/utils/websocket/WebSocketSessionManager.java
View File

@@ -2,6 +2,7 @@ package xin.merlin.myplayerbackend.utils.websocket;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
@@ -37,5 +38,12 @@ public class WebSocketSessionManager {
}
}
public void closeAll() {
websocketSessions.forEach((uid, session) -> {
try {
session.close(CloseStatus.GOING_AWAY);
} catch (Exception ignored) {}
});
}
}

10
src/main/java/xin/merlin/myplayerbackend/utils/websocket/command/BaseCommandHandler.java Normal file
View File

@@ -0,0 +1,10 @@
package xin.merlin.myplayerbackend.utils.websocket.command;
import com.alibaba.fastjson2.JSONObject;
import java.io.IOException;
public interface BaseCommandHandler {
void handle(JSONObject msg) throws IOException;
}

34
src/main/java/xin/merlin/myplayerbackend/utils/websocket/command/CommandDispatcher.java Normal file
View File

@@ -0,0 +1,34 @@
package xin.merlin.myplayerbackend.utils.websocket.command;
import com.alibaba.fastjson2.JSONObject;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;
import xin.merlin.myplayerbackend.utils.websocket.command.impl.*;
import java.io.IOException;
@Component
@RequiredArgsConstructor
public class CommandDispatcher {
private final Ping pingCommand;
private final Message messageCommand;
private final Typing typingCommand;
private final Heartbeat heartbeatCommand;
private final SystemNotify systemNotifyCommand;
public void dispatch(JSONObject msg) throws IOException {
String cmd = msg.getString("cmd");
switch (cmd) {
case "PING" -> pingCommand.handle(msg);
case "MESSAGE" -> messageCommand.handle(msg);
case "TYPING" -> typingCommand.handle(msg);
case "HEARTBEAT" -> heartbeatCommand.handle(msg);
case "SYSTEM_NOTIFY" -> systemNotifyCommand.handle(msg);
default -> {
System.err.println("Unknown command: " + cmd);
}
}
}
}

22
src/main/java/xin/merlin/myplayerbackend/utils/websocket/command/impl/Heartbeat.java Normal file
View File

@@ -0,0 +1,22 @@
package xin.merlin.myplayerbackend.utils.websocket.command.impl;
import com.alibaba.fastjson2.JSONObject;
import lombok.RequiredArgsConstructor;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;
import xin.merlin.myplayerbackend.utils.websocket.command.BaseCommandHandler;
import java.time.Duration;
@Component
@RequiredArgsConstructor
public class Heartbeat implements BaseCommandHandler {
private final RedisTemplate<String, String> redis;
@Override
public void handle(JSONObject msg) {
String userId = msg.getString("from");
redis.opsForValue().set("online:" + userId, "1", Duration.ofMinutes(5));
}
}

23
src/main/java/xin/merlin/myplayerbackend/utils/websocket/command/impl/Message.java Normal file
View File

@@ -0,0 +1,23 @@
package xin.merlin.myplayerbackend.utils.websocket.command.impl;
import com.alibaba.fastjson2.JSONObject;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;
import xin.merlin.myplayerbackend.utils.websocket.WebSocketSessionManager;
import xin.merlin.myplayerbackend.utils.websocket.command.BaseCommandHandler;
import java.io.IOException;
@Component
@RequiredArgsConstructor
public class Message implements BaseCommandHandler {
private final WebSocketSessionManager sessionManager;
@Override
public void handle(JSONObject msg) throws IOException {
String to = msg.getString("to");
sessionManager.sendToUser(Integer.valueOf(to), msg.toJSONString());
}
}

24
src/main/java/xin/merlin/myplayerbackend/utils/websocket/command/impl/Ping.java Normal file
View File

@@ -0,0 +1,24 @@
package xin.merlin.myplayerbackend.utils.websocket.command.impl;
import com.alibaba.fastjson2.JSONObject;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;
import xin.merlin.myplayerbackend.utils.websocket.WebSocketSessionManager;
import xin.merlin.myplayerbackend.utils.websocket.command.BaseCommandHandler;
import java.io.IOException;
@Component
@RequiredArgsConstructor
public class Ping implements BaseCommandHandler {
private final WebSocketSessionManager sessionManager;
@Override
public void handle(JSONObject msg) throws IOException {
String from = msg.getString("from");
msg.put("cmd", "PONG");
sessionManager.sendToUser(Integer.valueOf(from), msg.toJSONString());
}
}

22
src/main/java/xin/merlin/myplayerbackend/utils/websocket/command/impl/SystemNotify.java Normal file
View File

@@ -0,0 +1,22 @@
package xin.merlin.myplayerbackend.utils.websocket.command.impl;
import com.alibaba.fastjson2.JSONObject;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;
import xin.merlin.myplayerbackend.utils.websocket.WebSocketSessionManager;
import xin.merlin.myplayerbackend.utils.websocket.command.BaseCommandHandler;
import java.io.IOException;
@Component
@RequiredArgsConstructor
public class SystemNotify implements BaseCommandHandler {
private final WebSocketSessionManager sessionManager;
@Override
public void handle(JSONObject msg) throws IOException {
sessionManager.broadcast(msg.toJSONString());
}
}

22
src/main/java/xin/merlin/myplayerbackend/utils/websocket/command/impl/Typing.java Normal file
View File

@@ -0,0 +1,22 @@
package xin.merlin.myplayerbackend.utils.websocket.command.impl;
import com.alibaba.fastjson2.JSONObject;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;
import xin.merlin.myplayerbackend.utils.websocket.WebSocketSessionManager;
import xin.merlin.myplayerbackend.utils.websocket.command.BaseCommandHandler;
import java.io.IOException;
@Component
@RequiredArgsConstructor
public class Typing implements BaseCommandHandler {
private final WebSocketSessionManager sessionManager;
@Override
public void handle(JSONObject msg) throws IOException {
String to = msg.getString("to");
sessionManager.sendToUser(Integer.valueOf(to), msg.toJSONString());
}
}