Basic data validation & fixed calculated field update in scheduled update msg

application/src/main/java/org/thingsboard/server/actors/calculatedField/CalculatedFieldManagerMessageProcessor.java

@@ -544,7 +544,7 @@ public class CalculatedFieldManagerMessageProcessor extends AbstractContextAware
             return;
         }
         long refreshDynamicSourceInterval = TimeUnit.SECONDS.toMillis(cfConfig.getScheduledUpdateIntervalSec());
-        var scheduledMsg = new CalculatedFieldScheduledCheckForUpdatesMsg(tenantId, cfCtx);
+        var scheduledMsg = new CalculatedFieldScheduledCheckForUpdatesMsg(tenantId, cfCtx.getCfId());
 
         ScheduledFuture<?> scheduledFuture = systemContext
                 .schedulePeriodicMsgWithDelay(ctx, scheduledMsg, refreshDynamicSourceInterval, refreshDynamicSourceInterval);
@@ -553,9 +553,14 @@ public class CalculatedFieldManagerMessageProcessor extends AbstractContextAware
     }
 
     public void onScheduledCheckForUpdatesMsg(CalculatedFieldScheduledCheckForUpdatesMsg msg) {
-        CalculatedFieldCtx cfCtx = msg.getCfCtx();
+        log.debug("[{}] [{}] Processing CF scheduled update msg.", tenantId, msg.getCfId());
+        CalculatedFieldCtx cfCtx = calculatedFields.get(msg.getCfId());
+        if (cfCtx == null) {
+            log.debug("[{}][{}] Failed to find CF context, going to stop scheduler updates.", tenantId, msg.getCfId());
+            cancelCfUpdateTaskIfExists(msg.getCfId(), true);
+            return;
+        }
         EntityId entityId = cfCtx.getEntityId();
-        log.debug("[{}] [{}] Processing CF scheduled update msg.", cfCtx.getCfId(), entityId);
         EntityType entityType = entityId.getEntityType();
         if (isProfileEntity(entityType)) {
             var entityIds = entityProfileCache.getEntityIdsByProfileId(entityId);

application/src/main/java/org/thingsboard/server/actors/calculatedField/CalculatedFieldScheduledCheckForUpdatesMsg.java

@@ -16,16 +16,16 @@
 package org.thingsboard.server.actors.calculatedField;
 
 import lombok.Data;
+import org.thingsboard.server.common.data.id.CalculatedFieldId;
 import org.thingsboard.server.common.data.id.TenantId;
 import org.thingsboard.server.common.msg.MsgType;
 import org.thingsboard.server.common.msg.ToCalculatedFieldSystemMsg;
-import org.thingsboard.server.service.cf.ctx.state.CalculatedFieldCtx;
 
 @Data
 public class CalculatedFieldScheduledCheckForUpdatesMsg implements ToCalculatedFieldSystemMsg {
 
     private final TenantId tenantId;
-    private final CalculatedFieldCtx cfCtx;
+    private final CalculatedFieldId cfId;
 
     @Override
     public MsgType getMsgType() {

application/src/main/java/org/thingsboard/server/service/cf/DefaultCalculatedFieldProcessingService.java

@@ -93,10 +93,10 @@ import java.util.concurrent.ExecutionException;
 import java.util.stream.Collectors;
 
 import static org.thingsboard.server.common.data.DataConstants.SCOPE;
-import static org.thingsboard.server.service.cf.ctx.state.GeofencingCalculatedFieldState.ENTITY_ID_LATITUDE_ARGUMENT_KEY;
-import static org.thingsboard.server.service.cf.ctx.state.GeofencingCalculatedFieldState.ENTITY_ID_LONGITUDE_ARGUMENT_KEY;
-import static org.thingsboard.server.service.cf.ctx.state.GeofencingCalculatedFieldState.RESTRICTED_ZONES_ARGUMENT_KEY;
-import static org.thingsboard.server.service.cf.ctx.state.GeofencingCalculatedFieldState.SAVE_ZONES_ARGUMENT_KEY;
+import static org.thingsboard.server.common.data.cf.configuration.GeofencingCalculatedFieldConfiguration.ENTITY_ID_LATITUDE_ARGUMENT_KEY;
+import static org.thingsboard.server.common.data.cf.configuration.GeofencingCalculatedFieldConfiguration.ENTITY_ID_LONGITUDE_ARGUMENT_KEY;
+import static org.thingsboard.server.common.data.cf.configuration.GeofencingCalculatedFieldConfiguration.RESTRICTED_ZONES_ARGUMENT_KEY;
+import static org.thingsboard.server.common.data.cf.configuration.GeofencingCalculatedFieldConfiguration.SAVE_ZONES_ARGUMENT_KEY;
 import static org.thingsboard.server.utils.CalculatedFieldUtils.toProto;
 
 @TbRuleEngineComponent

application/src/main/java/org/thingsboard/server/service/cf/ctx/state/GeofencingCalculatedFieldState.java

@@ -31,14 +31,14 @@ import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 
+import static org.thingsboard.server.common.data.cf.configuration.GeofencingCalculatedFieldConfiguration.ENTITY_ID_LATITUDE_ARGUMENT_KEY;
+import static org.thingsboard.server.common.data.cf.configuration.GeofencingCalculatedFieldConfiguration.ENTITY_ID_LONGITUDE_ARGUMENT_KEY;
+import static org.thingsboard.server.common.data.cf.configuration.GeofencingCalculatedFieldConfiguration.RESTRICTED_ZONES_ARGUMENT_KEY;
+import static org.thingsboard.server.common.data.cf.configuration.GeofencingCalculatedFieldConfiguration.SAVE_ZONES_ARGUMENT_KEY;
+
 @Data
 public class GeofencingCalculatedFieldState implements CalculatedFieldState {
 
-    public static final String ENTITY_ID_LATITUDE_ARGUMENT_KEY = "latitude";
-    public static final String ENTITY_ID_LONGITUDE_ARGUMENT_KEY = "longitude";
-    public static final String SAVE_ZONES_ARGUMENT_KEY = "saveZones";
-    public static final String RESTRICTED_ZONES_ARGUMENT_KEY = "restrictedZones";
-
     private List<String> requiredArguments;
     private Map<String, ArgumentEntry> arguments;
 
@@ -73,7 +73,7 @@ public class GeofencingCalculatedFieldState implements CalculatedFieldState {
             String key = entry.getKey();
             ArgumentEntry newEntry = entry.getValue();
 
-             checkArgumentSize(key, newEntry, ctx);
+            checkArgumentSize(key, newEntry, ctx);
 
             ArgumentEntry existingEntry = arguments.get(key);
             boolean entryUpdated;

common/data/src/main/java/org/thingsboard/server/common/data/cf/configuration/BaseCalculatedFieldConfiguration.java

@@ -33,6 +33,8 @@ public abstract class BaseCalculatedFieldConfiguration implements CalculatedFiel
     protected String expression;
     protected Output output;
 
+    protected int scheduledUpdateIntervalSec;
+
     @Override
     public List<EntityId> getReferencedEntities() {
         return arguments.values().stream()
@@ -58,9 +60,19 @@ public abstract class BaseCalculatedFieldConfiguration implements CalculatedFiel
         return link;
     }
 
+    // TODO: update validate method in PE version.
     @Override
-    public boolean hasDynamicSourceArguments() {
-        return arguments.values().stream().anyMatch(arg -> arg.getRefDynamicSource() != null);
+    public void validate() {
+        boolean hasDynamicSourceRelationQuery = arguments.values()
+                .stream()
+                .anyMatch(arg -> CFArgumentDynamicSourceType.RELATION_QUERY.equals(arg.getRefDynamicSource()));
+        if (hasDynamicSourceRelationQuery) {
+            throw new IllegalArgumentException("Calculated field with type: '" + getType() + "' doesn't support arguments with 'RELATION_QUERY' dynamic source type!");
+        }
+    }
+
+    public boolean isScheduledUpdateEnabled() {
+        return scheduledUpdateIntervalSec > 0 && arguments.values().stream().anyMatch(arg -> arg.getRefDynamicSource() != null);
     }
 
 }

common/data/src/main/java/org/thingsboard/server/common/data/cf/configuration/CalculatedFieldConfiguration.java

@@ -59,16 +59,15 @@ public interface CalculatedFieldConfiguration {
 
     CalculatedFieldLink buildCalculatedFieldLink(TenantId tenantId, EntityId referencedEntityId, CalculatedFieldId calculatedFieldId);
 
-    @JsonIgnore
-    boolean hasDynamicSourceArguments();
+    void validate();
 
     @JsonIgnore
     default boolean isScheduledUpdateEnabled() {
-        return hasDynamicSourceArguments() && getScheduledUpdateIntervalSec() > 0;
+        return false;
     }
 
-    default int getScheduledUpdateIntervalSec() {
-        return 0;
-    }
+    void setScheduledUpdateIntervalSec(int scheduledUpdateIntervalSec);
+
+    int getScheduledUpdateIntervalSec();
 
 }

common/data/src/main/java/org/thingsboard/server/common/data/cf/configuration/GeofencingCalculatedFieldConfiguration.java

@@ -19,19 +19,77 @@ import lombok.Data;
 import lombok.EqualsAndHashCode;
 import org.thingsboard.server.common.data.cf.CalculatedFieldType;
 
+import java.util.Set;
+
+import static org.thingsboard.server.common.data.cf.configuration.CFArgumentDynamicSourceType.RELATION_QUERY;
+
 @Data
 @EqualsAndHashCode(callSuper = true)
 public class GeofencingCalculatedFieldConfiguration extends BaseCalculatedFieldConfiguration implements CalculatedFieldConfiguration {
 
-    private int refreshIntervalSec;
+    public static final String ENTITY_ID_LATITUDE_ARGUMENT_KEY = "latitude";
+    public static final String ENTITY_ID_LONGITUDE_ARGUMENT_KEY = "longitude";
+    public static final String SAVE_ZONES_ARGUMENT_KEY = "saveZones";
+    public static final String RESTRICTED_ZONES_ARGUMENT_KEY = "restrictedZones";
+
+    private static final Set<String> requiredKeys = Set.of(
+            ENTITY_ID_LATITUDE_ARGUMENT_KEY,
+            ENTITY_ID_LONGITUDE_ARGUMENT_KEY,
+            SAVE_ZONES_ARGUMENT_KEY,
+            RESTRICTED_ZONES_ARGUMENT_KEY
+    );
 
     @Override
     public CalculatedFieldType getType() {
         return CalculatedFieldType.GEOFENCING;
     }
 
-    public boolean isScheduledUpdateEnabled() {
-        return refreshIntervalSec > 0;
+    // TODO: update validate method in PE version.
+    @Override
+    public void validate() {
+        if (arguments == null || arguments.size() != 4) {
+            throw new IllegalArgumentException("Geofencing calculated field configuration must contain exactly 4 arguments: " + requiredKeys);
+        }
+        for (String requiredKey : requiredKeys) {
+            Argument argument = arguments.get(requiredKey);
+            if (argument == null) {
+                throw new IllegalArgumentException("Missing required argument: " + requiredKey);
+            }
+            ReferencedEntityKey refEntityKey = argument.getRefEntityKey();
+            if (refEntityKey == null || refEntityKey.getType() == null) {
+                throw new IllegalArgumentException("Missing or invalid reference entity key for argument: " + requiredKey);
+            }
+            switch (requiredKey) {
+                case ENTITY_ID_LATITUDE_ARGUMENT_KEY, ENTITY_ID_LONGITUDE_ARGUMENT_KEY -> {
+                    if (!ArgumentType.TS_LATEST.equals(refEntityKey.getType())) {
+                        throw new IllegalArgumentException("Argument: '" + requiredKey + "' must be set to " + ArgumentType.TS_LATEST + " type!");
+                    }
+                    var dynamicSource = argument.getRefDynamicSource();
+                    if (dynamicSource != null) {
+                        String test = "test";
+                        throw new IllegalArgumentException("Dynamic source configuration is forbidden for '" + requiredKey + "' argument. " +
+                                                           "Only '" + SAVE_ZONES_ARGUMENT_KEY + "' and '" + RESTRICTED_ZONES_ARGUMENT_KEY + "' " +
+                                                           "may use dynamic source configuration.");
+                    }
+                }
+                case SAVE_ZONES_ARGUMENT_KEY, RESTRICTED_ZONES_ARGUMENT_KEY -> {
+                    if (!ArgumentType.ATTRIBUTE.equals(refEntityKey.getType())) {
+                        throw new IllegalArgumentException("Argument: '" + requiredKey + "' must be set to " + ArgumentType.ATTRIBUTE + " type!");
+                    }
+                    var dynamicSource = argument.getRefDynamicSource();
+                    if (dynamicSource == null) {
+                        continue;
+                    }
+                    if (!RELATION_QUERY.equals(dynamicSource)) {
+                        throw new IllegalArgumentException("Only relation query dynamic source is supported for argument: " + requiredKey);
+                    }
+                    if (argument.getRefDynamicSourceConfiguration() == null) {
+                        throw new IllegalArgumentException("Missing dynamic source configuration for: " + requiredKey);
+                    }
+                    argument.getRefDynamicSourceConfiguration().validate();
+                }
+            }
+        }
     }
 
 }

common/data/src/main/java/org/thingsboard/server/common/data/cf/configuration/RelationQueryDynamicSourceConfiguration.java

@@ -41,6 +41,19 @@ public class RelationQueryDynamicSourceConfiguration implements CfArgumentDynami
         return CFArgumentDynamicSourceType.RELATION_QUERY;
     }
 
+    @Override
+    public void validate() {
+        if (maxLevel > 2) {
+            throw new IllegalArgumentException("Relation query dynamic source configuration max relation level can't be greater than 2!");
+        }
+        if (direction == null) {
+            throw new IllegalArgumentException("Relation query dynamic source configuration direction must be specified!");
+        }
+        if (relationType == null) {
+            throw new IllegalArgumentException("Relation query dynamic source configuration relation type must be specified!");
+        }
+    }
+
     @Override
     public boolean isSimpleRelation() {
         return maxLevel == 1 && (profiles == null || profiles.isEmpty());

common/data/src/main/java/org/thingsboard/server/common/data/tenant/profile/DefaultTenantProfileConfiguration.java

@@ -172,6 +172,10 @@ public class DefaultTenantProfileConfiguration implements TenantProfileConfigura
     private long maxCalculatedFieldsPerEntity = 5;
     @Schema(example = "10")
     private long maxArgumentsPerCF = 10;
+    @Schema(example = "300")
+    private int minAllowedScheduledUpdateIntervalInSecForCF = 60;
+    @Schema(example = "300")
+    private int maxAllowedScheduledUpdateIntervalInSecForCF = 3600;
     @Builder.Default
     @Min(value = 1, message = "must be at least 1")
     @Schema(example = "1000")

dao/src/main/java/org/thingsboard/server/dao/cf/BaseCalculatedFieldService.java

@@ -78,6 +78,7 @@ public class BaseCalculatedFieldService extends AbstractEntityService implements
             TenantId tenantId = calculatedField.getTenantId();
             log.trace("Executing save calculated field, [{}]", calculatedField);
             updateDebugSettings(tenantId, calculatedField, System.currentTimeMillis());
+            updatedSchedulingConfiguration(calculatedField);
             CalculatedField savedCalculatedField = calculatedFieldDao.save(tenantId, calculatedField);
             createOrUpdateCalculatedFieldLink(tenantId, savedCalculatedField);
             eventPublisher.publishEvent(SaveEntityEvent.builder().tenantId(savedCalculatedField.getTenantId()).entityId(savedCalculatedField.getId())
@@ -91,6 +92,17 @@ public class BaseCalculatedFieldService extends AbstractEntityService implements
         }
     }
 
+    private void updatedSchedulingConfiguration(CalculatedField calculatedField) {
+        CalculatedFieldConfiguration configuration = calculatedField.getConfiguration();
+        if (!configuration.isScheduledUpdateEnabled()) {
+            return;
+        }
+        var defaultProfileConfiguration = tbTenantProfileCache.get(calculatedField.getTenantId()).getDefaultProfileConfiguration();
+        int min = defaultProfileConfiguration.getMinAllowedScheduledUpdateIntervalInSecForCF();
+        int max = defaultProfileConfiguration.getMaxAllowedScheduledUpdateIntervalInSecForCF();
+        configuration.setScheduledUpdateIntervalSec(Math.max(min, Math.min(configuration.getScheduledUpdateIntervalSec(), max)));
+    }
+
     @Override
     public CalculatedField findById(TenantId tenantId, CalculatedFieldId calculatedFieldId) {
         log.trace("Executing findById, tenantId [{}], calculatedFieldId [{}]", tenantId, calculatedFieldId);

dao/src/main/java/org/thingsboard/server/dao/entity/AbstractEntityService.java

@@ -81,7 +81,7 @@ public abstract class AbstractEntityService {
 
     @Autowired
     @Lazy
-    private TbTenantProfileCache tbTenantProfileCache;
+    protected TbTenantProfileCache tbTenantProfileCache;
 
     @Value("${debug.settings.default_duration:15}")
     private int defaultDebugDurationMinutes;

dao/src/main/java/org/thingsboard/server/dao/service/validator/CalculatedFieldDataValidator.java

@@ -18,6 +18,8 @@ package org.thingsboard.server.dao.service.validator;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 import org.thingsboard.server.common.data.cf.CalculatedField;
+import org.thingsboard.server.common.data.cf.CalculatedFieldType;
+import org.thingsboard.server.common.data.cf.configuration.CalculatedFieldConfiguration;
 import org.thingsboard.server.common.data.id.EntityId;
 import org.thingsboard.server.common.data.id.TenantId;
 import org.thingsboard.server.common.data.tenant.profile.DefaultTenantProfileConfiguration;
@@ -39,7 +41,7 @@ public class CalculatedFieldDataValidator extends DataValidator<CalculatedField>
     protected void validateCreate(TenantId tenantId, CalculatedField calculatedField) {
         validateNumberOfCFsPerEntity(tenantId, calculatedField.getEntityId());
         validateNumberOfArgumentsPerCF(tenantId, calculatedField);
-        validateArgumentNames(calculatedField);
+        validateCalculatedFieldConfiguration(calculatedField);
     }
 
     @Override
@@ -49,7 +51,7 @@ public class CalculatedFieldDataValidator extends DataValidator<CalculatedField>
             throw new DataValidationException("Can't update non existing calculated field!");
         }
         validateNumberOfArgumentsPerCF(tenantId, calculatedField);
-        validateArgumentNames(calculatedField);
+        validateCalculatedFieldConfiguration(calculatedField);
         return old;
     }
 
@@ -68,15 +70,26 @@ public class CalculatedFieldDataValidator extends DataValidator<CalculatedField>
         if (maxArgumentsPerCF <= 0) {
             return;
         }
+        if (CalculatedFieldType.GEOFENCING.equals(calculatedField.getType()) && maxArgumentsPerCF < 4) {
+            throw new DataValidationException("Geofencing calculated field requires 4 arguments, but the system limit is " +
+                                              maxArgumentsPerCF + ". Contact your administrator to increase the limit."
+            );
+        }
         if (calculatedField.getConfiguration().getArguments().size() > maxArgumentsPerCF) {
             throw new DataValidationException("Calculated field arguments limit reached!");
         }
     }
 
-    private void validateArgumentNames(CalculatedField calculatedField) {
-        if (calculatedField.getConfiguration().getArguments().containsKey("ctx")) {
+    private void validateCalculatedFieldConfiguration(CalculatedField calculatedField) {
+        CalculatedFieldConfiguration configuration = calculatedField.getConfiguration();
+        if (configuration.getArguments().containsKey("ctx")) {
             throw new DataValidationException("Argument name 'ctx' is reserved and cannot be used.");
         }
+        try {
+            configuration.validate();
+        } catch (IllegalArgumentException e) {
+            throw new DataValidationException(e.getMessage(), e);
+        }
     }
 
 }