|
|
@@ -1,9 +1,28 @@
|
|
|
package cn.iocoder.yudao.module.iot.service.rule.action;
|
|
|
|
|
|
+import cn.hutool.core.collection.CollUtil;
|
|
|
+import cn.hutool.core.lang.Assert;
|
|
|
+import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
|
|
|
+import cn.iocoder.yudao.framework.common.util.http.HttpUtils;
|
|
|
+import cn.iocoder.yudao.framework.common.util.json.JsonUtils;
|
|
|
+import cn.iocoder.yudao.module.iot.dal.dataobject.rule.IotDataBridgeDO;
|
|
|
import cn.iocoder.yudao.module.iot.dal.dataobject.rule.IotRuleSceneDO;
|
|
|
+import cn.iocoder.yudao.module.iot.enums.rule.IotDataBridgTypeEnum;
|
|
|
import cn.iocoder.yudao.module.iot.enums.rule.IotRuleSceneActionTypeEnum;
|
|
|
import cn.iocoder.yudao.module.iot.mq.message.IotDeviceMessage;
|
|
|
+import cn.iocoder.yudao.module.iot.service.rule.IotDataBridgeService;
|
|
|
+import com.fasterxml.jackson.databind.ObjectMapper;
|
|
|
+import jakarta.annotation.Resource;
|
|
|
+import lombok.extern.slf4j.Slf4j;
|
|
|
+import org.springframework.http.*;
|
|
|
import org.springframework.stereotype.Component;
|
|
|
+import org.springframework.web.client.RestTemplate;
|
|
|
+import org.springframework.web.util.UriComponentsBuilder;
|
|
|
+
|
|
|
+import java.util.HashMap;
|
|
|
+import java.util.Map;
|
|
|
+
|
|
|
+import static cn.iocoder.yudao.framework.web.core.util.WebFrameworkUtils.HEADER_TENANT_ID;
|
|
|
|
|
|
/**
|
|
|
* IoT 数据桥梁的 {@link IotRuleSceneAction} 实现类
|
|
|
@@ -11,11 +30,38 @@ import org.springframework.stereotype.Component;
|
|
|
* @author 芋道源码
|
|
|
*/
|
|
|
@Component
|
|
|
+@Slf4j
|
|
|
public class IotRuleSceneDataBridgeAction implements IotRuleSceneAction {
|
|
|
|
|
|
+ @Resource
|
|
|
+ private RestTemplate restTemplate;
|
|
|
+
|
|
|
+ @Resource
|
|
|
+ private IotDataBridgeService dataBridgeService;
|
|
|
+
|
|
|
+ private final ObjectMapper objectMapper = new ObjectMapper();
|
|
|
+
|
|
|
@Override
|
|
|
public void execute(IotDeviceMessage message, IotRuleSceneDO.ActionConfig config) {
|
|
|
- // TODO @芋艿：http
|
|
|
+ // 1. 获得数据桥梁
|
|
|
+ Assert.notNull(config.getDataBridgeId(), "数据桥梁编号不能为空");
|
|
|
+ IotDataBridgeDO dataBridge = dataBridgeService.getIotDataBridge(config.getDataBridgeId());
|
|
|
+ if (dataBridge == null || dataBridge.getConfig() == null) {
|
|
|
+ log.error("[execute][message({}) config({}) 对应的数据桥梁不存在]", message, config);
|
|
|
+ return;
|
|
|
+ }
|
|
|
+ if (CommonStatusEnum.isDisable(dataBridge.getStatus())) {
|
|
|
+ log.info("[execute][message({}) config({}) 对应的数据桥梁({}) 状态为禁用]", message, config, dataBridge);
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
+ // 2.1 执行 HTTP 请求
|
|
|
+ // TODO @芋艿：groovy 或者 javascript 实现数据的转换；可以考虑基于 hutool 的 ScriptUtil 做
|
|
|
+ if (IotDataBridgTypeEnum.HTTP.getType().equals(dataBridge.getType())) {
|
|
|
+ executeHttp(message, (IotDataBridgeDO.HttpConfig) dataBridge.getConfig());
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
// TODO @芋艿：mq-redis
|
|
|
// TODO @芋艿：mq-数据库
|
|
|
// TODO @芋艿：kafka
|
|
|
@@ -23,6 +69,7 @@ public class IotRuleSceneDataBridgeAction implements IotRuleSceneAction {
|
|
|
// TODO @芋艿：rabbitmq
|
|
|
// TODO @芋艿：mqtt
|
|
|
// TODO @芋艿：tcp
|
|
|
+ // TODO @芋艿：websocket
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
@@ -30,4 +77,54 @@ public class IotRuleSceneDataBridgeAction implements IotRuleSceneAction {
|
|
|
return IotRuleSceneActionTypeEnum.DATA_BRIDGE;
|
|
|
}
|
|
|
|
|
|
+ @SuppressWarnings({"unchecked", "deprecation"})
|
|
|
+ private void executeHttp(IotDeviceMessage message, IotDataBridgeDO.HttpConfig config) {
|
|
|
+ String url = null;
|
|
|
+ HttpMethod method = HttpMethod.valueOf(config.getMethod().toUpperCase());
|
|
|
+ HttpEntity<String> requestEntity = null;
|
|
|
+ ResponseEntity<String> responseEntity = null;
|
|
|
+ try {
|
|
|
+ // 1.1 构建 Header
|
|
|
+ HttpHeaders headers = new HttpHeaders();
|
|
|
+ if (CollUtil.isNotEmpty(config.getHeaders())) {
|
|
|
+ config.getHeaders().putAll(config.getHeaders());
|
|
|
+ }
|
|
|
+ headers.add(HEADER_TENANT_ID, message.getTenantId().toString());
|
|
|
+ // 1.2 构建 URL
|
|
|
+ UriComponentsBuilder uriBuilder = UriComponentsBuilder.fromUriString(config.getUrl());
|
|
|
+ if (CollUtil.isNotEmpty(config.getQuery())) {
|
|
|
+ config.getQuery().forEach(uriBuilder::queryParam);
|
|
|
+ }
|
|
|
+ // 1.3 构建请求体
|
|
|
+ if (method == HttpMethod.GET) {
|
|
|
+ uriBuilder.queryParam("message", HttpUtils.encodeUtf8(JsonUtils.toJsonString(message)));
|
|
|
+ url = uriBuilder.build().toUriString();
|
|
|
+ requestEntity = new HttpEntity<>(headers);
|
|
|
+ } else {
|
|
|
+ url = uriBuilder.build().toUriString();
|
|
|
+ Map<String, Object> requestBody = JsonUtils.parseObject(config.getBody(), Map.class);
|
|
|
+ if (requestBody == null) {
|
|
|
+ requestBody = new HashMap<>();
|
|
|
+ }
|
|
|
+ requestBody.put("message", message);
|
|
|
+ headers.add(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON_UTF8_VALUE);
|
|
|
+ requestEntity = new HttpEntity<>(JsonUtils.toJsonString(requestBody), headers);
|
|
|
+ }
|
|
|
+
|
|
|
+ // 2.1 发送请求
|
|
|
+ responseEntity = restTemplate.exchange(url, method, requestEntity, String.class);
|
|
|
+ // 2.2 记录日志
|
|
|
+ if (responseEntity.getStatusCode().is2xxSuccessful()) {
|
|
|
+ log.info("[executeHttp][message({}) config({}) url({}) method({}) requestEntity({}) 请求成功({})]",
|
|
|
+ message, config, url, method, requestEntity, responseEntity);
|
|
|
+ } else {
|
|
|
+ log.error("[executeHttp][message({}) config({}) url({}) method({}) requestEntity({}) 请求失败({})]",
|
|
|
+ message, config, url, method, requestEntity, responseEntity);
|
|
|
+ }
|
|
|
+ } catch (Exception e) {
|
|
|
+ log.error("[executeHttp][message({}) config({}) url({}) method({}) requestEntity({}) 请求异常({})]",
|
|
|
+ message, config, url, method, requestEntity, responseEntity, e);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
}
|
YunaiV