Commit 5086e1225f55d18491b04b8f5c8638bde4c710c7
1 parent
d6e33484
review:【iot】“修复了一些 Iot 模块 TODO 提到的问题”
Showing
5 changed files
with
10 additions
and
7 deletions
yudao-module-iot/yudao-module-iot-biz/src/main/java/cn/iocoder/yudao/module/iot/service/rule/data/action/IotWebSocketDataRuleAction.java
| 1 | 1 | package cn.iocoder.yudao.module.iot.service.rule.data.action; |
| 2 | 2 | |
| 3 | +import cn.hutool.core.util.StrUtil; | |
| 3 | 4 | import cn.iocoder.yudao.module.iot.core.mq.message.IotDeviceMessage; |
| 4 | 5 | import cn.iocoder.yudao.module.iot.dal.dataobject.rule.config.IotDataSinkWebSocketConfig; |
| 5 | 6 | import cn.iocoder.yudao.module.iot.enums.rule.IotDataSinkTypeEnum; |
| ... | ... | @@ -28,11 +29,11 @@ public class IotWebSocketDataRuleAction extends |
| 28 | 29 | |
| 29 | 30 | @Override |
| 30 | 31 | protected IotWebSocketClient initProducer(IotDataSinkWebSocketConfig config) throws Exception { |
| 31 | - // 1.1 参数校验 | |
| 32 | - if (config.getServerUrl() == null || config.getServerUrl().trim().isEmpty()) { | |
| 32 | + // 1. 参数校验 | |
| 33 | + if (StrUtil.isBlank(config.getServerUrl())) { | |
| 33 | 34 | throw new IllegalArgumentException("WebSocket 服务器地址不能为空"); |
| 34 | 35 | } |
| 35 | - if (!config.getServerUrl().startsWith("ws://") && !config.getServerUrl().startsWith("wss://")) { | |
| 36 | + if (!StrUtil.startWithAny(config.getServerUrl(), "ws://", "wss://")) { | |
| 36 | 37 | throw new IllegalArgumentException("WebSocket 服务器地址必须以 ws:// 或 wss:// 开头"); |
| 37 | 38 | } |
| 38 | 39 | |
| ... | ... | @@ -61,6 +62,7 @@ public class IotWebSocketDataRuleAction extends |
| 61 | 62 | protected void execute(IotDeviceMessage message, IotDataSinkWebSocketConfig config) throws Exception { |
| 62 | 63 | try { |
| 63 | 64 | // 1.1 获取或创建 WebSocket 客户端 |
| 65 | + // TODO @puhui999:需要加锁,保证必须连接上; | |
| 64 | 66 | IotWebSocketClient webSocketClient = getProducer(config); |
| 65 | 67 | // 1.2 检查连接状态,如果断开则重新连接 |
| 66 | 68 | if (!webSocketClient.isConnected()) { | ... | ... |
yudao-module-iot/yudao-module-iot-biz/src/main/java/cn/iocoder/yudao/module/iot/service/rule/data/action/tcp/IotTcpClient.java
| ... | ... | @@ -31,6 +31,7 @@ public class IotTcpClient { |
| 31 | 31 | private final Integer connectTimeoutMs; |
| 32 | 32 | private final Integer readTimeoutMs; |
| 33 | 33 | private final Boolean ssl; |
| 34 | + // TODO @puhui999:sslCertPath 是不是没在用? | |
| 34 | 35 | private final String sslCertPath; |
| 35 | 36 | private final String dataFormat; |
| 36 | 37 | |
| ... | ... | @@ -47,6 +48,7 @@ public class IotTcpClient { |
| 47 | 48 | this.readTimeoutMs = readTimeoutMs != null ? readTimeoutMs : IotDataSinkTcpConfig.DEFAULT_READ_TIMEOUT_MS; |
| 48 | 49 | this.ssl = ssl != null ? ssl : IotDataSinkTcpConfig.DEFAULT_SSL; |
| 49 | 50 | this.sslCertPath = sslCertPath; |
| 51 | + // TODO @puhui999:可以使用 StrUtil.defaultIfBlank 方法简化 | |
| 50 | 52 | this.dataFormat = dataFormat != null ? dataFormat : IotDataSinkTcpConfig.DEFAULT_DATA_FORMAT; |
| 51 | 53 | } |
| 52 | 54 | ... | ... |
yudao-module-iot/yudao-module-iot-biz/src/main/java/cn/iocoder/yudao/module/iot/service/rule/data/action/websocket/IotWebSocketClient.java
| ... | ... | @@ -45,6 +45,7 @@ public class IotWebSocketClient implements WebSocket.Listener { |
| 45 | 45 | /** |
| 46 | 46 | * 连接到 WebSocket 服务器 |
| 47 | 47 | */ |
| 48 | + @SuppressWarnings("resource") | |
| 48 | 49 | public void connect() throws Exception { |
| 49 | 50 | if (connected.get()) { |
| 50 | 51 | log.warn("[connect][WebSocket 客户端已经连接,无需重复连接]"); | ... | ... |
yudao-module-iot/yudao-module-iot-biz/src/main/java/cn/iocoder/yudao/module/iot/service/rule/scene/IotSceneRuleServiceImpl.java
| ... | ... | @@ -406,10 +406,7 @@ public class IotSceneRuleServiceImpl implements IotSceneRuleService { |
| 406 | 406 | */ |
| 407 | 407 | private void updateLastTriggerTime(Long id) { |
| 408 | 408 | try { |
| 409 | - IotSceneRuleDO updateObj = new IotSceneRuleDO() | |
| 410 | - .setId(id) | |
| 411 | - .setLastTriggerTime(LocalDateTime.now()); | |
| 412 | - sceneRuleMapper.updateById(updateObj); | |
| 409 | + sceneRuleMapper.updateById(new IotSceneRuleDO().setId(id).setLastTriggerTime(LocalDateTime.now())); | |
| 413 | 410 | } catch (Exception e) { |
| 414 | 411 | log.error("[updateLastTriggerTime][规则场景编号({}) 更新最后触发时间异常]", id, e); |
| 415 | 412 | } | ... | ... |
yudao-module-iot/yudao-module-iot-biz/src/main/java/cn/iocoder/yudao/module/iot/service/rule/scene/matcher/trigger/IotDevicePropertyPostTriggerMatcher.java
| ... | ... | @@ -38,6 +38,7 @@ public class IotDevicePropertyPostTriggerMatcher implements IotSceneRuleTriggerM |
| 38 | 38 | |
| 39 | 39 | // 1.3 检查消息中是否包含触发器指定的属性标识符 |
| 40 | 40 | // 注意:属性上报可能同时上报多个属性,所以需要判断 trigger.getIdentifier() 是否在 message 的 params 中 |
| 41 | + // TODO @puhui999:可以考虑 notXXX 方法,简化代码(尽量取反) | |
| 41 | 42 | if (!IotDeviceMessageUtils.containsIdentifier(message, trigger.getIdentifier())) { |
| 42 | 43 | IotSceneRuleMatcherHelper.logTriggerMatchFailure(message, trigger, "消息中不包含属性: " + |
| 43 | 44 | trigger.getIdentifier()); | ... | ... |