diff --git a/src/main/java/com/qyft/gd/model/vo/WebsocketResult.java b/src/main/java/com/qyft/gd/model/vo/WebsocketResult.java index 62eaa2e..ed42fc0 100644 --- a/src/main/java/com/qyft/gd/model/vo/WebsocketResult.java +++ b/src/main/java/com/qyft/gd/model/vo/WebsocketResult.java @@ -1,5 +1,6 @@ package com.qyft.gd.model.vo; +import io.swagger.v3.oas.annotations.media.Schema; import lombok.Data; @Data @@ -7,9 +8,11 @@ public class WebsocketResult { /** * 推送类型(指令 cmd,报警 warn ,状态 status) */ + @Schema(description = "推送类型(指令 cmd,报警 warn ,状态 status)") private String type; /** * 执行结果 */ + @Schema(description = "推送数据") private Object data; } \ No newline at end of file diff --git a/src/main/java/com/qyft/gd/service/CMDService.java b/src/main/java/com/qyft/gd/service/CMDService.java index 9766b05..1c2b3e2 100644 --- a/src/main/java/com/qyft/gd/service/CMDService.java +++ b/src/main/java/com/qyft/gd/service/CMDService.java @@ -22,7 +22,6 @@ public class CMDService { DeviceService deviceService; DeviceStateService deviceStateService; Map> commandMap; - CMDForm form = new CMDForm(); @Autowired public CMDService(DeviceService deviceService) { @@ -164,28 +163,24 @@ public class CMDService { } public boolean executeCommand(CMDForm cmdForm) { - form = cmdForm; String commandName = cmdForm.getCommand(); Function command = commandMap.get(commandName); if (command == null) { return false; } - WebsocketResult websocketResult = new WebsocketResult(); - websocketResult.setType("warn"); - ExecutionResult executionResult = new ExecutionResult(); - executionResult.setCommandId(cmdForm.getCommandId()); - executionResult.setCommandName(cmdForm.getCommand()); - boolean result =command.apply(cmdForm); - if(result) { - executionResult.setStatus(CMDResultCode.SUCCESS.getCode()); - executionResult.setMessage(CMDResultCode.SUCCESS.getMsg()); - }else { - executionResult.setStatus(CMDResultCode.FAILURE.getCode()); - executionResult.setMessage(CMDResultCode.FAILURE.getMsg()); - } - websocketResult.setData(executionResult); - // 发送消息给客户端 - WebSocketServer.sendMessageToClients(JSONUtil.toJsonStr(websocketResult)); + command.apply(cmdForm); return true; + +// ExecutionResult executionResult = new ExecutionResult(); +// executionResult.setCommandId(cmdForm.getCommandId()); +// executionResult.setCommandName(cmdForm.getCommand()); +// boolean result =command.apply(cmdForm); +// if(result) { +// executionResult.setStatus(CMDResultCode.SUCCESS.getCode()); +// executionResult.setMessage(CMDResultCode.SUCCESS.getMsg()); +// }else { +// executionResult.setStatus(CMDResultCode.FAILURE.getCode()); +// executionResult.setMessage(CMDResultCode.FAILURE.getMsg()); +// } } } diff --git a/src/main/java/com/qyft/gd/service/warnService.java b/src/main/java/com/qyft/gd/service/WebSocketService.java similarity index 65% rename from src/main/java/com/qyft/gd/service/warnService.java rename to src/main/java/com/qyft/gd/service/WebSocketService.java index 5cbf5e5..be2b9f7 100644 --- a/src/main/java/com/qyft/gd/service/warnService.java +++ b/src/main/java/com/qyft/gd/service/WebSocketService.java @@ -2,17 +2,16 @@ package com.qyft.gd.service; import cn.hutool.json.JSONUtil; import com.qyft.gd.config.WebSocketServer; -import com.qyft.gd.device.model.bo.DeviceAlarm; import com.qyft.gd.model.vo.WebsocketResult; import org.springframework.stereotype.Service; @Service -public class warnService { - public boolean pushWarnMsg(DeviceAlarm deviceAlarm) { +public class WebSocketService { + public boolean pushMsg(String type, Object result) { WebsocketResult websocketResult = new WebsocketResult(); - websocketResult.setType("warn"); - websocketResult.setData(deviceAlarm); + websocketResult.setType(type); + websocketResult.setData(result); // 发送消息给客户端 WebSocketServer.sendMessageToClients(JSONUtil.toJsonStr(websocketResult)); return true;