Fix entities ids replaced with tenant id in dashboard config when importing
This commit is contained in:
parent
a137243301
commit
dae9a422a1
@ -23,7 +23,6 @@ import lombok.extern.slf4j.Slf4j;
|
|||||||
import org.checkerframework.checker.nullness.qual.Nullable;
|
import org.checkerframework.checker.nullness.qual.Nullable;
|
||||||
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
import org.springframework.context.annotation.Lazy;
|
import org.springframework.context.annotation.Lazy;
|
||||||
import org.springframework.transaction.annotation.Transactional;
|
|
||||||
import org.thingsboard.common.util.JacksonUtil;
|
import org.thingsboard.common.util.JacksonUtil;
|
||||||
import org.thingsboard.server.cluster.TbClusterService;
|
import org.thingsboard.server.cluster.TbClusterService;
|
||||||
import org.thingsboard.server.common.data.EntityType;
|
import org.thingsboard.server.common.data.EntityType;
|
||||||
@ -87,7 +86,6 @@ public abstract class BaseEntityImportService<I extends EntityId, E extends Expo
|
|||||||
@Autowired
|
@Autowired
|
||||||
protected TbNotificationEntityService entityNotificationService;
|
protected TbNotificationEntityService entityNotificationService;
|
||||||
|
|
||||||
@Transactional(rollbackFor = Exception.class)
|
|
||||||
@Override
|
@Override
|
||||||
public EntityImportResult<E> importEntity(EntitiesImportCtx ctx, D exportData) throws ThingsboardException {
|
public EntityImportResult<E> importEntity(EntitiesImportCtx ctx, D exportData) throws ThingsboardException {
|
||||||
EntityImportResult<E> importResult = new EntityImportResult<>();
|
EntityImportResult<E> importResult = new EntityImportResult<>();
|
||||||
@ -336,27 +334,37 @@ public abstract class BaseEntityImportService<I extends EntityId, E extends Expo
|
|||||||
if (externalUuid.equals(EntityId.NULL_UUID)) return Optional.empty();
|
if (externalUuid.equals(EntityId.NULL_UUID)) return Optional.empty();
|
||||||
|
|
||||||
for (EntityType entityType : EntityType.values()) {
|
for (EntityType entityType : EntityType.values()) {
|
||||||
Optional<EntityId> externalIdOpt = buildEntityId(entityType, externalUuid);
|
Optional<EntityId> externalId = buildEntityId(entityType, externalUuid);
|
||||||
if (!externalIdOpt.isPresent()) {
|
if (externalId.isEmpty()) {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
EntityId internalId = ctx.getInternalId(externalIdOpt.get());
|
EntityId internalId = ctx.getInternalId(externalId.get());
|
||||||
if (internalId != null) {
|
if (internalId != null) {
|
||||||
return Optional.of(internalId);
|
return Optional.of(internalId);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (fetchAllUUIDs) {
|
if (fetchAllUUIDs) {
|
||||||
for (EntityType entityType : hints) {
|
Set<EntityType> processLast = Set.of(EntityType.TENANT);
|
||||||
Optional<EntityId> internalId = lookupInDb(externalUuid, entityType);
|
List<EntityType> entityTypes = new ArrayList<>(hints);
|
||||||
if (internalId.isPresent()) return internalId;
|
|
||||||
}
|
|
||||||
for (EntityType entityType : EntityType.values()) {
|
for (EntityType entityType : EntityType.values()) {
|
||||||
if (hints.contains(entityType)) {
|
if (!hints.contains(entityType) && !processLast.contains(entityType)) {
|
||||||
|
entityTypes.add(entityType);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
entityTypes.addAll(processLast);
|
||||||
|
|
||||||
|
for (EntityType entityType : entityTypes) {
|
||||||
|
Optional<EntityId> externalId = buildEntityId(entityType, externalUuid);
|
||||||
|
if (externalId.isEmpty() || ctx.isNotFound(externalId.get())) {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
Optional<EntityId> internalId = lookupInDb(externalUuid, entityType);
|
EntityId internalId = getInternalId(externalId.get(), false);
|
||||||
if (internalId.isPresent()) return internalId;
|
if (internalId != null) {
|
||||||
|
return Optional.of(internalId);
|
||||||
|
} else {
|
||||||
|
ctx.registerNotFound(externalId.get());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -364,20 +372,6 @@ public abstract class BaseEntityImportService<I extends EntityId, E extends Expo
|
|||||||
return Optional.empty();
|
return Optional.empty();
|
||||||
}
|
}
|
||||||
|
|
||||||
private Optional<EntityId> lookupInDb(UUID externalUuid, EntityType entityType) {
|
|
||||||
Optional<EntityId> externalIdOpt = buildEntityId(entityType, externalUuid);
|
|
||||||
if (externalIdOpt.isEmpty() || ctx.isNotFound(externalIdOpt.get())) {
|
|
||||||
return Optional.empty();
|
|
||||||
}
|
|
||||||
EntityId internalId = getInternalId(externalIdOpt.get(), false);
|
|
||||||
if (internalId != null) {
|
|
||||||
return Optional.of(internalId);
|
|
||||||
} else {
|
|
||||||
ctx.registerNotFound(externalIdOpt.get());
|
|
||||||
}
|
|
||||||
return Optional.empty();
|
|
||||||
}
|
|
||||||
|
|
||||||
private Optional<EntityId> buildEntityId(EntityType entityType, UUID externalUuid) {
|
private Optional<EntityId> buildEntityId(EntityType entityType, UUID externalUuid) {
|
||||||
try {
|
try {
|
||||||
return Optional.of(EntityIdFactory.getByTypeAndUuid(entityType, externalUuid));
|
return Optional.of(EntityIdFactory.getByTypeAndUuid(entityType, externalUuid));
|
||||||
|
|||||||
@ -65,10 +65,10 @@ public class DashboardImportService extends BaseEntityImportService<DashboardId,
|
|||||||
@Override
|
@Override
|
||||||
protected Dashboard prepare(EntitiesImportCtx ctx, Dashboard dashboard, Dashboard old, EntityExportData<Dashboard> exportData, IdProvider idProvider) {
|
protected Dashboard prepare(EntitiesImportCtx ctx, Dashboard dashboard, Dashboard old, EntityExportData<Dashboard> exportData, IdProvider idProvider) {
|
||||||
for (JsonNode entityAlias : dashboard.getEntityAliasesConfig()) {
|
for (JsonNode entityAlias : dashboard.getEntityAliasesConfig()) {
|
||||||
replaceIdsRecursively(ctx, idProvider, entityAlias, Collections.emptySet(), HINTS);
|
replaceIdsRecursively(ctx, idProvider, entityAlias, Set.of("id"), HINTS);
|
||||||
}
|
}
|
||||||
for (JsonNode widgetConfig : dashboard.getWidgetsConfig()) {
|
for (JsonNode widgetConfig : dashboard.getWidgetsConfig()) {
|
||||||
replaceIdsRecursively(ctx, idProvider, JacksonUtil.getSafely(widgetConfig, "config", "actions"), Collections.singleton("id"), HINTS);
|
replaceIdsRecursively(ctx, idProvider, JacksonUtil.getSafely(widgetConfig, "config", "actions"), Set.of("id"), HINTS);
|
||||||
}
|
}
|
||||||
return dashboard;
|
return dashboard;
|
||||||
}
|
}
|
||||||
|
|||||||
@ -44,7 +44,6 @@ import org.thingsboard.server.common.data.id.HasId;
|
|||||||
import org.thingsboard.server.common.data.id.TenantId;
|
import org.thingsboard.server.common.data.id.TenantId;
|
||||||
import org.thingsboard.server.common.data.page.PageData;
|
import org.thingsboard.server.common.data.page.PageData;
|
||||||
import org.thingsboard.server.common.data.page.PageLink;
|
import org.thingsboard.server.common.data.page.PageLink;
|
||||||
import org.thingsboard.server.common.data.util.ThrowingRunnable;
|
|
||||||
import org.thingsboard.server.common.data.sync.ie.EntityExportData;
|
import org.thingsboard.server.common.data.sync.ie.EntityExportData;
|
||||||
import org.thingsboard.server.common.data.sync.ie.EntityExportSettings;
|
import org.thingsboard.server.common.data.sync.ie.EntityExportSettings;
|
||||||
import org.thingsboard.server.common.data.sync.ie.EntityImportResult;
|
import org.thingsboard.server.common.data.sync.ie.EntityImportResult;
|
||||||
@ -69,6 +68,7 @@ import org.thingsboard.server.common.data.sync.vc.request.load.EntityTypeVersion
|
|||||||
import org.thingsboard.server.common.data.sync.vc.request.load.SingleEntityVersionLoadRequest;
|
import org.thingsboard.server.common.data.sync.vc.request.load.SingleEntityVersionLoadRequest;
|
||||||
import org.thingsboard.server.common.data.sync.vc.request.load.VersionLoadConfig;
|
import org.thingsboard.server.common.data.sync.vc.request.load.VersionLoadConfig;
|
||||||
import org.thingsboard.server.common.data.sync.vc.request.load.VersionLoadRequest;
|
import org.thingsboard.server.common.data.sync.vc.request.load.VersionLoadRequest;
|
||||||
|
import org.thingsboard.server.common.data.util.ThrowingRunnable;
|
||||||
import org.thingsboard.server.dao.DaoUtil;
|
import org.thingsboard.server.dao.DaoUtil;
|
||||||
import org.thingsboard.server.dao.edge.EdgeService;
|
import org.thingsboard.server.dao.edge.EdgeService;
|
||||||
import org.thingsboard.server.dao.exception.DeviceCredentialsValidationException;
|
import org.thingsboard.server.dao.exception.DeviceCredentialsValidationException;
|
||||||
@ -259,7 +259,7 @@ public class DefaultEntitiesVersionControlService implements EntitiesVersionCont
|
|||||||
return gitServiceQueue.listEntitiesAtVersion(tenantId, versionId);
|
return gitServiceQueue.listEntitiesAtVersion(tenantId, versionId);
|
||||||
}
|
}
|
||||||
|
|
||||||
@SuppressWarnings({"UnstableApiUsage", "rawtypes"})
|
@SuppressWarnings({"rawtypes"})
|
||||||
@Override
|
@Override
|
||||||
public UUID loadEntitiesVersion(User user, VersionLoadRequest request) throws Exception {
|
public UUID loadEntitiesVersion(User user, VersionLoadRequest request) throws Exception {
|
||||||
EntitiesImportCtx ctx = new EntitiesImportCtx(UUID.randomUUID(), user, request.getVersionId());
|
EntitiesImportCtx ctx = new EntitiesImportCtx(UUID.randomUUID(), user, request.getVersionId());
|
||||||
@ -402,7 +402,7 @@ public class DefaultEntitiesVersionControlService implements EntitiesVersionCont
|
|||||||
ctx.getImportedEntities().computeIfAbsent(entityType, t -> new HashSet<>())
|
ctx.getImportedEntities().computeIfAbsent(entityType, t -> new HashSet<>())
|
||||||
.add(importResult.getSavedEntity().getId());
|
.add(importResult.getSavedEntity().getId());
|
||||||
}
|
}
|
||||||
log.debug("Imported {} pack for tenant {}", entityType, ctx.getTenantId());
|
log.debug("Imported {} pack ({}) for tenant {}", entityType, entityDataList.size(), ctx.getTenantId());
|
||||||
offset += limit;
|
offset += limit;
|
||||||
} while (entityDataList.size() == limit);
|
} while (entityDataList.size() == limit);
|
||||||
}
|
}
|
||||||
|
|||||||
@ -68,7 +68,6 @@ import org.thingsboard.server.service.ota.OtaPackageStateService;
|
|||||||
import java.util.HashMap;
|
import java.util.HashMap;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
import java.util.Set;
|
|
||||||
import java.util.stream.Collectors;
|
import java.util.stream.Collectors;
|
||||||
import java.util.stream.Stream;
|
import java.util.stream.Stream;
|
||||||
|
|
||||||
@ -254,21 +253,25 @@ public class ExportImportServiceSqlTest extends BaseExportImportServiceTest {
|
|||||||
Asset asset1 = createAsset(tenantId1, null, assetProfile.getId(), "Asset 1");
|
Asset asset1 = createAsset(tenantId1, null, assetProfile.getId(), "Asset 1");
|
||||||
Asset asset2 = createAsset(tenantId1, null, assetProfile.getId(), "Asset 2");
|
Asset asset2 = createAsset(tenantId1, null, assetProfile.getId(), "Asset 2");
|
||||||
Dashboard dashboard = createDashboard(tenantId1, null, "Dashboard 1");
|
Dashboard dashboard = createDashboard(tenantId1, null, "Dashboard 1");
|
||||||
|
DeviceProfile existingDeviceProfile = createDeviceProfile(tenantId2, null, null, "Existing");
|
||||||
|
|
||||||
|
String aliasId = "23c4185d-1497-9457-30b2-6d91e69a5b2c";
|
||||||
|
String unknownUuid = "ea0dc8b0-3d85-11ed-9200-77fc04fa14fa";
|
||||||
String entityAliases = "{\n" +
|
String entityAliases = "{\n" +
|
||||||
"\t\"23c4185d-1497-9457-30b2-6d91e69a5b2c\": {\n" +
|
"\"" + aliasId + "\": {\n" +
|
||||||
"\t\t\"alias\": \"assets\",\n" +
|
"\"alias\": \"assets\",\n" +
|
||||||
"\t\t\"filter\": {\n" +
|
"\"filter\": {\n" +
|
||||||
"\t\t\t\"entityList\": [\n" +
|
"\"entityList\": [\n" +
|
||||||
"\t\t\t\t\"" + asset1.getId().toString() + "\",\n" +
|
"\"" + asset1.getId().toString() + "\",\n" +
|
||||||
"\t\t\t\t\"" + asset2.getId().toString() + "\"\n" +
|
"\"" + asset2.getId().toString() + "\",\n" +
|
||||||
"\t\t\t],\n" +
|
"\"" + tenantId1.getId().toString() + "\",\n" +
|
||||||
"\t\t\t\"entityType\": \"ASSET\",\n" +
|
"\"" + existingDeviceProfile.getId().toString() + "\",\n" +
|
||||||
"\t\t\t\"resolveMultiple\": true,\n" +
|
"\"" + unknownUuid + "\"\n" +
|
||||||
"\t\t\t\"type\": \"entityList\"\n" +
|
"],\n" +
|
||||||
"\t\t},\n" +
|
"\"resolveMultiple\": true\n" +
|
||||||
"\t\t\"id\": \"23c4185d-1497-9457-30b2-6d91e69a5b2c\"\n" +
|
"},\n" +
|
||||||
"\t}\n" +
|
"\"id\": \"" + aliasId + "\"\n" +
|
||||||
|
"}\n" +
|
||||||
"}";
|
"}";
|
||||||
ObjectNode dashboardConfiguration = JacksonUtil.newObjectNode();
|
ObjectNode dashboardConfiguration = JacksonUtil.newObjectNode();
|
||||||
dashboardConfiguration.set("entityAliases", JacksonUtil.toJsonNode(entityAliases));
|
dashboardConfiguration.set("entityAliases", JacksonUtil.toJsonNode(entityAliases));
|
||||||
@ -287,11 +290,23 @@ public class ExportImportServiceSqlTest extends BaseExportImportServiceTest {
|
|||||||
Asset importedAsset2 = importEntity(tenantAdmin2, asset2ExportData).getSavedEntity();
|
Asset importedAsset2 = importEntity(tenantAdmin2, asset2ExportData).getSavedEntity();
|
||||||
Dashboard importedDashboard = importEntity(tenantAdmin2, dashboardExportData).getSavedEntity();
|
Dashboard importedDashboard = importEntity(tenantAdmin2, dashboardExportData).getSavedEntity();
|
||||||
|
|
||||||
Set<String> entityAliasEntitiesIds = Streams.stream(importedDashboard.getConfiguration()
|
Map.Entry<String, JsonNode> entityAlias = importedDashboard.getConfiguration().get("entityAliases").fields().next();
|
||||||
.get("entityAliases").elements().next().get("filter").get("entityList").elements())
|
assertThat(entityAlias.getKey()).isEqualTo(aliasId);
|
||||||
.map(JsonNode::asText).collect(Collectors.toSet());
|
assertThat(entityAlias.getValue().get("id").asText()).isEqualTo(aliasId);
|
||||||
assertThat(entityAliasEntitiesIds).doesNotContain(asset1.getId().toString(), asset2.getId().toString());
|
|
||||||
assertThat(entityAliasEntitiesIds).contains(importedAsset1.getId().toString(), importedAsset2.getId().toString());
|
List<String> aliasEntitiesIds = Streams.stream(entityAlias.getValue().get("filter").get("entityList").elements())
|
||||||
|
.map(JsonNode::asText).collect(Collectors.toList());
|
||||||
|
assertThat(aliasEntitiesIds).size().isEqualTo(5);
|
||||||
|
assertThat(aliasEntitiesIds).element(0).as("external asset 1 was replaced with imported one")
|
||||||
|
.isEqualTo(importedAsset1.getId().toString());
|
||||||
|
assertThat(aliasEntitiesIds).element(1).as("external asset 2 was replaced with imported one")
|
||||||
|
.isEqualTo(importedAsset2.getId().toString());
|
||||||
|
assertThat(aliasEntitiesIds).element(2).as("external tenant id was replaced with new tenant id")
|
||||||
|
.isEqualTo(tenantId2.toString());
|
||||||
|
assertThat(aliasEntitiesIds).element(3).as("existing device profile id was left as is")
|
||||||
|
.isEqualTo(existingDeviceProfile.getId().toString());
|
||||||
|
assertThat(aliasEntitiesIds).element(4).as("unresolved uuid was replaced with tenant id")
|
||||||
|
.isEqualTo(tenantId2.toString());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@ -469,7 +484,7 @@ public class ExportImportServiceSqlTest extends BaseExportImportServiceTest {
|
|||||||
Device device = createDevice(tenantId1, null, deviceProfile.getId(), "Device 1");
|
Device device = createDevice(tenantId1, null, deviceProfile.getId(), "Device 1");
|
||||||
|
|
||||||
Map<EntityType, EntityExportData> entitiesExportData = Stream.of(customer.getId(), asset.getId(), device.getId(),
|
Map<EntityType, EntityExportData> entitiesExportData = Stream.of(customer.getId(), asset.getId(), device.getId(),
|
||||||
ruleChain.getId(), dashboard.getId(), assetProfile.getId(), deviceProfile.getId())
|
ruleChain.getId(), dashboard.getId(), assetProfile.getId(), deviceProfile.getId())
|
||||||
.map(entityId -> {
|
.map(entityId -> {
|
||||||
try {
|
try {
|
||||||
return exportEntity(tenantAdmin1, entityId, EntityExportSettings.builder()
|
return exportEntity(tenantAdmin1, entityId, EntityExportSettings.builder()
|
||||||
|
|||||||
@ -33,14 +33,19 @@
|
|||||||
<pattern>%d{ISO8601} [%thread] %-5level %logger{36} - %msg%n</pattern>
|
<pattern>%d{ISO8601} [%thread] %-5level %logger{36} - %msg%n</pattern>
|
||||||
</encoder>
|
</encoder>
|
||||||
</appender>
|
</appender>
|
||||||
|
<appender name="STDOUT" class="ch.qos.logback.core.ConsoleAppender">
|
||||||
|
<encoder>
|
||||||
|
<pattern>%d{ISO8601} [%thread] %-5level %logger{36} - %msg%n</pattern>
|
||||||
|
</encoder>
|
||||||
|
</appender>
|
||||||
|
|
||||||
<logger name="org" level="WARN"/>
|
<logger name="org" level="WARN"/>
|
||||||
<logger name="org.thingsboard.server" level="INFO"/>
|
|
||||||
<logger name="org.thingsboard.monitoring" level="DEBUG"/>
|
<logger name="org.thingsboard.monitoring" level="DEBUG"/>
|
||||||
<logger name="org.thingsboard.monitoring.client" level="INFO"/>
|
<logger name="org.thingsboard.monitoring.client" level="INFO"/>
|
||||||
|
|
||||||
<root level="INFO">
|
<root level="INFO">
|
||||||
<appender-ref ref="fileLogAppender"/>
|
<appender-ref ref="fileLogAppender"/>
|
||||||
|
<appender-ref ref="STDOUT"/>
|
||||||
</root>
|
</root>
|
||||||
|
|
||||||
</configuration>
|
</configuration>
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user