package org.jeecg.modules.message.controller;
|
|
import org.jeecg.common.api.vo.Result;
|
import org.jeecg.common.constant.WebsocketConst;
|
import org.jeecg.modules.message.websocket.WebSocket;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.PostMapping;
|
import org.springframework.web.bind.annotation.RequestBody;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
|
import com.alibaba.fastjson.JSONObject;
|
|
/**
|
* @Description: TestSocketController
|
* @author: jeecg-boot
|
*/
|
@RestController
|
@RequestMapping("/sys/socketTest")
|
public class TestSocketController {
|
|
@Autowired
|
private WebSocket webSocket;
|
|
@PostMapping("/sendAll")
|
public Result<String> sendAll(@RequestBody JSONObject jsonObject) {
|
Result<String> result = new Result<String>();
|
String message = jsonObject.getString("message");
|
JSONObject obj = new JSONObject();
|
obj.put(WebsocketConst.MSG_CMD, WebsocketConst.CMD_TOPIC);
|
obj.put(WebsocketConst.MSG_ID, "M0001");
|
obj.put(WebsocketConst.MSG_TXT, message);
|
webSocket.sendMessage(obj.toJSONString());
|
result.setResult("群发！");
|
return result;
|
}
|
|
@PostMapping("/sendUser")
|
public Result<String> sendUser(@RequestBody JSONObject jsonObject) {
|
Result<String> result = new Result<String>();
|
String userId = jsonObject.getString("userId");
|
String message = jsonObject.getString("message");
|
JSONObject obj = new JSONObject();
|
obj.put(WebsocketConst.MSG_CMD, WebsocketConst.CMD_USER);
|
obj.put(WebsocketConst.MSG_USER_ID, userId);
|
obj.put(WebsocketConst.MSG_ID, "M0001");
|
obj.put(WebsocketConst.MSG_TXT, message);
|
webSocket.sendMessage(userId, obj.toJSONString());
|
result.setResult("单发");
|
return result;
|
}
|
|
}
|
