package com.qianwen.smartman.modules.cps.controller;
|
|
import com.alibaba.fastjson.JSONObject;
|
import com.google.common.collect.ImmutableMap;
|
import io.swagger.annotations.Api;
|
import java.util.Collections;
|
import com.qianwen.smartman.common.websocket.entity.FmsArea;
|
import com.qianwen.smartman.common.websocket.entity.FmsWorkstation;
|
import com.qianwen.smartman.common.websocket.fms.FmsVisualDataResponseMessage;
|
import com.qianwen.core.scanner.modular.stereotype.ApiResource;
|
import com.qianwen.core.tool.api.R;
|
import com.qianwen.core.tool.utils.SpringUtil;
|
import com.qianwen.core.websocket.distribute.MessageDO;
|
import com.qianwen.core.websocket.distribute.RedisMessageDistributor;
|
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;
|
|
@ApiResource({"smis/test"})
|
@Api(value = "测试websocket", tags = {"测试websocket"})
|
@RestController
|
public class WebSocketTestController {
|
@RequestMapping({"/send-message"})
|
public R<Boolean> sendMessage() {
|
RedisMessageDistributor messageDistributor = (RedisMessageDistributor) SpringUtil.getBean(RedisMessageDistributor.class);
|
MessageDO messageOne = new MessageDO();
|
messageOne.setNeedBroadcast(Boolean.FALSE);
|
JSONObject json = new JSONObject();
|
json.put("type", "realTimeData");
|
json.put("id", "111");
|
json.put("data", ImmutableMap.of("data", "hello realTimeData"));
|
messageOne.setMessageText(json.toJSONString());
|
messageDistributor.distribute(messageOne);
|
JSONObject typeTwo = new JSONObject();
|
typeTwo.put("type", "internalMessageData");
|
typeTwo.put("data", "hello internalMessageData");
|
MessageDO messageTwo = new MessageDO();
|
messageTwo.setNeedBroadcast(false);
|
messageTwo.setMessageText(typeTwo.toJSONString());
|
messageDistributor.distribute(messageTwo);
|
return R.success("");
|
}
|
|
@PostMapping({"/send-fms-area"})
|
public R<Boolean> sendFmsMessage(@RequestBody FmsArea fmsArea) {
|
FmsVisualDataResponseMessage responseMessage = new FmsVisualDataResponseMessage();
|
responseMessage.setFmsAreaList(Collections.singletonList(fmsArea));
|
responseMessage.setFmsAreaList(Collections.singletonList(fmsArea));
|
responseMessage.setFmsAreaList(Collections.singletonList(fmsArea));
|
responseMessage.setFmsAreaList(Collections.singletonList(fmsArea));
|
MessageDO messageDO = new MessageDO();
|
messageDO.setNeedBroadcast(true);
|
messageDO.setMessageText(JSONObject.toJSONString(responseMessage));
|
RedisMessageDistributor messageDistributor = (RedisMessageDistributor) SpringUtil.getBean(RedisMessageDistributor.class);
|
messageDistributor.distribute(messageDO);
|
return R.success("");
|
}
|
|
@PostMapping({"/send-fms-carry"})
|
public R<Boolean> sendFmsCarry(@RequestBody FmsWorkstation fmsCarry) {
|
FmsVisualDataResponseMessage responseMessage = new FmsVisualDataResponseMessage();
|
responseMessage.setFmsCarryList(Collections.singletonList(fmsCarry));
|
MessageDO messageDO = new MessageDO();
|
messageDO.setNeedBroadcast(false);
|
messageDO.setMessageText(JSONObject.toJSONString(responseMessage));
|
RedisMessageDistributor messageDistributor = (RedisMessageDistributor) SpringUtil.getBean(RedisMessageDistributor.class);
|
messageDistributor.distribute(messageDO);
|
return R.success("");
|
}
|
|
@PostMapping({"/send-fms-manual"})
|
public R<Boolean> sendFmsManual(@RequestBody FmsWorkstation station) {
|
FmsVisualDataResponseMessage responseMessage = new FmsVisualDataResponseMessage();
|
responseMessage.setFmsManualStationList(Collections.singletonList(station));
|
MessageDO messageDO = new MessageDO();
|
messageDO.setNeedBroadcast(false);
|
messageDO.setMessageText(JSONObject.toJSONString(responseMessage));
|
RedisMessageDistributor messageDistributor = (RedisMessageDistributor) SpringUtil.getBean(RedisMessageDistributor.class);
|
messageDistributor.distribute(messageDO);
|
return R.success("");
|
}
|
|
@PostMapping({"/send-fms-station"})
|
public R<Boolean> sendFmsStation(@RequestBody FmsWorkstation workstation) {
|
FmsVisualDataResponseMessage responseMessage = new FmsVisualDataResponseMessage();
|
responseMessage.setFmsWorkstationList(Collections.singletonList(workstation));
|
MessageDO messageDO = new MessageDO();
|
messageDO.setNeedBroadcast(false);
|
messageDO.setMessageText(JSONObject.toJSONString(responseMessage));
|
RedisMessageDistributor messageDistributor = (RedisMessageDistributor) SpringUtil.getBean(RedisMessageDistributor.class);
|
messageDistributor.distribute(messageDO);
|
return R.success("");
|
}
|
}
|
