refactor: 完善websocket测试用例
This commit is contained in:
@@ -1,5 +1,6 @@
|
|||||||
package com.youlai.system.controller.demo;
|
package com.youlai.system.websocket;
|
||||||
|
|
||||||
|
import com.youlai.system.websocket.dto.MessageDTO;
|
||||||
import lombok.RequiredArgsConstructor;
|
import lombok.RequiredArgsConstructor;
|
||||||
import lombok.extern.slf4j.Slf4j;
|
import lombok.extern.slf4j.Slf4j;
|
||||||
import org.springframework.messaging.handler.annotation.DestinationVariable;
|
import org.springframework.messaging.handler.annotation.DestinationVariable;
|
||||||
@@ -49,8 +50,8 @@ public class WebsocketController {
|
|||||||
@MessageMapping("/sendToUser/{username}")
|
@MessageMapping("/sendToUser/{username}")
|
||||||
//@SendToUser(value = "/queue/greeting")
|
//@SendToUser(value = "/queue/greeting")
|
||||||
public void sendToUser(Principal principal, @DestinationVariable String username, String message) {
|
public void sendToUser(Principal principal, @DestinationVariable String username, String message) {
|
||||||
log.info("sender:{};receiver:{}", username, principal.getName());
|
log.info("sender:{};receiver:{}", principal.getName(), username);
|
||||||
messagingTemplate.convertAndSendToUser(username, "/queue/greeting", "服务端点对点:" + message);
|
messagingTemplate.convertAndSendToUser(username, "/queue/greeting", new MessageDTO(principal.getName(), message));
|
||||||
/// return "Hello, " + message;
|
/// return "Hello, " + message;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -0,0 +1,16 @@
|
|||||||
|
package com.youlai.system.websocket.dto;
|
||||||
|
|
||||||
|
import lombok.AllArgsConstructor;
|
||||||
|
import lombok.Data;
|
||||||
|
import lombok.NoArgsConstructor;
|
||||||
|
|
||||||
|
@Data
|
||||||
|
@AllArgsConstructor
|
||||||
|
@NoArgsConstructor
|
||||||
|
public class MessageDTO {
|
||||||
|
|
||||||
|
private String sender;
|
||||||
|
|
||||||
|
private String content;
|
||||||
|
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user