Andrii Landiak
1 year ago
32 changed files with 524 additions and 221 deletions
@ -0,0 +1,51 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2025 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.sync.ie.exporting.impl; |
||||
|
|
||||
|
import org.thingsboard.server.common.data.ExportableEntity; |
||||
|
import org.thingsboard.server.common.data.HasTenantId; |
||||
|
import org.thingsboard.server.common.data.cf.CalculatedField; |
||||
|
import org.thingsboard.server.common.data.id.EntityId; |
||||
|
import org.thingsboard.server.common.data.sync.ie.CalculatedFieldExportData; |
||||
|
import org.thingsboard.server.dao.cf.CalculatedFieldService; |
||||
|
import org.thingsboard.server.service.sync.vc.data.EntitiesExportCtx; |
||||
|
|
||||
|
import java.util.List; |
||||
|
|
||||
|
public abstract class BaseCalculatedFieldsExportService<ID extends EntityId, E extends ExportableEntity<ID> & HasTenantId, D extends CalculatedFieldExportData<E>> extends BaseEntityExportService<ID, E, D> { |
||||
|
|
||||
|
protected final CalculatedFieldService calculatedFieldService; |
||||
|
|
||||
|
protected BaseCalculatedFieldsExportService(CalculatedFieldService calculatedFieldService) { |
||||
|
this.calculatedFieldService = calculatedFieldService; |
||||
|
} |
||||
|
|
||||
|
protected void setCalculatedFields(EntitiesExportCtx<?> ctx, E entity, D exportData) { |
||||
|
if (ctx.getSettings().isExportCalculatedFields()) { |
||||
|
List<CalculatedField> calculatedFields = calculatedFieldService.findCalculatedFieldsByEntityId(ctx.getTenantId(), entity.getId()); |
||||
|
calculatedFields.forEach(calculatedField -> { |
||||
|
calculatedField.getConfiguration().getArguments().values().forEach(argument -> { |
||||
|
if (argument.getRefEntityId() != null) { |
||||
|
EntityId externalId = getExternalIdOrElseInternal(ctx, argument.getRefEntityId()); |
||||
|
argument.setRefEntityId(externalId); |
||||
|
} |
||||
|
}); |
||||
|
}); |
||||
|
exportData.setCalculatedFields(calculatedFields); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -1,49 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2025 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.sync.ie.exporting.impl; |
|
||||
|
|
||||
import org.springframework.stereotype.Service; |
|
||||
import org.thingsboard.server.common.data.EntityType; |
|
||||
import org.thingsboard.server.common.data.cf.CalculatedField; |
|
||||
import org.thingsboard.server.common.data.id.CalculatedFieldId; |
|
||||
import org.thingsboard.server.common.data.id.EntityId; |
|
||||
import org.thingsboard.server.common.data.sync.ie.EntityExportData; |
|
||||
import org.thingsboard.server.queue.util.TbCoreComponent; |
|
||||
import org.thingsboard.server.service.sync.vc.data.EntitiesExportCtx; |
|
||||
|
|
||||
import java.util.Set; |
|
||||
|
|
||||
@Service |
|
||||
@TbCoreComponent |
|
||||
public class CalculatedFieldExportService extends BaseEntityExportService<CalculatedFieldId, CalculatedField, EntityExportData<CalculatedField>> { |
|
||||
|
|
||||
@Override |
|
||||
protected void setRelatedEntities(EntitiesExportCtx<?> ctx, CalculatedField calculatedField, EntityExportData<CalculatedField> exportData) { |
|
||||
calculatedField.setEntityId(getExternalIdOrElseInternal(ctx, calculatedField.getEntityId())); |
|
||||
calculatedField.getConfiguration().getArguments().values().forEach(argument -> { |
|
||||
if (argument.getRefEntityId() != null) { |
|
||||
EntityId internalEntityId = getExternalIdOrElseInternal(ctx, argument.getRefEntityId()); |
|
||||
argument.setRefEntityId(internalEntityId); |
|
||||
} |
|
||||
}); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public Set<EntityType> getSupportedEntityTypes() { |
|
||||
return Set.of(EntityType.CALCULATED_FIELD); |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
@ -0,0 +1,123 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2025 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.sync.ie.importing.impl; |
||||
|
|
||||
|
import org.thingsboard.server.common.data.ExportableEntity; |
||||
|
import org.thingsboard.server.common.data.HasTenantId; |
||||
|
import org.thingsboard.server.common.data.cf.CalculatedField; |
||||
|
import org.thingsboard.server.common.data.id.CalculatedFieldId; |
||||
|
import org.thingsboard.server.common.data.id.EntityId; |
||||
|
import org.thingsboard.server.common.data.sync.ie.CalculatedFieldExportData; |
||||
|
import org.thingsboard.server.dao.cf.CalculatedFieldService; |
||||
|
import org.thingsboard.server.service.sync.vc.data.EntitiesImportCtx; |
||||
|
|
||||
|
import java.util.LinkedHashMap; |
||||
|
import java.util.List; |
||||
|
import java.util.Map; |
||||
|
import java.util.function.Function; |
||||
|
import java.util.stream.Collectors; |
||||
|
|
||||
|
public abstract class BaseCalculatedFieldsImportService<ID extends EntityId, E extends ExportableEntity<ID> & HasTenantId, D extends CalculatedFieldExportData<E>> extends BaseEntityImportService<ID, E, D> { |
||||
|
|
||||
|
private final CalculatedFieldService calculatedFieldService; |
||||
|
|
||||
|
protected BaseCalculatedFieldsImportService(CalculatedFieldService calculatedFieldService) { |
||||
|
this.calculatedFieldService = calculatedFieldService; |
||||
|
} |
||||
|
|
||||
|
protected E saveOrUpdateEntity(EntitiesImportCtx ctx, E entity, D exportData, IdProvider idProvider, Function<E, E> saveFunction) { |
||||
|
E savedEntity = saveFunction.apply(entity); |
||||
|
|
||||
|
if (ctx.isFinalImportAttempt() || ctx.getCurrentImportResult().isUpdatedAllExternalIds()) { |
||||
|
saveCalculatedFields(ctx, savedEntity, exportData, idProvider); |
||||
|
} |
||||
|
return savedEntity; |
||||
|
} |
||||
|
|
||||
|
protected void saveCalculatedFields(EntitiesImportCtx ctx, E savedEntity, D exportData, IdProvider idProvider) { |
||||
|
if (exportData.getCalculatedFields() == null || !ctx.isSaveCalculatedFields()) { |
||||
|
return; |
||||
|
} |
||||
|
|
||||
|
exportData.getCalculatedFields().forEach(calculatedField -> { |
||||
|
calculatedField.setTenantId(savedEntity.getTenantId()); |
||||
|
calculatedField.setExternalId(calculatedField.getId()); |
||||
|
calculatedField.setId(idProvider.getInternalId(calculatedField.getId(), false)); |
||||
|
calculatedField.setEntityId(savedEntity.getId()); |
||||
|
|
||||
|
calculatedField.getConfiguration().getArguments().values().forEach(argument -> { |
||||
|
if (argument.getRefEntityId() != null) { |
||||
|
argument.setRefEntityId(idProvider.getInternalId(argument.getRefEntityId(), false)); |
||||
|
} |
||||
|
}); |
||||
|
|
||||
|
calculatedFieldService.save(calculatedField); |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
protected boolean updateRelatedEntitiesIfUnmodified(EntitiesImportCtx ctx, E prepared, D exportData, IdProvider idProvider) { |
||||
|
boolean updated = super.updateRelatedEntitiesIfUnmodified(ctx, prepared, exportData, idProvider); |
||||
|
updated |= updateCalculatedFields(ctx, prepared, exportData, idProvider); |
||||
|
return updated; |
||||
|
} |
||||
|
|
||||
|
private boolean updateCalculatedFields(EntitiesImportCtx ctx, E prepared, D exportData, IdProvider idProvider) { |
||||
|
var calculatedFields = exportData.getCalculatedFields(); |
||||
|
if (calculatedFields == null || !ctx.isSaveCalculatedFields()) { |
||||
|
return false; |
||||
|
} |
||||
|
Map<CalculatedFieldId, CalculatedField> calculatedFieldMap = calculatedFields.stream() |
||||
|
.peek(newField -> { |
||||
|
newField.setTenantId(ctx.getTenantId()); |
||||
|
newField.setExternalId(newField.getId()); |
||||
|
newField.setId(idProvider.getInternalId(newField.getId(), false)); |
||||
|
newField.setEntityId(prepared.getId()); |
||||
|
newField.getConfiguration().getArguments().values().forEach(argument -> { |
||||
|
argument.setRefEntityId(idProvider.getInternalId(argument.getRefEntityId(), false)); |
||||
|
}); |
||||
|
}) |
||||
|
.collect(Collectors.toMap(CalculatedField::getId, field -> field)); |
||||
|
|
||||
|
List<CalculatedField> existingFields = calculatedFieldService.findCalculatedFieldsByEntityId(ctx.getTenantId(), prepared.getId()); |
||||
|
boolean updated = false; |
||||
|
|
||||
|
Map<CalculatedField, Boolean> result = new LinkedHashMap<>(); |
||||
|
for (CalculatedField existingField : existingFields) { |
||||
|
if (calculatedFieldMap.containsKey(existingField.getId())) { |
||||
|
CalculatedField newField = calculatedFieldMap.get(existingField.getId()); |
||||
|
if (!newField.equals(existingField)) { |
||||
|
result.put(newField, false); |
||||
|
} |
||||
|
calculatedFieldMap.remove(existingField.getId()); |
||||
|
} else { |
||||
|
updated = true; |
||||
|
calculatedFieldService.deleteCalculatedField(ctx.getTenantId(), existingField.getId()); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
for (CalculatedField newField : calculatedFieldMap.values()) { |
||||
|
result.put(newField, true); |
||||
|
} |
||||
|
|
||||
|
if (!result.isEmpty()) { |
||||
|
updated = true; |
||||
|
ctx.addCalculatedFields(result); |
||||
|
} |
||||
|
return updated; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -1,85 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2025 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.sync.ie.importing.impl; |
|
||||
|
|
||||
import lombok.RequiredArgsConstructor; |
|
||||
import org.springframework.stereotype.Service; |
|
||||
import org.thingsboard.server.common.data.EntityType; |
|
||||
import org.thingsboard.server.common.data.User; |
|
||||
import org.thingsboard.server.common.data.audit.ActionType; |
|
||||
import org.thingsboard.server.common.data.cf.CalculatedField; |
|
||||
import org.thingsboard.server.common.data.exception.ThingsboardException; |
|
||||
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.sync.ie.EntityExportData; |
|
||||
import org.thingsboard.server.dao.cf.CalculatedFieldService; |
|
||||
import org.thingsboard.server.dao.service.ConstraintValidator; |
|
||||
import org.thingsboard.server.queue.util.TbCoreComponent; |
|
||||
import org.thingsboard.server.service.sync.vc.data.EntitiesImportCtx; |
|
||||
|
|
||||
@Service |
|
||||
@TbCoreComponent |
|
||||
@RequiredArgsConstructor |
|
||||
public class CalculatedFieldImportService extends BaseEntityImportService<CalculatedFieldId, CalculatedField, EntityExportData<CalculatedField>> { |
|
||||
|
|
||||
private final CalculatedFieldService calculatedFieldService; |
|
||||
|
|
||||
@Override |
|
||||
protected void setOwner(TenantId tenantId, CalculatedField calculatedField, IdProvider idProvider) { |
|
||||
calculatedField.setTenantId(tenantId); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected CalculatedField prepare(EntitiesImportCtx ctx, CalculatedField calculatedField, CalculatedField oldEntity, EntityExportData<CalculatedField> exportData, IdProvider idProvider) { |
|
||||
calculatedField.setEntityId(idProvider.getInternalId(calculatedField.getEntityId())); |
|
||||
calculatedField.getConfiguration().getArguments().values().forEach(argument -> { |
|
||||
if (argument.getRefEntityId() != null) { |
|
||||
EntityId internalEntityId = idProvider.getInternalId(argument.getRefEntityId()); |
|
||||
argument.setRefEntityId(internalEntityId); |
|
||||
} |
|
||||
}); |
|
||||
return calculatedField; |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected CalculatedField saveOrUpdate(EntitiesImportCtx ctx, CalculatedField calculatedField, EntityExportData<CalculatedField> exportData, IdProvider idProvider) { |
|
||||
ConstraintValidator.validateFields(calculatedField); |
|
||||
return calculatedFieldService.save(calculatedField); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected CalculatedField deepCopy(CalculatedField calculatedField) { |
|
||||
return new CalculatedField(calculatedField); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void onEntitySaved(User user, CalculatedField savedEntity, CalculatedField oldEntity) throws ThingsboardException { |
|
||||
entityActionService.logEntityAction(user, savedEntity.getId(), savedEntity, null, |
|
||||
oldEntity == null ? ActionType.ADDED : ActionType.UPDATED, null); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void cleanupForComparison(CalculatedField e) { |
|
||||
super.cleanupForComparison(e); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public EntityType getEntityType() { |
|
||||
return EntityType.CALCULATED_FIELD; |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
@ -0,0 +1,28 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2025 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.sync.ie; |
||||
|
|
||||
|
import lombok.Data; |
||||
|
import lombok.EqualsAndHashCode; |
||||
|
import lombok.ToString; |
||||
|
import org.thingsboard.server.common.data.asset.Asset; |
||||
|
|
||||
|
@EqualsAndHashCode(callSuper = true) |
||||
|
@ToString(callSuper = true) |
||||
|
@Data |
||||
|
public class AssetExportData extends CalculatedFieldExportData<Asset> { |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,28 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2025 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.sync.ie; |
||||
|
|
||||
|
import lombok.Data; |
||||
|
import lombok.EqualsAndHashCode; |
||||
|
import lombok.ToString; |
||||
|
import org.thingsboard.server.common.data.asset.AssetProfile; |
||||
|
|
||||
|
@EqualsAndHashCode(callSuper = true) |
||||
|
@ToString(callSuper = true) |
||||
|
@Data |
||||
|
public class AssetProfileExportData extends CalculatedFieldExportData<AssetProfile> { |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,56 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2025 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.sync.ie; |
||||
|
|
||||
|
import com.fasterxml.jackson.annotation.JsonIgnore; |
||||
|
import com.fasterxml.jackson.annotation.JsonIgnoreProperties; |
||||
|
import com.fasterxml.jackson.annotation.JsonProperty; |
||||
|
import lombok.Data; |
||||
|
import lombok.EqualsAndHashCode; |
||||
|
import lombok.ToString; |
||||
|
import org.thingsboard.server.common.data.ExportableEntity; |
||||
|
import org.thingsboard.server.common.data.cf.CalculatedField; |
||||
|
|
||||
|
import java.util.Comparator; |
||||
|
import java.util.List; |
||||
|
|
||||
|
@Data |
||||
|
@EqualsAndHashCode(callSuper = true) |
||||
|
@ToString(callSuper = true) |
||||
|
public class CalculatedFieldExportData<E extends ExportableEntity<?>> extends EntityExportData<E> { |
||||
|
|
||||
|
public static final Comparator<CalculatedField> calculatedFieldsComparator = Comparator.comparing(CalculatedField::getName); |
||||
|
|
||||
|
@JsonProperty(index = 102) |
||||
|
@JsonIgnoreProperties({"entityId", "createdTime", "version"}) |
||||
|
private List<CalculatedField> calculatedFields; |
||||
|
|
||||
|
@JsonIgnore |
||||
|
@Override |
||||
|
public boolean hasCalculatedFields() { |
||||
|
return calculatedFields != null; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public CalculatedFieldExportData<E> sort() { |
||||
|
super.sort(); |
||||
|
if (calculatedFields != null && !calculatedFields.isEmpty()) { |
||||
|
calculatedFields.sort(calculatedFieldsComparator); |
||||
|
} |
||||
|
return this; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,28 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2025 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.sync.ie; |
||||
|
|
||||
|
import lombok.Data; |
||||
|
import lombok.EqualsAndHashCode; |
||||
|
import lombok.ToString; |
||||
|
import org.thingsboard.server.common.data.DeviceProfile; |
||||
|
|
||||
|
@EqualsAndHashCode(callSuper = true) |
||||
|
@ToString(callSuper = true) |
||||
|
@Data |
||||
|
public class DeviceProfileExportData extends CalculatedFieldExportData<DeviceProfile> { |
||||
|
|
||||
|
} |
||||
Loading…
Reference in new issue