潘志宝
2024-11-21 d338b50afd6504a9676f0a26b3ecbcc844483e7c
提交 | 用户 | 时间
e7c126 1 package com.iailab.module.infra.api.websocket;
H 2
3 import com.iailab.framework.common.pojo.CommonResult;
4 import com.iailab.framework.common.util.json.JsonUtils;
5 import com.iailab.module.infra.api.websocket.dto.WebSocketSendReqDTO;
6 import com.iailab.module.infra.enums.ApiConstants;
7 import io.swagger.v3.oas.annotations.Operation;
8 import io.swagger.v3.oas.annotations.tags.Tag;
9 import org.springframework.cloud.openfeign.FeignClient;
10 import org.springframework.web.bind.annotation.PostMapping;
11 import org.springframework.web.bind.annotation.RequestBody;
12
13 import javax.validation.Valid;
14
15 @FeignClient(name = ApiConstants.NAME) // TODO iailab:fallbackFactory =
16 @Tag(name = "RPC 服务 - WebSocket 发送器的") // 对 WebSocketMessageSender 进行封装,提供给其它模块使用
17 public interface WebSocketSenderApi {
18
19     String PREFIX = ApiConstants.PREFIX + "/websocket";
20
21     @PostMapping(PREFIX + "/send")
22     @Operation(summary = "发送 WebSocket 消息")
23     CommonResult<Boolean> send(@Valid @RequestBody WebSocketSendReqDTO message);
24
25     /**
26      * 发送消息给指定用户
27      *
28      * @param userType 用户类型
29      * @param userId 用户编号
30      * @param messageType 消息类型
31      * @param messageContent 消息内容,JSON 格式
32      */
33     default void send(Integer userType, Long userId, String messageType, String messageContent) {
34         send(new WebSocketSendReqDTO().setUserType(userType).setUserId(userId)
35                 .setMessageType(messageType).setMessageContent(messageContent)).checkError();
36     }
37
38     /**
39      * 发送消息给指定用户类型
40      *
41      * @param userType 用户类型
42      * @param messageType 消息类型
43      * @param messageContent 消息内容,JSON 格式
44      */
45     default void send(Integer userType, String messageType, String messageContent) {
46         send(new WebSocketSendReqDTO().setUserType(userType)
47                 .setMessageType(messageType).setMessageContent(messageContent)).checkError();
48     }
49
50     /**
51      * 发送消息给指定 Session
52      *
53      * @param sessionId Session 编号
54      * @param messageType 消息类型
55      * @param messageContent 消息内容,JSON 格式
56      */
57     default void send(String sessionId, String messageType, String messageContent) {
58         send(new WebSocketSendReqDTO().setSessionId(sessionId)
59                 .setMessageType(messageType).setMessageContent(messageContent)).checkError();
60     }
61
62     default void sendObject(Integer userType, Long userId, String messageType, Object messageContent) {
63         send(userType, userId, messageType, JsonUtils.toJsonString(messageContent));
64     }
65
66     default void sendObject(Integer userType, String messageType, Object messageContent) {
67         send(userType, messageType, JsonUtils.toJsonString(messageContent));
68     }
69
70     default void sendObject(String sessionId, String messageType, Object messageContent) {
71         send(sessionId, messageType, JsonUtils.toJsonString(messageContent));
72     }
73
74 }