Andrew Shvayka
5 years ago
committed by
GitHub
GitHub
126 changed files with 2257 additions and 2241 deletions
@ -1,222 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2021 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.controller; |
|
||||
|
|
||||
import lombok.extern.slf4j.Slf4j; |
|
||||
import org.apache.commons.lang3.StringUtils; |
|
||||
import org.springframework.core.io.ByteArrayResource; |
|
||||
import org.springframework.http.HttpHeaders; |
|
||||
import org.springframework.http.ResponseEntity; |
|
||||
import org.springframework.security.access.prepost.PreAuthorize; |
|
||||
import org.springframework.web.bind.annotation.PathVariable; |
|
||||
import org.springframework.web.bind.annotation.RequestBody; |
|
||||
import org.springframework.web.bind.annotation.RequestMapping; |
|
||||
import org.springframework.web.bind.annotation.RequestMethod; |
|
||||
import org.springframework.web.bind.annotation.RequestParam; |
|
||||
import org.springframework.web.bind.annotation.ResponseBody; |
|
||||
import org.springframework.web.bind.annotation.RestController; |
|
||||
import org.springframework.web.multipart.MultipartFile; |
|
||||
import org.thingsboard.server.common.data.EntityType; |
|
||||
import org.thingsboard.server.common.data.Firmware; |
|
||||
import org.thingsboard.server.common.data.FirmwareInfo; |
|
||||
import org.thingsboard.server.common.data.audit.ActionType; |
|
||||
import org.thingsboard.server.common.data.exception.ThingsboardException; |
|
||||
import org.thingsboard.server.common.data.firmware.ChecksumAlgorithm; |
|
||||
import org.thingsboard.server.common.data.firmware.FirmwareType; |
|
||||
import org.thingsboard.server.common.data.id.DeviceProfileId; |
|
||||
import org.thingsboard.server.common.data.id.FirmwareId; |
|
||||
import org.thingsboard.server.common.data.page.PageData; |
|
||||
import org.thingsboard.server.common.data.page.PageLink; |
|
||||
import org.thingsboard.server.queue.util.TbCoreComponent; |
|
||||
import org.thingsboard.server.service.security.permission.Operation; |
|
||||
import org.thingsboard.server.service.security.permission.Resource; |
|
||||
|
|
||||
import java.nio.ByteBuffer; |
|
||||
|
|
||||
@Slf4j |
|
||||
@RestController |
|
||||
@TbCoreComponent |
|
||||
@RequestMapping("/api") |
|
||||
public class FirmwareController extends BaseController { |
|
||||
|
|
||||
public static final String FIRMWARE_ID = "firmwareId"; |
|
||||
public static final String CHECKSUM_ALGORITHM = "checksumAlgorithm"; |
|
||||
|
|
||||
@PreAuthorize("hasAnyAuthority( 'TENANT_ADMIN')") |
|
||||
@RequestMapping(value = "/firmware/{firmwareId}/download", method = RequestMethod.GET) |
|
||||
@ResponseBody |
|
||||
public ResponseEntity<org.springframework.core.io.Resource> downloadFirmware(@PathVariable(FIRMWARE_ID) String strFirmwareId) throws ThingsboardException { |
|
||||
checkParameter(FIRMWARE_ID, strFirmwareId); |
|
||||
try { |
|
||||
FirmwareId firmwareId = new FirmwareId(toUUID(strFirmwareId)); |
|
||||
Firmware firmware = checkFirmwareId(firmwareId, Operation.READ); |
|
||||
|
|
||||
ByteArrayResource resource = new ByteArrayResource(firmware.getData().array()); |
|
||||
return ResponseEntity.ok() |
|
||||
.header(HttpHeaders.CONTENT_DISPOSITION, "attachment;filename=" + firmware.getFileName()) |
|
||||
.header("x-filename", firmware.getFileName()) |
|
||||
.contentLength(resource.contentLength()) |
|
||||
.contentType(parseMediaType(firmware.getContentType())) |
|
||||
.body(resource); |
|
||||
} catch (Exception e) { |
|
||||
throw handleException(e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN', 'CUSTOMER_USER')") |
|
||||
@RequestMapping(value = "/firmware/info/{firmwareId}", method = RequestMethod.GET) |
|
||||
@ResponseBody |
|
||||
public FirmwareInfo getFirmwareInfoById(@PathVariable(FIRMWARE_ID) String strFirmwareId) throws ThingsboardException { |
|
||||
checkParameter(FIRMWARE_ID, strFirmwareId); |
|
||||
try { |
|
||||
FirmwareId firmwareId = new FirmwareId(toUUID(strFirmwareId)); |
|
||||
return checkNotNull(firmwareService.findFirmwareInfoById(getTenantId(), firmwareId)); |
|
||||
} catch (Exception e) { |
|
||||
throw handleException(e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN')") |
|
||||
@RequestMapping(value = "/firmware/{firmwareId}", method = RequestMethod.GET) |
|
||||
@ResponseBody |
|
||||
public Firmware getFirmwareById(@PathVariable(FIRMWARE_ID) String strFirmwareId) throws ThingsboardException { |
|
||||
checkParameter(FIRMWARE_ID, strFirmwareId); |
|
||||
try { |
|
||||
FirmwareId firmwareId = new FirmwareId(toUUID(strFirmwareId)); |
|
||||
return checkFirmwareId(firmwareId, Operation.READ); |
|
||||
} catch (Exception e) { |
|
||||
throw handleException(e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN')") |
|
||||
@RequestMapping(value = "/firmware", method = RequestMethod.POST) |
|
||||
@ResponseBody |
|
||||
public FirmwareInfo saveFirmwareInfo(@RequestBody FirmwareInfo firmwareInfo) throws ThingsboardException { |
|
||||
boolean created = firmwareInfo.getId() == null; |
|
||||
try { |
|
||||
firmwareInfo.setTenantId(getTenantId()); |
|
||||
checkEntity(firmwareInfo.getId(), firmwareInfo, Resource.FIRMWARE); |
|
||||
FirmwareInfo savedFirmwareInfo = firmwareService.saveFirmwareInfo(firmwareInfo); |
|
||||
logEntityAction(savedFirmwareInfo.getId(), savedFirmwareInfo, |
|
||||
null, created ? ActionType.ADDED : ActionType.UPDATED, null); |
|
||||
return savedFirmwareInfo; |
|
||||
} catch (Exception e) { |
|
||||
logEntityAction(emptyId(EntityType.FIRMWARE), firmwareInfo, |
|
||||
null, created ? ActionType.ADDED : ActionType.UPDATED, e); |
|
||||
throw handleException(e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN')") |
|
||||
@RequestMapping(value = "/firmware/{firmwareId}", method = RequestMethod.POST) |
|
||||
@ResponseBody |
|
||||
public Firmware saveFirmwareData(@PathVariable(FIRMWARE_ID) String strFirmwareId, |
|
||||
@RequestParam(required = false) String checksum, |
|
||||
@RequestParam(CHECKSUM_ALGORITHM) String checksumAlgorithmStr, |
|
||||
@RequestBody MultipartFile file) throws ThingsboardException { |
|
||||
checkParameter(FIRMWARE_ID, strFirmwareId); |
|
||||
checkParameter(CHECKSUM_ALGORITHM, checksumAlgorithmStr); |
|
||||
try { |
|
||||
FirmwareId firmwareId = new FirmwareId(toUUID(strFirmwareId)); |
|
||||
FirmwareInfo info = checkFirmwareInfoId(firmwareId, Operation.READ); |
|
||||
|
|
||||
Firmware firmware = new Firmware(firmwareId); |
|
||||
firmware.setCreatedTime(info.getCreatedTime()); |
|
||||
firmware.setTenantId(getTenantId()); |
|
||||
firmware.setDeviceProfileId(info.getDeviceProfileId()); |
|
||||
firmware.setType(info.getType()); |
|
||||
firmware.setTitle(info.getTitle()); |
|
||||
firmware.setVersion(info.getVersion()); |
|
||||
firmware.setAdditionalInfo(info.getAdditionalInfo()); |
|
||||
|
|
||||
ChecksumAlgorithm checksumAlgorithm = ChecksumAlgorithm.valueOf(checksumAlgorithmStr.toUpperCase()); |
|
||||
|
|
||||
byte[] bytes = file.getBytes(); |
|
||||
if (StringUtils.isEmpty(checksum)) { |
|
||||
checksum = firmwareService.generateChecksum(checksumAlgorithm, ByteBuffer.wrap(bytes)); |
|
||||
} |
|
||||
|
|
||||
firmware.setChecksumAlgorithm(checksumAlgorithm); |
|
||||
firmware.setChecksum(checksum); |
|
||||
firmware.setFileName(file.getOriginalFilename()); |
|
||||
firmware.setContentType(file.getContentType()); |
|
||||
firmware.setData(ByteBuffer.wrap(bytes)); |
|
||||
firmware.setDataSize((long) bytes.length); |
|
||||
Firmware savedFirmware = firmwareService.saveFirmware(firmware); |
|
||||
logEntityAction(savedFirmware.getId(), savedFirmware, null, ActionType.UPDATED, null); |
|
||||
return savedFirmware; |
|
||||
} catch (Exception e) { |
|
||||
logEntityAction(emptyId(EntityType.FIRMWARE), null, null, ActionType.UPDATED, e, strFirmwareId); |
|
||||
throw handleException(e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN', 'CUSTOMER_USER')") |
|
||||
@RequestMapping(value = "/firmwares", method = RequestMethod.GET) |
|
||||
@ResponseBody |
|
||||
public PageData<FirmwareInfo> getFirmwares(@RequestParam int pageSize, |
|
||||
@RequestParam int page, |
|
||||
@RequestParam(required = false) String textSearch, |
|
||||
@RequestParam(required = false) String sortProperty, |
|
||||
@RequestParam(required = false) String sortOrder) throws ThingsboardException { |
|
||||
try { |
|
||||
PageLink pageLink = createPageLink(pageSize, page, textSearch, sortProperty, sortOrder); |
|
||||
return checkNotNull(firmwareService.findTenantFirmwaresByTenantId(getTenantId(), pageLink)); |
|
||||
} catch (Exception e) { |
|
||||
throw handleException(e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN', 'CUSTOMER_USER')") |
|
||||
@RequestMapping(value = "/firmwares/{deviceProfileId}/{type}/{hasData}", method = RequestMethod.GET) |
|
||||
@ResponseBody |
|
||||
public PageData<FirmwareInfo> getFirmwares(@PathVariable("deviceProfileId") String strDeviceProfileId, |
|
||||
@PathVariable("type") String strType, |
|
||||
@PathVariable("hasData") boolean hasData, |
|
||||
@RequestParam int pageSize, |
|
||||
@RequestParam int page, |
|
||||
@RequestParam(required = false) String textSearch, |
|
||||
@RequestParam(required = false) String sortProperty, |
|
||||
@RequestParam(required = false) String sortOrder) throws ThingsboardException { |
|
||||
checkParameter("deviceProfileId", strDeviceProfileId); |
|
||||
checkParameter("type", strType); |
|
||||
try { |
|
||||
PageLink pageLink = createPageLink(pageSize, page, textSearch, sortProperty, sortOrder); |
|
||||
return checkNotNull(firmwareService.findTenantFirmwaresByTenantIdAndDeviceProfileIdAndTypeAndHasData(getTenantId(), |
|
||||
new DeviceProfileId(toUUID(strDeviceProfileId)), FirmwareType.valueOf(strType), hasData, pageLink)); |
|
||||
} catch (Exception e) { |
|
||||
throw handleException(e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN')") |
|
||||
@RequestMapping(value = "/firmware/{firmwareId}", method = RequestMethod.DELETE) |
|
||||
@ResponseBody |
|
||||
public void deleteFirmware(@PathVariable("firmwareId") String strFirmwareId) throws ThingsboardException { |
|
||||
checkParameter(FIRMWARE_ID, strFirmwareId); |
|
||||
try { |
|
||||
FirmwareId firmwareId = new FirmwareId(toUUID(strFirmwareId)); |
|
||||
FirmwareInfo info = checkFirmwareInfoId(firmwareId, Operation.DELETE); |
|
||||
firmwareService.deleteFirmware(getTenantId(), firmwareId); |
|
||||
logEntityAction(firmwareId, info, null, ActionType.DELETED, null, strFirmwareId); |
|
||||
} catch (Exception e) { |
|
||||
logEntityAction(emptyId(EntityType.FIRMWARE), null, null, ActionType.DELETED, e, strFirmwareId); |
|
||||
throw handleException(e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
@ -0,0 +1,222 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2021 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.controller; |
||||
|
|
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.apache.commons.lang3.StringUtils; |
||||
|
import org.springframework.core.io.ByteArrayResource; |
||||
|
import org.springframework.http.HttpHeaders; |
||||
|
import org.springframework.http.ResponseEntity; |
||||
|
import org.springframework.security.access.prepost.PreAuthorize; |
||||
|
import org.springframework.web.bind.annotation.PathVariable; |
||||
|
import org.springframework.web.bind.annotation.RequestBody; |
||||
|
import org.springframework.web.bind.annotation.RequestMapping; |
||||
|
import org.springframework.web.bind.annotation.RequestMethod; |
||||
|
import org.springframework.web.bind.annotation.RequestParam; |
||||
|
import org.springframework.web.bind.annotation.ResponseBody; |
||||
|
import org.springframework.web.bind.annotation.RestController; |
||||
|
import org.springframework.web.multipart.MultipartFile; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
import org.thingsboard.server.common.data.OtaPackage; |
||||
|
import org.thingsboard.server.common.data.OtaPackageInfo; |
||||
|
import org.thingsboard.server.common.data.audit.ActionType; |
||||
|
import org.thingsboard.server.common.data.exception.ThingsboardException; |
||||
|
import org.thingsboard.server.common.data.ota.ChecksumAlgorithm; |
||||
|
import org.thingsboard.server.common.data.ota.OtaPackageType; |
||||
|
import org.thingsboard.server.common.data.id.DeviceProfileId; |
||||
|
import org.thingsboard.server.common.data.id.OtaPackageId; |
||||
|
import org.thingsboard.server.common.data.page.PageData; |
||||
|
import org.thingsboard.server.common.data.page.PageLink; |
||||
|
import org.thingsboard.server.queue.util.TbCoreComponent; |
||||
|
import org.thingsboard.server.service.security.permission.Operation; |
||||
|
import org.thingsboard.server.service.security.permission.Resource; |
||||
|
|
||||
|
import java.nio.ByteBuffer; |
||||
|
|
||||
|
@Slf4j |
||||
|
@RestController |
||||
|
@TbCoreComponent |
||||
|
@RequestMapping("/api") |
||||
|
public class OtaPackageController extends BaseController { |
||||
|
|
||||
|
public static final String OTA_PACKAGE_ID = "otaPackageId"; |
||||
|
public static final String CHECKSUM_ALGORITHM = "checksumAlgorithm"; |
||||
|
|
||||
|
@PreAuthorize("hasAnyAuthority( 'TENANT_ADMIN')") |
||||
|
@RequestMapping(value = "/otaPackage/{otaPackageId}/download", method = RequestMethod.GET) |
||||
|
@ResponseBody |
||||
|
public ResponseEntity<org.springframework.core.io.Resource> downloadOtaPackage(@PathVariable(OTA_PACKAGE_ID) String strOtaPackageId) throws ThingsboardException { |
||||
|
checkParameter(OTA_PACKAGE_ID, strOtaPackageId); |
||||
|
try { |
||||
|
OtaPackageId otaPackageId = new OtaPackageId(toUUID(strOtaPackageId)); |
||||
|
OtaPackage otaPackage = checkOtaPackageId(otaPackageId, Operation.READ); |
||||
|
|
||||
|
ByteArrayResource resource = new ByteArrayResource(otaPackage.getData().array()); |
||||
|
return ResponseEntity.ok() |
||||
|
.header(HttpHeaders.CONTENT_DISPOSITION, "attachment;filename=" + otaPackage.getFileName()) |
||||
|
.header("x-filename", otaPackage.getFileName()) |
||||
|
.contentLength(resource.contentLength()) |
||||
|
.contentType(parseMediaType(otaPackage.getContentType())) |
||||
|
.body(resource); |
||||
|
} catch (Exception e) { |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN', 'CUSTOMER_USER')") |
||||
|
@RequestMapping(value = "/otaPackage/info/{otaPackageId}", method = RequestMethod.GET) |
||||
|
@ResponseBody |
||||
|
public OtaPackageInfo getOtaPackageInfoById(@PathVariable(OTA_PACKAGE_ID) String strOtaPackageId) throws ThingsboardException { |
||||
|
checkParameter(OTA_PACKAGE_ID, strOtaPackageId); |
||||
|
try { |
||||
|
OtaPackageId otaPackageId = new OtaPackageId(toUUID(strOtaPackageId)); |
||||
|
return checkNotNull(otaPackageService.findOtaPackageInfoById(getTenantId(), otaPackageId)); |
||||
|
} catch (Exception e) { |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN')") |
||||
|
@RequestMapping(value = "/otaPackage/{otaPackageId}", method = RequestMethod.GET) |
||||
|
@ResponseBody |
||||
|
public OtaPackage getOtaPackageById(@PathVariable(OTA_PACKAGE_ID) String strOtaPackageId) throws ThingsboardException { |
||||
|
checkParameter(OTA_PACKAGE_ID, strOtaPackageId); |
||||
|
try { |
||||
|
OtaPackageId otaPackageId = new OtaPackageId(toUUID(strOtaPackageId)); |
||||
|
return checkOtaPackageId(otaPackageId, Operation.READ); |
||||
|
} catch (Exception e) { |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN')") |
||||
|
@RequestMapping(value = "/otaPackage", method = RequestMethod.POST) |
||||
|
@ResponseBody |
||||
|
public OtaPackageInfo saveOtaPackageInfo(@RequestBody OtaPackageInfo otaPackageInfo) throws ThingsboardException { |
||||
|
boolean created = otaPackageInfo.getId() == null; |
||||
|
try { |
||||
|
otaPackageInfo.setTenantId(getTenantId()); |
||||
|
checkEntity(otaPackageInfo.getId(), otaPackageInfo, Resource.OTA_PACKAGE); |
||||
|
OtaPackageInfo savedOtaPackageInfo = otaPackageService.saveOtaPackageInfo(otaPackageInfo); |
||||
|
logEntityAction(savedOtaPackageInfo.getId(), savedOtaPackageInfo, |
||||
|
null, created ? ActionType.ADDED : ActionType.UPDATED, null); |
||||
|
return savedOtaPackageInfo; |
||||
|
} catch (Exception e) { |
||||
|
logEntityAction(emptyId(EntityType.OTA_PACKAGE), otaPackageInfo, |
||||
|
null, created ? ActionType.ADDED : ActionType.UPDATED, e); |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN')") |
||||
|
@RequestMapping(value = "/otaPackage/{otaPackageId}", method = RequestMethod.POST) |
||||
|
@ResponseBody |
||||
|
public OtaPackage saveOtaPackageData(@PathVariable(OTA_PACKAGE_ID) String strOtaPackageId, |
||||
|
@RequestParam(required = false) String checksum, |
||||
|
@RequestParam(CHECKSUM_ALGORITHM) String checksumAlgorithmStr, |
||||
|
@RequestBody MultipartFile file) throws ThingsboardException { |
||||
|
checkParameter(OTA_PACKAGE_ID, strOtaPackageId); |
||||
|
checkParameter(CHECKSUM_ALGORITHM, checksumAlgorithmStr); |
||||
|
try { |
||||
|
OtaPackageId otaPackageId = new OtaPackageId(toUUID(strOtaPackageId)); |
||||
|
OtaPackageInfo info = checkOtaPackageInfoId(otaPackageId, Operation.READ); |
||||
|
|
||||
|
OtaPackage otaPackage = new OtaPackage(otaPackageId); |
||||
|
otaPackage.setCreatedTime(info.getCreatedTime()); |
||||
|
otaPackage.setTenantId(getTenantId()); |
||||
|
otaPackage.setDeviceProfileId(info.getDeviceProfileId()); |
||||
|
otaPackage.setType(info.getType()); |
||||
|
otaPackage.setTitle(info.getTitle()); |
||||
|
otaPackage.setVersion(info.getVersion()); |
||||
|
otaPackage.setAdditionalInfo(info.getAdditionalInfo()); |
||||
|
|
||||
|
ChecksumAlgorithm checksumAlgorithm = ChecksumAlgorithm.valueOf(checksumAlgorithmStr.toUpperCase()); |
||||
|
|
||||
|
byte[] bytes = file.getBytes(); |
||||
|
if (StringUtils.isEmpty(checksum)) { |
||||
|
checksum = otaPackageService.generateChecksum(checksumAlgorithm, ByteBuffer.wrap(bytes)); |
||||
|
} |
||||
|
|
||||
|
otaPackage.setChecksumAlgorithm(checksumAlgorithm); |
||||
|
otaPackage.setChecksum(checksum); |
||||
|
otaPackage.setFileName(file.getOriginalFilename()); |
||||
|
otaPackage.setContentType(file.getContentType()); |
||||
|
otaPackage.setData(ByteBuffer.wrap(bytes)); |
||||
|
otaPackage.setDataSize((long) bytes.length); |
||||
|
OtaPackage savedOtaPackage = otaPackageService.saveOtaPackage(otaPackage); |
||||
|
logEntityAction(savedOtaPackage.getId(), savedOtaPackage, null, ActionType.UPDATED, null); |
||||
|
return savedOtaPackage; |
||||
|
} catch (Exception e) { |
||||
|
logEntityAction(emptyId(EntityType.OTA_PACKAGE), null, null, ActionType.UPDATED, e, strOtaPackageId); |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN', 'CUSTOMER_USER')") |
||||
|
@RequestMapping(value = "/otaPackages", method = RequestMethod.GET) |
||||
|
@ResponseBody |
||||
|
public PageData<OtaPackageInfo> getOtaPackages(@RequestParam int pageSize, |
||||
|
@RequestParam int page, |
||||
|
@RequestParam(required = false) String textSearch, |
||||
|
@RequestParam(required = false) String sortProperty, |
||||
|
@RequestParam(required = false) String sortOrder) throws ThingsboardException { |
||||
|
try { |
||||
|
PageLink pageLink = createPageLink(pageSize, page, textSearch, sortProperty, sortOrder); |
||||
|
return checkNotNull(otaPackageService.findTenantOtaPackagesByTenantId(getTenantId(), pageLink)); |
||||
|
} catch (Exception e) { |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN', 'CUSTOMER_USER')") |
||||
|
@RequestMapping(value = "/otaPackages/{deviceProfileId}/{type}/{hasData}", method = RequestMethod.GET) |
||||
|
@ResponseBody |
||||
|
public PageData<OtaPackageInfo> getOtaPackages(@PathVariable("deviceProfileId") String strDeviceProfileId, |
||||
|
@PathVariable("type") String strType, |
||||
|
@PathVariable("hasData") boolean hasData, |
||||
|
@RequestParam int pageSize, |
||||
|
@RequestParam int page, |
||||
|
@RequestParam(required = false) String textSearch, |
||||
|
@RequestParam(required = false) String sortProperty, |
||||
|
@RequestParam(required = false) String sortOrder) throws ThingsboardException { |
||||
|
checkParameter("deviceProfileId", strDeviceProfileId); |
||||
|
checkParameter("type", strType); |
||||
|
try { |
||||
|
PageLink pageLink = createPageLink(pageSize, page, textSearch, sortProperty, sortOrder); |
||||
|
return checkNotNull(otaPackageService.findTenantOtaPackagesByTenantIdAndDeviceProfileIdAndTypeAndHasData(getTenantId(), |
||||
|
new DeviceProfileId(toUUID(strDeviceProfileId)), OtaPackageType.valueOf(strType), hasData, pageLink)); |
||||
|
} catch (Exception e) { |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN')") |
||||
|
@RequestMapping(value = "/otaPackage/{otaPackageId}", method = RequestMethod.DELETE) |
||||
|
@ResponseBody |
||||
|
public void deleteOtaPackage(@PathVariable("otaPackageId") String strOtaPackageId) throws ThingsboardException { |
||||
|
checkParameter(OTA_PACKAGE_ID, strOtaPackageId); |
||||
|
try { |
||||
|
OtaPackageId otaPackageId = new OtaPackageId(toUUID(strOtaPackageId)); |
||||
|
OtaPackageInfo info = checkOtaPackageInfoId(otaPackageId, Operation.DELETE); |
||||
|
otaPackageService.deleteOtaPackage(getTenantId(), otaPackageId); |
||||
|
logEntityAction(otaPackageId, info, null, ActionType.DELETED, null, strOtaPackageId); |
||||
|
} catch (Exception e) { |
||||
|
logEntityAction(emptyId(EntityType.OTA_PACKAGE), null, null, ActionType.DELETED, e, strOtaPackageId); |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -1,52 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2021 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.dao.firmware; |
|
||||
|
|
||||
import com.google.common.util.concurrent.ListenableFuture; |
|
||||
import org.thingsboard.server.common.data.Firmware; |
|
||||
import org.thingsboard.server.common.data.FirmwareInfo; |
|
||||
import org.thingsboard.server.common.data.firmware.ChecksumAlgorithm; |
|
||||
import org.thingsboard.server.common.data.firmware.FirmwareType; |
|
||||
import org.thingsboard.server.common.data.id.DeviceProfileId; |
|
||||
import org.thingsboard.server.common.data.id.FirmwareId; |
|
||||
import org.thingsboard.server.common.data.id.TenantId; |
|
||||
import org.thingsboard.server.common.data.page.PageData; |
|
||||
import org.thingsboard.server.common.data.page.PageLink; |
|
||||
|
|
||||
import java.nio.ByteBuffer; |
|
||||
|
|
||||
public interface FirmwareService { |
|
||||
|
|
||||
FirmwareInfo saveFirmwareInfo(FirmwareInfo firmwareInfo); |
|
||||
|
|
||||
Firmware saveFirmware(Firmware firmware); |
|
||||
|
|
||||
String generateChecksum(ChecksumAlgorithm checksumAlgorithm, ByteBuffer data); |
|
||||
|
|
||||
Firmware findFirmwareById(TenantId tenantId, FirmwareId firmwareId); |
|
||||
|
|
||||
FirmwareInfo findFirmwareInfoById(TenantId tenantId, FirmwareId firmwareId); |
|
||||
|
|
||||
ListenableFuture<FirmwareInfo> findFirmwareInfoByIdAsync(TenantId tenantId, FirmwareId firmwareId); |
|
||||
|
|
||||
PageData<FirmwareInfo> findTenantFirmwaresByTenantId(TenantId tenantId, PageLink pageLink); |
|
||||
|
|
||||
PageData<FirmwareInfo> findTenantFirmwaresByTenantIdAndDeviceProfileIdAndTypeAndHasData(TenantId tenantId, DeviceProfileId deviceProfileId, FirmwareType firmwareType, boolean hasData, PageLink pageLink); |
|
||||
|
|
||||
void deleteFirmware(TenantId tenantId, FirmwareId firmwareId); |
|
||||
|
|
||||
void deleteFirmwaresByTenantId(TenantId tenantId); |
|
||||
} |
|
||||
@ -0,0 +1,52 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2021 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.dao.ota; |
||||
|
|
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
import org.thingsboard.server.common.data.OtaPackage; |
||||
|
import org.thingsboard.server.common.data.OtaPackageInfo; |
||||
|
import org.thingsboard.server.common.data.ota.ChecksumAlgorithm; |
||||
|
import org.thingsboard.server.common.data.ota.OtaPackageType; |
||||
|
import org.thingsboard.server.common.data.id.DeviceProfileId; |
||||
|
import org.thingsboard.server.common.data.id.OtaPackageId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.page.PageData; |
||||
|
import org.thingsboard.server.common.data.page.PageLink; |
||||
|
|
||||
|
import java.nio.ByteBuffer; |
||||
|
|
||||
|
public interface OtaPackageService { |
||||
|
|
||||
|
OtaPackageInfo saveOtaPackageInfo(OtaPackageInfo otaPackageInfo); |
||||
|
|
||||
|
OtaPackage saveOtaPackage(OtaPackage otaPackage); |
||||
|
|
||||
|
String generateChecksum(ChecksumAlgorithm checksumAlgorithm, ByteBuffer data); |
||||
|
|
||||
|
OtaPackage findOtaPackageById(TenantId tenantId, OtaPackageId otaPackageId); |
||||
|
|
||||
|
OtaPackageInfo findOtaPackageInfoById(TenantId tenantId, OtaPackageId otaPackageId); |
||||
|
|
||||
|
ListenableFuture<OtaPackageInfo> findOtaPackageInfoByIdAsync(TenantId tenantId, OtaPackageId otaPackageId); |
||||
|
|
||||
|
PageData<OtaPackageInfo> findTenantOtaPackagesByTenantId(TenantId tenantId, PageLink pageLink); |
||||
|
|
||||
|
PageData<OtaPackageInfo> findTenantOtaPackagesByTenantIdAndDeviceProfileIdAndTypeAndHasData(TenantId tenantId, DeviceProfileId deviceProfileId, OtaPackageType otaPackageType, boolean hasData, PageLink pageLink); |
||||
|
|
||||
|
void deleteOtaPackage(TenantId tenantId, OtaPackageId otaPackageId); |
||||
|
|
||||
|
void deleteOtaPackagesByTenantId(TenantId tenantId); |
||||
|
} |
||||
@ -1,379 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2021 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.dao.firmware; |
|
||||
|
|
||||
import com.google.common.hash.HashFunction; |
|
||||
import com.google.common.hash.Hashing; |
|
||||
import com.google.common.util.concurrent.ListenableFuture; |
|
||||
import lombok.RequiredArgsConstructor; |
|
||||
import lombok.extern.slf4j.Slf4j; |
|
||||
import org.apache.commons.lang3.StringUtils; |
|
||||
import org.hibernate.exception.ConstraintViolationException; |
|
||||
import org.springframework.cache.Cache; |
|
||||
import org.springframework.cache.CacheManager; |
|
||||
import org.springframework.cache.annotation.Cacheable; |
|
||||
import org.springframework.stereotype.Service; |
|
||||
import org.thingsboard.server.cache.firmware.FirmwareDataCache; |
|
||||
import org.thingsboard.server.common.data.DeviceProfile; |
|
||||
import org.thingsboard.server.common.data.Firmware; |
|
||||
import org.thingsboard.server.common.data.FirmwareInfo; |
|
||||
import org.thingsboard.server.common.data.Tenant; |
|
||||
import org.thingsboard.server.common.data.exception.ThingsboardErrorCode; |
|
||||
import org.thingsboard.server.common.data.exception.ThingsboardException; |
|
||||
import org.thingsboard.server.common.data.firmware.ChecksumAlgorithm; |
|
||||
import org.thingsboard.server.common.data.firmware.FirmwareType; |
|
||||
import org.thingsboard.server.common.data.id.DeviceProfileId; |
|
||||
import org.thingsboard.server.common.data.id.FirmwareId; |
|
||||
import org.thingsboard.server.common.data.id.TenantId; |
|
||||
import org.thingsboard.server.common.data.page.PageData; |
|
||||
import org.thingsboard.server.common.data.page.PageLink; |
|
||||
import org.thingsboard.server.dao.device.DeviceProfileDao; |
|
||||
import org.thingsboard.server.dao.exception.DataValidationException; |
|
||||
import org.thingsboard.server.dao.service.DataValidator; |
|
||||
import org.thingsboard.server.dao.service.PaginatedRemover; |
|
||||
import org.thingsboard.server.dao.tenant.TenantDao; |
|
||||
|
|
||||
import java.lang.reflect.InvocationTargetException; |
|
||||
import java.lang.reflect.Method; |
|
||||
import java.nio.ByteBuffer; |
|
||||
import java.util.ArrayList; |
|
||||
import java.util.Arrays; |
|
||||
import java.util.Collections; |
|
||||
import java.util.List; |
|
||||
import java.util.Optional; |
|
||||
|
|
||||
import static org.thingsboard.server.common.data.CacheConstants.FIRMWARE_CACHE; |
|
||||
import static org.thingsboard.server.dao.service.Validator.validateId; |
|
||||
import static org.thingsboard.server.dao.service.Validator.validatePageLink; |
|
||||
|
|
||||
@Service |
|
||||
@Slf4j |
|
||||
@RequiredArgsConstructor |
|
||||
public class BaseFirmwareService implements FirmwareService { |
|
||||
public static final String INCORRECT_FIRMWARE_ID = "Incorrect firmwareId "; |
|
||||
public static final String INCORRECT_TENANT_ID = "Incorrect tenantId "; |
|
||||
|
|
||||
private final TenantDao tenantDao; |
|
||||
private final DeviceProfileDao deviceProfileDao; |
|
||||
private final FirmwareDao firmwareDao; |
|
||||
private final FirmwareInfoDao firmwareInfoDao; |
|
||||
private final CacheManager cacheManager; |
|
||||
private final FirmwareDataCache firmwareDataCache; |
|
||||
|
|
||||
@Override |
|
||||
public FirmwareInfo saveFirmwareInfo(FirmwareInfo firmwareInfo) { |
|
||||
log.trace("Executing saveFirmwareInfo [{}]", firmwareInfo); |
|
||||
firmwareInfoValidator.validate(firmwareInfo, FirmwareInfo::getTenantId); |
|
||||
try { |
|
||||
FirmwareId firmwareId = firmwareInfo.getId(); |
|
||||
if (firmwareId != null) { |
|
||||
Cache cache = cacheManager.getCache(FIRMWARE_CACHE); |
|
||||
cache.evict(toFirmwareInfoKey(firmwareId)); |
|
||||
firmwareDataCache.evict(firmwareId.toString()); |
|
||||
} |
|
||||
return firmwareInfoDao.save(firmwareInfo.getTenantId(), firmwareInfo); |
|
||||
} catch (Exception t) { |
|
||||
ConstraintViolationException e = extractConstraintViolationException(t).orElse(null); |
|
||||
if (e != null && e.getConstraintName() != null && e.getConstraintName().equalsIgnoreCase("firmware_tenant_title_version_unq_key")) { |
|
||||
throw new DataValidationException("Firmware with such title and version already exists!"); |
|
||||
} else { |
|
||||
throw t; |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public Firmware saveFirmware(Firmware firmware) { |
|
||||
log.trace("Executing saveFirmware [{}]", firmware); |
|
||||
firmwareValidator.validate(firmware, FirmwareInfo::getTenantId); |
|
||||
try { |
|
||||
FirmwareId firmwareId = firmware.getId(); |
|
||||
if (firmwareId != null) { |
|
||||
Cache cache = cacheManager.getCache(FIRMWARE_CACHE); |
|
||||
cache.evict(toFirmwareInfoKey(firmwareId)); |
|
||||
firmwareDataCache.evict(firmwareId.toString()); |
|
||||
} |
|
||||
return firmwareDao.save(firmware.getTenantId(), firmware); |
|
||||
} catch (Exception t) { |
|
||||
ConstraintViolationException e = extractConstraintViolationException(t).orElse(null); |
|
||||
if (e != null && e.getConstraintName() != null && e.getConstraintName().equalsIgnoreCase("firmware_tenant_title_version_unq_key")) { |
|
||||
throw new DataValidationException("Firmware with such title and version already exists!"); |
|
||||
} else { |
|
||||
throw t; |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public String generateChecksum(ChecksumAlgorithm checksumAlgorithm, ByteBuffer data) { |
|
||||
if (data == null || !data.hasArray() || data.array().length == 0) { |
|
||||
throw new DataValidationException("Firmware data should be specified!"); |
|
||||
} |
|
||||
|
|
||||
return getHashFunction(checksumAlgorithm).hashBytes(data.array()).toString(); |
|
||||
} |
|
||||
|
|
||||
private HashFunction getHashFunction(ChecksumAlgorithm checksumAlgorithm) { |
|
||||
switch (checksumAlgorithm) { |
|
||||
case MD5: |
|
||||
return Hashing.md5(); |
|
||||
case SHA256: |
|
||||
return Hashing.sha256(); |
|
||||
case SHA384: |
|
||||
return Hashing.sha384(); |
|
||||
case SHA512: |
|
||||
return Hashing.sha512(); |
|
||||
case CRC32: |
|
||||
return Hashing.crc32(); |
|
||||
case MURMUR3_32: |
|
||||
return Hashing.murmur3_32(); |
|
||||
case MURMUR3_128: |
|
||||
return Hashing.murmur3_128(); |
|
||||
default: |
|
||||
throw new DataValidationException("Unknown checksum algorithm!"); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public Firmware findFirmwareById(TenantId tenantId, FirmwareId firmwareId) { |
|
||||
log.trace("Executing findFirmwareById [{}]", firmwareId); |
|
||||
validateId(firmwareId, INCORRECT_FIRMWARE_ID + firmwareId); |
|
||||
return firmwareDao.findById(tenantId, firmwareId.getId()); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
@Cacheable(cacheNames = FIRMWARE_CACHE, key = "{#firmwareId}") |
|
||||
public FirmwareInfo findFirmwareInfoById(TenantId tenantId, FirmwareId firmwareId) { |
|
||||
log.trace("Executing findFirmwareInfoById [{}]", firmwareId); |
|
||||
validateId(firmwareId, INCORRECT_FIRMWARE_ID + firmwareId); |
|
||||
return firmwareInfoDao.findById(tenantId, firmwareId.getId()); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public ListenableFuture<FirmwareInfo> findFirmwareInfoByIdAsync(TenantId tenantId, FirmwareId firmwareId) { |
|
||||
log.trace("Executing findFirmwareInfoByIdAsync [{}]", firmwareId); |
|
||||
validateId(firmwareId, INCORRECT_FIRMWARE_ID + firmwareId); |
|
||||
return firmwareInfoDao.findByIdAsync(tenantId, firmwareId.getId()); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public PageData<FirmwareInfo> findTenantFirmwaresByTenantId(TenantId tenantId, PageLink pageLink) { |
|
||||
log.trace("Executing findTenantFirmwaresByTenantId, tenantId [{}], pageLink [{}]", tenantId, pageLink); |
|
||||
validateId(tenantId, INCORRECT_TENANT_ID + tenantId); |
|
||||
validatePageLink(pageLink); |
|
||||
return firmwareInfoDao.findFirmwareInfoByTenantId(tenantId, pageLink); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public PageData<FirmwareInfo> findTenantFirmwaresByTenantIdAndDeviceProfileIdAndTypeAndHasData(TenantId tenantId, DeviceProfileId deviceProfileId, FirmwareType firmwareType, boolean hasData, PageLink pageLink) { |
|
||||
log.trace("Executing findTenantFirmwaresByTenantIdAndHasData, tenantId [{}], hasData [{}] pageLink [{}]", tenantId, hasData, pageLink); |
|
||||
validateId(tenantId, INCORRECT_TENANT_ID + tenantId); |
|
||||
validatePageLink(pageLink); |
|
||||
return firmwareInfoDao.findFirmwareInfoByTenantIdAndDeviceProfileIdAndTypeAndHasData(tenantId, deviceProfileId, firmwareType, hasData, pageLink); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void deleteFirmware(TenantId tenantId, FirmwareId firmwareId) { |
|
||||
log.trace("Executing deleteFirmware [{}]", firmwareId); |
|
||||
validateId(firmwareId, INCORRECT_FIRMWARE_ID + firmwareId); |
|
||||
try { |
|
||||
Cache cache = cacheManager.getCache(FIRMWARE_CACHE); |
|
||||
cache.evict(toFirmwareInfoKey(firmwareId)); |
|
||||
firmwareDataCache.evict(firmwareId.toString()); |
|
||||
firmwareDao.removeById(tenantId, firmwareId.getId()); |
|
||||
} catch (Exception t) { |
|
||||
ConstraintViolationException e = extractConstraintViolationException(t).orElse(null); |
|
||||
if (e != null && e.getConstraintName() != null && e.getConstraintName().equalsIgnoreCase("fk_firmware_device")) { |
|
||||
throw new DataValidationException("The firmware referenced by the devices cannot be deleted!"); |
|
||||
} else if (e != null && e.getConstraintName() != null && e.getConstraintName().equalsIgnoreCase("fk_firmware_device_profile")) { |
|
||||
throw new DataValidationException("The firmware referenced by the device profile cannot be deleted!"); |
|
||||
} else if (e != null && e.getConstraintName() != null && e.getConstraintName().equalsIgnoreCase("fk_software_device")) { |
|
||||
throw new DataValidationException("The software referenced by the devices cannot be deleted!"); |
|
||||
} else if (e != null && e.getConstraintName() != null && e.getConstraintName().equalsIgnoreCase("fk_software_device_profile")) { |
|
||||
throw new DataValidationException("The software referenced by the device profile cannot be deleted!"); |
|
||||
} else { |
|
||||
throw t; |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void deleteFirmwaresByTenantId(TenantId tenantId) { |
|
||||
log.trace("Executing deleteFirmwaresByTenantId, tenantId [{}]", tenantId); |
|
||||
validateId(tenantId, INCORRECT_TENANT_ID + tenantId); |
|
||||
tenantFirmwareRemover.removeEntities(tenantId, tenantId); |
|
||||
} |
|
||||
|
|
||||
private DataValidator<FirmwareInfo> firmwareInfoValidator = new DataValidator<>() { |
|
||||
|
|
||||
@Override |
|
||||
protected void validateDataImpl(TenantId tenantId, FirmwareInfo firmwareInfo) { |
|
||||
validateImpl(firmwareInfo); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void validateUpdate(TenantId tenantId, FirmwareInfo firmware) { |
|
||||
FirmwareInfo firmwareOld = firmwareInfoDao.findById(tenantId, firmware.getUuidId()); |
|
||||
|
|
||||
validateUpdateDeviceProfile(firmware, firmwareOld); |
|
||||
BaseFirmwareService.validateUpdate(firmware, firmwareOld); |
|
||||
} |
|
||||
}; |
|
||||
|
|
||||
private DataValidator<Firmware> firmwareValidator = new DataValidator<>() { |
|
||||
|
|
||||
@Override |
|
||||
protected void validateDataImpl(TenantId tenantId, Firmware firmware) { |
|
||||
validateImpl(firmware); |
|
||||
|
|
||||
if (StringUtils.isEmpty(firmware.getFileName())) { |
|
||||
throw new DataValidationException("Firmware file name should be specified!"); |
|
||||
} |
|
||||
|
|
||||
if (StringUtils.isEmpty(firmware.getContentType())) { |
|
||||
throw new DataValidationException("Firmware content type should be specified!"); |
|
||||
} |
|
||||
|
|
||||
if (firmware.getChecksumAlgorithm() == null) { |
|
||||
throw new DataValidationException("Firmware checksum algorithm should be specified!"); |
|
||||
} |
|
||||
if (StringUtils.isEmpty(firmware.getChecksum())) { |
|
||||
throw new DataValidationException("Firmware checksum should be specified!"); |
|
||||
} |
|
||||
|
|
||||
String currentChecksum; |
|
||||
|
|
||||
currentChecksum = generateChecksum(firmware.getChecksumAlgorithm(), firmware.getData()); |
|
||||
|
|
||||
if (!currentChecksum.equals(firmware.getChecksum())) { |
|
||||
throw new DataValidationException("Wrong firmware file!"); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void validateUpdate(TenantId tenantId, Firmware firmware) { |
|
||||
Firmware firmwareOld = firmwareDao.findById(tenantId, firmware.getUuidId()); |
|
||||
|
|
||||
validateUpdateDeviceProfile(firmware, firmwareOld); |
|
||||
BaseFirmwareService.validateUpdate(firmware, firmwareOld); |
|
||||
|
|
||||
if (firmwareOld.getData() != null && !firmwareOld.getData().equals(firmware.getData())) { |
|
||||
throw new DataValidationException("Updating firmware data is prohibited!"); |
|
||||
} |
|
||||
} |
|
||||
}; |
|
||||
|
|
||||
private void validateUpdateDeviceProfile(FirmwareInfo firmware, FirmwareInfo firmwareOld) { |
|
||||
if (firmwareOld.getDeviceProfileId() != null && !firmwareOld.getDeviceProfileId().equals(firmware.getDeviceProfileId())) { |
|
||||
if (firmwareInfoDao.isFirmwareUsed(firmwareOld.getId(), firmware.getType(), firmwareOld.getDeviceProfileId())) { |
|
||||
throw new DataValidationException("Can`t update deviceProfileId because firmware is already in use!"); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private static void validateUpdate(FirmwareInfo firmware, FirmwareInfo firmwareOld) { |
|
||||
if (!firmwareOld.getType().equals(firmware.getType())) { |
|
||||
throw new DataValidationException("Updating type is prohibited!"); |
|
||||
} |
|
||||
|
|
||||
if (!firmwareOld.getTitle().equals(firmware.getTitle())) { |
|
||||
throw new DataValidationException("Updating firmware title is prohibited!"); |
|
||||
} |
|
||||
|
|
||||
if (!firmwareOld.getVersion().equals(firmware.getVersion())) { |
|
||||
throw new DataValidationException("Updating firmware version is prohibited!"); |
|
||||
} |
|
||||
|
|
||||
if (firmwareOld.getFileName() != null && !firmwareOld.getFileName().equals(firmware.getFileName())) { |
|
||||
throw new DataValidationException("Updating firmware file name is prohibited!"); |
|
||||
} |
|
||||
|
|
||||
if (firmwareOld.getContentType() != null && !firmwareOld.getContentType().equals(firmware.getContentType())) { |
|
||||
throw new DataValidationException("Updating firmware content type is prohibited!"); |
|
||||
} |
|
||||
|
|
||||
if (firmwareOld.getChecksumAlgorithm() != null && !firmwareOld.getChecksumAlgorithm().equals(firmware.getChecksumAlgorithm())) { |
|
||||
throw new DataValidationException("Updating firmware content type is prohibited!"); |
|
||||
} |
|
||||
|
|
||||
if (firmwareOld.getChecksum() != null && !firmwareOld.getChecksum().equals(firmware.getChecksum())) { |
|
||||
throw new DataValidationException("Updating firmware content type is prohibited!"); |
|
||||
} |
|
||||
|
|
||||
if (firmwareOld.getDataSize() != null && !firmwareOld.getDataSize().equals(firmware.getDataSize())) { |
|
||||
throw new DataValidationException("Updating firmware data size is prohibited!"); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private void validateImpl(FirmwareInfo firmwareInfo) { |
|
||||
if (firmwareInfo.getTenantId() == null) { |
|
||||
throw new DataValidationException("Firmware should be assigned to tenant!"); |
|
||||
} else { |
|
||||
Tenant tenant = tenantDao.findById(firmwareInfo.getTenantId(), firmwareInfo.getTenantId().getId()); |
|
||||
if (tenant == null) { |
|
||||
throw new DataValidationException("Firmware is referencing to non-existent tenant!"); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
if (firmwareInfo.getDeviceProfileId() != null) { |
|
||||
DeviceProfile deviceProfile = deviceProfileDao.findById(firmwareInfo.getTenantId(), firmwareInfo.getDeviceProfileId().getId()); |
|
||||
if (deviceProfile == null) { |
|
||||
throw new DataValidationException("Firmware is referencing to non-existent device profile!"); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
if (firmwareInfo.getType() == null) { |
|
||||
throw new DataValidationException("Type should be specified!"); |
|
||||
} |
|
||||
|
|
||||
if (StringUtils.isEmpty(firmwareInfo.getTitle())) { |
|
||||
throw new DataValidationException("Firmware title should be specified!"); |
|
||||
} |
|
||||
|
|
||||
if (StringUtils.isEmpty(firmwareInfo.getVersion())) { |
|
||||
throw new DataValidationException("Firmware version should be specified!"); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private PaginatedRemover<TenantId, FirmwareInfo> tenantFirmwareRemover = |
|
||||
new PaginatedRemover<>() { |
|
||||
|
|
||||
@Override |
|
||||
protected PageData<FirmwareInfo> findEntities(TenantId tenantId, TenantId id, PageLink pageLink) { |
|
||||
return firmwareInfoDao.findFirmwareInfoByTenantId(id, pageLink); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void removeEntity(TenantId tenantId, FirmwareInfo entity) { |
|
||||
deleteFirmware(tenantId, entity.getId()); |
|
||||
} |
|
||||
}; |
|
||||
|
|
||||
protected Optional<ConstraintViolationException> extractConstraintViolationException(Exception t) { |
|
||||
if (t instanceof ConstraintViolationException) { |
|
||||
return Optional.of((ConstraintViolationException) t); |
|
||||
} else if (t.getCause() instanceof ConstraintViolationException) { |
|
||||
return Optional.of((ConstraintViolationException) (t.getCause())); |
|
||||
} else { |
|
||||
return Optional.empty(); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private static List<FirmwareId> toFirmwareInfoKey(FirmwareId firmwareId) { |
|
||||
return Collections.singletonList(firmwareId); |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
@ -0,0 +1,366 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2021 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.dao.ota; |
||||
|
|
||||
|
import com.google.common.hash.HashFunction; |
||||
|
import com.google.common.hash.Hashing; |
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.apache.commons.lang3.StringUtils; |
||||
|
import org.hibernate.exception.ConstraintViolationException; |
||||
|
import org.springframework.cache.Cache; |
||||
|
import org.springframework.cache.CacheManager; |
||||
|
import org.springframework.cache.annotation.Cacheable; |
||||
|
import org.springframework.stereotype.Service; |
||||
|
import org.thingsboard.server.cache.ota.OtaPackageDataCache; |
||||
|
import org.thingsboard.server.common.data.DeviceProfile; |
||||
|
import org.thingsboard.server.common.data.OtaPackage; |
||||
|
import org.thingsboard.server.common.data.OtaPackageInfo; |
||||
|
import org.thingsboard.server.common.data.Tenant; |
||||
|
import org.thingsboard.server.common.data.ota.ChecksumAlgorithm; |
||||
|
import org.thingsboard.server.common.data.ota.OtaPackageType; |
||||
|
import org.thingsboard.server.common.data.id.DeviceProfileId; |
||||
|
import org.thingsboard.server.common.data.id.OtaPackageId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.page.PageData; |
||||
|
import org.thingsboard.server.common.data.page.PageLink; |
||||
|
import org.thingsboard.server.dao.device.DeviceProfileDao; |
||||
|
import org.thingsboard.server.dao.exception.DataValidationException; |
||||
|
import org.thingsboard.server.dao.service.DataValidator; |
||||
|
import org.thingsboard.server.dao.service.PaginatedRemover; |
||||
|
import org.thingsboard.server.dao.tenant.TenantDao; |
||||
|
|
||||
|
import java.nio.ByteBuffer; |
||||
|
import java.util.Collections; |
||||
|
import java.util.List; |
||||
|
import java.util.Optional; |
||||
|
|
||||
|
import static org.thingsboard.server.common.data.CacheConstants.OTA_PACKAGE_CACHE; |
||||
|
import static org.thingsboard.server.dao.service.Validator.validateId; |
||||
|
import static org.thingsboard.server.dao.service.Validator.validatePageLink; |
||||
|
|
||||
|
@Service |
||||
|
@Slf4j |
||||
|
@RequiredArgsConstructor |
||||
|
public class BaseOtaPackageService implements OtaPackageService { |
||||
|
public static final String INCORRECT_OTA_PACKAGE_ID = "Incorrect otaPackageId "; |
||||
|
public static final String INCORRECT_TENANT_ID = "Incorrect tenantId "; |
||||
|
|
||||
|
private final TenantDao tenantDao; |
||||
|
private final DeviceProfileDao deviceProfileDao; |
||||
|
private final OtaPackageDao otaPackageDao; |
||||
|
private final OtaPackageInfoDao otaPackageInfoDao; |
||||
|
private final CacheManager cacheManager; |
||||
|
private final OtaPackageDataCache otaPackageDataCache; |
||||
|
|
||||
|
@Override |
||||
|
public OtaPackageInfo saveOtaPackageInfo(OtaPackageInfo otaPackageInfo) { |
||||
|
log.trace("Executing saveOtaPackageInfo [{}]", otaPackageInfo); |
||||
|
otaPackageInfoValidator.validate(otaPackageInfo, OtaPackageInfo::getTenantId); |
||||
|
try { |
||||
|
OtaPackageId otaPackageId = otaPackageInfo.getId(); |
||||
|
if (otaPackageId != null) { |
||||
|
Cache cache = cacheManager.getCache(OTA_PACKAGE_CACHE); |
||||
|
cache.evict(toOtaPackageInfoKey(otaPackageId)); |
||||
|
otaPackageDataCache.evict(otaPackageId.toString()); |
||||
|
} |
||||
|
return otaPackageInfoDao.save(otaPackageInfo.getTenantId(), otaPackageInfo); |
||||
|
} catch (Exception t) { |
||||
|
ConstraintViolationException e = extractConstraintViolationException(t).orElse(null); |
||||
|
if (e != null && e.getConstraintName() != null && e.getConstraintName().equalsIgnoreCase("ota_package_tenant_title_version_unq_key")) { |
||||
|
throw new DataValidationException("OtaPackage with such title and version already exists!"); |
||||
|
} else { |
||||
|
throw t; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public OtaPackage saveOtaPackage(OtaPackage otaPackage) { |
||||
|
log.trace("Executing saveOtaPackage [{}]", otaPackage); |
||||
|
otaPackageValidator.validate(otaPackage, OtaPackageInfo::getTenantId); |
||||
|
try { |
||||
|
OtaPackageId otaPackageId = otaPackage.getId(); |
||||
|
if (otaPackageId != null) { |
||||
|
Cache cache = cacheManager.getCache(OTA_PACKAGE_CACHE); |
||||
|
cache.evict(toOtaPackageInfoKey(otaPackageId)); |
||||
|
otaPackageDataCache.evict(otaPackageId.toString()); |
||||
|
} |
||||
|
return otaPackageDao.save(otaPackage.getTenantId(), otaPackage); |
||||
|
} catch (Exception t) { |
||||
|
ConstraintViolationException e = extractConstraintViolationException(t).orElse(null); |
||||
|
if (e != null && e.getConstraintName() != null && e.getConstraintName().equalsIgnoreCase("ota_package_tenant_title_version_unq_key")) { |
||||
|
throw new DataValidationException("OtaPackage with such title and version already exists!"); |
||||
|
} else { |
||||
|
throw t; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public String generateChecksum(ChecksumAlgorithm checksumAlgorithm, ByteBuffer data) { |
||||
|
if (data == null || !data.hasArray() || data.array().length == 0) { |
||||
|
throw new DataValidationException("OtaPackage data should be specified!"); |
||||
|
} |
||||
|
|
||||
|
return getHashFunction(checksumAlgorithm).hashBytes(data.array()).toString(); |
||||
|
} |
||||
|
|
||||
|
private HashFunction getHashFunction(ChecksumAlgorithm checksumAlgorithm) { |
||||
|
switch (checksumAlgorithm) { |
||||
|
case MD5: |
||||
|
return Hashing.md5(); |
||||
|
case SHA256: |
||||
|
return Hashing.sha256(); |
||||
|
case SHA384: |
||||
|
return Hashing.sha384(); |
||||
|
case SHA512: |
||||
|
return Hashing.sha512(); |
||||
|
case CRC32: |
||||
|
return Hashing.crc32(); |
||||
|
case MURMUR3_32: |
||||
|
return Hashing.murmur3_32(); |
||||
|
case MURMUR3_128: |
||||
|
return Hashing.murmur3_128(); |
||||
|
default: |
||||
|
throw new DataValidationException("Unknown checksum algorithm!"); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public OtaPackage findOtaPackageById(TenantId tenantId, OtaPackageId otaPackageId) { |
||||
|
log.trace("Executing findOtaPackageById [{}]", otaPackageId); |
||||
|
validateId(otaPackageId, INCORRECT_OTA_PACKAGE_ID + otaPackageId); |
||||
|
return otaPackageDao.findById(tenantId, otaPackageId.getId()); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
@Cacheable(cacheNames = OTA_PACKAGE_CACHE, key = "{#otaPackageId}") |
||||
|
public OtaPackageInfo findOtaPackageInfoById(TenantId tenantId, OtaPackageId otaPackageId) { |
||||
|
log.trace("Executing findOtaPackageInfoById [{}]", otaPackageId); |
||||
|
validateId(otaPackageId, INCORRECT_OTA_PACKAGE_ID + otaPackageId); |
||||
|
return otaPackageInfoDao.findById(tenantId, otaPackageId.getId()); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public ListenableFuture<OtaPackageInfo> findOtaPackageInfoByIdAsync(TenantId tenantId, OtaPackageId otaPackageId) { |
||||
|
log.trace("Executing findOtaPackageInfoByIdAsync [{}]", otaPackageId); |
||||
|
validateId(otaPackageId, INCORRECT_OTA_PACKAGE_ID + otaPackageId); |
||||
|
return otaPackageInfoDao.findByIdAsync(tenantId, otaPackageId.getId()); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public PageData<OtaPackageInfo> findTenantOtaPackagesByTenantId(TenantId tenantId, PageLink pageLink) { |
||||
|
log.trace("Executing findTenantOtaPackagesByTenantId, tenantId [{}], pageLink [{}]", tenantId, pageLink); |
||||
|
validateId(tenantId, INCORRECT_TENANT_ID + tenantId); |
||||
|
validatePageLink(pageLink); |
||||
|
return otaPackageInfoDao.findOtaPackageInfoByTenantId(tenantId, pageLink); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public PageData<OtaPackageInfo> findTenantOtaPackagesByTenantIdAndDeviceProfileIdAndTypeAndHasData(TenantId tenantId, DeviceProfileId deviceProfileId, OtaPackageType otaPackageType, boolean hasData, PageLink pageLink) { |
||||
|
log.trace("Executing findTenantOtaPackagesByTenantIdAndHasData, tenantId [{}], hasData [{}] pageLink [{}]", tenantId, hasData, pageLink); |
||||
|
validateId(tenantId, INCORRECT_TENANT_ID + tenantId); |
||||
|
validatePageLink(pageLink); |
||||
|
return otaPackageInfoDao.findOtaPackageInfoByTenantIdAndDeviceProfileIdAndTypeAndHasData(tenantId, deviceProfileId, otaPackageType, hasData, pageLink); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void deleteOtaPackage(TenantId tenantId, OtaPackageId otaPackageId) { |
||||
|
log.trace("Executing deleteOtaPackage [{}]", otaPackageId); |
||||
|
validateId(otaPackageId, INCORRECT_OTA_PACKAGE_ID + otaPackageId); |
||||
|
try { |
||||
|
Cache cache = cacheManager.getCache(OTA_PACKAGE_CACHE); |
||||
|
cache.evict(toOtaPackageInfoKey(otaPackageId)); |
||||
|
otaPackageDataCache.evict(otaPackageId.toString()); |
||||
|
otaPackageDao.removeById(tenantId, otaPackageId.getId()); |
||||
|
} catch (Exception t) { |
||||
|
ConstraintViolationException e = extractConstraintViolationException(t).orElse(null); |
||||
|
if (e != null && e.getConstraintName() != null && e.getConstraintName().equalsIgnoreCase("fk_firmware_device")) { |
||||
|
throw new DataValidationException("The otaPackage referenced by the devices cannot be deleted!"); |
||||
|
} else if (e != null && e.getConstraintName() != null && e.getConstraintName().equalsIgnoreCase("fk_firmware_device_profile")) { |
||||
|
throw new DataValidationException("The otaPackage referenced by the device profile cannot be deleted!"); |
||||
|
} else if (e != null && e.getConstraintName() != null && e.getConstraintName().equalsIgnoreCase("fk_software_device")) { |
||||
|
throw new DataValidationException("The software referenced by the devices cannot be deleted!"); |
||||
|
} else if (e != null && e.getConstraintName() != null && e.getConstraintName().equalsIgnoreCase("fk_software_device_profile")) { |
||||
|
throw new DataValidationException("The software referenced by the device profile cannot be deleted!"); |
||||
|
} else { |
||||
|
throw t; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void deleteOtaPackagesByTenantId(TenantId tenantId) { |
||||
|
log.trace("Executing deleteOtaPackagesByTenantId, tenantId [{}]", tenantId); |
||||
|
validateId(tenantId, INCORRECT_TENANT_ID + tenantId); |
||||
|
tenantOtaPackageRemover.removeEntities(tenantId, tenantId); |
||||
|
} |
||||
|
|
||||
|
private DataValidator<OtaPackageInfo> otaPackageInfoValidator = new DataValidator<>() { |
||||
|
|
||||
|
@Override |
||||
|
protected void validateDataImpl(TenantId tenantId, OtaPackageInfo otaPackageInfo) { |
||||
|
validateImpl(otaPackageInfo); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
protected void validateUpdate(TenantId tenantId, OtaPackageInfo otaPackage) { |
||||
|
OtaPackageInfo otaPackageOld = otaPackageInfoDao.findById(tenantId, otaPackage.getUuidId()); |
||||
|
BaseOtaPackageService.validateUpdate(otaPackage, otaPackageOld); |
||||
|
} |
||||
|
}; |
||||
|
|
||||
|
private DataValidator<OtaPackage> otaPackageValidator = new DataValidator<>() { |
||||
|
|
||||
|
@Override |
||||
|
protected void validateDataImpl(TenantId tenantId, OtaPackage otaPackage) { |
||||
|
validateImpl(otaPackage); |
||||
|
|
||||
|
if (StringUtils.isEmpty(otaPackage.getFileName())) { |
||||
|
throw new DataValidationException("OtaPackage file name should be specified!"); |
||||
|
} |
||||
|
|
||||
|
if (StringUtils.isEmpty(otaPackage.getContentType())) { |
||||
|
throw new DataValidationException("OtaPackage content type should be specified!"); |
||||
|
} |
||||
|
|
||||
|
if (otaPackage.getChecksumAlgorithm() == null) { |
||||
|
throw new DataValidationException("OtaPackage checksum algorithm should be specified!"); |
||||
|
} |
||||
|
if (StringUtils.isEmpty(otaPackage.getChecksum())) { |
||||
|
throw new DataValidationException("OtaPackage checksum should be specified!"); |
||||
|
} |
||||
|
|
||||
|
String currentChecksum; |
||||
|
|
||||
|
currentChecksum = generateChecksum(otaPackage.getChecksumAlgorithm(), otaPackage.getData()); |
||||
|
|
||||
|
if (!currentChecksum.equals(otaPackage.getChecksum())) { |
||||
|
throw new DataValidationException("Wrong otaPackage file!"); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
protected void validateUpdate(TenantId tenantId, OtaPackage otaPackage) { |
||||
|
OtaPackage otaPackageOld = otaPackageDao.findById(tenantId, otaPackage.getUuidId()); |
||||
|
|
||||
|
BaseOtaPackageService.validateUpdate(otaPackage, otaPackageOld); |
||||
|
|
||||
|
if (otaPackageOld.getData() != null && !otaPackageOld.getData().equals(otaPackage.getData())) { |
||||
|
throw new DataValidationException("Updating otaPackage data is prohibited!"); |
||||
|
} |
||||
|
} |
||||
|
}; |
||||
|
|
||||
|
private static void validateUpdate(OtaPackageInfo otaPackage, OtaPackageInfo otaPackageOld) { |
||||
|
if (!otaPackageOld.getType().equals(otaPackage.getType())) { |
||||
|
throw new DataValidationException("Updating type is prohibited!"); |
||||
|
} |
||||
|
|
||||
|
if (!otaPackageOld.getTitle().equals(otaPackage.getTitle())) { |
||||
|
throw new DataValidationException("Updating otaPackage title is prohibited!"); |
||||
|
} |
||||
|
|
||||
|
if (!otaPackageOld.getVersion().equals(otaPackage.getVersion())) { |
||||
|
throw new DataValidationException("Updating otaPackage version is prohibited!"); |
||||
|
} |
||||
|
|
||||
|
if (!otaPackageOld.getDeviceProfileId().equals(otaPackage.getDeviceProfileId())) { |
||||
|
throw new DataValidationException("Updating otaPackage deviceProfile is prohibited!"); |
||||
|
} |
||||
|
|
||||
|
if (otaPackageOld.getFileName() != null && !otaPackageOld.getFileName().equals(otaPackage.getFileName())) { |
||||
|
throw new DataValidationException("Updating otaPackage file name is prohibited!"); |
||||
|
} |
||||
|
|
||||
|
if (otaPackageOld.getContentType() != null && !otaPackageOld.getContentType().equals(otaPackage.getContentType())) { |
||||
|
throw new DataValidationException("Updating otaPackage content type is prohibited!"); |
||||
|
} |
||||
|
|
||||
|
if (otaPackageOld.getChecksumAlgorithm() != null && !otaPackageOld.getChecksumAlgorithm().equals(otaPackage.getChecksumAlgorithm())) { |
||||
|
throw new DataValidationException("Updating otaPackage content type is prohibited!"); |
||||
|
} |
||||
|
|
||||
|
if (otaPackageOld.getChecksum() != null && !otaPackageOld.getChecksum().equals(otaPackage.getChecksum())) { |
||||
|
throw new DataValidationException("Updating otaPackage content type is prohibited!"); |
||||
|
} |
||||
|
|
||||
|
if (otaPackageOld.getDataSize() != null && !otaPackageOld.getDataSize().equals(otaPackage.getDataSize())) { |
||||
|
throw new DataValidationException("Updating otaPackage data size is prohibited!"); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void validateImpl(OtaPackageInfo otaPackageInfo) { |
||||
|
if (otaPackageInfo.getTenantId() == null) { |
||||
|
throw new DataValidationException("OtaPackage should be assigned to tenant!"); |
||||
|
} else { |
||||
|
Tenant tenant = tenantDao.findById(otaPackageInfo.getTenantId(), otaPackageInfo.getTenantId().getId()); |
||||
|
if (tenant == null) { |
||||
|
throw new DataValidationException("OtaPackage is referencing to non-existent tenant!"); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
if (otaPackageInfo.getDeviceProfileId() != null) { |
||||
|
DeviceProfile deviceProfile = deviceProfileDao.findById(otaPackageInfo.getTenantId(), otaPackageInfo.getDeviceProfileId().getId()); |
||||
|
if (deviceProfile == null) { |
||||
|
throw new DataValidationException("OtaPackage is referencing to non-existent device profile!"); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
if (otaPackageInfo.getType() == null) { |
||||
|
throw new DataValidationException("Type should be specified!"); |
||||
|
} |
||||
|
|
||||
|
if (StringUtils.isEmpty(otaPackageInfo.getTitle())) { |
||||
|
throw new DataValidationException("OtaPackage title should be specified!"); |
||||
|
} |
||||
|
|
||||
|
if (StringUtils.isEmpty(otaPackageInfo.getVersion())) { |
||||
|
throw new DataValidationException("OtaPackage version should be specified!"); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private PaginatedRemover<TenantId, OtaPackageInfo> tenantOtaPackageRemover = |
||||
|
new PaginatedRemover<>() { |
||||
|
|
||||
|
@Override |
||||
|
protected PageData<OtaPackageInfo> findEntities(TenantId tenantId, TenantId id, PageLink pageLink) { |
||||
|
return otaPackageInfoDao.findOtaPackageInfoByTenantId(id, pageLink); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
protected void removeEntity(TenantId tenantId, OtaPackageInfo entity) { |
||||
|
deleteOtaPackage(tenantId, entity.getId()); |
||||
|
} |
||||
|
}; |
||||
|
|
||||
|
protected Optional<ConstraintViolationException> extractConstraintViolationException(Exception t) { |
||||
|
if (t instanceof ConstraintViolationException) { |
||||
|
return Optional.of((ConstraintViolationException) t); |
||||
|
} else if (t.getCause() instanceof ConstraintViolationException) { |
||||
|
return Optional.of((ConstraintViolationException) (t.getCause())); |
||||
|
} else { |
||||
|
return Optional.empty(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private static List<OtaPackageId> toOtaPackageInfoKey(OtaPackageId otaPackageId) { |
||||
|
return Collections.singletonList(otaPackageId); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -1,60 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2021 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.dao.sql.firmware; |
|
||||
|
|
||||
import org.springframework.data.domain.Page; |
|
||||
import org.springframework.data.domain.Pageable; |
|
||||
import org.springframework.data.jpa.repository.Query; |
|
||||
import org.springframework.data.repository.CrudRepository; |
|
||||
import org.springframework.data.repository.query.Param; |
|
||||
import org.thingsboard.server.common.data.firmware.FirmwareType; |
|
||||
import org.thingsboard.server.dao.model.sql.FirmwareInfoEntity; |
|
||||
|
|
||||
import java.util.UUID; |
|
||||
|
|
||||
public interface FirmwareInfoRepository extends CrudRepository<FirmwareInfoEntity, UUID> { |
|
||||
@Query("SELECT new FirmwareInfoEntity(f.id, f.createdTime, f.tenantId, f.deviceProfileId, f.type, f.title, f.version, f.fileName, f.contentType, f.checksumAlgorithm, f.checksum, f.dataSize, f.additionalInfo, f.data IS NOT NULL) FROM FirmwareEntity f WHERE " + |
|
||||
"f.tenantId = :tenantId " + |
|
||||
"AND LOWER(f.searchText) LIKE LOWER(CONCAT(:searchText, '%'))") |
|
||||
Page<FirmwareInfoEntity> findAllByTenantId(@Param("tenantId") UUID tenantId, |
|
||||
@Param("searchText") String searchText, |
|
||||
Pageable pageable); |
|
||||
|
|
||||
@Query("SELECT new FirmwareInfoEntity(f.id, f.createdTime, f.tenantId, f.deviceProfileId, f.type, f.title, f.version, f.fileName, f.contentType, f.checksumAlgorithm, f.checksum, f.dataSize, f.additionalInfo, f.data IS NOT NULL) FROM FirmwareEntity f WHERE " + |
|
||||
"f.tenantId = :tenantId " + |
|
||||
"AND f.deviceProfileId = :deviceProfileId " + |
|
||||
"AND f.type = :type " + |
|
||||
"AND ((f.data IS NOT NULL AND :hasData = true) OR (f.data IS NULL AND :hasData = false ))" + |
|
||||
"AND LOWER(f.searchText) LIKE LOWER(CONCAT(:searchText, '%'))") |
|
||||
Page<FirmwareInfoEntity> findAllByTenantIdAndTypeAndDeviceProfileIdAndHasData(@Param("tenantId") UUID tenantId, |
|
||||
@Param("deviceProfileId") UUID deviceProfileId, |
|
||||
@Param("type") FirmwareType type, |
|
||||
@Param("hasData") boolean hasData, |
|
||||
@Param("searchText") String searchText, |
|
||||
Pageable pageable); |
|
||||
|
|
||||
@Query("SELECT new FirmwareInfoEntity(f.id, f.createdTime, f.tenantId, f.deviceProfileId, f.type, f.title, f.version, f.fileName, f.contentType, f.checksumAlgorithm, f.checksum, f.dataSize, f.additionalInfo, f.data IS NOT NULL) FROM FirmwareEntity f WHERE f.id = :id") |
|
||||
FirmwareInfoEntity findFirmwareInfoById(@Param("id") UUID id); |
|
||||
|
|
||||
@Query(value = "SELECT exists(SELECT * " + |
|
||||
"FROM device_profile AS dp " + |
|
||||
"LEFT JOIN device AS d ON dp.id = d.device_profile_id " + |
|
||||
"WHERE dp.id = :deviceProfileId AND " + |
|
||||
"(('FIRMWARE' = :type AND (dp.firmware_id = :firmwareId OR d.firmware_id = :firmwareId)) " + |
|
||||
"OR ('SOFTWARE' = :type AND (dp.software_id = :firmwareId or d.software_id = :firmwareId))))", nativeQuery = true) |
|
||||
boolean isFirmwareUsed(@Param("firmwareId") UUID firmwareId, @Param("deviceProfileId") UUID deviceProfileId, @Param("type") String type); |
|
||||
|
|
||||
} |
|
||||
@ -1,94 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2021 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.dao.sql.firmware; |
|
||||
|
|
||||
import lombok.extern.slf4j.Slf4j; |
|
||||
import org.springframework.beans.factory.annotation.Autowired; |
|
||||
import org.springframework.data.repository.CrudRepository; |
|
||||
import org.springframework.stereotype.Component; |
|
||||
import org.thingsboard.server.common.data.FirmwareInfo; |
|
||||
import org.thingsboard.server.common.data.firmware.FirmwareType; |
|
||||
import org.thingsboard.server.common.data.id.DeviceProfileId; |
|
||||
import org.thingsboard.server.common.data.id.FirmwareId; |
|
||||
import org.thingsboard.server.common.data.id.TenantId; |
|
||||
import org.thingsboard.server.common.data.page.PageData; |
|
||||
import org.thingsboard.server.common.data.page.PageLink; |
|
||||
import org.thingsboard.server.dao.DaoUtil; |
|
||||
import org.thingsboard.server.dao.firmware.FirmwareInfoDao; |
|
||||
import org.thingsboard.server.dao.model.sql.FirmwareInfoEntity; |
|
||||
import org.thingsboard.server.dao.sql.JpaAbstractSearchTextDao; |
|
||||
|
|
||||
import java.util.Objects; |
|
||||
import java.util.UUID; |
|
||||
|
|
||||
@Slf4j |
|
||||
@Component |
|
||||
public class JpaFirmwareInfoDao extends JpaAbstractSearchTextDao<FirmwareInfoEntity, FirmwareInfo> implements FirmwareInfoDao { |
|
||||
|
|
||||
@Autowired |
|
||||
private FirmwareInfoRepository firmwareInfoRepository; |
|
||||
|
|
||||
@Override |
|
||||
protected Class<FirmwareInfoEntity> getEntityClass() { |
|
||||
return FirmwareInfoEntity.class; |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected CrudRepository<FirmwareInfoEntity, UUID> getCrudRepository() { |
|
||||
return firmwareInfoRepository; |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public FirmwareInfo findById(TenantId tenantId, UUID id) { |
|
||||
return DaoUtil.getData(firmwareInfoRepository.findFirmwareInfoById(id)); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public FirmwareInfo save(TenantId tenantId, FirmwareInfo firmwareInfo) { |
|
||||
FirmwareInfo savedFirmware = super.save(tenantId, firmwareInfo); |
|
||||
if (firmwareInfo.getId() == null) { |
|
||||
return savedFirmware; |
|
||||
} else { |
|
||||
return findById(tenantId, savedFirmware.getId().getId()); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public PageData<FirmwareInfo> findFirmwareInfoByTenantId(TenantId tenantId, PageLink pageLink) { |
|
||||
return DaoUtil.toPageData(firmwareInfoRepository |
|
||||
.findAllByTenantId( |
|
||||
tenantId.getId(), |
|
||||
Objects.toString(pageLink.getTextSearch(), ""), |
|
||||
DaoUtil.toPageable(pageLink))); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public PageData<FirmwareInfo> findFirmwareInfoByTenantIdAndDeviceProfileIdAndTypeAndHasData(TenantId tenantId, DeviceProfileId deviceProfileId, FirmwareType firmwareType, boolean hasData, PageLink pageLink) { |
|
||||
return DaoUtil.toPageData(firmwareInfoRepository |
|
||||
.findAllByTenantIdAndTypeAndDeviceProfileIdAndHasData( |
|
||||
tenantId.getId(), |
|
||||
deviceProfileId.getId(), |
|
||||
firmwareType, |
|
||||
hasData, |
|
||||
Objects.toString(pageLink.getTextSearch(), ""), |
|
||||
DaoUtil.toPageable(pageLink))); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public boolean isFirmwareUsed(FirmwareId firmwareId, FirmwareType type, DeviceProfileId deviceProfileId) { |
|
||||
return firmwareInfoRepository.isFirmwareUsed(firmwareId.getId(), deviceProfileId.getId(), type.name()); |
|
||||
} |
|
||||
} |
|
||||
@ -0,0 +1,94 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2021 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.dao.sql.ota; |
||||
|
|
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||
|
import org.springframework.data.repository.CrudRepository; |
||||
|
import org.springframework.stereotype.Component; |
||||
|
import org.thingsboard.server.common.data.OtaPackageInfo; |
||||
|
import org.thingsboard.server.common.data.ota.OtaPackageType; |
||||
|
import org.thingsboard.server.common.data.id.DeviceProfileId; |
||||
|
import org.thingsboard.server.common.data.id.OtaPackageId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.page.PageData; |
||||
|
import org.thingsboard.server.common.data.page.PageLink; |
||||
|
import org.thingsboard.server.dao.DaoUtil; |
||||
|
import org.thingsboard.server.dao.ota.OtaPackageInfoDao; |
||||
|
import org.thingsboard.server.dao.model.sql.OtaPackageInfoEntity; |
||||
|
import org.thingsboard.server.dao.sql.JpaAbstractSearchTextDao; |
||||
|
|
||||
|
import java.util.Objects; |
||||
|
import java.util.UUID; |
||||
|
|
||||
|
@Slf4j |
||||
|
@Component |
||||
|
public class JpaOtaPackageInfoDao extends JpaAbstractSearchTextDao<OtaPackageInfoEntity, OtaPackageInfo> implements OtaPackageInfoDao { |
||||
|
|
||||
|
@Autowired |
||||
|
private OtaPackageInfoRepository otaPackageInfoRepository; |
||||
|
|
||||
|
@Override |
||||
|
protected Class<OtaPackageInfoEntity> getEntityClass() { |
||||
|
return OtaPackageInfoEntity.class; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
protected CrudRepository<OtaPackageInfoEntity, UUID> getCrudRepository() { |
||||
|
return otaPackageInfoRepository; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public OtaPackageInfo findById(TenantId tenantId, UUID id) { |
||||
|
return DaoUtil.getData(otaPackageInfoRepository.findOtaPackageInfoById(id)); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public OtaPackageInfo save(TenantId tenantId, OtaPackageInfo otaPackageInfo) { |
||||
|
OtaPackageInfo savedOtaPackage = super.save(tenantId, otaPackageInfo); |
||||
|
if (otaPackageInfo.getId() == null) { |
||||
|
return savedOtaPackage; |
||||
|
} else { |
||||
|
return findById(tenantId, savedOtaPackage.getId().getId()); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public PageData<OtaPackageInfo> findOtaPackageInfoByTenantId(TenantId tenantId, PageLink pageLink) { |
||||
|
return DaoUtil.toPageData(otaPackageInfoRepository |
||||
|
.findAllByTenantId( |
||||
|
tenantId.getId(), |
||||
|
Objects.toString(pageLink.getTextSearch(), ""), |
||||
|
DaoUtil.toPageable(pageLink))); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public PageData<OtaPackageInfo> findOtaPackageInfoByTenantIdAndDeviceProfileIdAndTypeAndHasData(TenantId tenantId, DeviceProfileId deviceProfileId, OtaPackageType otaPackageType, boolean hasData, PageLink pageLink) { |
||||
|
return DaoUtil.toPageData(otaPackageInfoRepository |
||||
|
.findAllByTenantIdAndTypeAndDeviceProfileIdAndHasData( |
||||
|
tenantId.getId(), |
||||
|
deviceProfileId.getId(), |
||||
|
otaPackageType, |
||||
|
hasData, |
||||
|
Objects.toString(pageLink.getTextSearch(), ""), |
||||
|
DaoUtil.toPageable(pageLink))); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public boolean isOtaPackageUsed(OtaPackageId otaPackageId, OtaPackageType otaPackageType, DeviceProfileId deviceProfileId) { |
||||
|
return otaPackageInfoRepository.isOtaPackageUsed(otaPackageId.getId(), deviceProfileId.getId(), otaPackageType.name()); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,60 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2021 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.dao.sql.ota; |
||||
|
|
||||
|
import org.springframework.data.domain.Page; |
||||
|
import org.springframework.data.domain.Pageable; |
||||
|
import org.springframework.data.jpa.repository.Query; |
||||
|
import org.springframework.data.repository.CrudRepository; |
||||
|
import org.springframework.data.repository.query.Param; |
||||
|
import org.thingsboard.server.common.data.ota.OtaPackageType; |
||||
|
import org.thingsboard.server.dao.model.sql.OtaPackageInfoEntity; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
public interface OtaPackageInfoRepository extends CrudRepository<OtaPackageInfoEntity, UUID> { |
||||
|
@Query("SELECT new OtaPackageInfoEntity(f.id, f.createdTime, f.tenantId, f.deviceProfileId, f.type, f.title, f.version, f.fileName, f.contentType, f.checksumAlgorithm, f.checksum, f.dataSize, f.additionalInfo, f.data IS NOT NULL) FROM OtaPackageEntity f WHERE " + |
||||
|
"f.tenantId = :tenantId " + |
||||
|
"AND LOWER(f.searchText) LIKE LOWER(CONCAT(:searchText, '%'))") |
||||
|
Page<OtaPackageInfoEntity> findAllByTenantId(@Param("tenantId") UUID tenantId, |
||||
|
@Param("searchText") String searchText, |
||||
|
Pageable pageable); |
||||
|
|
||||
|
@Query("SELECT new OtaPackageInfoEntity(f.id, f.createdTime, f.tenantId, f.deviceProfileId, f.type, f.title, f.version, f.fileName, f.contentType, f.checksumAlgorithm, f.checksum, f.dataSize, f.additionalInfo, f.data IS NOT NULL) FROM OtaPackageEntity f WHERE " + |
||||
|
"f.tenantId = :tenantId " + |
||||
|
"AND f.deviceProfileId = :deviceProfileId " + |
||||
|
"AND f.type = :type " + |
||||
|
"AND ((f.data IS NOT NULL AND :hasData = true) OR (f.data IS NULL AND :hasData = false ))" + |
||||
|
"AND LOWER(f.searchText) LIKE LOWER(CONCAT(:searchText, '%'))") |
||||
|
Page<OtaPackageInfoEntity> findAllByTenantIdAndTypeAndDeviceProfileIdAndHasData(@Param("tenantId") UUID tenantId, |
||||
|
@Param("deviceProfileId") UUID deviceProfileId, |
||||
|
@Param("type") OtaPackageType type, |
||||
|
@Param("hasData") boolean hasData, |
||||
|
@Param("searchText") String searchText, |
||||
|
Pageable pageable); |
||||
|
|
||||
|
@Query("SELECT new OtaPackageInfoEntity(f.id, f.createdTime, f.tenantId, f.deviceProfileId, f.type, f.title, f.version, f.fileName, f.contentType, f.checksumAlgorithm, f.checksum, f.dataSize, f.additionalInfo, f.data IS NOT NULL) FROM OtaPackageEntity f WHERE f.id = :id") |
||||
|
OtaPackageInfoEntity findOtaPackageInfoById(@Param("id") UUID id); |
||||
|
|
||||
|
@Query(value = "SELECT exists(SELECT * " + |
||||
|
"FROM device_profile AS dp " + |
||||
|
"LEFT JOIN device AS d ON dp.id = d.device_profile_id " + |
||||
|
"WHERE dp.id = :deviceProfileId AND " + |
||||
|
"(('FIRMWARE' = :type AND (dp.firmware_id = :otaPackageId OR d.firmware_id = :otaPackageId)) " + |
||||
|
"OR ('SOFTWARE' = :type AND (dp.software_id = :otaPackageId or d.software_id = :otaPackageId))))", nativeQuery = true) |
||||
|
boolean isOtaPackageUsed(@Param("otaPackageId") UUID otaPackageId, @Param("deviceProfileId") UUID deviceProfileId, @Param("type") String type); |
||||
|
|
||||
|
} |
||||
Some files were not shown because too many files changed in this diff
Loading…
Reference in new issue