refactor: 完善websocket配置和测试用例

This commit is contained in:
haoxr
2023-09-07 18:42:18 +08:00
parent 09bfa1876d
commit a7a4175a57

View File

@@ -0,0 +1,58 @@
package com.youlai.system.controller.demo;
import com.youlai.system.common.result.Result;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.messaging.simp.SimpMessagingTemplate;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
/**
* WebSocket 测试控制器
*
* @author haoxr
* @since 2.3.0
*/
@RestController
@RequestMapping("/websocket")
@RequiredArgsConstructor
@Slf4j
public class WebsocketController {
private final SimpMessagingTemplate messagingTemplate;
/**
* 广播发送消息
*
* @param message 消息内容
*/
@PostMapping("/sendToAll")
// @SendTo("/topic/all") // 使用SimpMessagingTemplate发送消息给所有订阅了"/topic/all"目标的客户端
public void sendToAll(String message) {
log.info("【广播消息请求接收】消息:{}", message);
// 处理接收到的消息逻辑
// ...
// 构造要发送的消息内容
String content = "服务端广播消息: " + message;
// 使用SimpMessagingTemplate发送消息给所有订阅了"/topic/all"目标的客户端
messagingTemplate.convertAndSend("/topic/all", content);
}
/**
* 点对点发送消息
*
* @param username 用户名
*/
@PostMapping("/sendToUser/{username}")
public Result sendToUser(@PathVariable String username) {
// 发送主题目的(destination)= /user/{userId}/message
messagingTemplate.convertAndSendToUser(username, "/queue/user", "hello "+username);
return Result.success("点对点消息发送成功");
}
}