Browse Source

Minor refactoring for resources import

pull/11873/head
ViacheslavKlimov 2 years ago
parent
commit
19941d9fd1
  1. 15
      application/src/main/java/org/thingsboard/server/service/install/update/DefaultDataUpdateService.java
  2. 8
      application/src/main/java/org/thingsboard/server/service/resource/DefaultTbImageService.java
  3. 11
      application/src/main/java/org/thingsboard/server/service/resource/DefaultTbResourceService.java

15
application/src/main/java/org/thingsboard/server/service/install/update/DefaultDataUpdateService.java

@ -35,15 +35,11 @@ import org.thingsboard.server.common.data.query.DynamicValue;
import org.thingsboard.server.common.data.query.FilterPredicateValue; import org.thingsboard.server.common.data.query.FilterPredicateValue;
import org.thingsboard.server.dao.customer.CustomerDao; import org.thingsboard.server.dao.customer.CustomerDao;
import org.thingsboard.server.dao.customer.CustomerService; import org.thingsboard.server.dao.customer.CustomerService;
import org.thingsboard.server.dao.dashboard.DashboardService;
import org.thingsboard.server.dao.device.DeviceConnectivityConfiguration; import org.thingsboard.server.dao.device.DeviceConnectivityConfiguration;
import org.thingsboard.server.dao.resource.ImageService;
import org.thingsboard.server.dao.resource.ResourceService;
import org.thingsboard.server.dao.rule.RuleChainService; import org.thingsboard.server.dao.rule.RuleChainService;
import org.thingsboard.server.dao.settings.AdminSettingsService; import org.thingsboard.server.dao.settings.AdminSettingsService;
import org.thingsboard.server.dao.sql.JpaExecutorService; import org.thingsboard.server.dao.sql.JpaExecutorService;
import org.thingsboard.server.dao.tenant.TenantProfileService; import org.thingsboard.server.dao.tenant.TenantProfileService;
import org.thingsboard.server.dao.widget.WidgetTypeService;
import org.thingsboard.server.service.component.ComponentDiscoveryService; import org.thingsboard.server.service.component.ComponentDiscoveryService;
import org.thingsboard.server.service.component.RuleNodeClassInfo; import org.thingsboard.server.service.component.RuleNodeClassInfo;
import org.thingsboard.server.utils.TbNodeUpgradeUtils; import org.thingsboard.server.utils.TbNodeUpgradeUtils;
@ -85,17 +81,6 @@ public class DefaultDataUpdateService implements DataUpdateService {
@Autowired @Autowired
private TenantProfileService tenantProfileService; private TenantProfileService tenantProfileService;
@Autowired
private ResourceService resourceService;
@Autowired
private ImageService imageService;
@Autowired
private DashboardService dashboardService;
@Autowired
private WidgetTypeService widgetTypeService;
@Override @Override
public void updateData(String fromVersion) throws Exception { public void updateData(String fromVersion) throws Exception {
switch (fromVersion) { switch (fromVersion) {

8
application/src/main/java/org/thingsboard/server/service/resource/DefaultTbImageService.java

@ -185,14 +185,12 @@ public class DefaultTbImageService extends AbstractTbEntityService implements Tb
@Override @Override
public TbResourceInfo importImage(ResourceExportData imageData, boolean checkExisting, SecurityUser user) throws Exception { public TbResourceInfo importImage(ResourceExportData imageData, boolean checkExisting, SecurityUser user) throws Exception {
TbResource image = new TbResource(); TbResource image = imageService.toImage(user.getTenantId(), imageData, checkExisting);
image.setTenantId(user.getTenantId());
accessControlService.checkPermission(user, Resource.TB_RESOURCE, Operation.CREATE, null, image);
image = imageService.toImage(user.getTenantId(), imageData, checkExisting);
if (checkExisting && image.getId() != null) { if (checkExisting && image.getId() != null) {
accessControlService.checkPermission(user, Resource.TB_RESOURCE, Operation.READ, image.getId(), image); accessControlService.checkPermission(user, Resource.TB_RESOURCE, Operation.READ, image.getId(), image);
return image; return image;
} else {
accessControlService.checkPermission(user, Resource.TB_RESOURCE, Operation.CREATE, null, image);
} }
return save(image, user); return save(image, user);
} }

11
application/src/main/java/org/thingsboard/server/service/resource/DefaultTbResourceService.java

@ -143,10 +143,6 @@ public class DefaultTbResourceService extends AbstractTbEntityService implements
@Override @Override
public void importResources(List<ResourceExportData> resources, SecurityUser user) throws Exception { public void importResources(List<ResourceExportData> resources, SecurityUser user) throws Exception {
for (ResourceExportData resourceData : resources) { for (ResourceExportData resourceData : resources) {
TbResource resource = new TbResource();
resource.setTenantId(user.getTenantId());
accessControlService.checkPermission(user, Resource.TB_RESOURCE, Operation.CREATE, null, resource);
TbResourceInfo resourceInfo; TbResourceInfo resourceInfo;
if (resourceData.getType() == ResourceType.IMAGE) { if (resourceData.getType() == ResourceType.IMAGE) {
resourceInfo = tbImageService.importImage(resourceData, true, user); resourceInfo = tbImageService.importImage(resourceData, true, user);
@ -172,12 +168,9 @@ public class DefaultTbResourceService extends AbstractTbEntityService implements
} }
private TbResourceInfo importResource(ResourceExportData resourceData, SecurityUser user) throws ThingsboardException { private TbResourceInfo importResource(ResourceExportData resourceData, SecurityUser user) throws ThingsboardException {
TbResource resource = new TbResource(); TbResource resource = resourceService.toResource(user.getTenantId(), resourceData);
resource.setTenantId(user.getTenantId());
accessControlService.checkPermission(user, Resource.TB_RESOURCE, Operation.CREATE, null, resource);
resource = resourceService.toResource(user.getTenantId(), resourceData);
if (resource.getData() != null) { if (resource.getData() != null) {
accessControlService.checkPermission(user, Resource.TB_RESOURCE, Operation.CREATE, null, resource);
return save(resource, user); return save(resource, user);
} else { } else {
accessControlService.checkPermission(user, Resource.TB_RESOURCE, Operation.READ, resource.getId(), resource); accessControlService.checkPermission(user, Resource.TB_RESOURCE, Operation.READ, resource.getId(), resource);

Loading…
Cancel
Save