Andrew Shvayka
2 years ago
committed by
GitHub
GitHub
52 changed files with 1791 additions and 443 deletions
@ -0,0 +1,37 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf; |
||||
|
|
||||
|
import org.thingsboard.server.common.data.id.CalculatedFieldId; |
||||
|
import org.thingsboard.server.common.data.id.EntityId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.kv.KvEntry; |
||||
|
import org.thingsboard.server.common.msg.queue.TbCallback; |
||||
|
import org.thingsboard.server.gen.transport.TransportProtos; |
||||
|
|
||||
|
import java.util.Map; |
||||
|
|
||||
|
public interface CalculatedFieldExecutionService { |
||||
|
|
||||
|
void onCalculatedFieldMsg(TransportProtos.CalculatedFieldMsgProto proto, TbCallback callback); |
||||
|
|
||||
|
void onTelemetryUpdate(TenantId tenantId, EntityId entityId, CalculatedFieldId calculatedFieldId, Map<String, KvEntry> updatedTelemetry); |
||||
|
|
||||
|
void onEntityProfileChanged(TransportProtos.EntityProfileUpdateMsgProto proto, TbCallback callback); |
||||
|
|
||||
|
void onProfileEntityMsg(TransportProtos.ProfileEntityMsgProto proto, TbCallback callback); |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,35 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf; |
||||
|
|
||||
|
import lombok.Data; |
||||
|
import org.thingsboard.server.common.data.AttributeScope; |
||||
|
|
||||
|
import java.util.Map; |
||||
|
|
||||
|
@Data |
||||
|
public final class CalculatedFieldResult { |
||||
|
|
||||
|
private String type; |
||||
|
private AttributeScope scope; |
||||
|
private Map<String, Object> resultMap; |
||||
|
|
||||
|
public CalculatedFieldResult(String type, AttributeScope scope, Map<String, Object> resultMap) { |
||||
|
this.type = type; |
||||
|
this.scope = scope; |
||||
|
this.resultMap = resultMap == null ? Map.of() : Map.copyOf(resultMap); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,553 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf; |
||||
|
|
||||
|
import com.fasterxml.jackson.databind.JsonNode; |
||||
|
import com.fasterxml.jackson.databind.node.ObjectNode; |
||||
|
import com.google.common.util.concurrent.FutureCallback; |
||||
|
import com.google.common.util.concurrent.Futures; |
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
import com.google.common.util.concurrent.ListeningExecutorService; |
||||
|
import com.google.common.util.concurrent.MoreExecutors; |
||||
|
import jakarta.annotation.PostConstruct; |
||||
|
import jakarta.annotation.PreDestroy; |
||||
|
import lombok.Getter; |
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.apache.commons.lang3.math.NumberUtils; |
||||
|
import org.springframework.beans.factory.annotation.Value; |
||||
|
import org.springframework.stereotype.Service; |
||||
|
import org.thingsboard.common.util.JacksonUtil; |
||||
|
import org.thingsboard.common.util.ThingsBoardExecutors; |
||||
|
import org.thingsboard.script.api.tbel.TbelInvokeService; |
||||
|
import org.thingsboard.server.cluster.TbClusterService; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
import org.thingsboard.server.common.data.cf.CalculatedField; |
||||
|
import org.thingsboard.server.common.data.cf.CalculatedFieldType; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.Argument; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.CalculatedFieldConfiguration; |
||||
|
import org.thingsboard.server.common.data.id.AssetProfileId; |
||||
|
import org.thingsboard.server.common.data.id.CalculatedFieldId; |
||||
|
import org.thingsboard.server.common.data.id.DeviceProfileId; |
||||
|
import org.thingsboard.server.common.data.id.EntityId; |
||||
|
import org.thingsboard.server.common.data.id.EntityIdFactory; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.kv.Aggregation; |
||||
|
import org.thingsboard.server.common.data.kv.BaseAttributeKvEntry; |
||||
|
import org.thingsboard.server.common.data.kv.BaseReadTsKvQuery; |
||||
|
import org.thingsboard.server.common.data.kv.BasicTsKvEntry; |
||||
|
import org.thingsboard.server.common.data.kv.BooleanDataEntry; |
||||
|
import org.thingsboard.server.common.data.kv.DoubleDataEntry; |
||||
|
import org.thingsboard.server.common.data.kv.KvEntry; |
||||
|
import org.thingsboard.server.common.data.kv.ReadTsKvQuery; |
||||
|
import org.thingsboard.server.common.data.kv.StringDataEntry; |
||||
|
import org.thingsboard.server.common.data.kv.TsKvEntry; |
||||
|
import org.thingsboard.server.common.data.msg.TbMsgType; |
||||
|
import org.thingsboard.server.common.data.page.PageDataIterable; |
||||
|
import org.thingsboard.server.common.msg.TbMsg; |
||||
|
import org.thingsboard.server.common.msg.TbMsgMetaData; |
||||
|
import org.thingsboard.server.common.msg.queue.TbCallback; |
||||
|
import org.thingsboard.server.common.msg.queue.TopicPartitionInfo; |
||||
|
import org.thingsboard.server.dao.asset.AssetService; |
||||
|
import org.thingsboard.server.dao.attributes.AttributesService; |
||||
|
import org.thingsboard.server.dao.cf.CalculatedFieldService; |
||||
|
import org.thingsboard.server.dao.device.DeviceService; |
||||
|
import org.thingsboard.server.dao.timeseries.TimeseriesService; |
||||
|
import org.thingsboard.server.gen.transport.TransportProtos; |
||||
|
import org.thingsboard.server.queue.util.TbCoreComponent; |
||||
|
import org.thingsboard.server.service.cf.ctx.CalculatedFieldEntityCtx; |
||||
|
import org.thingsboard.server.service.cf.ctx.CalculatedFieldEntityCtxId; |
||||
|
import org.thingsboard.server.service.cf.ctx.state.ArgumentEntry; |
||||
|
import org.thingsboard.server.service.cf.ctx.state.CalculatedFieldCtx; |
||||
|
import org.thingsboard.server.service.cf.ctx.state.CalculatedFieldState; |
||||
|
import org.thingsboard.server.service.cf.ctx.state.ScriptCalculatedFieldState; |
||||
|
import org.thingsboard.server.service.cf.ctx.state.SimpleCalculatedFieldState; |
||||
|
import org.thingsboard.server.service.partition.AbstractPartitionBasedService; |
||||
|
|
||||
|
import java.util.ArrayList; |
||||
|
import java.util.Collections; |
||||
|
import java.util.HashMap; |
||||
|
import java.util.HashSet; |
||||
|
import java.util.List; |
||||
|
import java.util.Map; |
||||
|
import java.util.Optional; |
||||
|
import java.util.Set; |
||||
|
import java.util.UUID; |
||||
|
import java.util.concurrent.ConcurrentHashMap; |
||||
|
import java.util.concurrent.ConcurrentMap; |
||||
|
import java.util.function.Consumer; |
||||
|
import java.util.stream.Collectors; |
||||
|
|
||||
|
import static org.thingsboard.server.common.data.DataConstants.SCOPE; |
||||
|
|
||||
|
@TbCoreComponent |
||||
|
@Service |
||||
|
@Slf4j |
||||
|
@RequiredArgsConstructor |
||||
|
public class DefaultCalculatedFieldExecutionService extends AbstractPartitionBasedService<CalculatedFieldId> implements CalculatedFieldExecutionService { |
||||
|
|
||||
|
private final CalculatedFieldService calculatedFieldService; |
||||
|
private final AssetService assetService; |
||||
|
private final DeviceService deviceService; |
||||
|
private final AttributesService attributesService; |
||||
|
private final TimeseriesService timeseriesService; |
||||
|
private final RocksDBService rocksDBService; |
||||
|
private final TbClusterService clusterService; |
||||
|
private final TbelInvokeService tbelInvokeService; |
||||
|
|
||||
|
private ListeningExecutorService calculatedFieldExecutor; |
||||
|
private ListeningExecutorService calculatedFieldCallbackExecutor; |
||||
|
|
||||
|
private final ConcurrentMap<CalculatedFieldId, CalculatedField> calculatedFields = new ConcurrentHashMap<>(); |
||||
|
private final ConcurrentMap<CalculatedFieldId, CalculatedFieldCtx> calculatedFieldsCtx = new ConcurrentHashMap<>(); |
||||
|
private final ConcurrentMap<CalculatedFieldEntityCtxId, CalculatedFieldEntityCtx> states = new ConcurrentHashMap<>(); |
||||
|
|
||||
|
private final ConcurrentMap<EntityId, Set<EntityId>> profileEntities = new ConcurrentHashMap<>(); |
||||
|
|
||||
|
private static final int MAX_LAST_RECORDS_VALUE = 1024; |
||||
|
|
||||
|
@Value("${calculatedField.initFetchPackSize:50000}") |
||||
|
@Getter |
||||
|
private int initFetchPackSize; |
||||
|
|
||||
|
@PostConstruct |
||||
|
public void init() { |
||||
|
super.init(); |
||||
|
calculatedFieldExecutor = MoreExecutors.listeningDecorator(ThingsBoardExecutors.newWorkStealingPool( |
||||
|
Math.max(4, Runtime.getRuntime().availableProcessors()), "calculated-field")); |
||||
|
calculatedFieldCallbackExecutor = MoreExecutors.listeningDecorator(ThingsBoardExecutors.newWorkStealingPool( |
||||
|
Math.max(4, Runtime.getRuntime().availableProcessors()), "calculated-field-callback")); |
||||
|
} |
||||
|
|
||||
|
@PreDestroy |
||||
|
public void stop() { |
||||
|
if (calculatedFieldExecutor != null) { |
||||
|
calculatedFieldExecutor.shutdownNow(); |
||||
|
} |
||||
|
if (calculatedFieldCallbackExecutor != null) { |
||||
|
calculatedFieldCallbackExecutor.shutdownNow(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
protected String getServiceName() { |
||||
|
return "Calculated Field Execution"; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
protected String getSchedulerExecutorName() { |
||||
|
return "calculated-field-scheduled"; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
protected Map<TopicPartitionInfo, List<ListenableFuture<?>>> onAddedPartitions(Set<TopicPartitionInfo> addedPartitions) { |
||||
|
// TODO: implementation for cluster mode
|
||||
|
return Map.of(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
protected void cleanupEntityOnPartitionRemoval(CalculatedFieldId entityId) { |
||||
|
// TODO: implementation for cluster mode
|
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onCalculatedFieldMsg(TransportProtos.CalculatedFieldMsgProto proto, TbCallback callback) { |
||||
|
try { |
||||
|
TenantId tenantId = TenantId.fromUUID(new UUID(proto.getTenantIdMSB(), proto.getTenantIdLSB())); |
||||
|
CalculatedFieldId calculatedFieldId = new CalculatedFieldId(new UUID(proto.getCalculatedFieldIdMSB(), proto.getCalculatedFieldIdLSB())); |
||||
|
log.info("Received CalculatedFieldMsgProto for processing: tenantId=[{}], calculatedFieldId=[{}]", tenantId, calculatedFieldId); |
||||
|
if (proto.getDeleted()) { |
||||
|
log.warn("Executing onCalculatedFieldDelete, calculatedFieldId=[{}]", calculatedFieldId); |
||||
|
onCalculatedFieldDelete(calculatedFieldId, callback); |
||||
|
callback.onSuccess(); |
||||
|
} |
||||
|
CalculatedField cf = getOrFetchFromDb(tenantId, calculatedFieldId); |
||||
|
if (proto.getUpdated()) { |
||||
|
log.info("Executing onCalculatedFieldUpdate, calculatedFieldId=[{}]", calculatedFieldId); |
||||
|
boolean shouldReinit = onCalculatedFieldUpdate(cf, callback); |
||||
|
if (!shouldReinit) { |
||||
|
return; |
||||
|
} |
||||
|
} |
||||
|
if (cf != null) { |
||||
|
EntityId entityId = cf.getEntityId(); |
||||
|
CalculatedFieldCtx calculatedFieldCtx = new CalculatedFieldCtx(cf, tbelInvokeService); |
||||
|
calculatedFieldsCtx.put(calculatedFieldId, calculatedFieldCtx); |
||||
|
switch (entityId.getEntityType()) { |
||||
|
case ASSET, DEVICE -> { |
||||
|
log.info("Initializing state for entity: tenantId=[{}], entityId=[{}]", tenantId, entityId); |
||||
|
initializeStateForEntity(calculatedFieldCtx, entityId, callback); |
||||
|
} |
||||
|
case ASSET_PROFILE, DEVICE_PROFILE -> { |
||||
|
log.info("Initializing state for all entities in profile: tenantId=[{}], profileId=[{}]", tenantId, entityId); |
||||
|
fetchCommonArguments(calculatedFieldCtx, callback, commonArguments -> { |
||||
|
getOrFetchFromDBProfileEntities(tenantId, entityId).forEach(assetId -> { |
||||
|
initializeStateForEntity(calculatedFieldCtx, assetId, commonArguments, callback); |
||||
|
}); |
||||
|
}); |
||||
|
} |
||||
|
default -> |
||||
|
throw new IllegalArgumentException("Entity type '" + calculatedFieldId.getEntityType() + "' does not support calculated fields."); |
||||
|
} |
||||
|
} else { |
||||
|
//Calculated field was probably deleted while message was in queue;
|
||||
|
log.warn("Calculated field not found, possibly deleted: {}", calculatedFieldId); |
||||
|
callback.onSuccess(); |
||||
|
} |
||||
|
callback.onSuccess(); |
||||
|
log.info("Successfully processed calculated field message for calculatedFieldId: [{}]", calculatedFieldId); |
||||
|
} catch (Exception e) { |
||||
|
log.trace("Failed to process calculated field msg: [{}]", proto, e); |
||||
|
callback.onFailure(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onTelemetryUpdate(TenantId tenantId, EntityId entityId, CalculatedFieldId calculatedFieldId, Map<String, KvEntry> updatedTelemetry) { |
||||
|
try { |
||||
|
log.info("Received telemetry update msg: tenantId=[{}], calculatedFieldId=[{}]", tenantId, calculatedFieldId); |
||||
|
CalculatedField calculatedField = getOrFetchFromDb(tenantId, calculatedFieldId); |
||||
|
CalculatedFieldCtx calculatedFieldCtx = calculatedFieldsCtx.computeIfAbsent(calculatedFieldId, id -> new CalculatedFieldCtx(calculatedField, tbelInvokeService)); |
||||
|
Map<String, ArgumentEntry> argumentValues = updatedTelemetry.entrySet().stream() |
||||
|
.collect(Collectors.toMap(Map.Entry::getKey, entry -> ArgumentEntry.createSingleValueArgument(entry.getValue()))); |
||||
|
|
||||
|
EntityId cfEntityId = calculatedField.getEntityId(); |
||||
|
switch (cfEntityId.getEntityType()) { |
||||
|
case ASSET_PROFILE, DEVICE_PROFILE -> { |
||||
|
boolean isCommonEntity = calculatedField.getConfiguration().getReferencedEntities().contains(entityId); |
||||
|
if (isCommonEntity) { |
||||
|
getOrFetchFromDBProfileEntities(tenantId, cfEntityId).forEach(id -> updateOrInitializeState(calculatedFieldCtx, id, argumentValues)); |
||||
|
} else { |
||||
|
updateOrInitializeState(calculatedFieldCtx, entityId, argumentValues); |
||||
|
} |
||||
|
} |
||||
|
default -> updateOrInitializeState(calculatedFieldCtx, cfEntityId, argumentValues); |
||||
|
} |
||||
|
log.info("Successfully updated telemetry for calculatedFieldId: [{}]", calculatedFieldId); |
||||
|
} catch (Exception e) { |
||||
|
log.trace("Failed to update telemetry for calculatedFieldId: [{}]", calculatedFieldId, e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onEntityProfileChanged(TransportProtos.EntityProfileUpdateMsgProto proto, TbCallback callback) { |
||||
|
try { |
||||
|
TenantId tenantId = TenantId.fromUUID(new UUID(proto.getTenantIdMSB(), proto.getTenantIdLSB())); |
||||
|
EntityId entityId = EntityIdFactory.getByTypeAndUuid(proto.getEntityType(), new UUID(proto.getEntityIdMSB(), proto.getEntityIdLSB())); |
||||
|
EntityId oldProfileId = EntityIdFactory.getByTypeAndUuid(proto.getEntityProfileType(), new UUID(proto.getOldProfileIdMSB(), proto.getOldProfileIdLSB())); |
||||
|
EntityId newProfileId = EntityIdFactory.getByTypeAndUuid(proto.getEntityProfileType(), new UUID(proto.getNewProfileIdMSB(), proto.getNewProfileIdLSB())); |
||||
|
log.info("Received EntityProfileUpdateMsgProto for processing: tenantId=[{}], entityId=[{}]", tenantId, entityId); |
||||
|
|
||||
|
profileEntities.get(oldProfileId).remove(entityId); |
||||
|
profileEntities.computeIfAbsent(newProfileId, id -> new HashSet<>()).add(entityId); |
||||
|
|
||||
|
calculatedFieldService.findCalculatedFieldIdsByEntityId(tenantId, oldProfileId) |
||||
|
.forEach(cfId -> { |
||||
|
CalculatedFieldEntityCtxId ctxId = new CalculatedFieldEntityCtxId(cfId.getId(), entityId.getId()); |
||||
|
states.remove(ctxId); |
||||
|
rocksDBService.delete(JacksonUtil.writeValueAsString(ctxId)); |
||||
|
}); |
||||
|
|
||||
|
initializeStateForEntityByProfile(tenantId, entityId, newProfileId, callback); |
||||
|
} catch (Exception e) { |
||||
|
log.trace("Failed to process entity type update msg: [{}]", proto, e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onProfileEntityMsg(TransportProtos.ProfileEntityMsgProto proto, TbCallback callback) { |
||||
|
try { |
||||
|
TenantId tenantId = TenantId.fromUUID(new UUID(proto.getTenantIdMSB(), proto.getTenantIdLSB())); |
||||
|
EntityId entityId = EntityIdFactory.getByTypeAndUuid(proto.getEntityType(), new UUID(proto.getEntityIdMSB(), proto.getEntityIdLSB())); |
||||
|
EntityId profileId = EntityIdFactory.getByTypeAndUuid(proto.getEntityProfileType(), new UUID(proto.getProfileIdMSB(), proto.getProfileIdLSB())); |
||||
|
log.info("Received ProfileEntityMsgProto for processing: tenantId=[{}], entityId=[{}]", tenantId, entityId); |
||||
|
if (proto.getDeleted()) { |
||||
|
log.info("Executing profile entity deleted msg, tenantId=[{}], entityId=[{}]", tenantId, entityId); |
||||
|
profileEntities.get(profileId).remove(entityId); |
||||
|
List<String> statesToRemove = states.keySet().stream() |
||||
|
.filter(ctxEntityId -> ctxEntityId.entityId().equals(entityId.getId())) |
||||
|
.map(JacksonUtil::writeValueAsString) |
||||
|
.toList(); |
||||
|
states.keySet().removeIf(ctxEntityId -> ctxEntityId.entityId().equals(entityId.getId())); |
||||
|
rocksDBService.deleteAll(statesToRemove); |
||||
|
} else { |
||||
|
log.info("Executing profile entity added msg, tenantId=[{}], entityId=[{}]", tenantId, entityId); |
||||
|
profileEntities.computeIfAbsent(profileId, id -> new HashSet<>()).add(entityId); |
||||
|
initializeStateForEntityByProfile(tenantId, entityId, profileId, callback); |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
log.trace("Failed to process profile entity msg: [{}]", proto, e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private boolean onCalculatedFieldUpdate(CalculatedField updatedCalculatedField, TbCallback callback) { |
||||
|
CalculatedField oldCalculatedField = getOrFetchFromDb(updatedCalculatedField.getTenantId(), updatedCalculatedField.getId()); |
||||
|
boolean shouldReinit = true; |
||||
|
if (hasSignificantChanges(oldCalculatedField, updatedCalculatedField)) { |
||||
|
onCalculatedFieldDelete(updatedCalculatedField.getId(), callback); |
||||
|
} else { |
||||
|
calculatedFields.put(updatedCalculatedField.getId(), updatedCalculatedField); |
||||
|
calculatedFieldsCtx.put(updatedCalculatedField.getId(), new CalculatedFieldCtx(updatedCalculatedField, tbelInvokeService)); |
||||
|
callback.onSuccess(); |
||||
|
shouldReinit = false; |
||||
|
} |
||||
|
return shouldReinit; |
||||
|
} |
||||
|
|
||||
|
private void onCalculatedFieldDelete(CalculatedFieldId calculatedFieldId, TbCallback callback) { |
||||
|
try { |
||||
|
calculatedFields.remove(calculatedFieldId); |
||||
|
calculatedFieldsCtx.remove(calculatedFieldId); |
||||
|
states.keySet().removeIf(ctxId -> ctxId.cfId().equals(calculatedFieldId.getId())); |
||||
|
List<String> statesToRemove = states.keySet().stream() |
||||
|
.filter(ctxId -> ctxId.cfId().equals(calculatedFieldId.getId())) |
||||
|
.map(JacksonUtil::writeValueAsString) |
||||
|
.toList(); |
||||
|
rocksDBService.deleteAll(statesToRemove); |
||||
|
} catch (Exception e) { |
||||
|
log.trace("Failed to delete calculated field: [{}]", calculatedFieldId, e); |
||||
|
callback.onFailure(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private CalculatedField getOrFetchFromDb(TenantId tenantId, CalculatedFieldId calculatedFieldId) { |
||||
|
return calculatedFields.computeIfAbsent(calculatedFieldId, cfId -> calculatedFieldService.findById(tenantId, calculatedFieldId)); |
||||
|
} |
||||
|
|
||||
|
private Set<EntityId> getOrFetchFromDBProfileEntities(TenantId tenantId, EntityId entityProfileId) { |
||||
|
return switch (entityProfileId.getEntityType()) { |
||||
|
case ASSET_PROFILE -> profileEntities.computeIfAbsent(entityProfileId, profileId -> { |
||||
|
Set<EntityId> assetIds = new HashSet<>(); |
||||
|
(new PageDataIterable<>(pageLink -> |
||||
|
assetService.findAssetIdsByTenantIdAndAssetProfileId(tenantId, (AssetProfileId) profileId, pageLink), initFetchPackSize)).forEach(assetIds::add); |
||||
|
return assetIds; |
||||
|
}); |
||||
|
case DEVICE_PROFILE -> profileEntities.computeIfAbsent(entityProfileId, profileId -> { |
||||
|
Set<EntityId> deviceIds = new HashSet<>(); |
||||
|
(new PageDataIterable<>(pageLink -> |
||||
|
deviceService.findDeviceIdsByTenantIdAndDeviceProfileId(tenantId, (DeviceProfileId) entityProfileId, pageLink), initFetchPackSize)).forEach(deviceIds::add); |
||||
|
return deviceIds; |
||||
|
}); |
||||
|
default -> throw new IllegalArgumentException("Entity type should be ASSET_PROFILE or DEVICE_PROFILE."); |
||||
|
}; |
||||
|
} |
||||
|
|
||||
|
private boolean hasSignificantChanges(CalculatedField oldCalculatedField, CalculatedField newCalculatedField) { |
||||
|
if (oldCalculatedField == null) { |
||||
|
return true; |
||||
|
} |
||||
|
boolean entityIdChanged = !oldCalculatedField.getEntityId().equals(newCalculatedField.getEntityId()); |
||||
|
boolean typeChanged = !oldCalculatedField.getType().equals(newCalculatedField.getType()); |
||||
|
CalculatedFieldConfiguration oldConfig = oldCalculatedField.getConfiguration(); |
||||
|
CalculatedFieldConfiguration newConfig = newCalculatedField.getConfiguration(); |
||||
|
boolean argumentsChanged = !oldConfig.getArguments().equals(newConfig.getArguments()); |
||||
|
boolean outputTypeChanged = !oldConfig.getOutput().getType().equals(newConfig.getOutput().getType()); |
||||
|
boolean expressionChanged = !oldConfig.getExpression().equals(newConfig.getExpression()); |
||||
|
|
||||
|
return entityIdChanged || typeChanged || argumentsChanged || outputTypeChanged || expressionChanged; |
||||
|
} |
||||
|
|
||||
|
private void initializeStateForEntityByProfile(TenantId tenantId, EntityId entityId, EntityId profileId, TbCallback callback) { |
||||
|
calculatedFieldService.findCalculatedFieldIdsByEntityId(tenantId, profileId) |
||||
|
.stream() |
||||
|
.map(cfId -> calculatedFieldsCtx.computeIfAbsent(cfId, id -> new CalculatedFieldCtx(calculatedFieldService.findById(tenantId, id), tbelInvokeService))) |
||||
|
.forEach(cfCtx -> initializeStateForEntity(cfCtx, entityId, callback)); |
||||
|
} |
||||
|
|
||||
|
private void fetchCommonArguments(CalculatedFieldCtx calculatedFieldCtx, TbCallback callback, Consumer<Map<String, ArgumentEntry>> onComplete) { |
||||
|
Map<String, ArgumentEntry> argumentValues = new HashMap<>(); |
||||
|
List<ListenableFuture<ArgumentEntry>> futures = new ArrayList<>(); |
||||
|
|
||||
|
calculatedFieldCtx.getArguments().forEach((key, argument) -> { |
||||
|
if (!EntityType.DEVICE_PROFILE.equals(argument.getEntityId().getEntityType()) && |
||||
|
!EntityType.ASSET_PROFILE.equals(argument.getEntityId().getEntityType())) { |
||||
|
futures.add(Futures.transform(fetchKvEntry(calculatedFieldCtx.getTenantId(), argument.getEntityId(), argument), |
||||
|
result -> { |
||||
|
argumentValues.put(key, result); |
||||
|
return result; |
||||
|
}, calculatedFieldCallbackExecutor)); |
||||
|
} |
||||
|
}); |
||||
|
|
||||
|
Futures.addCallback(Futures.allAsList(futures), new FutureCallback<>() { |
||||
|
@Override |
||||
|
public void onSuccess(List<ArgumentEntry> results) { |
||||
|
onComplete.accept(argumentValues); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onFailure(Throwable t) { |
||||
|
log.error("Failed to fetch common arguments", t); |
||||
|
callback.onFailure(t); |
||||
|
} |
||||
|
}, calculatedFieldCallbackExecutor); |
||||
|
} |
||||
|
|
||||
|
private void initializeStateForEntity(CalculatedFieldCtx calculatedFieldCtx, EntityId entityId, TbCallback callback) { |
||||
|
initializeStateForEntity(calculatedFieldCtx, entityId, new HashMap<>(), callback); |
||||
|
} |
||||
|
|
||||
|
private void initializeStateForEntity(CalculatedFieldCtx calculatedFieldCtx, EntityId entityId, Map<String, ArgumentEntry> commonArguments, TbCallback callback) { |
||||
|
Map<String, ArgumentEntry> argumentValues = new HashMap<>(commonArguments); |
||||
|
List<ListenableFuture<ArgumentEntry>> futures = new ArrayList<>(); |
||||
|
|
||||
|
calculatedFieldCtx.getArguments().forEach((key, argument) -> { |
||||
|
if (!commonArguments.containsKey(key)) { |
||||
|
futures.add(Futures.transform(fetchArgumentValue(calculatedFieldCtx, entityId, argument), |
||||
|
result -> { |
||||
|
argumentValues.put(key, result); |
||||
|
return result; |
||||
|
}, calculatedFieldCallbackExecutor)); |
||||
|
} |
||||
|
}); |
||||
|
|
||||
|
Futures.addCallback(Futures.allAsList(futures), new FutureCallback<>() { |
||||
|
@Override |
||||
|
public void onSuccess(List<ArgumentEntry> results) { |
||||
|
updateOrInitializeState(calculatedFieldCtx, entityId, argumentValues); |
||||
|
callback.onSuccess(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onFailure(Throwable t) { |
||||
|
log.error("Failed to initialize state for entity: [{}]", entityId, t); |
||||
|
callback.onFailure(t); |
||||
|
} |
||||
|
}, calculatedFieldCallbackExecutor); |
||||
|
} |
||||
|
|
||||
|
private ListenableFuture<ArgumentEntry> fetchArgumentValue(CalculatedFieldCtx calculatedFieldCtx, EntityId targetEntityId, Argument argument) { |
||||
|
TenantId tenantId = calculatedFieldCtx.getTenantId(); |
||||
|
EntityId argumentEntityId = argument.getEntityId(); |
||||
|
EntityId entityId = EntityType.DEVICE_PROFILE.equals(argumentEntityId.getEntityType()) || EntityType.ASSET_PROFILE.equals(argumentEntityId.getEntityType()) |
||||
|
? targetEntityId |
||||
|
: argumentEntityId; |
||||
|
return fetchKvEntry(tenantId, entityId, argument); |
||||
|
} |
||||
|
|
||||
|
private ListenableFuture<ArgumentEntry> fetchKvEntry(TenantId tenantId, EntityId entityId, Argument argument) { |
||||
|
return switch (argument.getType()) { |
||||
|
case "TS_ROLLING" -> fetchTsRolling(tenantId, entityId, argument); |
||||
|
case "ATTRIBUTE" -> transformSingleValueArgument( |
||||
|
Futures.transform( |
||||
|
attributesService.find(tenantId, entityId, argument.getScope(), argument.getKey()), |
||||
|
result -> result.or(() -> Optional.of(new BaseAttributeKvEntry(System.currentTimeMillis(), createDefaultKvEntry(argument)))), |
||||
|
calculatedFieldCallbackExecutor) |
||||
|
); |
||||
|
case "TS_LATEST" -> transformSingleValueArgument( |
||||
|
Futures.transform( |
||||
|
timeseriesService.findLatest(tenantId, entityId, argument.getKey()), |
||||
|
result -> result.or(() -> Optional.of(new BasicTsKvEntry(System.currentTimeMillis(), createDefaultKvEntry(argument)))), |
||||
|
calculatedFieldCallbackExecutor)); |
||||
|
default -> throw new IllegalArgumentException("Invalid argument type '" + argument.getType() + "'."); |
||||
|
}; |
||||
|
} |
||||
|
|
||||
|
private ListenableFuture<ArgumentEntry> fetchTsRolling(TenantId tenantId, EntityId entityId, Argument argument) { |
||||
|
long currentTime = System.currentTimeMillis(); |
||||
|
long timeWindow = argument.getTimeWindow() == 0 ? System.currentTimeMillis() : argument.getTimeWindow(); |
||||
|
long startTs = currentTime - timeWindow; |
||||
|
int limit = argument.getLimit() == 0 ? MAX_LAST_RECORDS_VALUE : argument.getLimit(); |
||||
|
|
||||
|
ReadTsKvQuery query = new BaseReadTsKvQuery(argument.getKey(), startTs, currentTime, 0, limit, Aggregation.NONE); |
||||
|
ListenableFuture<List<TsKvEntry>> tsRollingFuture = timeseriesService.findAll(tenantId, entityId, List.of(query)); |
||||
|
|
||||
|
return Futures.transform(tsRollingFuture, tsRolling -> tsRolling == null ? ArgumentEntry.createTsRollingArgument(Collections.emptyList()) : ArgumentEntry.createTsRollingArgument(tsRolling), calculatedFieldCallbackExecutor); |
||||
|
} |
||||
|
|
||||
|
private ListenableFuture<ArgumentEntry> transformSingleValueArgument(ListenableFuture<Optional<? extends KvEntry>> kvEntryFuture) { |
||||
|
return Futures.transform(kvEntryFuture, kvEntry -> ArgumentEntry.createSingleValueArgument(kvEntry.orElse(null)), calculatedFieldCallbackExecutor); |
||||
|
} |
||||
|
|
||||
|
private KvEntry createDefaultKvEntry(Argument argument) { |
||||
|
String key = argument.getKey(); |
||||
|
String defaultValue = argument.getDefaultValue(); |
||||
|
if (NumberUtils.isParsable(defaultValue)) { |
||||
|
return new DoubleDataEntry(key, Double.parseDouble(defaultValue)); |
||||
|
} |
||||
|
if ("true".equalsIgnoreCase(defaultValue) || "false".equalsIgnoreCase(defaultValue)) { |
||||
|
return new BooleanDataEntry(key, Boolean.parseBoolean(defaultValue)); |
||||
|
} |
||||
|
return new StringDataEntry(key, defaultValue); |
||||
|
} |
||||
|
|
||||
|
private void updateOrInitializeState(CalculatedFieldCtx calculatedFieldCtx, EntityId entityId, Map<String, ArgumentEntry> argumentValues) { |
||||
|
CalculatedFieldEntityCtxId entityCtxId = new CalculatedFieldEntityCtxId(calculatedFieldCtx.getCfId().getId(), entityId.getId()); |
||||
|
CalculatedFieldEntityCtx calculatedFieldEntityCtx = states.computeIfAbsent(entityCtxId, this::fetchCalculatedFieldEntityState); |
||||
|
|
||||
|
CalculatedFieldState state = calculatedFieldEntityCtx.getState(); |
||||
|
|
||||
|
if (state == null) { |
||||
|
state = createStateByType(calculatedFieldCtx.getCfType()); |
||||
|
} |
||||
|
state.initState(argumentValues); |
||||
|
calculatedFieldEntityCtx.setState(state); |
||||
|
states.put(entityCtxId, calculatedFieldEntityCtx); |
||||
|
rocksDBService.put(JacksonUtil.writeValueAsString(entityCtxId), JacksonUtil.writeValueAsString(calculatedFieldEntityCtx)); |
||||
|
|
||||
|
ListenableFuture<CalculatedFieldResult> resultFuture = state.performCalculation(calculatedFieldCtx); |
||||
|
Futures.addCallback(resultFuture, new FutureCallback<>() { |
||||
|
@Override |
||||
|
public void onSuccess(CalculatedFieldResult result) { |
||||
|
if (result != null) { |
||||
|
pushMsgToRuleEngine(calculatedFieldCtx.getTenantId(), entityId, result); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onFailure(Throwable t) { |
||||
|
log.warn("[{}] Failed to perform calculation. entityId: [{}]", calculatedFieldCtx.getCfId(), entityId, t); |
||||
|
} |
||||
|
}, MoreExecutors.directExecutor()); |
||||
|
|
||||
|
} |
||||
|
|
||||
|
private CalculatedFieldEntityCtx fetchCalculatedFieldEntityState(CalculatedFieldEntityCtxId entityCtxId) { |
||||
|
String stateStr = rocksDBService.get(JacksonUtil.writeValueAsString(entityCtxId)); |
||||
|
if (stateStr == null) { |
||||
|
return new CalculatedFieldEntityCtx(entityCtxId, null); |
||||
|
} |
||||
|
return JacksonUtil.fromString(rocksDBService.get(JacksonUtil.writeValueAsString(entityCtxId)), CalculatedFieldEntityCtx.class); |
||||
|
} |
||||
|
|
||||
|
private void pushMsgToRuleEngine(TenantId tenantId, EntityId originatorId, CalculatedFieldResult calculatedFieldResult) { |
||||
|
try { |
||||
|
String type = calculatedFieldResult.getType(); |
||||
|
TbMsgType msgType = "ATTRIBUTES".equals(type) ? TbMsgType.POST_ATTRIBUTES_REQUEST : TbMsgType.POST_TELEMETRY_REQUEST; |
||||
|
TbMsgMetaData md = "ATTRIBUTES".equals(type) ? new TbMsgMetaData(Map.of(SCOPE, calculatedFieldResult.getScope().name())) : TbMsgMetaData.EMPTY; |
||||
|
ObjectNode payload = createJsonPayload(calculatedFieldResult); |
||||
|
TbMsg msg = TbMsg.newMsg(msgType, originatorId, md, JacksonUtil.writeValueAsString(payload)); |
||||
|
clusterService.pushMsgToRuleEngine(tenantId, originatorId, msg, null); |
||||
|
} catch (Exception e) { |
||||
|
log.warn("[{}] Failed to push message to rule engine. CalculatedFieldResult: {}", originatorId, calculatedFieldResult, e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private ObjectNode createJsonPayload(CalculatedFieldResult calculatedFieldResult) { |
||||
|
ObjectNode payload = JacksonUtil.newObjectNode(); |
||||
|
Map<String, Object> resultMap = calculatedFieldResult.getResultMap(); |
||||
|
resultMap.forEach((k, v) -> payload.set(k, JacksonUtil.convertValue(v, JsonNode.class))); |
||||
|
return payload; |
||||
|
} |
||||
|
|
||||
|
private CalculatedFieldState createStateByType(CalculatedFieldType calculatedFieldType) { |
||||
|
return switch (calculatedFieldType) { |
||||
|
case SIMPLE -> new SimpleCalculatedFieldState(); |
||||
|
case SCRIPT -> new ScriptCalculatedFieldState(); |
||||
|
}; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,35 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf.ctx; |
||||
|
|
||||
|
import lombok.Data; |
||||
|
import org.thingsboard.server.service.cf.ctx.state.CalculatedFieldState; |
||||
|
|
||||
|
@Data |
||||
|
public class CalculatedFieldEntityCtx { |
||||
|
|
||||
|
private CalculatedFieldEntityCtxId id; |
||||
|
private CalculatedFieldState state; |
||||
|
|
||||
|
public CalculatedFieldEntityCtx() { |
||||
|
} |
||||
|
|
||||
|
public CalculatedFieldEntityCtx(CalculatedFieldEntityCtxId id, CalculatedFieldState state) { |
||||
|
this.id = id; |
||||
|
this.state = state; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,21 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf.ctx; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
public record CalculatedFieldEntityCtxId(UUID cfId, UUID entityId) { |
||||
|
} |
||||
@ -0,0 +1,53 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf.ctx.state; |
||||
|
|
||||
|
import com.fasterxml.jackson.annotation.JsonIgnore; |
||||
|
import com.fasterxml.jackson.annotation.JsonSubTypes; |
||||
|
import com.fasterxml.jackson.annotation.JsonTypeInfo; |
||||
|
import org.thingsboard.server.common.data.kv.KvEntry; |
||||
|
import org.thingsboard.server.common.data.kv.TsKvEntry; |
||||
|
|
||||
|
import java.util.List; |
||||
|
import java.util.TreeMap; |
||||
|
import java.util.stream.Collectors; |
||||
|
|
||||
|
@JsonTypeInfo( |
||||
|
use = JsonTypeInfo.Id.NAME, |
||||
|
include = JsonTypeInfo.As.PROPERTY, |
||||
|
property = "type" |
||||
|
) |
||||
|
@JsonSubTypes({ |
||||
|
@JsonSubTypes.Type(value = SingleValueArgumentEntry.class, name = "SINGLE_VALUE"), |
||||
|
@JsonSubTypes.Type(value = TsRollingArgumentEntry.class, name = "TS_ROLLING") |
||||
|
}) |
||||
|
public interface ArgumentEntry { |
||||
|
|
||||
|
@JsonIgnore |
||||
|
ArgumentType getType(); |
||||
|
|
||||
|
Object getValue(); |
||||
|
|
||||
|
static ArgumentEntry createSingleValueArgument(KvEntry kvEntry) { |
||||
|
return new SingleValueArgumentEntry(kvEntry); |
||||
|
} |
||||
|
|
||||
|
static ArgumentEntry createTsRollingArgument(List<TsKvEntry> kvEntries) { |
||||
|
return new TsRollingArgumentEntry(kvEntries.stream(). |
||||
|
collect(Collectors.toMap(TsKvEntry::getTs, TsKvEntry::getValue, (oldValue, newValue) -> newValue, TreeMap::new))); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,20 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf.ctx.state; |
||||
|
|
||||
|
public enum ArgumentType { |
||||
|
SINGLE_VALUE, TS_ROLLING |
||||
|
} |
||||
@ -0,0 +1,56 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf.ctx.state; |
||||
|
|
||||
|
import java.util.HashMap; |
||||
|
import java.util.Map; |
||||
|
|
||||
|
public abstract class BaseCalculatedFieldState implements CalculatedFieldState { |
||||
|
|
||||
|
protected Map<String, ArgumentEntry> arguments; |
||||
|
|
||||
|
public BaseCalculatedFieldState() { |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public Map<String, ArgumentEntry> getArguments() { |
||||
|
return this.arguments; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void initState(Map<String, ArgumentEntry> argumentValues) { |
||||
|
if (arguments == null) { |
||||
|
arguments = new HashMap<>(); |
||||
|
} |
||||
|
argumentValues.forEach((key, argumentEntry) -> { |
||||
|
ArgumentEntry existingArgumentEntry = arguments.get(key); |
||||
|
if (existingArgumentEntry != null) { |
||||
|
if (existingArgumentEntry instanceof SingleValueArgumentEntry) { |
||||
|
arguments.put(key, argumentEntry); |
||||
|
} else if (existingArgumentEntry instanceof TsRollingArgumentEntry existingTsRollingArgumentEntry) { |
||||
|
if (argumentEntry instanceof TsRollingArgumentEntry tsRollingArgumentEntry) { |
||||
|
existingTsRollingArgumentEntry.getTsRecords().putAll(tsRollingArgumentEntry.getTsRecords()); |
||||
|
} else if (argumentEntry instanceof SingleValueArgumentEntry singleValueArgumentEntry) { |
||||
|
existingTsRollingArgumentEntry.getTsRecords().put(singleValueArgumentEntry.getTs(), singleValueArgumentEntry.getValue()); |
||||
|
} |
||||
|
} |
||||
|
} else { |
||||
|
arguments.put(key, argumentEntry); |
||||
|
} |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,72 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf.ctx.state; |
||||
|
|
||||
|
import lombok.Data; |
||||
|
import org.thingsboard.script.api.tbel.TbelInvokeService; |
||||
|
import org.thingsboard.server.common.data.cf.CalculatedField; |
||||
|
import org.thingsboard.server.common.data.cf.CalculatedFieldType; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.Argument; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.CalculatedFieldConfiguration; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.Output; |
||||
|
import org.thingsboard.server.common.data.id.CalculatedFieldId; |
||||
|
import org.thingsboard.server.common.data.id.EntityId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
|
||||
|
import java.util.Map; |
||||
|
|
||||
|
@Data |
||||
|
public class CalculatedFieldCtx { |
||||
|
|
||||
|
private CalculatedFieldId cfId; |
||||
|
private TenantId tenantId; |
||||
|
private EntityId entityId; |
||||
|
private CalculatedFieldType cfType; |
||||
|
private Map<String, Argument> arguments; |
||||
|
private Output output; |
||||
|
private String expression; |
||||
|
private TbelInvokeService tbelInvokeService; |
||||
|
private CalculatedFieldScriptEngine calculatedFieldScriptEngine; |
||||
|
|
||||
|
public CalculatedFieldCtx(CalculatedField calculatedField, TbelInvokeService tbelInvokeService) { |
||||
|
this.cfId = calculatedField.getId(); |
||||
|
this.tenantId = calculatedField.getTenantId(); |
||||
|
this.entityId = calculatedField.getEntityId(); |
||||
|
this.cfType = calculatedField.getType(); |
||||
|
CalculatedFieldConfiguration configuration = calculatedField.getConfiguration(); |
||||
|
this.arguments = configuration.getArguments(); |
||||
|
this.output = configuration.getOutput(); |
||||
|
this.expression = configuration.getExpression(); |
||||
|
this.tbelInvokeService = tbelInvokeService; |
||||
|
if (!CalculatedFieldType.SIMPLE.equals(calculatedField.getType())) { |
||||
|
this.calculatedFieldScriptEngine = initEngine(tenantId, expression, tbelInvokeService); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private CalculatedFieldScriptEngine initEngine(TenantId tenantId, String expression, TbelInvokeService tbelInvokeService) { |
||||
|
if (tbelInvokeService == null) { |
||||
|
throw new IllegalArgumentException("TBEL script engine is disabled!"); |
||||
|
} |
||||
|
|
||||
|
return new CalculatedFieldTbelScriptEngine( |
||||
|
tenantId, |
||||
|
tbelInvokeService, |
||||
|
expression, |
||||
|
arguments.keySet().toArray(new String[0]) |
||||
|
); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,32 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf.ctx.state; |
||||
|
|
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
|
||||
|
import java.util.Map; |
||||
|
|
||||
|
public interface CalculatedFieldScriptEngine { |
||||
|
|
||||
|
ListenableFuture<Object> executeScriptAsync(Object[] args); |
||||
|
|
||||
|
ListenableFuture<Map<String, Object>> executeToMapAsync(Object[] args); |
||||
|
|
||||
|
ListenableFuture<Map<String, Object>> executeToMapTransform(Object result); |
||||
|
|
||||
|
void destroy(); |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,89 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf.ctx.state; |
||||
|
|
||||
|
import com.google.common.util.concurrent.Futures; |
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
import com.google.common.util.concurrent.MoreExecutors; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.thingsboard.script.api.ScriptType; |
||||
|
import org.thingsboard.script.api.tbel.TbelInvokeService; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
|
||||
|
import javax.script.ScriptException; |
||||
|
import java.util.Map; |
||||
|
import java.util.UUID; |
||||
|
import java.util.concurrent.ExecutionException; |
||||
|
|
||||
|
@Slf4j |
||||
|
public class CalculatedFieldTbelScriptEngine implements CalculatedFieldScriptEngine { |
||||
|
|
||||
|
private final TbelInvokeService tbelInvokeService; |
||||
|
|
||||
|
private final UUID scriptId; |
||||
|
private final TenantId tenantId; |
||||
|
|
||||
|
public CalculatedFieldTbelScriptEngine(TenantId tenantId, TbelInvokeService tbelInvokeService, String script, String... argNames) { |
||||
|
this.tenantId = tenantId; |
||||
|
this.tbelInvokeService = tbelInvokeService; |
||||
|
try { |
||||
|
this.scriptId = this.tbelInvokeService.eval(tenantId, ScriptType.CALCULATED_FIELD_SCRIPT, script, argNames).get(); |
||||
|
} catch (Exception e) { |
||||
|
Throwable t = e; |
||||
|
if (e instanceof ExecutionException) { |
||||
|
t = e.getCause(); |
||||
|
} |
||||
|
throw new IllegalArgumentException("Can't compile script: " + t.getMessage(), t); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public ListenableFuture<Object> executeScriptAsync(Object[] args) { |
||||
|
log.trace("Executing script async, args {}", args); |
||||
|
return Futures.transformAsync(tbelInvokeService.invokeScript(tenantId, null, this.scriptId, args), |
||||
|
o -> { |
||||
|
try { |
||||
|
return Futures.immediateFuture(o); |
||||
|
} catch (Exception e) { |
||||
|
if (e.getCause() instanceof ScriptException) { |
||||
|
return Futures.immediateFailedFuture(e.getCause()); |
||||
|
} else if (e.getCause() instanceof RuntimeException) { |
||||
|
return Futures.immediateFailedFuture(new ScriptException(e.getCause().getMessage())); |
||||
|
} else { |
||||
|
return Futures.immediateFailedFuture(new ScriptException(e)); |
||||
|
} |
||||
|
} |
||||
|
}, MoreExecutors.directExecutor()); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public ListenableFuture<Map<String, Object>> executeToMapAsync(Object[] args) { |
||||
|
return Futures.transformAsync(executeScriptAsync(args), this::executeToMapTransform, MoreExecutors.directExecutor()); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public ListenableFuture<Map<String, Object>> executeToMapTransform(Object result) { |
||||
|
if (result instanceof Map) { |
||||
|
return Futures.immediateFuture((Map<String, Object>) result); |
||||
|
} |
||||
|
throw new IllegalArgumentException("Wrong result type: [" + result.getClass().getName() + "]"); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void destroy() { |
||||
|
tbelInvokeService.release(this.scriptId); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,64 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf.ctx.state; |
||||
|
|
||||
|
import com.google.common.util.concurrent.Futures; |
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
import com.google.common.util.concurrent.MoreExecutors; |
||||
|
import lombok.Data; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.thingsboard.server.common.data.cf.CalculatedFieldType; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.Argument; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.Output; |
||||
|
import org.thingsboard.server.service.cf.CalculatedFieldResult; |
||||
|
|
||||
|
import java.util.Map; |
||||
|
import java.util.TreeMap; |
||||
|
|
||||
|
@Data |
||||
|
@Slf4j |
||||
|
public class ScriptCalculatedFieldState extends BaseCalculatedFieldState { |
||||
|
|
||||
|
@Override |
||||
|
public CalculatedFieldType getType() { |
||||
|
return CalculatedFieldType.SCRIPT; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public ListenableFuture<CalculatedFieldResult> performCalculation(CalculatedFieldCtx ctx) { |
||||
|
if (isValid(ctx.getArguments())) { |
||||
|
arguments.forEach((key, argumentEntry) -> { |
||||
|
if (argumentEntry instanceof TsRollingArgumentEntry) { |
||||
|
Argument argument = ctx.getArguments().get(key); |
||||
|
TreeMap<Long, Object> tsRecords = ((TsRollingArgumentEntry) argumentEntry).getTsRecords(); |
||||
|
if (tsRecords.size() > argument.getLimit()) { |
||||
|
tsRecords.pollFirstEntry(); |
||||
|
} |
||||
|
tsRecords.entrySet().removeIf(tsRecord -> tsRecord.getKey() < System.currentTimeMillis() - argument.getTimeWindow()); |
||||
|
} |
||||
|
}); |
||||
|
Object[] args = arguments.values().stream().map(ArgumentEntry::getValue).toArray(); |
||||
|
ListenableFuture<Map<String, Object>> resultFuture = ctx.getCalculatedFieldScriptEngine().executeToMapAsync(args); |
||||
|
Output output = ctx.getOutput(); |
||||
|
return Futures.transform(resultFuture, |
||||
|
result -> new CalculatedFieldResult(output.getType(), output.getScope(), result), |
||||
|
MoreExecutors.directExecutor() |
||||
|
); |
||||
|
} |
||||
|
return Futures.immediateFuture(null); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,63 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf.ctx.state; |
||||
|
|
||||
|
import com.google.common.util.concurrent.Futures; |
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
import lombok.Data; |
||||
|
import net.objecthunter.exp4j.Expression; |
||||
|
import net.objecthunter.exp4j.ExpressionBuilder; |
||||
|
import org.thingsboard.server.common.data.cf.CalculatedFieldType; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.Output; |
||||
|
import org.thingsboard.server.service.cf.CalculatedFieldResult; |
||||
|
|
||||
|
import java.util.HashMap; |
||||
|
import java.util.Map; |
||||
|
|
||||
|
@Data |
||||
|
public class SimpleCalculatedFieldState extends BaseCalculatedFieldState { |
||||
|
|
||||
|
@Override |
||||
|
public CalculatedFieldType getType() { |
||||
|
return CalculatedFieldType.SIMPLE; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public ListenableFuture<CalculatedFieldResult> performCalculation(CalculatedFieldCtx ctx) { |
||||
|
if (isValid(ctx.getArguments())) { |
||||
|
String expression = ctx.getExpression(); |
||||
|
ThreadLocal<Expression> customExpression = new ThreadLocal<>(); |
||||
|
var expr = customExpression.get(); |
||||
|
if (expr == null) { |
||||
|
expr = new ExpressionBuilder(expression) |
||||
|
.implicitMultiplication(true) |
||||
|
.variables(this.arguments.keySet()) |
||||
|
.build(); |
||||
|
customExpression.set(expr); |
||||
|
} |
||||
|
Map<String, Double> variables = new HashMap<>(); |
||||
|
this.arguments.forEach((k, v) -> variables.put(k, Double.parseDouble(v.getValue().toString()))); |
||||
|
expr.setVariables(variables); |
||||
|
|
||||
|
double expressionResult = expr.evaluate(); |
||||
|
|
||||
|
Output output = ctx.getOutput(); |
||||
|
return Futures.immediateFuture(new CalculatedFieldResult(output.getType(), output.getScope(), Map.of(output.getName(), expressionResult))); |
||||
|
} |
||||
|
return Futures.immediateFuture(null); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,51 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf.ctx.state; |
||||
|
|
||||
|
import lombok.Data; |
||||
|
import org.thingsboard.server.common.data.kv.AttributeKvEntry; |
||||
|
import org.thingsboard.server.common.data.kv.KvEntry; |
||||
|
import org.thingsboard.server.common.data.kv.TsKvEntry; |
||||
|
|
||||
|
@Data |
||||
|
public class SingleValueArgumentEntry implements ArgumentEntry { |
||||
|
|
||||
|
private long ts; |
||||
|
private Object value; |
||||
|
|
||||
|
public SingleValueArgumentEntry() { |
||||
|
} |
||||
|
|
||||
|
public SingleValueArgumentEntry(KvEntry entry) { |
||||
|
if (entry instanceof TsKvEntry) { |
||||
|
this.ts = ((TsKvEntry) entry).getTs(); |
||||
|
} else if (entry instanceof AttributeKvEntry) { |
||||
|
this.ts = ((AttributeKvEntry) entry).getLastUpdateTs(); |
||||
|
} |
||||
|
this.value = entry.getValue(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public ArgumentType getType() { |
||||
|
return ArgumentType.SINGLE_VALUE; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public Object getValue() { |
||||
|
return value; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,43 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf.ctx.state; |
||||
|
|
||||
|
import com.fasterxml.jackson.annotation.JsonIgnore; |
||||
|
import lombok.AllArgsConstructor; |
||||
|
import lombok.Data; |
||||
|
import lombok.NoArgsConstructor; |
||||
|
|
||||
|
import java.util.TreeMap; |
||||
|
|
||||
|
@Data |
||||
|
@NoArgsConstructor |
||||
|
@AllArgsConstructor |
||||
|
public class TsRollingArgumentEntry implements ArgumentEntry { |
||||
|
|
||||
|
private TreeMap<Long, Object> tsRecords; |
||||
|
|
||||
|
@Override |
||||
|
public ArgumentType getType() { |
||||
|
return ArgumentType.TS_ROLLING; |
||||
|
} |
||||
|
|
||||
|
@JsonIgnore |
||||
|
@Override |
||||
|
public Object getValue() { |
||||
|
return tsRecords; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -1,49 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2024 The Thingsboard Authors |
|
||||
* |
|
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|
||||
* you may not use this file except in compliance with the License. |
|
||||
* You may obtain a copy of the License at |
|
||||
* |
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||
* |
|
||||
* Unless required by applicable law or agreed to in writing, software |
|
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
||||
* See the License for the specific language governing permissions and |
|
||||
* limitations under the License. |
|
||||
*/ |
|
||||
package org.thingsboard.server.service.entitiy.cf; |
|
||||
|
|
||||
import lombok.Data; |
|
||||
import org.thingsboard.server.common.data.cf.CalculatedFieldConfiguration; |
|
||||
import org.thingsboard.server.common.data.cf.CalculatedFieldType; |
|
||||
|
|
||||
import java.util.HashMap; |
|
||||
import java.util.Map; |
|
||||
|
|
||||
@Data |
|
||||
public class SimpleCalculatedFieldState implements CalculatedFieldState { |
|
||||
|
|
||||
// TODO: use value object(TsKv) instead of string
|
|
||||
Map<String, String> arguments = new HashMap<>(); |
|
||||
String result; |
|
||||
|
|
||||
@Override |
|
||||
public CalculatedFieldType getType() { |
|
||||
return CalculatedFieldType.SIMPLE; |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void performCalculation(Map<String, String> argumentValues, CalculatedFieldConfiguration calculatedFieldConfiguration, boolean initialCalculation) { |
|
||||
if (initialCalculation) { |
|
||||
// todo: perform initial calculation
|
|
||||
this.arguments = argumentValues; |
|
||||
} else { |
|
||||
// todo: perform calculation based on previous data
|
|
||||
this.arguments.putAll(argumentValues); |
|
||||
} |
|
||||
this.result = "result"; |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
@ -0,0 +1,28 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.common.data.cf.configuration; |
||||
|
|
||||
|
import lombok.Data; |
||||
|
import org.thingsboard.server.common.data.AttributeScope; |
||||
|
|
||||
|
@Data |
||||
|
public class Output { |
||||
|
|
||||
|
private String name; |
||||
|
private String type; |
||||
|
private AttributeScope scope; |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,40 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.common.data.cf.configuration; |
||||
|
|
||||
|
import com.fasterxml.jackson.databind.JsonNode; |
||||
|
import lombok.Data; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
import org.thingsboard.server.common.data.cf.CalculatedFieldType; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
@Data |
||||
|
public class ScriptCalculatedFieldConfiguration extends BaseCalculatedFieldConfiguration implements CalculatedFieldConfiguration { |
||||
|
|
||||
|
public ScriptCalculatedFieldConfiguration() { |
||||
|
super(); |
||||
|
} |
||||
|
|
||||
|
public ScriptCalculatedFieldConfiguration(JsonNode config, EntityType entityType, UUID entityId) { |
||||
|
super(config, entityType, entityId); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public CalculatedFieldType getType() { |
||||
|
return CalculatedFieldType.SCRIPT; |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue