Andrew Shvayka
2 years ago
committed by
GitHub
GitHub
29 changed files with 1547 additions and 6 deletions
@ -0,0 +1,95 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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 io.swagger.v3.oas.annotations.Parameter; |
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.http.HttpStatus; |
||||
|
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.ResponseBody; |
||||
|
import org.springframework.web.bind.annotation.ResponseStatus; |
||||
|
import org.springframework.web.bind.annotation.RestController; |
||||
|
import org.thingsboard.server.common.data.calculated_field.CalculatedField; |
||||
|
import org.thingsboard.server.common.data.exception.ThingsboardException; |
||||
|
import org.thingsboard.server.common.data.id.CalculatedFieldId; |
||||
|
import org.thingsboard.server.config.annotations.ApiOperation; |
||||
|
import org.thingsboard.server.dao.calculated_field.CalculatedFieldService; |
||||
|
import org.thingsboard.server.queue.util.TbCoreComponent; |
||||
|
import org.thingsboard.server.service.security.permission.Operation; |
||||
|
import org.thingsboard.server.service.security.permission.Resource; |
||||
|
|
||||
|
import static org.thingsboard.server.controller.ControllerConstants.TENANT_OR_CUSTOMER_AUTHORITY_PARAGRAPH; |
||||
|
import static org.thingsboard.server.controller.ControllerConstants.UUID_WIKI_LINK; |
||||
|
|
||||
|
@RestController |
||||
|
@TbCoreComponent |
||||
|
@RequestMapping("/api") |
||||
|
@RequiredArgsConstructor |
||||
|
@Slf4j |
||||
|
public class CalculatedFieldController extends BaseController { |
||||
|
|
||||
|
private final CalculatedFieldService calculatedFieldService; |
||||
|
|
||||
|
public static final String CALCULATED_FIELD_ID = "calculatedFieldId"; |
||||
|
|
||||
|
@ApiOperation(value = "Create Or Update Calculated Field (saveCalculatedField)", |
||||
|
notes = "Creates or Updates the Calculated Field. When creating calculated field, platform generates Calculated Field Id as " + UUID_WIKI_LINK + |
||||
|
"The newly created Calculated Field Id will be present in the response. " + |
||||
|
"Specify existing Calculated Field Id to update the calculated field. " + |
||||
|
"Referencing non-existing Calculated Field Id will cause 'Not Found' error. " + |
||||
|
"Remove 'id', 'tenantId' from the request body example (below) to create new Calculated Field entity. " |
||||
|
+ TENANT_OR_CUSTOMER_AUTHORITY_PARAGRAPH) |
||||
|
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN', 'CUSTOMER_USER')") |
||||
|
@RequestMapping(value = "/calculatedField", method = RequestMethod.POST) |
||||
|
@ResponseBody |
||||
|
public CalculatedField saveCalculatedField(@io.swagger.v3.oas.annotations.parameters.RequestBody(description = "A JSON value representing the calculated field.") |
||||
|
@RequestBody CalculatedField calculatedField) throws Exception { |
||||
|
calculatedField.setTenantId(getTenantId()); |
||||
|
checkEntity(calculatedField.getId(), calculatedField, Resource.CALCULATED_FIELD); |
||||
|
return calculatedFieldService.save(calculatedField); |
||||
|
} |
||||
|
|
||||
|
@ApiOperation(value = "Get Calculated Field (getCalculatedFieldById)", |
||||
|
notes = "Fetch the Calculated Field object based on the provided Calculated Field Id." |
||||
|
) |
||||
|
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN', 'CUSTOMER_USER')") |
||||
|
@RequestMapping(value = "/calculatedField/{calculatedFieldId}", method = RequestMethod.GET) |
||||
|
@ResponseBody |
||||
|
public CalculatedField getCalculatedFieldById(@Parameter @PathVariable(CALCULATED_FIELD_ID) String strCalculatedFieldId) throws ThingsboardException { |
||||
|
checkParameter(CALCULATED_FIELD_ID, strCalculatedFieldId); |
||||
|
CalculatedFieldId calculatedFieldId = new CalculatedFieldId(toUUID(strCalculatedFieldId)); |
||||
|
return checkCalculatedFieldId(calculatedFieldId, Operation.READ); |
||||
|
} |
||||
|
|
||||
|
|
||||
|
@ApiOperation(value = "Delete Calculated Field (deleteCalculatedField)", |
||||
|
notes = "Deletes the calculated field. Referencing non-existing Calculated Field Id will cause an error." + TENANT_OR_CUSTOMER_AUTHORITY_PARAGRAPH) |
||||
|
@PreAuthorize("hasAuthority('TENANT_ADMIN')") |
||||
|
@RequestMapping(value = "/calculatedField/{calculatedFieldId}", method = RequestMethod.DELETE) |
||||
|
@ResponseStatus(value = HttpStatus.OK) |
||||
|
public void deleteCalculatedField(@PathVariable(CALCULATED_FIELD_ID) String strCalculatedField) throws Exception { |
||||
|
checkParameter(CALCULATED_FIELD_ID, strCalculatedField); |
||||
|
CalculatedFieldId calculatedFieldId = new CalculatedFieldId(toUUID(strCalculatedField)); |
||||
|
checkCalculatedFieldId(calculatedFieldId, Operation.DELETE); |
||||
|
calculatedFieldService.deleteCalculatedField(getTenantId(), calculatedFieldId); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,142 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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 org.junit.After; |
||||
|
import org.junit.Before; |
||||
|
import org.junit.Test; |
||||
|
import org.thingsboard.common.util.JacksonUtil; |
||||
|
import org.thingsboard.server.common.data.Device; |
||||
|
import org.thingsboard.server.common.data.Tenant; |
||||
|
import org.thingsboard.server.common.data.User; |
||||
|
import org.thingsboard.server.common.data.calculated_field.CalculatedField; |
||||
|
import org.thingsboard.server.common.data.id.DeviceId; |
||||
|
import org.thingsboard.server.common.data.security.Authority; |
||||
|
import org.thingsboard.server.dao.service.DaoSqlTest; |
||||
|
|
||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status; |
||||
|
|
||||
|
@DaoSqlTest |
||||
|
public class CalculatedFieldControllerTest extends AbstractControllerTest { |
||||
|
|
||||
|
private Tenant savedTenant; |
||||
|
|
||||
|
@Before |
||||
|
public void beforeTest() throws Exception { |
||||
|
loginSysAdmin(); |
||||
|
|
||||
|
Tenant tenant = new Tenant(); |
||||
|
tenant.setTitle("My tenant"); |
||||
|
savedTenant = saveTenant(tenant); |
||||
|
assertThat(savedTenant).isNotNull(); |
||||
|
|
||||
|
User tenantAdmin = new User(); |
||||
|
tenantAdmin.setAuthority(Authority.TENANT_ADMIN); |
||||
|
tenantAdmin.setTenantId(savedTenant.getId()); |
||||
|
tenantAdmin.setEmail("tenant2@thingsboard.org"); |
||||
|
tenantAdmin.setFirstName("Joe"); |
||||
|
tenantAdmin.setLastName("Downs"); |
||||
|
|
||||
|
createUserAndLogin(tenantAdmin, "testPassword1"); |
||||
|
} |
||||
|
|
||||
|
@After |
||||
|
public void afterTest() throws Exception { |
||||
|
loginSysAdmin(); |
||||
|
|
||||
|
deleteTenant(savedTenant.getId()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testSaveCalculatedField() throws Exception { |
||||
|
Device testDevice = createDevice("Test device", "1234567890"); |
||||
|
CalculatedField calculatedField = getCalculatedField(testDevice.getId()); |
||||
|
|
||||
|
CalculatedField savedCalculatedField = doPost("/api/calculatedField", calculatedField, CalculatedField.class); |
||||
|
|
||||
|
assertThat(savedCalculatedField).isNotNull(); |
||||
|
assertThat(savedCalculatedField.getId()).isNotNull(); |
||||
|
assertThat(savedCalculatedField.getCreatedTime()).isGreaterThan(0); |
||||
|
assertThat(savedCalculatedField.getTenantId()).isEqualTo(savedTenant.getId()); |
||||
|
assertThat(savedCalculatedField.getEntityId()).isEqualTo(calculatedField.getEntityId()); |
||||
|
assertThat(savedCalculatedField.getType()).isEqualTo(calculatedField.getType()); |
||||
|
assertThat(savedCalculatedField.getName()).isEqualTo(calculatedField.getName()); |
||||
|
assertThat(savedCalculatedField.getConfiguration()).isEqualTo(calculatedField.getConfiguration()); |
||||
|
assertThat(savedCalculatedField.getVersion()).isEqualTo(calculatedField.getVersion()); |
||||
|
|
||||
|
savedCalculatedField.setName("Test CF"); |
||||
|
|
||||
|
CalculatedField updatedCalculatedField = doPost("/api/calculatedField", savedCalculatedField, CalculatedField.class); |
||||
|
|
||||
|
assertThat(updatedCalculatedField).isEqualTo(savedCalculatedField); |
||||
|
|
||||
|
doDelete("/api/calculatedField/" + savedCalculatedField.getId().getId().toString()) |
||||
|
.andExpect(status().isOk()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testGetCalculatedFieldById() throws Exception { |
||||
|
Device testDevice = createDevice("Test device", "1234567890"); |
||||
|
CalculatedField calculatedField = getCalculatedField(testDevice.getId()); |
||||
|
|
||||
|
CalculatedField savedCalculatedField = doPost("/api/calculatedField", calculatedField, CalculatedField.class); |
||||
|
CalculatedField fetchedCalculatedField = doGet("/api/calculatedField/" + savedCalculatedField.getId().getId(), CalculatedField.class); |
||||
|
|
||||
|
assertThat(fetchedCalculatedField).isNotNull(); |
||||
|
assertThat(fetchedCalculatedField).isEqualTo(savedCalculatedField); |
||||
|
|
||||
|
doDelete("/api/calculatedField/" + savedCalculatedField.getId().getId().toString()) |
||||
|
.andExpect(status().isOk()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testDeleteCalculatedField() throws Exception { |
||||
|
Device testDevice = createDevice("Test device", "1234567890"); |
||||
|
CalculatedField calculatedField = getCalculatedField(testDevice.getId()); |
||||
|
|
||||
|
CalculatedField savedCalculatedField = doPost("/api/calculatedField", calculatedField, CalculatedField.class); |
||||
|
|
||||
|
assertThat(savedCalculatedField).isNotNull(); |
||||
|
|
||||
|
doDelete("/api/calculatedField/" + savedCalculatedField.getId().getId().toString()) |
||||
|
.andExpect(status().isOk()); |
||||
|
doGet("/api/calculatedField/" + savedCalculatedField.getId().getId()).andExpect(status().isNotFound()); |
||||
|
|
||||
|
} |
||||
|
|
||||
|
private CalculatedField getCalculatedField(DeviceId deviceId) { |
||||
|
CalculatedField calculatedField = new CalculatedField(); |
||||
|
calculatedField.setEntityId(deviceId); |
||||
|
calculatedField.setType("Simple"); |
||||
|
calculatedField.setName("Test Calculated Field"); |
||||
|
calculatedField.setConfigurationVersion(1); |
||||
|
calculatedField.setConfiguration(JacksonUtil.toJsonNode("{\n" + |
||||
|
" \"T\": {\n" + |
||||
|
" \"key\": \"temperature\",\n" + |
||||
|
" \"type\": \"TIME_SERIES\"\n" + |
||||
|
" },\n" + |
||||
|
" \"H\": {\n" + |
||||
|
" \"key\": \"humidity\",\n" + |
||||
|
" \"type\": \"TIME_SERIES\",\n" + |
||||
|
" \"defaultValue\": 50\n" + |
||||
|
" }\n" + |
||||
|
" }\n")); |
||||
|
calculatedField.setVersion(1L); |
||||
|
return calculatedField; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,34 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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.calculated_field; |
||||
|
|
||||
|
import org.thingsboard.server.common.data.calculated_field.CalculatedField; |
||||
|
import org.thingsboard.server.common.data.calculated_field.CalculatedFieldLink; |
||||
|
import org.thingsboard.server.common.data.id.CalculatedFieldId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.dao.entity.EntityDaoService; |
||||
|
|
||||
|
public interface CalculatedFieldService extends EntityDaoService { |
||||
|
|
||||
|
CalculatedField save(CalculatedField calculatedField); |
||||
|
|
||||
|
CalculatedField findById(TenantId tenantId, CalculatedFieldId calculatedFieldId); |
||||
|
|
||||
|
void deleteCalculatedField(TenantId tenantId, CalculatedFieldId calculatedFieldId); |
||||
|
|
||||
|
CalculatedFieldLink saveCalculatedFieldLink(TenantId tenantId, CalculatedFieldLink calculatedFieldLink); |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,108 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.common.data.calculated_field; |
||||
|
|
||||
|
import com.fasterxml.jackson.databind.JsonNode; |
||||
|
import io.swagger.v3.oas.annotations.media.Schema; |
||||
|
import lombok.Data; |
||||
|
import lombok.EqualsAndHashCode; |
||||
|
import lombok.Getter; |
||||
|
import lombok.Setter; |
||||
|
import org.thingsboard.server.common.data.*; |
||||
|
import org.thingsboard.server.common.data.id.CalculatedFieldId; |
||||
|
import org.thingsboard.server.common.data.id.EntityId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.validation.Length; |
||||
|
import org.thingsboard.server.common.data.validation.NoXss; |
||||
|
|
||||
|
@Schema |
||||
|
@Data |
||||
|
@EqualsAndHashCode(callSuper = true) |
||||
|
public class CalculatedField extends BaseData<CalculatedFieldId> implements HasName, HasTenantId, HasVersion, ExportableEntity<CalculatedFieldId> { |
||||
|
|
||||
|
private static final long serialVersionUID = 4491966747773381420L; |
||||
|
|
||||
|
private TenantId tenantId; |
||||
|
private EntityId entityId; |
||||
|
|
||||
|
@NoXss |
||||
|
@Length(fieldName = "type") |
||||
|
private String type; |
||||
|
@NoXss |
||||
|
@Length(fieldName = "name") |
||||
|
@Schema(description = "User defined name of the calculated field.") |
||||
|
private String name; |
||||
|
@Schema(description = "Version of calculated field configuration.", example = "0") |
||||
|
private int configurationVersion; |
||||
|
@Schema(description = "JSON with the calculated field configuration.", implementation = com.fasterxml.jackson.databind.JsonNode.class) |
||||
|
private transient JsonNode configuration; |
||||
|
@Getter |
||||
|
@Setter |
||||
|
private Long version; |
||||
|
@Getter |
||||
|
@Setter |
||||
|
private CalculatedFieldId externalId; |
||||
|
|
||||
|
public CalculatedField() { |
||||
|
super(); |
||||
|
} |
||||
|
|
||||
|
public CalculatedField(CalculatedFieldId id) { |
||||
|
super(id); |
||||
|
} |
||||
|
|
||||
|
public CalculatedField(TenantId tenantId, EntityId entityId, String type, String name, int configurationVersion, JsonNode configuration, Long version, CalculatedFieldId externalId) { |
||||
|
super(); |
||||
|
this.tenantId = tenantId; |
||||
|
this.entityId = entityId; |
||||
|
this.type = type; |
||||
|
this.name = name; |
||||
|
this.configurationVersion = configurationVersion; |
||||
|
this.configuration = configuration; |
||||
|
this.version = version; |
||||
|
this.externalId = externalId; |
||||
|
} |
||||
|
|
||||
|
@Schema(description = "JSON object with the Calculated Field Id. Referencing non-existing Calculated Field Id will cause error.") |
||||
|
@Override |
||||
|
public CalculatedFieldId getId() { |
||||
|
return super.getId(); |
||||
|
} |
||||
|
|
||||
|
@Schema(description = "Timestamp of the calculated field creation, in milliseconds", example = "1609459200000", accessMode = Schema.AccessMode.READ_ONLY) |
||||
|
@Override |
||||
|
public long getCreatedTime() { |
||||
|
return super.getCreatedTime(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public String toString() { |
||||
|
return new StringBuilder() |
||||
|
.append("CalculatedField[") |
||||
|
.append("tenantId=").append(tenantId) |
||||
|
.append(", entityId=").append(entityId) |
||||
|
.append(", type='").append(type) |
||||
|
.append(", name='").append(name) |
||||
|
.append(", configurationVersion=").append(configurationVersion) |
||||
|
.append(", configuration=").append(configuration) |
||||
|
.append(", version=").append(version) |
||||
|
.append(", externalId=").append(externalId) |
||||
|
.append(", createdTime=").append(createdTime) |
||||
|
.append(", id=").append(id).append(']') |
||||
|
.toString(); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,71 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.common.data.calculated_field; |
||||
|
|
||||
|
import com.fasterxml.jackson.databind.JsonNode; |
||||
|
import io.swagger.v3.oas.annotations.media.Schema; |
||||
|
import lombok.Data; |
||||
|
import lombok.EqualsAndHashCode; |
||||
|
import org.thingsboard.server.common.data.BaseData; |
||||
|
import org.thingsboard.server.common.data.id.CalculatedFieldId; |
||||
|
import org.thingsboard.server.common.data.id.CalculatedFieldLinkId; |
||||
|
import org.thingsboard.server.common.data.id.EntityId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
|
||||
|
@Schema |
||||
|
@Data |
||||
|
@EqualsAndHashCode(callSuper = true) |
||||
|
public class CalculatedFieldLink extends BaseData<CalculatedFieldLinkId> { |
||||
|
|
||||
|
private static final long serialVersionUID = 6492846246722091530L; |
||||
|
|
||||
|
private TenantId tenantId; |
||||
|
private EntityId entityId; |
||||
|
|
||||
|
@Schema(description = "JSON object with the Calculated Field Id. ", accessMode = Schema.AccessMode.READ_ONLY) |
||||
|
private CalculatedFieldId calculatedFieldId; |
||||
|
@Schema(description = "JSON with the calculated field link configuration.", implementation = com.fasterxml.jackson.databind.JsonNode.class) |
||||
|
private transient JsonNode configuration; |
||||
|
|
||||
|
public CalculatedFieldLink() { |
||||
|
super(); |
||||
|
} |
||||
|
|
||||
|
public CalculatedFieldLink(CalculatedFieldLinkId id) { |
||||
|
super(id); |
||||
|
} |
||||
|
|
||||
|
public CalculatedFieldLink(TenantId tenantId, EntityId entityId, JsonNode configuration, CalculatedFieldId calculatedFieldId) { |
||||
|
this.tenantId = tenantId; |
||||
|
this.entityId = entityId; |
||||
|
this.configuration = configuration; |
||||
|
this.calculatedFieldId = calculatedFieldId; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public String toString() { |
||||
|
return new StringBuilder() |
||||
|
.append("CalculatedFieldLink[") |
||||
|
.append("tenantId=").append(tenantId) |
||||
|
.append(", entityId=").append(entityId) |
||||
|
.append(", calculatedFieldId=").append(calculatedFieldId) |
||||
|
.append(", configuration=").append(configuration) |
||||
|
.append(", createdTime=").append(createdTime) |
||||
|
.append(", id=").append(id).append(']') |
||||
|
.toString(); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,44 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.common.data.id; |
||||
|
|
||||
|
import com.fasterxml.jackson.annotation.JsonCreator; |
||||
|
import com.fasterxml.jackson.annotation.JsonProperty; |
||||
|
import io.swagger.v3.oas.annotations.media.Schema; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
@Schema |
||||
|
public class CalculatedFieldId extends UUIDBased implements EntityId { |
||||
|
|
||||
|
private static final long serialVersionUID = 1L; |
||||
|
|
||||
|
@JsonCreator |
||||
|
public CalculatedFieldId(@JsonProperty("id") UUID id) { |
||||
|
super(id); |
||||
|
} |
||||
|
|
||||
|
public static CalculatedFieldId fromString(String calculatedFieldId) { |
||||
|
return new CalculatedFieldId(UUID.fromString(calculatedFieldId)); |
||||
|
} |
||||
|
|
||||
|
@Schema(requiredMode = Schema.RequiredMode.REQUIRED, description = "string", example = "CALCULATED_FIELD", allowableValues = "CALCULATED_FIELD") |
||||
|
@Override |
||||
|
public EntityType getEntityType() { |
||||
|
return EntityType.CALCULATED_FIELD; |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,45 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.common.data.id; |
||||
|
|
||||
|
import com.fasterxml.jackson.annotation.JsonCreator; |
||||
|
import com.fasterxml.jackson.annotation.JsonProperty; |
||||
|
import io.swagger.v3.oas.annotations.media.Schema; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
@Schema |
||||
|
public class CalculatedFieldLinkId extends UUIDBased implements EntityId { |
||||
|
|
||||
|
private static final long serialVersionUID = 1L; |
||||
|
|
||||
|
@JsonCreator |
||||
|
public CalculatedFieldLinkId(@JsonProperty("id") UUID id) { |
||||
|
super(id); |
||||
|
} |
||||
|
|
||||
|
public static CalculatedFieldLinkId fromString(String calculatedFieldLinkId) { |
||||
|
return new CalculatedFieldLinkId(UUID.fromString(calculatedFieldLinkId)); |
||||
|
} |
||||
|
|
||||
|
@Schema(requiredMode = Schema.RequiredMode.REQUIRED, description = "string", example = "CALCULATED_FIELD_LINK", allowableValues = "CALCULATED_FIELD_LINK") |
||||
|
@Override |
||||
|
public EntityType getEntityType() { |
||||
|
return EntityType.CALCULATED_FIELD_LINK; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,136 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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.calculated_field; |
||||
|
|
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.stereotype.Service; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
import org.thingsboard.server.common.data.calculated_field.CalculatedField; |
||||
|
import org.thingsboard.server.common.data.calculated_field.CalculatedFieldLink; |
||||
|
import org.thingsboard.server.common.data.id.AssetId; |
||||
|
import org.thingsboard.server.common.data.id.CalculatedFieldId; |
||||
|
import org.thingsboard.server.common.data.id.DeviceId; |
||||
|
import org.thingsboard.server.common.data.id.EntityId; |
||||
|
import org.thingsboard.server.common.data.id.HasId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.dao.asset.AssetService; |
||||
|
import org.thingsboard.server.dao.device.DeviceService; |
||||
|
import org.thingsboard.server.dao.service.DataValidator; |
||||
|
|
||||
|
import java.util.Objects; |
||||
|
import java.util.Optional; |
||||
|
|
||||
|
import static org.thingsboard.server.dao.entity.AbstractEntityService.checkConstraintViolation; |
||||
|
import static org.thingsboard.server.dao.service.Validator.validateId; |
||||
|
|
||||
|
@Service("CalculatedFieldDaoService") |
||||
|
@Slf4j |
||||
|
@RequiredArgsConstructor |
||||
|
public class BaseCalculatedFieldService implements CalculatedFieldService { |
||||
|
|
||||
|
public static final String INCORRECT_TENANT_ID = "Incorrect tenantId "; |
||||
|
public static final String INCORRECT_CALCULATED_FIELD_ID = "Incorrect calculatedFieldId "; |
||||
|
|
||||
|
private final CalculatedFieldDao calculatedFieldDao; |
||||
|
private final CalculatedFieldLinkDao calculatedFieldLinkDao; |
||||
|
private final DeviceService deviceService; |
||||
|
private final AssetService assetService; |
||||
|
private final DataValidator<CalculatedField> calculatedFieldDataValidator; |
||||
|
private final DataValidator<CalculatedFieldLink> calculatedFieldLinkDataValidator; |
||||
|
|
||||
|
@Override |
||||
|
public CalculatedField save(CalculatedField calculatedField) { |
||||
|
calculatedFieldDataValidator.validate(calculatedField, CalculatedField::getTenantId); |
||||
|
try { |
||||
|
TenantId tenantId = calculatedField.getTenantId(); |
||||
|
checkEntityExistence(tenantId, calculatedField.getEntityId()); |
||||
|
log.trace("Executing save calculated field, [{}]", calculatedField); |
||||
|
CalculatedField savedCalculatedField = calculatedFieldDao.save(tenantId, calculatedField); |
||||
|
createOrUpdateCalculatedFieldLink(tenantId, savedCalculatedField); |
||||
|
return savedCalculatedField; |
||||
|
} catch (Exception e) { |
||||
|
checkConstraintViolation(e, |
||||
|
"calculated_field_unq_key", "Calculated Field with such name is already in exists!", |
||||
|
"calculated_field_external_id_unq_key", "Calculated Field with such external id already exists!"); |
||||
|
throw e; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public CalculatedField findById(TenantId tenantId, CalculatedFieldId calculatedFieldId) { |
||||
|
log.trace("Executing findById, tenantId [{}], rpcId [{}]", tenantId, calculatedFieldId); |
||||
|
validateId(tenantId, id -> INCORRECT_TENANT_ID + id); |
||||
|
validateId(calculatedFieldId, id -> INCORRECT_CALCULATED_FIELD_ID + id); |
||||
|
return calculatedFieldDao.findById(tenantId, calculatedFieldId.getId()); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void deleteCalculatedField(TenantId tenantId, CalculatedFieldId calculatedFieldId) { |
||||
|
log.trace("Executing deleteRpc, tenantId [{}], rpcId [{}]", tenantId, calculatedFieldId); |
||||
|
validateId(tenantId, id -> INCORRECT_TENANT_ID + id); |
||||
|
validateId(calculatedFieldId, id -> INCORRECT_CALCULATED_FIELD_ID + id); |
||||
|
calculatedFieldDao.removeById(tenantId, calculatedFieldId.getId()); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public CalculatedFieldLink saveCalculatedFieldLink(TenantId tenantId, CalculatedFieldLink calculatedFieldLink) { |
||||
|
calculatedFieldLinkDataValidator.validate(calculatedFieldLink, CalculatedFieldLink::getTenantId); |
||||
|
try { |
||||
|
log.trace("Executing save calculated field link, [{}]", calculatedFieldLink); |
||||
|
return calculatedFieldLinkDao.save(tenantId, calculatedFieldLink); |
||||
|
} catch (Exception e) { |
||||
|
checkConstraintViolation(e, "calculated_field_link_unq_key", "Calculated Field for such entity id is already exists!"); |
||||
|
throw e; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public Optional<HasId<?>> findEntity(TenantId tenantId, EntityId entityId) { |
||||
|
return Optional.ofNullable(findById(tenantId, new CalculatedFieldId(entityId.getId()))); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public EntityType getEntityType() { |
||||
|
return EntityType.CALCULATED_FIELD; |
||||
|
} |
||||
|
|
||||
|
private void checkEntityExistence(TenantId tenantId, EntityId entityId) { |
||||
|
switch (entityId.getEntityType()) { |
||||
|
case ASSET -> Optional.ofNullable(assetService.findAssetById(tenantId, (AssetId) entityId)) |
||||
|
.orElseThrow(() -> new IllegalArgumentException("Asset with id [" + entityId.getId() + "] does not exist.")); |
||||
|
case DEVICE -> Optional.ofNullable(deviceService.findDeviceById(tenantId, (DeviceId) entityId)) |
||||
|
.orElseThrow(() -> new IllegalArgumentException("Device with id [" + entityId.getId() + "] does not exist.")); |
||||
|
default -> |
||||
|
throw new IllegalArgumentException("Entity type '" + entityId.getEntityType() + "' is not supported."); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void createOrUpdateCalculatedFieldLink(TenantId tenantId, CalculatedField calculatedField) { |
||||
|
CalculatedFieldLink calculatedFieldLink = calculatedFieldLinkDao.findCalculatedFieldLinkByEntityId(tenantId.getId(), calculatedField.getEntityId().getId()); |
||||
|
saveCalculatedFieldLink(tenantId, Objects.requireNonNullElseGet(calculatedFieldLink, () -> createCalculatedFieldLink(tenantId, calculatedField))); |
||||
|
} |
||||
|
|
||||
|
private CalculatedFieldLink createCalculatedFieldLink(TenantId tenantId, CalculatedField calculatedField) { |
||||
|
CalculatedFieldLink calculatedFieldLink = new CalculatedFieldLink(); |
||||
|
calculatedFieldLink.setTenantId(tenantId); |
||||
|
calculatedFieldLink.setEntityId(calculatedField.getEntityId()); |
||||
|
calculatedFieldLink.setCalculatedFieldId(calculatedField.getId()); |
||||
|
calculatedFieldLink.setConfiguration(calculatedField.getConfiguration()); |
||||
|
return calculatedFieldLink; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,22 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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.calculated_field; |
||||
|
|
||||
|
import org.thingsboard.server.common.data.calculated_field.CalculatedField; |
||||
|
import org.thingsboard.server.dao.Dao; |
||||
|
|
||||
|
public interface CalculatedFieldDao extends Dao<CalculatedField> { |
||||
|
} |
||||
@ -0,0 +1,27 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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.calculated_field; |
||||
|
|
||||
|
import org.thingsboard.server.common.data.calculated_field.CalculatedFieldLink; |
||||
|
import org.thingsboard.server.dao.Dao; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
public interface CalculatedFieldLinkDao extends Dao<CalculatedFieldLink> { |
||||
|
|
||||
|
CalculatedFieldLink findCalculatedFieldLinkByEntityId(UUID tenantId, UUID entityId); |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,118 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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.model.sql; |
||||
|
|
||||
|
import com.fasterxml.jackson.databind.JsonNode; |
||||
|
import jakarta.persistence.Column; |
||||
|
import jakarta.persistence.Convert; |
||||
|
import jakarta.persistence.Entity; |
||||
|
import jakarta.persistence.Table; |
||||
|
import lombok.Data; |
||||
|
import lombok.EqualsAndHashCode; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
import org.thingsboard.server.common.data.calculated_field.CalculatedField; |
||||
|
import org.thingsboard.server.common.data.id.CalculatedFieldId; |
||||
|
import org.thingsboard.server.common.data.id.EntityIdFactory; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.dao.model.BaseEntity; |
||||
|
import org.thingsboard.server.dao.model.BaseSqlEntity; |
||||
|
import org.thingsboard.server.dao.util.mapping.JsonConverter; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.CALCULATED_FIELD_CONFIGURATION; |
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.CALCULATED_FIELD_CONFIGURATION_VERSION; |
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.CALCULATED_FIELD_ENTITY_ID; |
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.CALCULATED_FIELD_ENTITY_TYPE; |
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.CALCULATED_FIELD_EXTERNAL_ID; |
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.CALCULATED_FIELD_NAME; |
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.CALCULATED_FIELD_TABLE_NAME; |
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.CALCULATED_FIELD_TENANT_ID_COLUMN; |
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.CALCULATED_FIELD_TYPE; |
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.CALCULATED_FIELD_VERSION; |
||||
|
|
||||
|
@Data |
||||
|
@EqualsAndHashCode(callSuper = true) |
||||
|
@Entity |
||||
|
@Table(name = CALCULATED_FIELD_TABLE_NAME) |
||||
|
public class CalculatedFieldEntity extends BaseSqlEntity<CalculatedField> implements BaseEntity<CalculatedField> { |
||||
|
|
||||
|
@Column(name = CALCULATED_FIELD_TENANT_ID_COLUMN) |
||||
|
private UUID tenantId; |
||||
|
|
||||
|
@Column(name = CALCULATED_FIELD_ENTITY_TYPE) |
||||
|
private EntityType entityType; |
||||
|
|
||||
|
@Column(name = CALCULATED_FIELD_ENTITY_ID) |
||||
|
private UUID entityId; |
||||
|
|
||||
|
@Column(name = CALCULATED_FIELD_TYPE) |
||||
|
private String type; |
||||
|
|
||||
|
@Column(name = CALCULATED_FIELD_NAME) |
||||
|
private String name; |
||||
|
|
||||
|
@Column(name = CALCULATED_FIELD_CONFIGURATION_VERSION) |
||||
|
private int configurationVersion; |
||||
|
|
||||
|
@Convert(converter = JsonConverter.class) |
||||
|
@Column(name = CALCULATED_FIELD_CONFIGURATION) |
||||
|
private JsonNode configuration; |
||||
|
|
||||
|
@Column(name = CALCULATED_FIELD_VERSION) |
||||
|
private Long version; |
||||
|
|
||||
|
@Column(name = CALCULATED_FIELD_EXTERNAL_ID) |
||||
|
private UUID externalId; |
||||
|
|
||||
|
public CalculatedFieldEntity() { |
||||
|
super(); |
||||
|
} |
||||
|
|
||||
|
public CalculatedFieldEntity(CalculatedField calculatedField) { |
||||
|
this.setUuid(calculatedField.getUuidId()); |
||||
|
this.createdTime = calculatedField.getCreatedTime(); |
||||
|
this.tenantId = calculatedField.getTenantId().getId(); |
||||
|
this.entityType = calculatedField.getEntityId().getEntityType(); |
||||
|
this.entityId = calculatedField.getEntityId().getId(); |
||||
|
this.type = calculatedField.getType(); |
||||
|
this.name = calculatedField.getName(); |
||||
|
this.configurationVersion = calculatedField.getConfigurationVersion(); |
||||
|
this.configuration = calculatedField.getConfiguration(); |
||||
|
this.version = calculatedField.getVersion(); |
||||
|
if (calculatedField.getExternalId() != null) { |
||||
|
this.externalId = calculatedField.getExternalId().getId(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public CalculatedField toData() { |
||||
|
CalculatedField calculatedField = new CalculatedField(new CalculatedFieldId(id)); |
||||
|
calculatedField.setCreatedTime(createdTime); |
||||
|
calculatedField.setTenantId(TenantId.fromUUID(tenantId)); |
||||
|
calculatedField.setEntityId(EntityIdFactory.getByTypeAndUuid(entityType, entityId)); |
||||
|
calculatedField.setType(type); |
||||
|
calculatedField.setName(name); |
||||
|
calculatedField.setConfigurationVersion(configurationVersion); |
||||
|
calculatedField.setConfiguration(configuration); |
||||
|
calculatedField.setVersion(version); |
||||
|
if (externalId != null) { |
||||
|
calculatedField.setExternalId(new CalculatedFieldId(externalId)); |
||||
|
} |
||||
|
return calculatedField; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,92 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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.model.sql; |
||||
|
|
||||
|
import com.fasterxml.jackson.databind.JsonNode; |
||||
|
import jakarta.persistence.Column; |
||||
|
import jakarta.persistence.Convert; |
||||
|
import jakarta.persistence.Entity; |
||||
|
import jakarta.persistence.Table; |
||||
|
import lombok.Data; |
||||
|
import lombok.EqualsAndHashCode; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
import org.thingsboard.server.common.data.calculated_field.CalculatedFieldLink; |
||||
|
import org.thingsboard.server.common.data.id.CalculatedFieldId; |
||||
|
import org.thingsboard.server.common.data.id.CalculatedFieldLinkId; |
||||
|
import org.thingsboard.server.common.data.id.EntityIdFactory; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.dao.model.BaseEntity; |
||||
|
import org.thingsboard.server.dao.model.BaseSqlEntity; |
||||
|
import org.thingsboard.server.dao.util.mapping.JsonConverter; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.CALCULATED_FIELD_LINK_CALCULATED_FIELD_ID; |
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.CALCULATED_FIELD_LINK_CONFIGURATION; |
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.CALCULATED_FIELD_LINK_ENTITY_ID; |
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.CALCULATED_FIELD_LINK_ENTITY_TYPE; |
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.CALCULATED_FIELD_LINK_TABLE_NAME; |
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.CALCULATED_FIELD_LINK_TENANT_ID_COLUMN; |
||||
|
|
||||
|
@Data |
||||
|
@EqualsAndHashCode(callSuper = true) |
||||
|
@Entity |
||||
|
@Table(name = CALCULATED_FIELD_LINK_TABLE_NAME) |
||||
|
public class CalculatedFieldLinkEntity extends BaseSqlEntity<CalculatedFieldLink> implements BaseEntity<CalculatedFieldLink> { |
||||
|
|
||||
|
@Column(name = CALCULATED_FIELD_LINK_TENANT_ID_COLUMN) |
||||
|
private UUID tenantId; |
||||
|
|
||||
|
@Column(name = CALCULATED_FIELD_LINK_ENTITY_TYPE) |
||||
|
private EntityType entityType; |
||||
|
|
||||
|
@Column(name = CALCULATED_FIELD_LINK_ENTITY_ID) |
||||
|
private UUID entityId; |
||||
|
|
||||
|
@Column(name = CALCULATED_FIELD_LINK_CALCULATED_FIELD_ID) |
||||
|
private UUID calculatedFieldId; |
||||
|
|
||||
|
@Convert(converter = JsonConverter.class) |
||||
|
@Column(name = CALCULATED_FIELD_LINK_CONFIGURATION) |
||||
|
private JsonNode configuration; |
||||
|
|
||||
|
|
||||
|
public CalculatedFieldLinkEntity() { |
||||
|
super(); |
||||
|
} |
||||
|
|
||||
|
public CalculatedFieldLinkEntity(CalculatedFieldLink calculatedFieldLink) { |
||||
|
this.setUuid(calculatedFieldLink.getUuidId()); |
||||
|
this.createdTime = calculatedFieldLink.getCreatedTime(); |
||||
|
this.tenantId = calculatedFieldLink.getTenantId().getId(); |
||||
|
this.entityType = calculatedFieldLink.getEntityId().getEntityType(); |
||||
|
this.entityId = calculatedFieldLink.getEntityId().getId(); |
||||
|
this.calculatedFieldId = calculatedFieldLink.getCalculatedFieldId().getId(); |
||||
|
this.configuration = calculatedFieldLink.getConfiguration(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public CalculatedFieldLink toData() { |
||||
|
CalculatedFieldLink calculatedFieldLink = new CalculatedFieldLink(new CalculatedFieldLinkId(id)); |
||||
|
calculatedFieldLink.setCreatedTime(createdTime); |
||||
|
calculatedFieldLink.setTenantId(TenantId.fromUUID(tenantId)); |
||||
|
calculatedFieldLink.setEntityId(EntityIdFactory.getByTypeAndUuid(entityType, entityId)); |
||||
|
calculatedFieldLink.setCalculatedFieldId(new CalculatedFieldId(calculatedFieldId)); |
||||
|
calculatedFieldLink.setConfiguration(configuration); |
||||
|
return calculatedFieldLink; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,41 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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.service.validator; |
||||
|
|
||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||
|
import org.springframework.stereotype.Component; |
||||
|
import org.thingsboard.server.common.data.calculated_field.CalculatedField; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.dao.calculated_field.CalculatedFieldDao; |
||||
|
import org.thingsboard.server.dao.exception.DataValidationException; |
||||
|
import org.thingsboard.server.dao.service.DataValidator; |
||||
|
|
||||
|
@Component |
||||
|
public class CalculatedFieldDataValidator extends DataValidator<CalculatedField> { |
||||
|
|
||||
|
@Autowired |
||||
|
private CalculatedFieldDao calculatedFieldDao; |
||||
|
|
||||
|
@Override |
||||
|
protected CalculatedField validateUpdate(TenantId tenantId, CalculatedField calculatedField) { |
||||
|
CalculatedField old = calculatedFieldDao.findById(calculatedField.getTenantId(), calculatedField.getId().getId()); |
||||
|
if (old == null) { |
||||
|
throw new DataValidationException("Can't update non existing calculated field!"); |
||||
|
} |
||||
|
return old; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,41 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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.service.validator; |
||||
|
|
||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||
|
import org.springframework.stereotype.Component; |
||||
|
import org.thingsboard.server.common.data.calculated_field.CalculatedFieldLink; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.dao.calculated_field.CalculatedFieldLinkDao; |
||||
|
import org.thingsboard.server.dao.exception.DataValidationException; |
||||
|
import org.thingsboard.server.dao.service.DataValidator; |
||||
|
|
||||
|
@Component |
||||
|
public class CalculatedFieldLinkDataValidator extends DataValidator<CalculatedFieldLink> { |
||||
|
|
||||
|
@Autowired |
||||
|
private CalculatedFieldLinkDao calculatedFieldLinkDao; |
||||
|
|
||||
|
@Override |
||||
|
protected CalculatedFieldLink validateUpdate(TenantId tenantId, CalculatedFieldLink calculatedFieldLink) { |
||||
|
CalculatedFieldLink old = calculatedFieldLinkDao.findById(calculatedFieldLink.getTenantId(), calculatedFieldLink.getId().getId()); |
||||
|
if (old == null) { |
||||
|
throw new DataValidationException("Can't update non existing calculated field link!"); |
||||
|
} |
||||
|
return old; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,27 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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.calculated_field; |
||||
|
|
||||
|
import org.springframework.data.jpa.repository.JpaRepository; |
||||
|
import org.thingsboard.server.dao.model.sql.CalculatedFieldLinkEntity; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
public interface CalculatedFieldLinkRepository extends JpaRepository<CalculatedFieldLinkEntity, UUID> { |
||||
|
|
||||
|
CalculatedFieldLinkEntity findByTenantIdAndEntityId(UUID tenantId, UUID entityId); |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,24 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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.calculated_field; |
||||
|
|
||||
|
import org.springframework.data.jpa.repository.JpaRepository; |
||||
|
import org.thingsboard.server.dao.model.sql.CalculatedFieldEntity; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
public interface CalculatedFieldRepository extends JpaRepository<CalculatedFieldEntity, UUID> { |
||||
|
} |
||||
@ -0,0 +1,47 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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.calculated_field; |
||||
|
|
||||
|
import lombok.AllArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.data.jpa.repository.JpaRepository; |
||||
|
import org.springframework.stereotype.Component; |
||||
|
import org.thingsboard.server.common.data.calculated_field.CalculatedField; |
||||
|
import org.thingsboard.server.dao.calculated_field.CalculatedFieldDao; |
||||
|
import org.thingsboard.server.dao.model.sql.CalculatedFieldEntity; |
||||
|
import org.thingsboard.server.dao.sql.JpaAbstractDao; |
||||
|
import org.thingsboard.server.dao.util.SqlDao; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
@Slf4j |
||||
|
@Component |
||||
|
@AllArgsConstructor |
||||
|
@SqlDao |
||||
|
public class JpaCalculatedFieldDao extends JpaAbstractDao<CalculatedFieldEntity, CalculatedField> implements CalculatedFieldDao { |
||||
|
|
||||
|
private final CalculatedFieldRepository calculatedFieldRepository; |
||||
|
|
||||
|
@Override |
||||
|
protected Class<CalculatedFieldEntity> getEntityClass() { |
||||
|
return CalculatedFieldEntity.class; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
protected JpaRepository<CalculatedFieldEntity, UUID> getRepository() { |
||||
|
return calculatedFieldRepository; |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,54 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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.calculated_field; |
||||
|
|
||||
|
import lombok.AllArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.data.jpa.repository.JpaRepository; |
||||
|
import org.springframework.stereotype.Component; |
||||
|
import org.thingsboard.server.common.data.calculated_field.CalculatedFieldLink; |
||||
|
import org.thingsboard.server.dao.DaoUtil; |
||||
|
import org.thingsboard.server.dao.calculated_field.CalculatedFieldLinkDao; |
||||
|
import org.thingsboard.server.dao.model.sql.CalculatedFieldLinkEntity; |
||||
|
import org.thingsboard.server.dao.sql.JpaAbstractDao; |
||||
|
import org.thingsboard.server.dao.util.SqlDao; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
@Slf4j |
||||
|
@Component |
||||
|
@AllArgsConstructor |
||||
|
@SqlDao |
||||
|
public class JpaCalculatedFieldLinkDao extends JpaAbstractDao<CalculatedFieldLinkEntity, CalculatedFieldLink> implements CalculatedFieldLinkDao { |
||||
|
|
||||
|
private final CalculatedFieldLinkRepository calculatedFieldLinkRepository; |
||||
|
|
||||
|
@Override |
||||
|
public CalculatedFieldLink findCalculatedFieldLinkByEntityId(UUID tenantId, UUID entityId) { |
||||
|
return DaoUtil.getData(calculatedFieldLinkRepository.findByTenantIdAndEntityId(tenantId, entityId)); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
protected Class<CalculatedFieldLinkEntity> getEntityClass() { |
||||
|
return CalculatedFieldLinkEntity.class; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
protected JpaRepository<CalculatedFieldLinkEntity, UUID> getRepository() { |
||||
|
return calculatedFieldLinkRepository; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,193 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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.calculated_field; |
||||
|
|
||||
|
import com.google.common.util.concurrent.ListeningExecutorService; |
||||
|
import com.google.common.util.concurrent.MoreExecutors; |
||||
|
import org.junit.After; |
||||
|
import org.junit.Before; |
||||
|
import org.junit.Test; |
||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||
|
import org.thingsboard.common.util.JacksonUtil; |
||||
|
import org.thingsboard.common.util.ThingsBoardExecutors; |
||||
|
import org.thingsboard.server.common.data.Device; |
||||
|
import org.thingsboard.server.common.data.calculated_field.CalculatedField; |
||||
|
import org.thingsboard.server.common.data.calculated_field.CalculatedFieldLink; |
||||
|
import org.thingsboard.server.common.data.id.CalculatedFieldId; |
||||
|
import org.thingsboard.server.common.data.id.DeviceId; |
||||
|
import org.thingsboard.server.dao.device.DeviceService; |
||||
|
import org.thingsboard.server.dao.exception.DataValidationException; |
||||
|
import org.thingsboard.server.dao.service.AbstractServiceTest; |
||||
|
import org.thingsboard.server.dao.service.DaoSqlTest; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
import static org.assertj.core.api.Assertions.assertThatThrownBy; |
||||
|
|
||||
|
@DaoSqlTest |
||||
|
public class CalculatedFieldServiceTest extends AbstractServiceTest { |
||||
|
|
||||
|
@Autowired |
||||
|
private CalculatedFieldService calculatedFieldService; |
||||
|
@Autowired |
||||
|
private DeviceService deviceService; |
||||
|
|
||||
|
private ListeningExecutorService executor; |
||||
|
|
||||
|
@Before |
||||
|
public void before() { |
||||
|
executor = MoreExecutors.listeningDecorator(ThingsBoardExecutors.newWorkStealingPool(8, getClass())); |
||||
|
} |
||||
|
|
||||
|
@After |
||||
|
public void after() { |
||||
|
executor.shutdownNow(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testSaveCalculatedField() { |
||||
|
Device device = createTestDevice(); |
||||
|
CalculatedField calculatedField = getCalculatedField(device.getId()); |
||||
|
CalculatedField savedCalculatedField = calculatedFieldService.save(calculatedField); |
||||
|
|
||||
|
assertThat(savedCalculatedField).isNotNull(); |
||||
|
assertThat(savedCalculatedField.getId()).isNotNull(); |
||||
|
assertThat(savedCalculatedField.getCreatedTime()).isGreaterThan(0); |
||||
|
assertThat(savedCalculatedField.getTenantId()).isEqualTo(calculatedField.getTenantId()); |
||||
|
assertThat(savedCalculatedField.getEntityId()).isEqualTo(calculatedField.getEntityId()); |
||||
|
assertThat(savedCalculatedField.getType()).isEqualTo(calculatedField.getType()); |
||||
|
assertThat(savedCalculatedField.getName()).isEqualTo(calculatedField.getName()); |
||||
|
assertThat(savedCalculatedField.getConfiguration()).isEqualTo(calculatedField.getConfiguration()); |
||||
|
assertThat(savedCalculatedField.getVersion()).isEqualTo(calculatedField.getVersion()); |
||||
|
|
||||
|
savedCalculatedField.setName("Test CF"); |
||||
|
|
||||
|
CalculatedField updatedCalculatedField = calculatedFieldService.save(savedCalculatedField); |
||||
|
|
||||
|
assertThat(updatedCalculatedField).isEqualTo(savedCalculatedField); |
||||
|
|
||||
|
calculatedFieldService.deleteCalculatedField(tenantId, savedCalculatedField.getId()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testSaveCalculatesFieldWithNonExistingDeviceId() { |
||||
|
CalculatedField calculatedField = getCalculatedField(new DeviceId(UUID.fromString("038f8668-c9fd-4f00-8501-ce20f2f93c22"))); |
||||
|
|
||||
|
assertThatThrownBy(() -> calculatedFieldService.save(calculatedField)) |
||||
|
.isInstanceOf(IllegalArgumentException.class) |
||||
|
.hasMessage("Device with id [" + calculatedField.getEntityId().getId() + "] does not exist."); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testSaveCalculatedFieldWithExistingName() { |
||||
|
Device device = createTestDevice(); |
||||
|
CalculatedField calculatedField = getCalculatedField(device.getId()); |
||||
|
calculatedFieldService.save(calculatedField); |
||||
|
|
||||
|
assertThatThrownBy(() -> calculatedFieldService.save(calculatedField)) |
||||
|
.isInstanceOf(DataValidationException.class) |
||||
|
.hasMessage("Calculated Field with such name is already in exists!"); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testSaveCalculatedFieldWithExistingExternalId() { |
||||
|
Device device = createTestDevice(); |
||||
|
CalculatedField calculatedField = getCalculatedField(device.getId()); |
||||
|
calculatedField.setExternalId(new CalculatedFieldId(UUID.fromString("2ef69d0a-89cf-4868-86f8-c50551d87ebe"))); |
||||
|
|
||||
|
calculatedFieldService.save(calculatedField); |
||||
|
|
||||
|
calculatedField.setName("Test 2"); |
||||
|
assertThatThrownBy(() -> calculatedFieldService.save(calculatedField)) |
||||
|
.isInstanceOf(DataValidationException.class) |
||||
|
.hasMessage("Calculated Field with such external id already exists!"); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindCalculatedFieldById() { |
||||
|
CalculatedField savedCalculatedField = saveValidCalculatedField(); |
||||
|
CalculatedField fetchedCalculatedField = calculatedFieldService.findById(tenantId, savedCalculatedField.getId()); |
||||
|
|
||||
|
assertThat(fetchedCalculatedField).isEqualTo(savedCalculatedField); |
||||
|
|
||||
|
calculatedFieldService.deleteCalculatedField(tenantId, savedCalculatedField.getId()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testDeleteCalculatedField() { |
||||
|
CalculatedField savedCalculatedField = saveValidCalculatedField(); |
||||
|
|
||||
|
calculatedFieldService.deleteCalculatedField(tenantId, savedCalculatedField.getId()); |
||||
|
|
||||
|
assertThat(calculatedFieldService.findById(tenantId, savedCalculatedField.getId())).isNull(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testSaveCalculatedFieldLinkIfCalculatedFieldForSuchEntityExists() { |
||||
|
CalculatedField savedCalculatedField = saveValidCalculatedField(); |
||||
|
CalculatedFieldLink calculatedFieldLink = getCalculatedFieldLink(savedCalculatedField); |
||||
|
|
||||
|
assertThatThrownBy(() -> calculatedFieldService.saveCalculatedFieldLink(tenantId, calculatedFieldLink)) |
||||
|
.isInstanceOf(DataValidationException.class) |
||||
|
.hasMessage("Calculated Field for such entity id is already exists!"); |
||||
|
} |
||||
|
|
||||
|
private CalculatedField saveValidCalculatedField() { |
||||
|
Device device = createTestDevice(); |
||||
|
CalculatedField calculatedField = getCalculatedField(device.getId()); |
||||
|
return calculatedFieldService.save(calculatedField); |
||||
|
} |
||||
|
|
||||
|
private CalculatedField getCalculatedField(DeviceId deviceId) { |
||||
|
CalculatedField calculatedField = new CalculatedField(); |
||||
|
calculatedField.setTenantId(tenantId); |
||||
|
calculatedField.setEntityId(deviceId); |
||||
|
calculatedField.setType("Simple"); |
||||
|
calculatedField.setName("Test Calculated Field"); |
||||
|
calculatedField.setConfigurationVersion(1); |
||||
|
calculatedField.setConfiguration(JacksonUtil.toJsonNode("{\n" + |
||||
|
" \"T\": {\n" + |
||||
|
" \"key\": \"temperature\",\n" + |
||||
|
" \"type\": \"TIME_SERIES\"\n" + |
||||
|
" },\n" + |
||||
|
" \"H\": {\n" + |
||||
|
" \"key\": \"humidity\",\n" + |
||||
|
" \"type\": \"TIME_SERIES\",\n" + |
||||
|
" \"defaultValue\": 50\n" + |
||||
|
" }\n" + |
||||
|
" }\n")); |
||||
|
calculatedField.setVersion(1L); |
||||
|
return calculatedField; |
||||
|
} |
||||
|
|
||||
|
private CalculatedFieldLink getCalculatedFieldLink(CalculatedField calculatedField) { |
||||
|
CalculatedFieldLink calculatedFieldLink = new CalculatedFieldLink(); |
||||
|
calculatedFieldLink.setTenantId(tenantId); |
||||
|
calculatedFieldLink.setEntityId(calculatedField.getEntityId()); |
||||
|
calculatedFieldLink.setConfiguration(calculatedField.getConfiguration()); |
||||
|
calculatedFieldLink.setCalculatedFieldId(calculatedField.getId()); |
||||
|
return calculatedFieldLink; |
||||
|
} |
||||
|
|
||||
|
private Device createTestDevice() { |
||||
|
Device device = new Device(); |
||||
|
device.setTenantId(tenantId); |
||||
|
device.setName("Test"); |
||||
|
return deviceService.saveDevice(device); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,57 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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.service.validator; |
||||
|
|
||||
|
import org.junit.jupiter.api.Test; |
||||
|
import org.springframework.boot.test.context.SpringBootTest; |
||||
|
import org.springframework.boot.test.mock.mockito.MockBean; |
||||
|
import org.springframework.boot.test.mock.mockito.SpyBean; |
||||
|
import org.thingsboard.server.common.data.calculated_field.CalculatedField; |
||||
|
import org.thingsboard.server.common.data.id.CalculatedFieldId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.dao.calculated_field.CalculatedFieldDao; |
||||
|
import org.thingsboard.server.dao.exception.DataValidationException; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
import static org.assertj.core.api.Assertions.assertThatThrownBy; |
||||
|
import static org.mockito.BDDMockito.given; |
||||
|
|
||||
|
@SpringBootTest(classes = CalculatedFieldDataValidator.class) |
||||
|
public class CalculatedFieldDataValidatorTest { |
||||
|
|
||||
|
private final TenantId TENANT_ID = TenantId.fromUUID(UUID.fromString("7b5229e9-166e-41a9-a257-3b1dafad1b04")); |
||||
|
private final CalculatedFieldId CALCULATED_FIELD_ID = new CalculatedFieldId(UUID.fromString("060fbe45-fbb2-4549-abf3-f72a6be3cb9f")); |
||||
|
|
||||
|
@MockBean |
||||
|
private CalculatedFieldDao calculatedFieldDao; |
||||
|
@SpyBean |
||||
|
private CalculatedFieldDataValidator validator; |
||||
|
|
||||
|
@Test |
||||
|
public void testUpdateNonExistingCalculatedField() { |
||||
|
CalculatedField calculatedField = new CalculatedField(CALCULATED_FIELD_ID); |
||||
|
calculatedField.setType("Simple"); |
||||
|
calculatedField.setName("Test"); |
||||
|
|
||||
|
given(calculatedFieldDao.findById(TENANT_ID, CALCULATED_FIELD_ID.getId())).willReturn(null); |
||||
|
|
||||
|
assertThatThrownBy(() -> validator.validateUpdate(TENANT_ID, calculatedField)) |
||||
|
.isInstanceOf(DataValidationException.class) |
||||
|
.hasMessage("Can't update non existing calculated field!"); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,57 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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.service.validator; |
||||
|
|
||||
|
import org.junit.jupiter.api.Test; |
||||
|
import org.springframework.boot.test.context.SpringBootTest; |
||||
|
import org.springframework.boot.test.mock.mockito.MockBean; |
||||
|
import org.springframework.boot.test.mock.mockito.SpyBean; |
||||
|
import org.thingsboard.server.common.data.calculated_field.CalculatedFieldLink; |
||||
|
import org.thingsboard.server.common.data.id.CalculatedFieldId; |
||||
|
import org.thingsboard.server.common.data.id.CalculatedFieldLinkId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.dao.calculated_field.CalculatedFieldLinkDao; |
||||
|
import org.thingsboard.server.dao.exception.DataValidationException; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
import static org.assertj.core.api.Assertions.assertThatThrownBy; |
||||
|
import static org.mockito.BDDMockito.given; |
||||
|
|
||||
|
@SpringBootTest(classes = CalculatedFieldLinkDataValidator.class) |
||||
|
public class CalculatedFieldLinkDataValidatorTest { |
||||
|
|
||||
|
private final TenantId TENANT_ID = TenantId.fromUUID(UUID.fromString("2ba09d99-6143-43dc-b645-381fc0c43ebe")); |
||||
|
private final CalculatedFieldLinkId CALCULATED_FIELD_LINK_ID = new CalculatedFieldLinkId(UUID.fromString("a5609ef4-cb42-43ce-9b23-e090a4878d1c")); |
||||
|
|
||||
|
@MockBean |
||||
|
private CalculatedFieldLinkDao calculatedFieldLinkDao; |
||||
|
@SpyBean |
||||
|
private CalculatedFieldLinkDataValidator validator; |
||||
|
|
||||
|
@Test |
||||
|
public void testUpdateNonExistingCalculatedField() { |
||||
|
CalculatedFieldLink calculatedFieldLink = new CalculatedFieldLink(CALCULATED_FIELD_LINK_ID); |
||||
|
calculatedFieldLink.setCalculatedFieldId(new CalculatedFieldId(UUID.fromString("136477af-fd07-4498-b9c9-54fe50e82992"))); |
||||
|
|
||||
|
given(calculatedFieldLinkDao.findById(TENANT_ID, CALCULATED_FIELD_LINK_ID.getId())).willReturn(null); |
||||
|
|
||||
|
assertThatThrownBy(() -> validator.validateUpdate(TENANT_ID, calculatedFieldLink)) |
||||
|
.isInstanceOf(DataValidationException.class) |
||||
|
.hasMessage("Can't update non existing calculated field link!"); |
||||
|
} |
||||
|
|
||||
|
} |
||||
Loading…
Reference in new issue