509 lines
20 KiB
Java
509 lines
20 KiB
Java
package com.ruoyi.device.domain.impl;
|
|
|
|
|
|
import com.ruoyi.device.config.DeviceCacheConfig;
|
|
import com.ruoyi.device.domain.api.IThingsBoardDomain;
|
|
import com.ruoyi.device.domain.model.thingsboard.*;
|
|
import com.ruoyi.device.domain.model.thingsboard.constants.DeviceAttributes;
|
|
import com.ruoyi.device.domain.model.thingsboard.constants.DeviceTelemetry;
|
|
import com.ruoyi.device.domain.model.thingsboard.tuoheng.constants.TuohengDeviceAttributes;
|
|
import com.ruoyi.device.domain.model.thingsboard.tuoheng.constants.TuohengDeviceTelemetry;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
import org.springframework.beans.factory.annotation.Value;
|
|
import org.springframework.cache.annotation.CacheEvict;
|
|
import org.springframework.cache.annotation.Cacheable;
|
|
import org.springframework.scheduling.annotation.Scheduled;
|
|
import org.springframework.stereotype.Component;
|
|
import org.thingsboard.rest.client.RestClient;
|
|
import org.thingsboard.server.common.data.Device;
|
|
import org.thingsboard.server.common.data.EntityType;
|
|
import org.thingsboard.server.common.data.id.DeviceId;
|
|
import org.thingsboard.server.common.data.id.EntityId;
|
|
import org.thingsboard.server.common.data.kv.AttributeKvEntry;
|
|
import org.thingsboard.server.common.data.kv.TsKvEntry;
|
|
import org.thingsboard.server.common.data.relation.EntityRelation;
|
|
import org.thingsboard.server.common.data.relation.EntityRelationInfo;
|
|
import org.thingsboard.server.common.data.relation.EntityRelationsQuery;
|
|
import org.thingsboard.server.common.data.relation.EntitySearchDirection;
|
|
import org.thingsboard.server.common.data.relation.RelationsSearchParameters;
|
|
import org.thingsboard.server.common.data.relation.RelationTypeGroup;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.Optional;
|
|
import java.util.UUID;
|
|
|
|
/**
|
|
* ThingsBoard设备服务实现类
|
|
*/
|
|
@Component
|
|
public class ThingsBoardDomainImpl implements IThingsBoardDomain {
|
|
|
|
|
|
|
|
private static final Logger log = LoggerFactory.getLogger(ThingsBoardDomainImpl.class);
|
|
|
|
private final RestClient client;
|
|
private final int pageSize;
|
|
|
|
/**
|
|
* 构造函数 - Spring 会自动装配
|
|
* @param clientManager RestClient 管理器
|
|
* @param pageSize 分页大小，从配置文件读取，默认值为 10
|
|
*/
|
|
public ThingsBoardDomainImpl(RestClientManager clientManager,
|
|
@Value("${thingsboard.page-size:10}") int pageSize) {
|
|
this.client = clientManager.getClient();
|
|
this.pageSize = pageSize;
|
|
}
|
|
|
|
@Override
|
|
public Iterable<List<DeviceInfo>> getAllDevices() {
|
|
return new DeviceIterator(client, pageSize);
|
|
}
|
|
|
|
@Override
|
|
public Iterable<List<DeviceInfo>> getAllGatewayDevices() {
|
|
return new GatewayDeviceIterator(client, pageSize);
|
|
}
|
|
|
|
@Override
|
|
public DeviceInfo getDeviceInfo(String deviceId) {
|
|
try {
|
|
DeviceId id = new DeviceId(UUID.fromString(deviceId));
|
|
Optional<Device> deviceOptional = client.getDeviceById(id);
|
|
if (deviceOptional.isEmpty()) {
|
|
log.warn("设备不存在: deviceId={}", deviceId);
|
|
return null;
|
|
}
|
|
|
|
return new DeviceInfo(
|
|
deviceOptional.get().getName(),
|
|
deviceOptional.get().getId().getId().toString(),
|
|
deviceOptional.get().getType(),
|
|
deviceOptional.get().getId(),
|
|
deviceOptional.get().getAdditionalInfo()
|
|
);
|
|
} catch (Exception e) {
|
|
log.error("获取设备信息失败: deviceId={}, error={}", deviceId, e.getMessage(), e);
|
|
return null;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
@Cacheable(value = DeviceCacheConfig.THINGSBOARD_ATTRIBUTES_CACHE, key = "#deviceId", unless = "#result == null || #result.isEmpty()")
|
|
public AttributeMap getDeviceAttributes(String deviceId) {
|
|
AttributeMap attributeMap = new AttributeMap();
|
|
|
|
try {
|
|
DeviceId deviceIdObj = new DeviceId(UUID.fromString(deviceId));
|
|
|
|
List<String> attributeKeys = client.getAttributeKeys(deviceIdObj);
|
|
if (attributeKeys == null || attributeKeys.isEmpty()) {
|
|
log.debug("设备 {} 没有属性", deviceId);
|
|
return attributeMap;
|
|
}
|
|
|
|
List<AttributeKvEntry> attributeKvEntries = client.getAttributeKvEntries(deviceIdObj, attributeKeys);
|
|
if (attributeKvEntries == null || attributeKvEntries.isEmpty()) {
|
|
log.debug("设备 {} 的属性值为空", deviceId);
|
|
return attributeMap;
|
|
}
|
|
|
|
for (AttributeKvEntry entry : attributeKvEntries) {
|
|
parseAndPutAttribute(attributeMap, entry);
|
|
}
|
|
|
|
} catch (Exception e) {
|
|
log.error("获取设备属性失败: deviceId={}, error={}", deviceId, e.getMessage(), e);
|
|
}
|
|
|
|
return attributeMap;
|
|
}
|
|
|
|
@Override
|
|
@Cacheable(value = DeviceCacheConfig.THINGSBOARD_TELEMETRY_CACHE, key = "#deviceId", unless = "#result == null || #result.isEmpty()")
|
|
public TelemetryMap getDeviceTelemetry(String deviceId) {
|
|
TelemetryMap telemetryMap = new TelemetryMap();
|
|
|
|
try {
|
|
DeviceId deviceIdObj = new DeviceId(UUID.fromString(deviceId));
|
|
|
|
List<String> timeseriesKeys = client.getTimeseriesKeys(deviceIdObj);
|
|
if (timeseriesKeys == null || timeseriesKeys.isEmpty()) {
|
|
log.debug("设备 {} 没有遥测数据", deviceId);
|
|
return telemetryMap;
|
|
}
|
|
|
|
List<TsKvEntry> latestTimeseries = client.getLatestTimeseries(deviceIdObj, timeseriesKeys);
|
|
if (latestTimeseries == null || latestTimeseries.isEmpty()) {
|
|
log.debug("设备 {} 的遥测数据为空", deviceId);
|
|
return telemetryMap;
|
|
}
|
|
|
|
for (TsKvEntry entry : latestTimeseries) {
|
|
parseAndPutTelemetry(telemetryMap, entry);
|
|
}
|
|
|
|
} catch (Exception e) {
|
|
log.error("获取设备遥测数据失败: deviceId={}, error={}", deviceId, e.getMessage(), e);
|
|
}
|
|
|
|
return telemetryMap;
|
|
}
|
|
|
|
|
|
@Override
|
|
public AttributeMap getPredefinedDeviceAttributes(String deviceId) {
|
|
// 先获取所有属性（已经处理了异常情况）
|
|
AttributeMap allAttributes = getDeviceAttributes(deviceId);
|
|
|
|
// 创建新的 AttributeMap 只包含预定义的键
|
|
AttributeMap predefinedAttributes = new AttributeMap();
|
|
|
|
// 获取预定义的键名称集合
|
|
List<String> predefinedKeyNames = DeviceAttributes.getPredefinedKeys()
|
|
.stream()
|
|
.map(AttributeKey::getName)
|
|
.toList();
|
|
|
|
// 过滤：只保留预定义的键
|
|
for (AttributeKey<?> key : allAttributes.keySet()) {
|
|
if (predefinedKeyNames.contains(key.getName())) {
|
|
// 复制到新的 map
|
|
allAttributes.get(key).ifPresent(value -> {
|
|
@SuppressWarnings("unchecked")
|
|
AttributeKey<Object> objKey = (AttributeKey<Object>) key;
|
|
predefinedAttributes.put(objKey, value);
|
|
});
|
|
}
|
|
}
|
|
|
|
return predefinedAttributes;
|
|
}
|
|
|
|
@Override
|
|
public TelemetryMap getPredefinedDeviceTelemetry(String deviceId) {
|
|
// 先获取所有遥测数据（已经处理了 null 值问题）
|
|
TelemetryMap allTelemetry = getDeviceTelemetry(deviceId);
|
|
|
|
// 创建新的 TelemetryMap 只包含预定义的键
|
|
TelemetryMap predefinedTelemetry = new TelemetryMap();
|
|
|
|
// 获取预定义的键名称集合
|
|
List<String> predefinedKeyNames = DeviceTelemetry.getPredefinedKeys()
|
|
.stream()
|
|
.map(TelemetryKey::getName)
|
|
.toList();
|
|
|
|
// 过滤：只保留预定义的键
|
|
for (TelemetryKey<?> key : allTelemetry.keySet()) {
|
|
if (predefinedKeyNames.contains(key.getName())) {
|
|
// 复制到新的 map
|
|
allTelemetry.get(key).ifPresent(telemetryValue -> {
|
|
@SuppressWarnings("unchecked")
|
|
TelemetryKey<Object> objKey = (TelemetryKey<Object>) key;
|
|
predefinedTelemetry.put(objKey, telemetryValue.getValue(), telemetryValue.getTimestamp());
|
|
});
|
|
}
|
|
}
|
|
|
|
return predefinedTelemetry;
|
|
}
|
|
|
|
@Override
|
|
@Deprecated
|
|
public String getDeviceGatewayId(String deviceId) {
|
|
try {
|
|
DeviceId id = new DeviceId(UUID.fromString(deviceId));
|
|
|
|
// 查询指向该设备的 "Contains" 关系（网关 -> 设备）
|
|
List<EntityRelation> relations = client.findByTo(
|
|
id,
|
|
EntityRelation.CONTAINS_TYPE,
|
|
RelationTypeGroup.COMMON
|
|
);
|
|
|
|
if (relations == null || relations.isEmpty()) {
|
|
log.debug("设备 {} 不属于任何网关", deviceId);
|
|
return null;
|
|
}
|
|
|
|
// 获取第一个关系的 from 实体（网关设备）
|
|
EntityId gatewayEntityId = relations.get(0).getFrom();
|
|
String gatewayId = gatewayEntityId.getId().toString();
|
|
|
|
log.debug("设备 {} 属于网关 {}", deviceId, gatewayId);
|
|
return gatewayId;
|
|
|
|
} catch (Exception e) {
|
|
log.error("获取设备网关关系失败: deviceId={}, error={}", deviceId, e.getMessage(), e);
|
|
return null;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public List<String> getGatewayChildDevices(String gatewayDeviceId) {
|
|
List<String> childDeviceIds = new ArrayList<>();
|
|
|
|
try {
|
|
|
|
EntityId gatewayId = new EntityId() {
|
|
@Override
|
|
public UUID getId() {
|
|
return UUID.fromString(gatewayDeviceId);
|
|
}
|
|
@Override
|
|
public EntityType getEntityType() {
|
|
return EntityType.DEVICE;
|
|
}
|
|
};
|
|
|
|
List<EntityRelationInfo> relationInfos = client.findInfoByFrom(
|
|
gatewayId,
|
|
RelationTypeGroup.COMMON
|
|
);
|
|
|
|
if (relationInfos == null || relationInfos.isEmpty()) {
|
|
log.debug("网关 {} 没有子设备", gatewayDeviceId);
|
|
return childDeviceIds;
|
|
}
|
|
|
|
// 提取所有子设备的ID
|
|
for (EntityRelationInfo relationInfo : relationInfos) {
|
|
EntityId childEntityId = relationInfo.getTo();
|
|
String childDeviceId = childEntityId.getId().toString();
|
|
childDeviceIds.add(childDeviceId);
|
|
}
|
|
|
|
log.debug("网关 {} 有 {} 个子设备", gatewayDeviceId, childDeviceIds.size());
|
|
return childDeviceIds;
|
|
|
|
} catch (Exception e) {
|
|
log.error("获取网关子设备失败: gatewayDeviceId={}, error={}", gatewayDeviceId, e.getMessage(), e);
|
|
return childDeviceIds;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 解析属性并添加到AttributeMap
|
|
* 使用延迟注册机制，自动处理所有属性
|
|
*/
|
|
@SuppressWarnings("unchecked")
|
|
private void parseAndPutAttribute(AttributeMap attributeMap, AttributeKvEntry entry) {
|
|
String keyName = entry.getKey();
|
|
Object value = entry.getValue();
|
|
|
|
try {
|
|
// 使用延迟注册机制：如果键不存在则自动创建
|
|
AttributeKey<?> key = AttributeKey.getOrCreate(keyName, value);
|
|
|
|
// 使用键的解析器解析值
|
|
Object parsedValue = ((AttributeKey<Object>) key).parse(value);
|
|
attributeMap.put((AttributeKey<Object>) key, parsedValue);
|
|
log.debug("成功解析属性: {} = {} (type: {})", keyName, parsedValue, key.getType().getSimpleName());
|
|
} catch (Exception e) {
|
|
log.warn("解析属性失败: key={}, value={}, error={}", keyName, value, e.getMessage());
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 解析遥测数据并添加到TelemetryMap
|
|
* 使用延迟注册机制，自动处理所有遥测数据
|
|
*/
|
|
@SuppressWarnings("unchecked")
|
|
private void parseAndPutTelemetry(TelemetryMap telemetryMap, TsKvEntry entry) {
|
|
String keyName = entry.getKey();
|
|
Object value = entry.getValue();
|
|
|
|
try {
|
|
// 使用延迟注册机制：如果键不存在则自动创建
|
|
TelemetryKey<?> key = TelemetryKey.getOrCreate(keyName, value);
|
|
|
|
// 使用键的解析器解析值
|
|
Object parsedValue = ((TelemetryKey<Object>) key).parse(value);
|
|
telemetryMap.put((TelemetryKey<Object>) key, parsedValue, entry.getTs());
|
|
log.debug("成功解析遥测数据: {} = {} (timestamp: {}, type: {})",
|
|
keyName, parsedValue, entry.getTs(), key.getType().getSimpleName());
|
|
} catch (Exception e) {
|
|
log.warn("解析遥测数据失败: key={}, value={}, error={}", keyName, value, e.getMessage());
|
|
}
|
|
}
|
|
|
|
@Override
|
|
@CacheEvict(value = DeviceCacheConfig.THINGSBOARD_ATTRIBUTES_CACHE, key = "#deviceId")
|
|
public void evictDeviceAttributesCache(String deviceId) {
|
|
// 空实现，仅用于清除缓存
|
|
}
|
|
|
|
@Override
|
|
@CacheEvict(value = DeviceCacheConfig.THINGSBOARD_TELEMETRY_CACHE, key = "#deviceId")
|
|
public void evictDeviceTelemetryCache(String deviceId) {
|
|
// 空实现，仅用于清除缓存
|
|
}
|
|
|
|
@Override
|
|
@Cacheable(value = DeviceCacheConfig.THINGSBOARD_ATTRIBUTES_CACHE, key = "#deviceId", unless = "#result == null || #result.isEmpty()")
|
|
public AttributeMap getTuohengDeviceAttributes(String deviceId) {
|
|
AttributeMap attributeMap = new AttributeMap();
|
|
|
|
try {
|
|
DeviceId deviceIdObj = new DeviceId(UUID.fromString(deviceId));
|
|
|
|
List<String> attributeKeys = client.getAttributeKeys(deviceIdObj);
|
|
if (attributeKeys == null || attributeKeys.isEmpty()) {
|
|
log.debug("拓恒设备 {} 没有属性", deviceId);
|
|
return attributeMap;
|
|
}
|
|
|
|
List<AttributeKvEntry> attributeKvEntries = client.getAttributeKvEntries(deviceIdObj, attributeKeys);
|
|
if (attributeKvEntries == null || attributeKvEntries.isEmpty()) {
|
|
log.debug("拓恒设备 {} 的属性值为空", deviceId);
|
|
return attributeMap;
|
|
}
|
|
|
|
for (AttributeKvEntry entry : attributeKvEntries) {
|
|
parseAndPutAttribute(attributeMap, entry);
|
|
}
|
|
|
|
} catch (Exception e) {
|
|
log.error("获取拓恒设备属性失败: deviceId={}, error={}", deviceId, e.getMessage(), e);
|
|
}
|
|
|
|
return attributeMap;
|
|
}
|
|
|
|
@Override
|
|
@Cacheable(value = DeviceCacheConfig.THINGSBOARD_TELEMETRY_CACHE, key = "#deviceId", unless = "#result == null || #result.isEmpty()")
|
|
public TelemetryMap getTuohengDeviceTelemetry(String deviceId) {
|
|
TelemetryMap telemetryMap = new TelemetryMap();
|
|
|
|
try {
|
|
DeviceId deviceIdObj = new DeviceId(UUID.fromString(deviceId));
|
|
|
|
List<String> timeseriesKeys = client.getTimeseriesKeys(deviceIdObj);
|
|
if (timeseriesKeys == null || timeseriesKeys.isEmpty()) {
|
|
log.debug("拓恒设备 {} 没有遥测数据", deviceId);
|
|
return telemetryMap;
|
|
}
|
|
|
|
List<TsKvEntry> latestTimeseries = client.getLatestTimeseries(deviceIdObj, timeseriesKeys);
|
|
if (latestTimeseries == null || latestTimeseries.isEmpty()) {
|
|
log.debug("拓恒设备 {} 的遥测数据为空", deviceId);
|
|
return telemetryMap;
|
|
}
|
|
|
|
for (TsKvEntry entry : latestTimeseries) {
|
|
parseAndPutTelemetry(telemetryMap, entry);
|
|
}
|
|
|
|
} catch (Exception e) {
|
|
log.error("获取拓恒设备遥测数据失败: deviceId={}, error={}", deviceId, e.getMessage(), e);
|
|
}
|
|
|
|
return telemetryMap;
|
|
}
|
|
|
|
@Override
|
|
public AttributeMap getPredefinedTuohengDeviceAttributes(String deviceId) {
|
|
// 先获取所有属性（已经处理了异常情况）
|
|
AttributeMap allAttributes = getTuohengDeviceAttributes(deviceId);
|
|
|
|
// 创建新的 AttributeMap 只包含预定义的键
|
|
AttributeMap predefinedAttributes = new AttributeMap();
|
|
|
|
// 获取预定义的键名称集合
|
|
List<String> predefinedKeyNames = TuohengDeviceAttributes.getPredefinedKeys()
|
|
.stream()
|
|
.map(AttributeKey::getName)
|
|
.toList();
|
|
|
|
// 过滤：只保留预定义的键
|
|
for (AttributeKey<?> key : allAttributes.keySet()) {
|
|
if (predefinedKeyNames.contains(key.getName())) {
|
|
// 复制到新的 map
|
|
allAttributes.get(key).ifPresent(value -> {
|
|
@SuppressWarnings("unchecked")
|
|
AttributeKey<Object> objKey = (AttributeKey<Object>) key;
|
|
predefinedAttributes.put(objKey, value);
|
|
});
|
|
}
|
|
}
|
|
|
|
return predefinedAttributes;
|
|
}
|
|
|
|
@Override
|
|
public TelemetryMap getPredefinedTuohengDeviceTelemetry(String deviceId) {
|
|
// 先获取所有遥测数据（已经处理了 null 值问题）
|
|
TelemetryMap allTelemetry = getTuohengDeviceTelemetry(deviceId);
|
|
|
|
// 创建新的 TelemetryMap 只包含预定义的键
|
|
TelemetryMap predefinedTelemetry = new TelemetryMap();
|
|
|
|
// 获取预定义的键名称集合
|
|
List<String> predefinedKeyNames = TuohengDeviceTelemetry.getPredefinedKeys()
|
|
.stream()
|
|
.map(TelemetryKey::getName)
|
|
.toList();
|
|
|
|
// 过滤：只保留预定义的键
|
|
for (TelemetryKey<?> key : allTelemetry.keySet()) {
|
|
if (predefinedKeyNames.contains(key.getName())) {
|
|
// 复制到新的 map
|
|
allTelemetry.get(key).ifPresent(telemetryValue -> {
|
|
@SuppressWarnings("unchecked")
|
|
TelemetryKey<Object> objKey = (TelemetryKey<Object>) key;
|
|
predefinedTelemetry.put(objKey, telemetryValue.getValue(), telemetryValue.getTimestamp());
|
|
});
|
|
}
|
|
}
|
|
|
|
return predefinedTelemetry;
|
|
}
|
|
|
|
@Override
|
|
@CacheEvict(value = DeviceCacheConfig.THINGSBOARD_ATTRIBUTES_CACHE,
|
|
allEntries = false,
|
|
key = "#deviceId")
|
|
public boolean setDeviceAttribute(String deviceId, String key, Object value) {
|
|
try {
|
|
log.info("设置设备属性: deviceId={}, key={}, value={}", deviceId, key, value);
|
|
|
|
// 将 deviceId 字符串转换为 DeviceId 对象
|
|
DeviceId deviceIdObj = new DeviceId(UUID.fromString(deviceId));
|
|
|
|
// 构建 JsonNode 对象
|
|
com.fasterxml.jackson.databind.ObjectMapper mapper = new com.fasterxml.jackson.databind.ObjectMapper();
|
|
com.fasterxml.jackson.databind.node.ObjectNode jsonNode = mapper.createObjectNode();
|
|
|
|
// 根据值的类型设置到 JsonNode
|
|
if (value instanceof String) {
|
|
jsonNode.put(key, (String) value);
|
|
} else if (value instanceof Integer) {
|
|
jsonNode.put(key, (Integer) value);
|
|
} else if (value instanceof Long) {
|
|
jsonNode.put(key, (Long) value);
|
|
} else if (value instanceof Double) {
|
|
jsonNode.put(key, (Double) value);
|
|
} else if (value instanceof Boolean) {
|
|
jsonNode.put(key, (Boolean) value);
|
|
} else {
|
|
jsonNode.put(key, value.toString());
|
|
}
|
|
|
|
// 调用 ThingsBoard REST API 保存属性
|
|
// saveDeviceAttributes 方法会将属性保存到 SERVER_SCOPE
|
|
client.saveDeviceAttributes(deviceIdObj, "SERVER_SCOPE", jsonNode);
|
|
|
|
log.info("设备属性设置成功: deviceId={}, key={}", deviceId, key);
|
|
|
|
return true;
|
|
} catch (Exception e) {
|
|
log.error("设置设备属性失败: deviceId=, key={}, error={}",
|
|
deviceId, key, e.getMessage(), e);
|
|
return false;
|
|
}
|
|
}
|
|
} |