205 changed files with 8167 additions and 3396 deletions
@ -0,0 +1,129 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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 com.google.common.util.concurrent.ListenableFuture; |
||||
|
import org.apache.commons.lang3.StringUtils; |
||||
|
import org.springframework.http.HttpStatus; |
||||
|
import org.springframework.security.access.prepost.PreAuthorize; |
||||
|
import org.springframework.web.bind.annotation.*; |
||||
|
import org.thingsboard.server.common.data.Customer; |
||||
|
import org.thingsboard.server.common.data.Event; |
||||
|
import org.thingsboard.server.common.data.alarm.Alarm; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmId; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmQuery; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmStatus; |
||||
|
import org.thingsboard.server.common.data.asset.Asset; |
||||
|
import org.thingsboard.server.common.data.id.*; |
||||
|
import org.thingsboard.server.common.data.page.TextPageData; |
||||
|
import org.thingsboard.server.common.data.page.TextPageLink; |
||||
|
import org.thingsboard.server.common.data.page.TimePageData; |
||||
|
import org.thingsboard.server.common.data.page.TimePageLink; |
||||
|
import org.thingsboard.server.dao.asset.AssetSearchQuery; |
||||
|
import org.thingsboard.server.dao.exception.IncorrectParameterException; |
||||
|
import org.thingsboard.server.dao.model.ModelConstants; |
||||
|
import org.thingsboard.server.exception.ThingsboardErrorCode; |
||||
|
import org.thingsboard.server.exception.ThingsboardException; |
||||
|
import org.thingsboard.server.service.security.model.SecurityUser; |
||||
|
|
||||
|
import java.util.ArrayList; |
||||
|
import java.util.List; |
||||
|
import java.util.stream.Collectors; |
||||
|
|
||||
|
@RestController |
||||
|
@RequestMapping("/api") |
||||
|
public class AlarmController extends BaseController { |
||||
|
|
||||
|
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN', 'CUSTOMER_USER')") |
||||
|
@RequestMapping(value = "/alarm/{alarmId}", method = RequestMethod.GET) |
||||
|
@ResponseBody |
||||
|
public Alarm getAlarmById(@PathVariable("alarmId") String strAlarmId) throws ThingsboardException { |
||||
|
checkParameter("alarmId", strAlarmId); |
||||
|
try { |
||||
|
AlarmId alarmId = new AlarmId(toUUID(strAlarmId)); |
||||
|
return checkAlarmId(alarmId); |
||||
|
} catch (Exception e) { |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN', 'CUSTOMER_USER')") |
||||
|
@RequestMapping(value = "/alarm", method = RequestMethod.POST) |
||||
|
@ResponseBody |
||||
|
public Alarm saveAlarm(@RequestBody Alarm alarm) throws ThingsboardException { |
||||
|
try { |
||||
|
alarm.setTenantId(getCurrentUser().getTenantId()); |
||||
|
return checkNotNull(alarmService.createOrUpdateAlarm(alarm)); |
||||
|
} catch (Exception e) { |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN', 'CUSTOMER_USER')") |
||||
|
@RequestMapping(value = "/alarm/{alarmId}/ack", method = RequestMethod.POST) |
||||
|
@ResponseStatus(value = HttpStatus.OK) |
||||
|
public void ackAlarm(@PathVariable("alarmId") String strAlarmId) throws ThingsboardException { |
||||
|
checkParameter("alarmId", strAlarmId); |
||||
|
try { |
||||
|
AlarmId alarmId = new AlarmId(toUUID(strAlarmId)); |
||||
|
checkAlarmId(alarmId); |
||||
|
alarmService.ackAlarm(alarmId, System.currentTimeMillis()).get(); |
||||
|
} catch (Exception e) { |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN', 'CUSTOMER_USER')") |
||||
|
@RequestMapping(value = "/alarm/{alarmId}/clear", method = RequestMethod.POST) |
||||
|
@ResponseStatus(value = HttpStatus.OK) |
||||
|
public void clearAlarm(@PathVariable("alarmId") String strAlarmId) throws ThingsboardException { |
||||
|
checkParameter("alarmId", strAlarmId); |
||||
|
try { |
||||
|
AlarmId alarmId = new AlarmId(toUUID(strAlarmId)); |
||||
|
checkAlarmId(alarmId); |
||||
|
alarmService.clearAlarm(alarmId, System.currentTimeMillis()).get(); |
||||
|
} catch (Exception e) { |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN', 'CUSTOMER_USER')") |
||||
|
@RequestMapping(value = "/alarm/{entityType}/{entityId}", method = RequestMethod.GET) |
||||
|
@ResponseBody |
||||
|
public TimePageData<Alarm> getAlarms( |
||||
|
@PathVariable("entityType") String strEntityType, |
||||
|
@PathVariable("entityId") String strEntityId, |
||||
|
@RequestParam(required = false) String status, |
||||
|
@RequestParam int limit, |
||||
|
@RequestParam(required = false) Long startTime, |
||||
|
@RequestParam(required = false) Long endTime, |
||||
|
@RequestParam(required = false, defaultValue = "false") boolean ascOrder, |
||||
|
@RequestParam(required = false) String offset |
||||
|
) throws ThingsboardException { |
||||
|
checkParameter("EntityId", strEntityId); |
||||
|
checkParameter("EntityType", strEntityType); |
||||
|
EntityId entityId = EntityIdFactory.getByTypeAndId(strEntityType, strEntityId); |
||||
|
AlarmStatus alarmStatus = StringUtils.isEmpty(status) ? null : AlarmStatus.valueOf(status); |
||||
|
checkEntityId(entityId); |
||||
|
try { |
||||
|
TimePageLink pageLink = createPageLink(limit, startTime, endTime, ascOrder, offset); |
||||
|
return checkNotNull(alarmService.findAlarms(new AlarmQuery(entityId, pageLink, alarmStatus)).get()); |
||||
|
} catch (Exception e) { |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,659 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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 static org.hamcrest.Matchers.containsString; |
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.NULL_UUID; |
||||
|
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status; |
||||
|
|
||||
|
import java.util.ArrayList; |
||||
|
import java.util.Collections; |
||||
|
import java.util.List; |
||||
|
|
||||
|
import org.apache.commons.lang3.RandomStringUtils; |
||||
|
import org.thingsboard.server.common.data.*; |
||||
|
import org.thingsboard.server.common.data.asset.Asset; |
||||
|
import org.thingsboard.server.common.data.asset.TenantAssetType; |
||||
|
import org.thingsboard.server.common.data.id.CustomerId; |
||||
|
import org.thingsboard.server.common.data.id.AssetId; |
||||
|
import org.thingsboard.server.common.data.page.TextPageData; |
||||
|
import org.thingsboard.server.common.data.page.TextPageLink; |
||||
|
import org.thingsboard.server.common.data.security.Authority; |
||||
|
import org.thingsboard.server.dao.model.ModelConstants; |
||||
|
import org.junit.After; |
||||
|
import org.junit.Assert; |
||||
|
import org.junit.Before; |
||||
|
import org.junit.Test; |
||||
|
|
||||
|
import com.datastax.driver.core.utils.UUIDs; |
||||
|
import com.fasterxml.jackson.core.type.TypeReference; |
||||
|
|
||||
|
public class AssetControllerTest extends AbstractControllerTest { |
||||
|
|
||||
|
private IdComparator<Asset> idComparator = new IdComparator<>(); |
||||
|
|
||||
|
private Tenant savedTenant; |
||||
|
private User tenantAdmin; |
||||
|
|
||||
|
@Before |
||||
|
public void beforeTest() throws Exception { |
||||
|
loginSysAdmin(); |
||||
|
|
||||
|
Tenant tenant = new Tenant(); |
||||
|
tenant.setTitle("My tenant"); |
||||
|
savedTenant = doPost("/api/tenant", tenant, Tenant.class); |
||||
|
Assert.assertNotNull(savedTenant); |
||||
|
|
||||
|
tenantAdmin = new User(); |
||||
|
tenantAdmin.setAuthority(Authority.TENANT_ADMIN); |
||||
|
tenantAdmin.setTenantId(savedTenant.getId()); |
||||
|
tenantAdmin.setEmail("tenant2@thingsboard.org"); |
||||
|
tenantAdmin.setFirstName("Joe"); |
||||
|
tenantAdmin.setLastName("Downs"); |
||||
|
|
||||
|
tenantAdmin = createUserAndLogin(tenantAdmin, "testPassword1"); |
||||
|
} |
||||
|
|
||||
|
@After |
||||
|
public void afterTest() throws Exception { |
||||
|
loginSysAdmin(); |
||||
|
|
||||
|
doDelete("/api/tenant/"+savedTenant.getId().getId().toString()) |
||||
|
.andExpect(status().isOk()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testSaveAsset() throws Exception { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setName("My asset"); |
||||
|
asset.setType("default"); |
||||
|
Asset savedAsset = doPost("/api/asset", asset, Asset.class); |
||||
|
|
||||
|
Assert.assertNotNull(savedAsset); |
||||
|
Assert.assertNotNull(savedAsset.getId()); |
||||
|
Assert.assertTrue(savedAsset.getCreatedTime() > 0); |
||||
|
Assert.assertEquals(savedTenant.getId(), savedAsset.getTenantId()); |
||||
|
Assert.assertNotNull(savedAsset.getCustomerId()); |
||||
|
Assert.assertEquals(NULL_UUID, savedAsset.getCustomerId().getId()); |
||||
|
Assert.assertEquals(asset.getName(), savedAsset.getName()); |
||||
|
|
||||
|
savedAsset.setName("My new asset"); |
||||
|
doPost("/api/asset", savedAsset, Asset.class); |
||||
|
|
||||
|
Asset foundAsset = doGet("/api/asset/" + savedAsset.getId().getId().toString(), Asset.class); |
||||
|
Assert.assertEquals(foundAsset.getName(), savedAsset.getName()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindAssetById() throws Exception { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setName("My asset"); |
||||
|
asset.setType("default"); |
||||
|
Asset savedAsset = doPost("/api/asset", asset, Asset.class); |
||||
|
Asset foundAsset = doGet("/api/asset/" + savedAsset.getId().getId().toString(), Asset.class); |
||||
|
Assert.assertNotNull(foundAsset); |
||||
|
Assert.assertEquals(savedAsset, foundAsset); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindAssetTypesByTenantId() throws Exception { |
||||
|
List<Asset> assets = new ArrayList<>(); |
||||
|
for (int i=0;i<3;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setName("My asset B"+i); |
||||
|
asset.setType("typeB"); |
||||
|
assets.add(doPost("/api/asset", asset, Asset.class)); |
||||
|
} |
||||
|
for (int i=0;i<7;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setName("My asset C"+i); |
||||
|
asset.setType("typeC"); |
||||
|
assets.add(doPost("/api/asset", asset, Asset.class)); |
||||
|
} |
||||
|
for (int i=0;i<9;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setName("My asset A"+i); |
||||
|
asset.setType("typeA"); |
||||
|
assets.add(doPost("/api/asset", asset, Asset.class)); |
||||
|
} |
||||
|
List<TenantAssetType> assetTypes = doGetTyped("/api/asset/types", |
||||
|
new TypeReference<List<TenantAssetType>>(){}); |
||||
|
|
||||
|
Assert.assertNotNull(assetTypes); |
||||
|
Assert.assertEquals(3, assetTypes.size()); |
||||
|
Assert.assertEquals("typeA", assetTypes.get(0).getType()); |
||||
|
Assert.assertEquals("typeB", assetTypes.get(1).getType()); |
||||
|
Assert.assertEquals("typeC", assetTypes.get(2).getType()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testDeleteAsset() throws Exception { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setName("My asset"); |
||||
|
asset.setType("default"); |
||||
|
Asset savedAsset = doPost("/api/asset", asset, Asset.class); |
||||
|
|
||||
|
doDelete("/api/asset/"+savedAsset.getId().getId().toString()) |
||||
|
.andExpect(status().isOk()); |
||||
|
|
||||
|
doGet("/api/asset/"+savedAsset.getId().getId().toString()) |
||||
|
.andExpect(status().isNotFound()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testSaveAssetWithEmptyType() throws Exception { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setName("My asset"); |
||||
|
doPost("/api/asset", asset) |
||||
|
.andExpect(status().isBadRequest()) |
||||
|
.andExpect(statusReason(containsString("Asset type should be specified"))); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testSaveAssetWithEmptyName() throws Exception { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setType("default"); |
||||
|
doPost("/api/asset", asset) |
||||
|
.andExpect(status().isBadRequest()) |
||||
|
.andExpect(statusReason(containsString("Asset name should be specified"))); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testAssignUnassignAssetToCustomer() throws Exception { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setName("My asset"); |
||||
|
asset.setType("default"); |
||||
|
Asset savedAsset = doPost("/api/asset", asset, Asset.class); |
||||
|
|
||||
|
Customer customer = new Customer(); |
||||
|
customer.setTitle("My customer"); |
||||
|
Customer savedCustomer = doPost("/api/customer", customer, Customer.class); |
||||
|
|
||||
|
Asset assignedAsset = doPost("/api/customer/" + savedCustomer.getId().getId().toString() |
||||
|
+ "/asset/" + savedAsset.getId().getId().toString(), Asset.class); |
||||
|
Assert.assertEquals(savedCustomer.getId(), assignedAsset.getCustomerId()); |
||||
|
|
||||
|
Asset foundAsset = doGet("/api/asset/" + savedAsset.getId().getId().toString(), Asset.class); |
||||
|
Assert.assertEquals(savedCustomer.getId(), foundAsset.getCustomerId()); |
||||
|
|
||||
|
Asset unassignedAsset = |
||||
|
doDelete("/api/customer/asset/" + savedAsset.getId().getId().toString(), Asset.class); |
||||
|
Assert.assertEquals(ModelConstants.NULL_UUID, unassignedAsset.getCustomerId().getId()); |
||||
|
|
||||
|
foundAsset = doGet("/api/asset/" + savedAsset.getId().getId().toString(), Asset.class); |
||||
|
Assert.assertEquals(ModelConstants.NULL_UUID, foundAsset.getCustomerId().getId()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testAssignAssetToNonExistentCustomer() throws Exception { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setName("My asset"); |
||||
|
asset.setType("default"); |
||||
|
Asset savedAsset = doPost("/api/asset", asset, Asset.class); |
||||
|
|
||||
|
doPost("/api/customer/" + UUIDs.timeBased().toString() |
||||
|
+ "/asset/" + savedAsset.getId().getId().toString()) |
||||
|
.andExpect(status().isNotFound()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testAssignAssetToCustomerFromDifferentTenant() throws Exception { |
||||
|
loginSysAdmin(); |
||||
|
|
||||
|
Tenant tenant2 = new Tenant(); |
||||
|
tenant2.setTitle("Different tenant"); |
||||
|
Tenant savedTenant2 = doPost("/api/tenant", tenant2, Tenant.class); |
||||
|
Assert.assertNotNull(savedTenant2); |
||||
|
|
||||
|
User tenantAdmin2 = new User(); |
||||
|
tenantAdmin2.setAuthority(Authority.TENANT_ADMIN); |
||||
|
tenantAdmin2.setTenantId(savedTenant2.getId()); |
||||
|
tenantAdmin2.setEmail("tenant3@thingsboard.org"); |
||||
|
tenantAdmin2.setFirstName("Joe"); |
||||
|
tenantAdmin2.setLastName("Downs"); |
||||
|
|
||||
|
tenantAdmin2 = createUserAndLogin(tenantAdmin2, "testPassword1"); |
||||
|
|
||||
|
Customer customer = new Customer(); |
||||
|
customer.setTitle("Different customer"); |
||||
|
Customer savedCustomer = doPost("/api/customer", customer, Customer.class); |
||||
|
|
||||
|
login(tenantAdmin.getEmail(), "testPassword1"); |
||||
|
|
||||
|
Asset asset = new Asset(); |
||||
|
asset.setName("My asset"); |
||||
|
asset.setType("default"); |
||||
|
Asset savedAsset = doPost("/api/asset", asset, Asset.class); |
||||
|
|
||||
|
doPost("/api/customer/" + savedCustomer.getId().getId().toString() |
||||
|
+ "/asset/" + savedAsset.getId().getId().toString()) |
||||
|
.andExpect(status().isForbidden()); |
||||
|
|
||||
|
loginSysAdmin(); |
||||
|
|
||||
|
doDelete("/api/tenant/"+savedTenant2.getId().getId().toString()) |
||||
|
.andExpect(status().isOk()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindTenantAssets() throws Exception { |
||||
|
List<Asset> assets = new ArrayList<>(); |
||||
|
for (int i=0;i<178;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setName("Asset"+i); |
||||
|
asset.setType("default"); |
||||
|
assets.add(doPost("/api/asset", asset, Asset.class)); |
||||
|
} |
||||
|
List<Asset> loadedAssets = new ArrayList<>(); |
||||
|
TextPageLink pageLink = new TextPageLink(23); |
||||
|
TextPageData<Asset> pageData = null; |
||||
|
do { |
||||
|
pageData = doGetTypedWithPageLink("/api/tenant/assets?", |
||||
|
new TypeReference<TextPageData<Asset>>(){}, pageLink); |
||||
|
loadedAssets.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assets, idComparator); |
||||
|
Collections.sort(loadedAssets, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assets, loadedAssets); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindTenantAssetsByName() throws Exception { |
||||
|
String title1 = "Asset title 1"; |
||||
|
List<Asset> assetsTitle1 = new ArrayList<>(); |
||||
|
for (int i=0;i<143;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
String suffix = RandomStringUtils.randomAlphanumeric(15); |
||||
|
String name = title1+suffix; |
||||
|
name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); |
||||
|
asset.setName(name); |
||||
|
asset.setType("default"); |
||||
|
assetsTitle1.add(doPost("/api/asset", asset, Asset.class)); |
||||
|
} |
||||
|
String title2 = "Asset title 2"; |
||||
|
List<Asset> assetsTitle2 = new ArrayList<>(); |
||||
|
for (int i=0;i<75;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
String suffix = RandomStringUtils.randomAlphanumeric(15); |
||||
|
String name = title2+suffix; |
||||
|
name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); |
||||
|
asset.setName(name); |
||||
|
asset.setType("default"); |
||||
|
assetsTitle2.add(doPost("/api/asset", asset, Asset.class)); |
||||
|
} |
||||
|
|
||||
|
List<Asset> loadedAssetsTitle1 = new ArrayList<>(); |
||||
|
TextPageLink pageLink = new TextPageLink(15, title1); |
||||
|
TextPageData<Asset> pageData = null; |
||||
|
do { |
||||
|
pageData = doGetTypedWithPageLink("/api/tenant/assets?", |
||||
|
new TypeReference<TextPageData<Asset>>(){}, pageLink); |
||||
|
loadedAssetsTitle1.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assetsTitle1, idComparator); |
||||
|
Collections.sort(loadedAssetsTitle1, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assetsTitle1, loadedAssetsTitle1); |
||||
|
|
||||
|
List<Asset> loadedAssetsTitle2 = new ArrayList<>(); |
||||
|
pageLink = new TextPageLink(4, title2); |
||||
|
do { |
||||
|
pageData = doGetTypedWithPageLink("/api/tenant/assets?", |
||||
|
new TypeReference<TextPageData<Asset>>(){}, pageLink); |
||||
|
loadedAssetsTitle2.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assetsTitle2, idComparator); |
||||
|
Collections.sort(loadedAssetsTitle2, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assetsTitle2, loadedAssetsTitle2); |
||||
|
|
||||
|
for (Asset asset : loadedAssetsTitle1) { |
||||
|
doDelete("/api/asset/"+asset.getId().getId().toString()) |
||||
|
.andExpect(status().isOk()); |
||||
|
} |
||||
|
|
||||
|
pageLink = new TextPageLink(4, title1); |
||||
|
pageData = doGetTypedWithPageLink("/api/tenant/assets?", |
||||
|
new TypeReference<TextPageData<Asset>>(){}, pageLink); |
||||
|
Assert.assertFalse(pageData.hasNext()); |
||||
|
Assert.assertEquals(0, pageData.getData().size()); |
||||
|
|
||||
|
for (Asset asset : loadedAssetsTitle2) { |
||||
|
doDelete("/api/asset/"+asset.getId().getId().toString()) |
||||
|
.andExpect(status().isOk()); |
||||
|
} |
||||
|
|
||||
|
pageLink = new TextPageLink(4, title2); |
||||
|
pageData = doGetTypedWithPageLink("/api/tenant/assets?", |
||||
|
new TypeReference<TextPageData<Asset>>(){}, pageLink); |
||||
|
Assert.assertFalse(pageData.hasNext()); |
||||
|
Assert.assertEquals(0, pageData.getData().size()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindTenantAssetsByType() throws Exception { |
||||
|
String title1 = "Asset title 1"; |
||||
|
String type1 = "typeA"; |
||||
|
List<Asset> assetsType1 = new ArrayList<>(); |
||||
|
for (int i=0;i<143;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
String suffix = RandomStringUtils.randomAlphanumeric(15); |
||||
|
String name = title1+suffix; |
||||
|
name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); |
||||
|
asset.setName(name); |
||||
|
asset.setType(type1); |
||||
|
assetsType1.add(doPost("/api/asset", asset, Asset.class)); |
||||
|
} |
||||
|
String title2 = "Asset title 2"; |
||||
|
String type2 = "typeB"; |
||||
|
List<Asset> assetsType2 = new ArrayList<>(); |
||||
|
for (int i=0;i<75;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
String suffix = RandomStringUtils.randomAlphanumeric(15); |
||||
|
String name = title2+suffix; |
||||
|
name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); |
||||
|
asset.setName(name); |
||||
|
asset.setType(type2); |
||||
|
assetsType2.add(doPost("/api/asset", asset, Asset.class)); |
||||
|
} |
||||
|
|
||||
|
List<Asset> loadedAssetsType1 = new ArrayList<>(); |
||||
|
TextPageLink pageLink = new TextPageLink(15); |
||||
|
TextPageData<Asset> pageData = null; |
||||
|
do { |
||||
|
pageData = doGetTypedWithPageLink("/api/tenant/assets?type={type}&", |
||||
|
new TypeReference<TextPageData<Asset>>(){}, pageLink, type1); |
||||
|
loadedAssetsType1.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assetsType1, idComparator); |
||||
|
Collections.sort(loadedAssetsType1, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assetsType1, loadedAssetsType1); |
||||
|
|
||||
|
List<Asset> loadedAssetsType2 = new ArrayList<>(); |
||||
|
pageLink = new TextPageLink(4); |
||||
|
do { |
||||
|
pageData = doGetTypedWithPageLink("/api/tenant/assets?type={type}&", |
||||
|
new TypeReference<TextPageData<Asset>>(){}, pageLink, type2); |
||||
|
loadedAssetsType2.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assetsType2, idComparator); |
||||
|
Collections.sort(loadedAssetsType2, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assetsType2, loadedAssetsType2); |
||||
|
|
||||
|
for (Asset asset : loadedAssetsType1) { |
||||
|
doDelete("/api/asset/"+asset.getId().getId().toString()) |
||||
|
.andExpect(status().isOk()); |
||||
|
} |
||||
|
|
||||
|
pageLink = new TextPageLink(4); |
||||
|
pageData = doGetTypedWithPageLink("/api/tenant/assets?type={type}&", |
||||
|
new TypeReference<TextPageData<Asset>>(){}, pageLink, type1); |
||||
|
Assert.assertFalse(pageData.hasNext()); |
||||
|
Assert.assertEquals(0, pageData.getData().size()); |
||||
|
|
||||
|
for (Asset asset : loadedAssetsType2) { |
||||
|
doDelete("/api/asset/"+asset.getId().getId().toString()) |
||||
|
.andExpect(status().isOk()); |
||||
|
} |
||||
|
|
||||
|
pageLink = new TextPageLink(4); |
||||
|
pageData = doGetTypedWithPageLink("/api/tenant/assets?type={type}&", |
||||
|
new TypeReference<TextPageData<Asset>>(){}, pageLink, type2); |
||||
|
Assert.assertFalse(pageData.hasNext()); |
||||
|
Assert.assertEquals(0, pageData.getData().size()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindCustomerAssets() throws Exception { |
||||
|
Customer customer = new Customer(); |
||||
|
customer.setTitle("Test customer"); |
||||
|
customer = doPost("/api/customer", customer, Customer.class); |
||||
|
CustomerId customerId = customer.getId(); |
||||
|
|
||||
|
List<Asset> assets = new ArrayList<>(); |
||||
|
for (int i=0;i<128;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setName("Asset"+i); |
||||
|
asset.setType("default"); |
||||
|
asset = doPost("/api/asset", asset, Asset.class); |
||||
|
assets.add(doPost("/api/customer/" + customerId.getId().toString() |
||||
|
+ "/asset/" + asset.getId().getId().toString(), Asset.class)); |
||||
|
} |
||||
|
|
||||
|
List<Asset> loadedAssets = new ArrayList<>(); |
||||
|
TextPageLink pageLink = new TextPageLink(23); |
||||
|
TextPageData<Asset> pageData = null; |
||||
|
do { |
||||
|
pageData = doGetTypedWithPageLink("/api/customer/" + customerId.getId().toString() + "/assets?", |
||||
|
new TypeReference<TextPageData<Asset>>(){}, pageLink); |
||||
|
loadedAssets.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assets, idComparator); |
||||
|
Collections.sort(loadedAssets, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assets, loadedAssets); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindCustomerAssetsByName() throws Exception { |
||||
|
Customer customer = new Customer(); |
||||
|
customer.setTitle("Test customer"); |
||||
|
customer = doPost("/api/customer", customer, Customer.class); |
||||
|
CustomerId customerId = customer.getId(); |
||||
|
|
||||
|
String title1 = "Asset title 1"; |
||||
|
List<Asset> assetsTitle1 = new ArrayList<>(); |
||||
|
for (int i=0;i<125;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
String suffix = RandomStringUtils.randomAlphanumeric(15); |
||||
|
String name = title1+suffix; |
||||
|
name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); |
||||
|
asset.setName(name); |
||||
|
asset.setType("default"); |
||||
|
asset = doPost("/api/asset", asset, Asset.class); |
||||
|
assetsTitle1.add(doPost("/api/customer/" + customerId.getId().toString() |
||||
|
+ "/asset/" + asset.getId().getId().toString(), Asset.class)); |
||||
|
} |
||||
|
String title2 = "Asset title 2"; |
||||
|
List<Asset> assetsTitle2 = new ArrayList<>(); |
||||
|
for (int i=0;i<143;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
String suffix = RandomStringUtils.randomAlphanumeric(15); |
||||
|
String name = title2+suffix; |
||||
|
name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); |
||||
|
asset.setName(name); |
||||
|
asset.setType("default"); |
||||
|
asset = doPost("/api/asset", asset, Asset.class); |
||||
|
assetsTitle2.add(doPost("/api/customer/" + customerId.getId().toString() |
||||
|
+ "/asset/" + asset.getId().getId().toString(), Asset.class)); |
||||
|
} |
||||
|
|
||||
|
List<Asset> loadedAssetsTitle1 = new ArrayList<>(); |
||||
|
TextPageLink pageLink = new TextPageLink(15, title1); |
||||
|
TextPageData<Asset> pageData = null; |
||||
|
do { |
||||
|
pageData = doGetTypedWithPageLink("/api/customer/" + customerId.getId().toString() + "/assets?", |
||||
|
new TypeReference<TextPageData<Asset>>(){}, pageLink); |
||||
|
loadedAssetsTitle1.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assetsTitle1, idComparator); |
||||
|
Collections.sort(loadedAssetsTitle1, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assetsTitle1, loadedAssetsTitle1); |
||||
|
|
||||
|
List<Asset> loadedAssetsTitle2 = new ArrayList<>(); |
||||
|
pageLink = new TextPageLink(4, title2); |
||||
|
do { |
||||
|
pageData = doGetTypedWithPageLink("/api/customer/" + customerId.getId().toString() + "/assets?", |
||||
|
new TypeReference<TextPageData<Asset>>(){}, pageLink); |
||||
|
loadedAssetsTitle2.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assetsTitle2, idComparator); |
||||
|
Collections.sort(loadedAssetsTitle2, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assetsTitle2, loadedAssetsTitle2); |
||||
|
|
||||
|
for (Asset asset : loadedAssetsTitle1) { |
||||
|
doDelete("/api/customer/asset/" + asset.getId().getId().toString()) |
||||
|
.andExpect(status().isOk()); |
||||
|
} |
||||
|
|
||||
|
pageLink = new TextPageLink(4, title1); |
||||
|
pageData = doGetTypedWithPageLink("/api/customer/" + customerId.getId().toString() + "/assets?", |
||||
|
new TypeReference<TextPageData<Asset>>(){}, pageLink); |
||||
|
Assert.assertFalse(pageData.hasNext()); |
||||
|
Assert.assertEquals(0, pageData.getData().size()); |
||||
|
|
||||
|
for (Asset asset : loadedAssetsTitle2) { |
||||
|
doDelete("/api/customer/asset/" + asset.getId().getId().toString()) |
||||
|
.andExpect(status().isOk()); |
||||
|
} |
||||
|
|
||||
|
pageLink = new TextPageLink(4, title2); |
||||
|
pageData = doGetTypedWithPageLink("/api/customer/" + customerId.getId().toString() + "/assets?", |
||||
|
new TypeReference<TextPageData<Asset>>(){}, pageLink); |
||||
|
Assert.assertFalse(pageData.hasNext()); |
||||
|
Assert.assertEquals(0, pageData.getData().size()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindCustomerAssetsByType() throws Exception { |
||||
|
Customer customer = new Customer(); |
||||
|
customer.setTitle("Test customer"); |
||||
|
customer = doPost("/api/customer", customer, Customer.class); |
||||
|
CustomerId customerId = customer.getId(); |
||||
|
|
||||
|
String title1 = "Asset title 1"; |
||||
|
String type1 = "typeC"; |
||||
|
List<Asset> assetsType1 = new ArrayList<>(); |
||||
|
for (int i=0;i<125;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
String suffix = RandomStringUtils.randomAlphanumeric(15); |
||||
|
String name = title1+suffix; |
||||
|
name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); |
||||
|
asset.setName(name); |
||||
|
asset.setType(type1); |
||||
|
asset = doPost("/api/asset", asset, Asset.class); |
||||
|
assetsType1.add(doPost("/api/customer/" + customerId.getId().toString() |
||||
|
+ "/asset/" + asset.getId().getId().toString(), Asset.class)); |
||||
|
} |
||||
|
String title2 = "Asset title 2"; |
||||
|
String type2 = "typeD"; |
||||
|
List<Asset> assetsType2 = new ArrayList<>(); |
||||
|
for (int i=0;i<143;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
String suffix = RandomStringUtils.randomAlphanumeric(15); |
||||
|
String name = title2+suffix; |
||||
|
name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); |
||||
|
asset.setName(name); |
||||
|
asset.setType(type2); |
||||
|
asset = doPost("/api/asset", asset, Asset.class); |
||||
|
assetsType2.add(doPost("/api/customer/" + customerId.getId().toString() |
||||
|
+ "/asset/" + asset.getId().getId().toString(), Asset.class)); |
||||
|
} |
||||
|
|
||||
|
List<Asset> loadedAssetsType1 = new ArrayList<>(); |
||||
|
TextPageLink pageLink = new TextPageLink(15); |
||||
|
TextPageData<Asset> pageData = null; |
||||
|
do { |
||||
|
pageData = doGetTypedWithPageLink("/api/customer/" + customerId.getId().toString() + "/assets?type={type}&", |
||||
|
new TypeReference<TextPageData<Asset>>(){}, pageLink, type1); |
||||
|
loadedAssetsType1.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assetsType1, idComparator); |
||||
|
Collections.sort(loadedAssetsType1, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assetsType1, loadedAssetsType1); |
||||
|
|
||||
|
List<Asset> loadedAssetsType2 = new ArrayList<>(); |
||||
|
pageLink = new TextPageLink(4); |
||||
|
do { |
||||
|
pageData = doGetTypedWithPageLink("/api/customer/" + customerId.getId().toString() + "/assets?type={type}&", |
||||
|
new TypeReference<TextPageData<Asset>>(){}, pageLink, type2); |
||||
|
loadedAssetsType2.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assetsType2, idComparator); |
||||
|
Collections.sort(loadedAssetsType2, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assetsType2, loadedAssetsType2); |
||||
|
|
||||
|
for (Asset asset : loadedAssetsType1) { |
||||
|
doDelete("/api/customer/asset/" + asset.getId().getId().toString()) |
||||
|
.andExpect(status().isOk()); |
||||
|
} |
||||
|
|
||||
|
pageLink = new TextPageLink(4); |
||||
|
pageData = doGetTypedWithPageLink("/api/customer/" + customerId.getId().toString() + "/assets?type={type}&", |
||||
|
new TypeReference<TextPageData<Asset>>(){}, pageLink, type1); |
||||
|
Assert.assertFalse(pageData.hasNext()); |
||||
|
Assert.assertEquals(0, pageData.getData().size()); |
||||
|
|
||||
|
for (Asset asset : loadedAssetsType2) { |
||||
|
doDelete("/api/customer/asset/" + asset.getId().getId().toString()) |
||||
|
.andExpect(status().isOk()); |
||||
|
} |
||||
|
|
||||
|
pageLink = new TextPageLink(4); |
||||
|
pageData = doGetTypedWithPageLink("/api/customer/" + customerId.getId().toString() + "/assets?type={type}&", |
||||
|
new TypeReference<TextPageData<Asset>>(){}, pageLink, type2); |
||||
|
Assert.assertFalse(pageData.hasNext()); |
||||
|
Assert.assertEquals(0, pageData.getData().size()); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,79 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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; |
||||
|
|
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
|
||||
|
public class TenantDeviceType { |
||||
|
|
||||
|
private static final long serialVersionUID = 8057240243859922101L; |
||||
|
|
||||
|
private String type; |
||||
|
private TenantId tenantId; |
||||
|
|
||||
|
public TenantDeviceType() { |
||||
|
super(); |
||||
|
} |
||||
|
|
||||
|
public TenantDeviceType(String type, TenantId tenantId) { |
||||
|
this.type = type; |
||||
|
this.tenantId = tenantId; |
||||
|
} |
||||
|
|
||||
|
public String getType() { |
||||
|
return type; |
||||
|
} |
||||
|
|
||||
|
public void setType(String type) { |
||||
|
this.type = type; |
||||
|
} |
||||
|
|
||||
|
public TenantId getTenantId() { |
||||
|
return tenantId; |
||||
|
} |
||||
|
|
||||
|
public void setTenantId(TenantId tenantId) { |
||||
|
this.tenantId = tenantId; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public int hashCode() { |
||||
|
int result = type != null ? type.hashCode() : 0; |
||||
|
result = 31 * result + (tenantId != null ? tenantId.hashCode() : 0); |
||||
|
return result; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public boolean equals(Object o) { |
||||
|
if (this == o) return true; |
||||
|
if (o == null || getClass() != o.getClass()) return false; |
||||
|
|
||||
|
TenantDeviceType that = (TenantDeviceType) o; |
||||
|
|
||||
|
if (type != null ? !type.equals(that.type) : that.type != null) return false; |
||||
|
return tenantId != null ? tenantId.equals(that.tenantId) : that.tenantId == null; |
||||
|
|
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public String toString() { |
||||
|
final StringBuilder sb = new StringBuilder("TenantDeviceType{"); |
||||
|
sb.append("type='").append(type).append('\''); |
||||
|
sb.append(", tenantId=").append(tenantId); |
||||
|
sb.append('}'); |
||||
|
return sb.toString(); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,79 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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.asset; |
||||
|
|
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
|
||||
|
public class TenantAssetType { |
||||
|
|
||||
|
private static final long serialVersionUID = 8057290243855622101L; |
||||
|
|
||||
|
private String type; |
||||
|
private TenantId tenantId; |
||||
|
|
||||
|
public TenantAssetType() { |
||||
|
super(); |
||||
|
} |
||||
|
|
||||
|
public TenantAssetType(String type, TenantId tenantId) { |
||||
|
this.type = type; |
||||
|
this.tenantId = tenantId; |
||||
|
} |
||||
|
|
||||
|
public String getType() { |
||||
|
return type; |
||||
|
} |
||||
|
|
||||
|
public void setType(String type) { |
||||
|
this.type = type; |
||||
|
} |
||||
|
|
||||
|
public TenantId getTenantId() { |
||||
|
return tenantId; |
||||
|
} |
||||
|
|
||||
|
public void setTenantId(TenantId tenantId) { |
||||
|
this.tenantId = tenantId; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public int hashCode() { |
||||
|
int result = type != null ? type.hashCode() : 0; |
||||
|
result = 31 * result + (tenantId != null ? tenantId.hashCode() : 0); |
||||
|
return result; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public boolean equals(Object o) { |
||||
|
if (this == o) return true; |
||||
|
if (o == null || getClass() != o.getClass()) return false; |
||||
|
|
||||
|
TenantAssetType that = (TenantAssetType) o; |
||||
|
|
||||
|
if (type != null ? !type.equals(that.type) : that.type != null) return false; |
||||
|
return tenantId != null ? tenantId.equals(that.tenantId) : that.tenantId == null; |
||||
|
|
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public String toString() { |
||||
|
final StringBuilder sb = new StringBuilder("TenantAssetType{"); |
||||
|
sb.append("type='").append(type).append('\''); |
||||
|
sb.append(", tenantId=").append(tenantId); |
||||
|
sb.append('}'); |
||||
|
return sb.toString(); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,68 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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.relation; |
||||
|
|
||||
|
public class EntityRelationInfo extends EntityRelation { |
||||
|
|
||||
|
private static final long serialVersionUID = 2807343097519543363L; |
||||
|
|
||||
|
private String fromName; |
||||
|
private String toName; |
||||
|
|
||||
|
public EntityRelationInfo() { |
||||
|
super(); |
||||
|
} |
||||
|
|
||||
|
public EntityRelationInfo(EntityRelation entityRelation) { |
||||
|
super(entityRelation); |
||||
|
} |
||||
|
|
||||
|
public String getFromName() { |
||||
|
return fromName; |
||||
|
} |
||||
|
|
||||
|
public void setFromName(String fromName) { |
||||
|
this.fromName = fromName; |
||||
|
} |
||||
|
|
||||
|
public String getToName() { |
||||
|
return toName; |
||||
|
} |
||||
|
|
||||
|
public void setToName(String toName) { |
||||
|
this.toName = toName; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public boolean equals(Object o) { |
||||
|
if (this == o) return true; |
||||
|
if (o == null || getClass() != o.getClass()) return false; |
||||
|
if (!super.equals(o)) return false; |
||||
|
|
||||
|
EntityRelationInfo that = (EntityRelationInfo) o; |
||||
|
|
||||
|
return toName != null ? toName.equals(that.toName) : that.toName == null; |
||||
|
|
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public int hashCode() { |
||||
|
int result = super.hashCode(); |
||||
|
result = 31 * result + (toName != null ? toName.hashCode() : 0); |
||||
|
return result; |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,23 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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.relation; |
||||
|
|
||||
|
public enum RelationTypeGroup { |
||||
|
|
||||
|
COMMON, |
||||
|
ALARM |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,110 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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.alarm; |
||||
|
|
||||
|
import com.datastax.driver.core.querybuilder.QueryBuilder; |
||||
|
import com.datastax.driver.core.querybuilder.Select; |
||||
|
import com.google.common.util.concurrent.AsyncFunction; |
||||
|
import com.google.common.util.concurrent.Futures; |
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||
|
import org.springframework.stereotype.Component; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
import org.thingsboard.server.common.data.alarm.Alarm; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmQuery; |
||||
|
import org.thingsboard.server.common.data.id.EntityId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.relation.EntityRelation; |
||||
|
import org.thingsboard.server.common.data.relation.RelationTypeGroup; |
||||
|
import org.thingsboard.server.dao.AbstractModelDao; |
||||
|
import org.thingsboard.server.dao.AbstractSearchTimeDao; |
||||
|
import org.thingsboard.server.dao.model.AlarmEntity; |
||||
|
import org.thingsboard.server.dao.model.ModelConstants; |
||||
|
import org.thingsboard.server.dao.relation.RelationDao; |
||||
|
|
||||
|
import java.util.ArrayList; |
||||
|
import java.util.List; |
||||
|
import java.util.UUID; |
||||
|
|
||||
|
import static com.datastax.driver.core.querybuilder.QueryBuilder.eq; |
||||
|
import static com.datastax.driver.core.querybuilder.QueryBuilder.select; |
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.*; |
||||
|
|
||||
|
@Component |
||||
|
@Slf4j |
||||
|
public class AlarmDaoImpl extends AbstractModelDao<AlarmEntity> implements AlarmDao { |
||||
|
|
||||
|
@Autowired |
||||
|
private RelationDao relationDao; |
||||
|
|
||||
|
@Override |
||||
|
protected Class<AlarmEntity> getColumnFamilyClass() { |
||||
|
return AlarmEntity.class; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
protected String getColumnFamilyName() { |
||||
|
return ALARM_COLUMN_FAMILY_NAME; |
||||
|
} |
||||
|
|
||||
|
protected boolean isDeleteOnSave() { |
||||
|
return false; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public AlarmEntity save(Alarm alarm) { |
||||
|
log.debug("Save asset [{}] ", alarm); |
||||
|
return save(new AlarmEntity(alarm)); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public ListenableFuture<Alarm> findLatestByOriginatorAndType(TenantId tenantId, EntityId originator, String type) { |
||||
|
Select select = select().from(ALARM_COLUMN_FAMILY_NAME); |
||||
|
Select.Where query = select.where(); |
||||
|
query.and(eq(ALARM_TENANT_ID_PROPERTY, tenantId.getId())); |
||||
|
query.and(eq(ALARM_ORIGINATOR_ID_PROPERTY, originator.getId())); |
||||
|
query.and(eq(ALARM_ORIGINATOR_TYPE_PROPERTY, originator.getEntityType())); |
||||
|
query.and(eq(ALARM_TYPE_PROPERTY, type)); |
||||
|
query.limit(1); |
||||
|
query.orderBy(QueryBuilder.asc(ModelConstants.ALARM_TYPE_PROPERTY), QueryBuilder.desc(ModelConstants.ID_PROPERTY)); |
||||
|
return Futures.transform(findOneByStatementAsync(query), toDataFunction()); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public ListenableFuture<Alarm> findAlarmByIdAsync(UUID key) { |
||||
|
log.debug("Get alarm by id {}", key); |
||||
|
Select.Where query = select().from(ALARM_BY_ID_VIEW_NAME).where(eq(ModelConstants.ID_PROPERTY, key)); |
||||
|
query.limit(1); |
||||
|
log.trace("Execute query {}", query); |
||||
|
return Futures.transform(findOneByStatementAsync(query), toDataFunction()); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public ListenableFuture<List<Alarm>> findAlarms(AlarmQuery query) { |
||||
|
log.trace("Try to find alarms by entity [{}], status [{}] and pageLink [{}]", query.getAffectedEntityId(), query.getStatus(), query.getPageLink()); |
||||
|
EntityId affectedEntity = query.getAffectedEntityId(); |
||||
|
String relationType = query.getStatus() == null ? BaseAlarmService.ALARM_RELATION : BaseAlarmService.ALARM_RELATION_PREFIX + query.getStatus().name(); |
||||
|
ListenableFuture<List<EntityRelation>> relations = relationDao.findRelations(affectedEntity, relationType, RelationTypeGroup.ALARM, EntityType.ALARM, query.getPageLink()); |
||||
|
return Futures.transform(relations, (AsyncFunction<List<EntityRelation>, List<Alarm>>) input -> { |
||||
|
List<ListenableFuture<Alarm>> alarmFutures = new ArrayList<>(input.size()); |
||||
|
for (EntityRelation relation : input) { |
||||
|
alarmFutures.add(findAlarmByIdAsync(relation.getTo().getId())); |
||||
|
} |
||||
|
return Futures.successfulAsList(alarmFutures); |
||||
|
}); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,292 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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.alarm; |
||||
|
|
||||
|
|
||||
|
import com.google.common.base.Function; |
||||
|
import com.google.common.util.concurrent.Futures; |
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||
|
import org.springframework.stereotype.Service; |
||||
|
import org.springframework.util.StringUtils; |
||||
|
import org.thingsboard.server.common.data.alarm.Alarm; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmId; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmQuery; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmStatus; |
||||
|
import org.thingsboard.server.common.data.id.EntityId; |
||||
|
import org.thingsboard.server.common.data.page.TimePageData; |
||||
|
import org.thingsboard.server.common.data.relation.EntityRelation; |
||||
|
import org.thingsboard.server.common.data.relation.RelationTypeGroup; |
||||
|
import org.thingsboard.server.dao.entity.AbstractEntityService; |
||||
|
import org.thingsboard.server.dao.entity.BaseEntityService; |
||||
|
import org.thingsboard.server.dao.exception.DataValidationException; |
||||
|
import org.thingsboard.server.dao.model.*; |
||||
|
import org.thingsboard.server.dao.relation.EntityRelationsQuery; |
||||
|
import org.thingsboard.server.dao.relation.EntitySearchDirection; |
||||
|
import org.thingsboard.server.dao.relation.RelationService; |
||||
|
import org.thingsboard.server.dao.relation.RelationsSearchParameters; |
||||
|
import org.thingsboard.server.dao.service.DataValidator; |
||||
|
import org.thingsboard.server.dao.tenant.TenantDao; |
||||
|
|
||||
|
import javax.annotation.Nullable; |
||||
|
import javax.annotation.PostConstruct; |
||||
|
import javax.annotation.PreDestroy; |
||||
|
import java.util.List; |
||||
|
import java.util.concurrent.ExecutionException; |
||||
|
import java.util.concurrent.ExecutorService; |
||||
|
import java.util.concurrent.Executors; |
||||
|
import java.util.stream.Collectors; |
||||
|
|
||||
|
import static org.thingsboard.server.dao.DaoUtil.*; |
||||
|
import static org.thingsboard.server.dao.service.Validator.*; |
||||
|
|
||||
|
@Service |
||||
|
@Slf4j |
||||
|
public class BaseAlarmService extends AbstractEntityService implements AlarmService { |
||||
|
|
||||
|
public static final String ALARM_RELATION_PREFIX = "ALARM_"; |
||||
|
public static final String ALARM_RELATION = "ALARM_ANY"; |
||||
|
|
||||
|
@Autowired |
||||
|
private AlarmDao alarmDao; |
||||
|
|
||||
|
@Autowired |
||||
|
private TenantDao tenantDao; |
||||
|
|
||||
|
@Autowired |
||||
|
private RelationService relationService; |
||||
|
|
||||
|
protected ExecutorService readResultsProcessingExecutor; |
||||
|
|
||||
|
@PostConstruct |
||||
|
public void startExecutor() { |
||||
|
readResultsProcessingExecutor = Executors.newCachedThreadPool(); |
||||
|
} |
||||
|
|
||||
|
@PreDestroy |
||||
|
public void stopExecutor() { |
||||
|
if (readResultsProcessingExecutor != null) { |
||||
|
readResultsProcessingExecutor.shutdownNow(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public Alarm createOrUpdateAlarm(Alarm alarm) { |
||||
|
alarmDataValidator.validate(alarm); |
||||
|
try { |
||||
|
if (alarm.getStartTs() == 0L) { |
||||
|
alarm.setStartTs(System.currentTimeMillis()); |
||||
|
} |
||||
|
if (alarm.getEndTs() == 0L) { |
||||
|
alarm.setEndTs(alarm.getStartTs()); |
||||
|
} |
||||
|
if (alarm.getId() == null) { |
||||
|
Alarm existing = alarmDao.findLatestByOriginatorAndType(alarm.getTenantId(), alarm.getOriginator(), alarm.getType()).get(); |
||||
|
if (existing == null || existing.getStatus().isCleared()) { |
||||
|
return createAlarm(alarm); |
||||
|
} else { |
||||
|
return updateAlarm(existing, alarm); |
||||
|
} |
||||
|
} else { |
||||
|
return updateAlarm(alarm).get(); |
||||
|
} |
||||
|
} catch (ExecutionException | InterruptedException e) { |
||||
|
throw new RuntimeException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private Alarm createAlarm(Alarm alarm) throws InterruptedException, ExecutionException { |
||||
|
log.debug("New Alarm : {}", alarm); |
||||
|
Alarm saved = getData(alarmDao.save(new AlarmEntity(alarm))); |
||||
|
EntityRelationsQuery query = new EntityRelationsQuery(); |
||||
|
query.setParameters(new RelationsSearchParameters(saved.getOriginator(), EntitySearchDirection.TO, Integer.MAX_VALUE)); |
||||
|
List<EntityId> parentEntities = relationService.findByQuery(query).get().stream().map(r -> r.getFrom()).collect(Collectors.toList()); |
||||
|
for (EntityId parentId : parentEntities) { |
||||
|
createRelation(new EntityRelation(parentId, saved.getId(), ALARM_RELATION, RelationTypeGroup.ALARM)); |
||||
|
createRelation(new EntityRelation(parentId, saved.getId(), ALARM_RELATION_PREFIX + saved.getStatus().name(), RelationTypeGroup.ALARM)); |
||||
|
} |
||||
|
createRelation(new EntityRelation(alarm.getOriginator(), saved.getId(), ALARM_RELATION, RelationTypeGroup.ALARM)); |
||||
|
createRelation(new EntityRelation(alarm.getOriginator(), saved.getId(), ALARM_RELATION_PREFIX + saved.getStatus().name(), RelationTypeGroup.ALARM)); |
||||
|
return saved; |
||||
|
} |
||||
|
|
||||
|
protected ListenableFuture<Alarm> updateAlarm(Alarm update) { |
||||
|
alarmDataValidator.validate(update); |
||||
|
return getAndUpdate(update.getId(), new Function<Alarm, Alarm>() { |
||||
|
@Nullable |
||||
|
@Override |
||||
|
public Alarm apply(@Nullable Alarm alarm) { |
||||
|
if (alarm == null) { |
||||
|
return null; |
||||
|
} else { |
||||
|
return updateAlarm(alarm, update); |
||||
|
} |
||||
|
} |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
private Alarm updateAlarm(Alarm oldAlarm, Alarm newAlarm) { |
||||
|
AlarmStatus oldStatus = oldAlarm.getStatus(); |
||||
|
AlarmStatus newStatus = newAlarm.getStatus(); |
||||
|
AlarmEntity result = alarmDao.save(new AlarmEntity(merge(oldAlarm, newAlarm))); |
||||
|
if (oldStatus != newStatus) { |
||||
|
updateRelations(oldAlarm, oldStatus, newStatus); |
||||
|
} |
||||
|
return result.toData(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public ListenableFuture<Boolean> ackAlarm(AlarmId alarmId, long ackTime) { |
||||
|
return getAndUpdate(alarmId, new Function<Alarm, Boolean>() { |
||||
|
@Nullable |
||||
|
@Override |
||||
|
public Boolean apply(@Nullable Alarm alarm) { |
||||
|
if (alarm == null || alarm.getStatus().isAck()) { |
||||
|
return false; |
||||
|
} else { |
||||
|
AlarmStatus oldStatus = alarm.getStatus(); |
||||
|
AlarmStatus newStatus = oldStatus.isCleared() ? AlarmStatus.CLEARED_ACK : AlarmStatus.ACTIVE_ACK; |
||||
|
alarm.setStatus(newStatus); |
||||
|
alarm.setAckTs(ackTime); |
||||
|
alarmDao.save(new AlarmEntity(alarm)); |
||||
|
updateRelations(alarm, oldStatus, newStatus); |
||||
|
return true; |
||||
|
} |
||||
|
} |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public ListenableFuture<Boolean> clearAlarm(AlarmId alarmId, long clearTime) { |
||||
|
return getAndUpdate(alarmId, new Function<Alarm, Boolean>() { |
||||
|
@Nullable |
||||
|
@Override |
||||
|
public Boolean apply(@Nullable Alarm alarm) { |
||||
|
if (alarm == null || alarm.getStatus().isCleared()) { |
||||
|
return false; |
||||
|
} else { |
||||
|
AlarmStatus oldStatus = alarm.getStatus(); |
||||
|
AlarmStatus newStatus = oldStatus.isAck() ? AlarmStatus.CLEARED_ACK : AlarmStatus.CLEARED_UNACK; |
||||
|
alarm.setStatus(newStatus); |
||||
|
alarm.setClearTs(clearTime); |
||||
|
alarmDao.save(new AlarmEntity(alarm)); |
||||
|
updateRelations(alarm, oldStatus, newStatus); |
||||
|
return true; |
||||
|
} |
||||
|
} |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public ListenableFuture<Alarm> findAlarmByIdAsync(AlarmId alarmId) { |
||||
|
log.trace("Executing findAlarmById [{}]", alarmId); |
||||
|
validateId(alarmId, "Incorrect alarmId " + alarmId); |
||||
|
return alarmDao.findAlarmByIdAsync(alarmId.getId()); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public ListenableFuture<TimePageData<Alarm>> findAlarms(AlarmQuery query) { |
||||
|
ListenableFuture<List<Alarm>> alarms = alarmDao.findAlarms(query); |
||||
|
return Futures.transform(alarms, new Function<List<Alarm>, TimePageData<Alarm>>() { |
||||
|
@Nullable |
||||
|
@Override |
||||
|
public TimePageData<Alarm> apply(@Nullable List<Alarm> alarms) { |
||||
|
return new TimePageData<>(alarms, query.getPageLink()); |
||||
|
} |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
private void deleteRelation(EntityRelation alarmRelation) throws ExecutionException, InterruptedException { |
||||
|
log.debug("Deleting Alarm relation: {}", alarmRelation); |
||||
|
relationService.deleteRelation(alarmRelation).get(); |
||||
|
} |
||||
|
|
||||
|
private void createRelation(EntityRelation alarmRelation) throws ExecutionException, InterruptedException { |
||||
|
log.debug("Creating Alarm relation: {}", alarmRelation); |
||||
|
relationService.saveRelation(alarmRelation).get(); |
||||
|
} |
||||
|
|
||||
|
private Alarm merge(Alarm existing, Alarm alarm) { |
||||
|
if (alarm.getStartTs() > existing.getEndTs()) { |
||||
|
existing.setEndTs(alarm.getStartTs()); |
||||
|
} |
||||
|
if (alarm.getEndTs() > existing.getEndTs()) { |
||||
|
existing.setEndTs(alarm.getEndTs()); |
||||
|
} |
||||
|
if (alarm.getClearTs() > existing.getClearTs()) { |
||||
|
existing.setClearTs(alarm.getClearTs()); |
||||
|
} |
||||
|
if (alarm.getAckTs() > existing.getAckTs()) { |
||||
|
existing.setAckTs(alarm.getAckTs()); |
||||
|
} |
||||
|
existing.setStatus(alarm.getStatus()); |
||||
|
existing.setSeverity(alarm.getSeverity()); |
||||
|
existing.setDetails(alarm.getDetails()); |
||||
|
return existing; |
||||
|
} |
||||
|
|
||||
|
private void updateRelations(Alarm alarm, AlarmStatus oldStatus, AlarmStatus newStatus) { |
||||
|
try { |
||||
|
EntityRelationsQuery query = new EntityRelationsQuery(); |
||||
|
query.setParameters(new RelationsSearchParameters(alarm.getOriginator(), EntitySearchDirection.TO, Integer.MAX_VALUE)); |
||||
|
List<EntityId> parentEntities = relationService.findByQuery(query).get().stream().map(r -> r.getFrom()).collect(Collectors.toList()); |
||||
|
for (EntityId parentId : parentEntities) { |
||||
|
deleteRelation(new EntityRelation(parentId, alarm.getId(), ALARM_RELATION_PREFIX + oldStatus.name(), RelationTypeGroup.ALARM)); |
||||
|
createRelation(new EntityRelation(parentId, alarm.getId(), ALARM_RELATION_PREFIX + newStatus.name(), RelationTypeGroup.ALARM)); |
||||
|
} |
||||
|
deleteRelation(new EntityRelation(alarm.getOriginator(), alarm.getId(), ALARM_RELATION_PREFIX + oldStatus.name(), RelationTypeGroup.ALARM)); |
||||
|
createRelation(new EntityRelation(alarm.getOriginator(), alarm.getId(), ALARM_RELATION_PREFIX + newStatus.name(), RelationTypeGroup.ALARM)); |
||||
|
} catch (ExecutionException | InterruptedException e) { |
||||
|
log.warn("[{}] Failed to update relations. Old status: [{}], New status: [{}]", alarm.getId(), oldStatus, newStatus); |
||||
|
throw new RuntimeException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private <T> ListenableFuture<T> getAndUpdate(AlarmId alarmId, Function<Alarm, T> function) { |
||||
|
validateId(alarmId, "Alarm id should be specified!"); |
||||
|
ListenableFuture<Alarm> entity = alarmDao.findAlarmByIdAsync(alarmId.getId()); |
||||
|
return Futures.transform(entity, function, readResultsProcessingExecutor); |
||||
|
} |
||||
|
|
||||
|
private DataValidator<Alarm> alarmDataValidator = |
||||
|
new DataValidator<Alarm>() { |
||||
|
|
||||
|
@Override |
||||
|
protected void validateDataImpl(Alarm alarm) { |
||||
|
if (StringUtils.isEmpty(alarm.getType())) { |
||||
|
throw new DataValidationException("Alarm type should be specified!"); |
||||
|
} |
||||
|
if (alarm.getOriginator() == null) { |
||||
|
throw new DataValidationException("Alarm originator should be specified!"); |
||||
|
} |
||||
|
if (alarm.getSeverity() == null) { |
||||
|
throw new DataValidationException("Alarm severity should be specified!"); |
||||
|
} |
||||
|
if (alarm.getStatus() == null) { |
||||
|
throw new DataValidationException("Alarm status should be specified!"); |
||||
|
} |
||||
|
if (alarm.getTenantId() == null) { |
||||
|
throw new DataValidationException("Alarm should be assigned to tenant!"); |
||||
|
} else { |
||||
|
TenantEntity tenant = tenantDao.findById(alarm.getTenantId().getId()); |
||||
|
if (tenant == null) { |
||||
|
throw new DataValidationException("Alarm is referencing to non-existent tenant!"); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
}; |
||||
|
} |
||||
@ -0,0 +1,28 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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.entity; |
||||
|
|
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
import org.thingsboard.server.common.data.id.EntityId; |
||||
|
|
||||
|
public interface EntityService { |
||||
|
|
||||
|
ListenableFuture<String> fetchEntityNameAsync(EntityId entityId); |
||||
|
|
||||
|
void deleteEntityRelations(EntityId entityId); |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,236 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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; |
||||
|
|
||||
|
import com.datastax.driver.core.utils.UUIDs; |
||||
|
import com.datastax.driver.mapping.annotations.*; |
||||
|
import com.fasterxml.jackson.databind.JsonNode; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
import org.thingsboard.server.common.data.alarm.Alarm; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmId; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmSeverity; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmStatus; |
||||
|
import org.thingsboard.server.common.data.id.EntityIdFactory; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.dao.model.type.AlarmSeverityCodec; |
||||
|
import org.thingsboard.server.dao.model.type.AlarmStatusCodec; |
||||
|
import org.thingsboard.server.dao.model.type.EntityTypeCodec; |
||||
|
import org.thingsboard.server.dao.model.type.JsonCodec; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.*; |
||||
|
|
||||
|
@Table(name = ALARM_COLUMN_FAMILY_NAME) |
||||
|
public final class AlarmEntity implements BaseEntity<Alarm> { |
||||
|
|
||||
|
@Transient |
||||
|
private static final long serialVersionUID = -1265181166886910152L; |
||||
|
|
||||
|
@ClusteringColumn(value = 1) |
||||
|
@Column(name = ID_PROPERTY) |
||||
|
private UUID id; |
||||
|
|
||||
|
@PartitionKey(value = 0) |
||||
|
@Column(name = ALARM_TENANT_ID_PROPERTY) |
||||
|
private UUID tenantId; |
||||
|
|
||||
|
@PartitionKey(value = 1) |
||||
|
@Column(name = ALARM_ORIGINATOR_ID_PROPERTY) |
||||
|
private UUID originatorId; |
||||
|
|
||||
|
@PartitionKey(value = 2) |
||||
|
@Column(name = ALARM_ORIGINATOR_TYPE_PROPERTY, codec = EntityTypeCodec.class) |
||||
|
private EntityType originatorType; |
||||
|
|
||||
|
@ClusteringColumn(value = 0) |
||||
|
@Column(name = ALARM_TYPE_PROPERTY) |
||||
|
private String type; |
||||
|
|
||||
|
@Column(name = ALARM_SEVERITY_PROPERTY, codec = AlarmSeverityCodec.class) |
||||
|
private AlarmSeverity severity; |
||||
|
|
||||
|
@Column(name = ALARM_STATUS_PROPERTY, codec = AlarmStatusCodec.class) |
||||
|
private AlarmStatus status; |
||||
|
|
||||
|
@Column(name = ALARM_START_TS_PROPERTY) |
||||
|
private Long startTs; |
||||
|
|
||||
|
@Column(name = ALARM_END_TS_PROPERTY) |
||||
|
private Long endTs; |
||||
|
|
||||
|
@Column(name = ALARM_ACK_TS_PROPERTY) |
||||
|
private Long ackTs; |
||||
|
|
||||
|
@Column(name = ALARM_CLEAR_TS_PROPERTY) |
||||
|
private Long clearTs; |
||||
|
|
||||
|
@Column(name = ALARM_DETAILS_PROPERTY, codec = JsonCodec.class) |
||||
|
private JsonNode details; |
||||
|
|
||||
|
@Column(name = ALARM_PROPAGATE_PROPERTY) |
||||
|
private Boolean propagate; |
||||
|
|
||||
|
public AlarmEntity() { |
||||
|
super(); |
||||
|
} |
||||
|
|
||||
|
public AlarmEntity(Alarm alarm) { |
||||
|
if (alarm.getId() != null) { |
||||
|
this.id = alarm.getId().getId(); |
||||
|
} |
||||
|
if (alarm.getTenantId() != null) { |
||||
|
this.tenantId = alarm.getTenantId().getId(); |
||||
|
} |
||||
|
this.type = alarm.getType(); |
||||
|
this.originatorId = alarm.getOriginator().getId(); |
||||
|
this.originatorType = alarm.getOriginator().getEntityType(); |
||||
|
this.type = alarm.getType(); |
||||
|
this.severity = alarm.getSeverity(); |
||||
|
this.status = alarm.getStatus(); |
||||
|
this.propagate = alarm.isPropagate(); |
||||
|
this.startTs = alarm.getStartTs(); |
||||
|
this.endTs = alarm.getEndTs(); |
||||
|
this.ackTs = alarm.getAckTs(); |
||||
|
this.clearTs = alarm.getClearTs(); |
||||
|
this.details = alarm.getDetails(); |
||||
|
} |
||||
|
|
||||
|
public UUID getId() { |
||||
|
return id; |
||||
|
} |
||||
|
|
||||
|
public void setId(UUID id) { |
||||
|
this.id = id; |
||||
|
} |
||||
|
|
||||
|
public UUID getTenantId() { |
||||
|
return tenantId; |
||||
|
} |
||||
|
|
||||
|
public void setTenantId(UUID tenantId) { |
||||
|
this.tenantId = tenantId; |
||||
|
} |
||||
|
|
||||
|
public UUID getOriginatorId() { |
||||
|
return originatorId; |
||||
|
} |
||||
|
|
||||
|
public void setOriginatorId(UUID originatorId) { |
||||
|
this.originatorId = originatorId; |
||||
|
} |
||||
|
|
||||
|
public EntityType getOriginatorType() { |
||||
|
return originatorType; |
||||
|
} |
||||
|
|
||||
|
public void setOriginatorType(EntityType originatorType) { |
||||
|
this.originatorType = originatorType; |
||||
|
} |
||||
|
|
||||
|
public String getType() { |
||||
|
return type; |
||||
|
} |
||||
|
|
||||
|
public void setType(String type) { |
||||
|
this.type = type; |
||||
|
} |
||||
|
|
||||
|
public AlarmSeverity getSeverity() { |
||||
|
return severity; |
||||
|
} |
||||
|
|
||||
|
public void setSeverity(AlarmSeverity severity) { |
||||
|
this.severity = severity; |
||||
|
} |
||||
|
|
||||
|
public AlarmStatus getStatus() { |
||||
|
return status; |
||||
|
} |
||||
|
|
||||
|
public void setStatus(AlarmStatus status) { |
||||
|
this.status = status; |
||||
|
} |
||||
|
|
||||
|
public Long getStartTs() { |
||||
|
return startTs; |
||||
|
} |
||||
|
|
||||
|
public void setStartTs(Long startTs) { |
||||
|
this.startTs = startTs; |
||||
|
} |
||||
|
|
||||
|
public Long getEndTs() { |
||||
|
return endTs; |
||||
|
} |
||||
|
|
||||
|
public void setEndTs(Long endTs) { |
||||
|
this.endTs = endTs; |
||||
|
} |
||||
|
|
||||
|
public Long getAckTs() { |
||||
|
return ackTs; |
||||
|
} |
||||
|
|
||||
|
public void setAckTs(Long ackTs) { |
||||
|
this.ackTs = ackTs; |
||||
|
} |
||||
|
|
||||
|
public Long getClearTs() { |
||||
|
return clearTs; |
||||
|
} |
||||
|
|
||||
|
public void setClearTs(Long clearTs) { |
||||
|
this.clearTs = clearTs; |
||||
|
} |
||||
|
|
||||
|
public JsonNode getDetails() { |
||||
|
return details; |
||||
|
} |
||||
|
|
||||
|
public void setDetails(JsonNode details) { |
||||
|
this.details = details; |
||||
|
} |
||||
|
|
||||
|
public Boolean getPropagate() { |
||||
|
return propagate; |
||||
|
} |
||||
|
|
||||
|
public void setPropagate(Boolean propagate) { |
||||
|
this.propagate = propagate; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public Alarm toData() { |
||||
|
Alarm alarm = new Alarm(new AlarmId(id)); |
||||
|
alarm.setCreatedTime(UUIDs.unixTimestamp(id)); |
||||
|
if (tenantId != null) { |
||||
|
alarm.setTenantId(new TenantId(tenantId)); |
||||
|
} |
||||
|
alarm.setOriginator(EntityIdFactory.getByTypeAndUuid(originatorType, originatorId)); |
||||
|
alarm.setType(type); |
||||
|
alarm.setSeverity(severity); |
||||
|
alarm.setStatus(status); |
||||
|
alarm.setPropagate(propagate); |
||||
|
alarm.setStartTs(startTs); |
||||
|
alarm.setEndTs(endTs); |
||||
|
alarm.setAckTs(ackTs); |
||||
|
alarm.setClearTs(clearTs); |
||||
|
alarm.setDetails(details); |
||||
|
return alarm; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,107 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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; |
||||
|
|
||||
|
import com.datastax.driver.mapping.annotations.Column; |
||||
|
import com.datastax.driver.mapping.annotations.PartitionKey; |
||||
|
import com.datastax.driver.mapping.annotations.Table; |
||||
|
import com.datastax.driver.mapping.annotations.Transient; |
||||
|
import org.thingsboard.server.common.data.asset.TenantAssetType; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.*; |
||||
|
|
||||
|
@Table(name = ASSET_TYPES_BY_TENANT_VIEW_NAME) |
||||
|
public class TenantAssetTypeEntity { |
||||
|
|
||||
|
@Transient |
||||
|
private static final long serialVersionUID = -1268181161886910152L; |
||||
|
|
||||
|
@PartitionKey(value = 0) |
||||
|
@Column(name = ASSET_TYPE_PROPERTY) |
||||
|
private String type; |
||||
|
|
||||
|
@PartitionKey(value = 1) |
||||
|
@Column(name = ASSET_TENANT_ID_PROPERTY) |
||||
|
private UUID tenantId; |
||||
|
|
||||
|
public TenantAssetTypeEntity() { |
||||
|
super(); |
||||
|
} |
||||
|
|
||||
|
public TenantAssetTypeEntity(TenantAssetType tenantAssetType) { |
||||
|
this.type = tenantAssetType.getType(); |
||||
|
if (tenantAssetType.getTenantId() != null) { |
||||
|
this.tenantId = tenantAssetType.getTenantId().getId(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
public String getType() { |
||||
|
return type; |
||||
|
} |
||||
|
|
||||
|
public void setType(String type) { |
||||
|
this.type = type; |
||||
|
} |
||||
|
|
||||
|
public UUID getTenantId() { |
||||
|
return tenantId; |
||||
|
} |
||||
|
|
||||
|
public void setTenantId(UUID tenantId) { |
||||
|
this.tenantId = tenantId; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public int hashCode() { |
||||
|
int result = type != null ? type.hashCode() : 0; |
||||
|
result = 31 * result + (tenantId != null ? tenantId.hashCode() : 0); |
||||
|
return result; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public boolean equals(Object o) { |
||||
|
if (this == o) return true; |
||||
|
if (o == null || getClass() != o.getClass()) return false; |
||||
|
|
||||
|
TenantAssetTypeEntity that = (TenantAssetTypeEntity) o; |
||||
|
|
||||
|
if (type != null ? !type.equals(that.type) : that.type != null) return false; |
||||
|
return tenantId != null ? tenantId.equals(that.tenantId) : that.tenantId == null; |
||||
|
|
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public String toString() { |
||||
|
final StringBuilder sb = new StringBuilder("TenantAssetTypeEntity{"); |
||||
|
sb.append("type='").append(type).append('\''); |
||||
|
sb.append(", tenantId=").append(tenantId); |
||||
|
sb.append('}'); |
||||
|
return sb.toString(); |
||||
|
} |
||||
|
|
||||
|
public TenantAssetType toTenantAssetType() { |
||||
|
TenantAssetType tenantAssetType = new TenantAssetType(); |
||||
|
tenantAssetType.setType(type); |
||||
|
if (tenantId != null) { |
||||
|
tenantAssetType.setTenantId(new TenantId(tenantId)); |
||||
|
} |
||||
|
return tenantAssetType; |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,107 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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; |
||||
|
|
||||
|
import com.datastax.driver.mapping.annotations.Column; |
||||
|
import com.datastax.driver.mapping.annotations.PartitionKey; |
||||
|
import com.datastax.driver.mapping.annotations.Table; |
||||
|
import com.datastax.driver.mapping.annotations.Transient; |
||||
|
import org.thingsboard.server.common.data.TenantDeviceType; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.*; |
||||
|
|
||||
|
@Table(name = DEVICE_TYPES_BY_TENANT_VIEW_NAME) |
||||
|
public class TenantDeviceTypeEntity { |
||||
|
|
||||
|
@Transient |
||||
|
private static final long serialVersionUID = -1268181166886910152L; |
||||
|
|
||||
|
@PartitionKey(value = 0) |
||||
|
@Column(name = DEVICE_TYPE_PROPERTY) |
||||
|
private String type; |
||||
|
|
||||
|
@PartitionKey(value = 1) |
||||
|
@Column(name = DEVICE_TENANT_ID_PROPERTY) |
||||
|
private UUID tenantId; |
||||
|
|
||||
|
public TenantDeviceTypeEntity() { |
||||
|
super(); |
||||
|
} |
||||
|
|
||||
|
public TenantDeviceTypeEntity(TenantDeviceType tenantDeviceType) { |
||||
|
this.type = tenantDeviceType.getType(); |
||||
|
if (tenantDeviceType.getTenantId() != null) { |
||||
|
this.tenantId = tenantDeviceType.getTenantId().getId(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
public String getType() { |
||||
|
return type; |
||||
|
} |
||||
|
|
||||
|
public void setType(String type) { |
||||
|
this.type = type; |
||||
|
} |
||||
|
|
||||
|
public UUID getTenantId() { |
||||
|
return tenantId; |
||||
|
} |
||||
|
|
||||
|
public void setTenantId(UUID tenantId) { |
||||
|
this.tenantId = tenantId; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public int hashCode() { |
||||
|
int result = type != null ? type.hashCode() : 0; |
||||
|
result = 31 * result + (tenantId != null ? tenantId.hashCode() : 0); |
||||
|
return result; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public boolean equals(Object o) { |
||||
|
if (this == o) return true; |
||||
|
if (o == null || getClass() != o.getClass()) return false; |
||||
|
|
||||
|
TenantDeviceTypeEntity that = (TenantDeviceTypeEntity) o; |
||||
|
|
||||
|
if (type != null ? !type.equals(that.type) : that.type != null) return false; |
||||
|
return tenantId != null ? tenantId.equals(that.tenantId) : that.tenantId == null; |
||||
|
|
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public String toString() { |
||||
|
final StringBuilder sb = new StringBuilder("TenantDeviceTypeEntity{"); |
||||
|
sb.append("type='").append(type).append('\''); |
||||
|
sb.append(", tenantId=").append(tenantId); |
||||
|
sb.append('}'); |
||||
|
return sb.toString(); |
||||
|
} |
||||
|
|
||||
|
public TenantDeviceType toTenantDeviceType() { |
||||
|
TenantDeviceType tenantDeviceType = new TenantDeviceType(); |
||||
|
tenantDeviceType.setType(type); |
||||
|
if (tenantId != null) { |
||||
|
tenantDeviceType.setTenantId(new TenantId(tenantId)); |
||||
|
} |
||||
|
return tenantDeviceType; |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,29 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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.type; |
||||
|
|
||||
|
import com.datastax.driver.extras.codecs.enums.EnumNameCodec; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmSeverity; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmStatus; |
||||
|
import org.thingsboard.server.dao.alarm.AlarmService; |
||||
|
|
||||
|
public class AlarmSeverityCodec extends EnumNameCodec<AlarmSeverity> { |
||||
|
|
||||
|
public AlarmSeverityCodec() { |
||||
|
super(AlarmSeverity.class); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,27 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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.type; |
||||
|
|
||||
|
import com.datastax.driver.extras.codecs.enums.EnumNameCodec; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmStatus; |
||||
|
|
||||
|
public class AlarmStatusCodec extends EnumNameCodec<AlarmStatus> { |
||||
|
|
||||
|
public AlarmStatusCodec() { |
||||
|
super(AlarmStatus.class); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,28 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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.type; |
||||
|
|
||||
|
import org.thingsboard.server.common.data.relation.RelationTypeGroup; |
||||
|
|
||||
|
import com.datastax.driver.extras.codecs.enums.EnumNameCodec; |
||||
|
|
||||
|
public class RelationTypeGroupCodec extends EnumNameCodec<RelationTypeGroup> { |
||||
|
|
||||
|
public RelationTypeGroupCodec() { |
||||
|
super(RelationTypeGroup.class); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,172 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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; |
||||
|
|
||||
|
import com.datastax.driver.core.utils.UUIDs; |
||||
|
import org.junit.After; |
||||
|
import org.junit.Assert; |
||||
|
import org.junit.Before; |
||||
|
import org.junit.Test; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
import org.thingsboard.server.common.data.Tenant; |
||||
|
import org.thingsboard.server.common.data.alarm.Alarm; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmQuery; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmSeverity; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmStatus; |
||||
|
import org.thingsboard.server.common.data.id.AssetId; |
||||
|
import org.thingsboard.server.common.data.id.DeviceId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.page.TimePageData; |
||||
|
import org.thingsboard.server.common.data.page.TimePageLink; |
||||
|
import org.thingsboard.server.common.data.relation.EntityRelation; |
||||
|
import org.thingsboard.server.dao.exception.DataValidationException; |
||||
|
import org.thingsboard.server.dao.relation.EntityRelationsQuery; |
||||
|
import org.thingsboard.server.dao.relation.EntitySearchDirection; |
||||
|
import org.thingsboard.server.dao.relation.EntityTypeFilter; |
||||
|
import org.thingsboard.server.dao.relation.RelationsSearchParameters; |
||||
|
|
||||
|
import java.util.Collections; |
||||
|
import java.util.List; |
||||
|
import java.util.concurrent.ExecutionException; |
||||
|
|
||||
|
public class AlarmServiceTest extends AbstractServiceTest { |
||||
|
|
||||
|
public static final String TEST_ALARM = "TEST_ALARM"; |
||||
|
private TenantId tenantId; |
||||
|
|
||||
|
@Before |
||||
|
public void before() { |
||||
|
Tenant tenant = new Tenant(); |
||||
|
tenant.setTitle("My tenant"); |
||||
|
Tenant savedTenant = tenantService.saveTenant(tenant); |
||||
|
Assert.assertNotNull(savedTenant); |
||||
|
tenantId = savedTenant.getId(); |
||||
|
} |
||||
|
|
||||
|
@After |
||||
|
public void after() { |
||||
|
tenantService.deleteTenant(tenantId); |
||||
|
} |
||||
|
|
||||
|
|
||||
|
@Test |
||||
|
public void testSaveAndFetchAlarm() throws ExecutionException, InterruptedException { |
||||
|
AssetId parentId = new AssetId(UUIDs.timeBased()); |
||||
|
AssetId childId = new AssetId(UUIDs.timeBased()); |
||||
|
|
||||
|
EntityRelation relation = new EntityRelation(parentId, childId, EntityRelation.CONTAINS_TYPE); |
||||
|
|
||||
|
Assert.assertTrue(relationService.saveRelation(relation).get()); |
||||
|
|
||||
|
long ts = System.currentTimeMillis(); |
||||
|
Alarm alarm = Alarm.builder().tenantId(tenantId).originator(childId) |
||||
|
.type(TEST_ALARM) |
||||
|
.severity(AlarmSeverity.CRITICAL).status(AlarmStatus.ACTIVE_UNACK) |
||||
|
.startTs(ts).build(); |
||||
|
|
||||
|
Alarm created = alarmService.createOrUpdateAlarm(alarm); |
||||
|
|
||||
|
Assert.assertNotNull(created); |
||||
|
Assert.assertNotNull(created.getId()); |
||||
|
Assert.assertNotNull(created.getOriginator()); |
||||
|
Assert.assertNotNull(created.getSeverity()); |
||||
|
Assert.assertNotNull(created.getStatus()); |
||||
|
|
||||
|
Assert.assertEquals(tenantId, created.getTenantId()); |
||||
|
Assert.assertEquals(childId, created.getOriginator()); |
||||
|
Assert.assertEquals(TEST_ALARM, created.getType()); |
||||
|
Assert.assertEquals(AlarmSeverity.CRITICAL, created.getSeverity()); |
||||
|
Assert.assertEquals(AlarmStatus.ACTIVE_UNACK, created.getStatus()); |
||||
|
Assert.assertEquals(ts, created.getStartTs()); |
||||
|
Assert.assertEquals(ts, created.getEndTs()); |
||||
|
Assert.assertEquals(0L, created.getAckTs()); |
||||
|
Assert.assertEquals(0L, created.getClearTs()); |
||||
|
|
||||
|
Alarm fetched = alarmService.findAlarmByIdAsync(created.getId()).get(); |
||||
|
Assert.assertEquals(created, fetched); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindAlarm() throws ExecutionException, InterruptedException { |
||||
|
AssetId parentId = new AssetId(UUIDs.timeBased()); |
||||
|
AssetId childId = new AssetId(UUIDs.timeBased()); |
||||
|
|
||||
|
EntityRelation relation = new EntityRelation(parentId, childId, EntityRelation.CONTAINS_TYPE); |
||||
|
|
||||
|
Assert.assertTrue(relationService.saveRelation(relation).get()); |
||||
|
|
||||
|
long ts = System.currentTimeMillis(); |
||||
|
Alarm alarm = Alarm.builder().tenantId(tenantId).originator(childId) |
||||
|
.type(TEST_ALARM) |
||||
|
.severity(AlarmSeverity.CRITICAL).status(AlarmStatus.ACTIVE_UNACK) |
||||
|
.startTs(ts).build(); |
||||
|
|
||||
|
Alarm created = alarmService.createOrUpdateAlarm(alarm); |
||||
|
|
||||
|
// Check child relation
|
||||
|
TimePageData<Alarm> alarms = alarmService.findAlarms(AlarmQuery.builder() |
||||
|
.affectedEntityId(childId) |
||||
|
.status(AlarmStatus.ACTIVE_UNACK).pageLink( |
||||
|
new TimePageLink(1, 0L, System.currentTimeMillis(), false) |
||||
|
).build()).get(); |
||||
|
Assert.assertNotNull(alarms.getData()); |
||||
|
Assert.assertEquals(1, alarms.getData().size()); |
||||
|
Assert.assertEquals(created, alarms.getData().get(0)); |
||||
|
|
||||
|
// Check parent relation
|
||||
|
alarms = alarmService.findAlarms(AlarmQuery.builder() |
||||
|
.affectedEntityId(parentId) |
||||
|
.status(AlarmStatus.ACTIVE_UNACK).pageLink( |
||||
|
new TimePageLink(1, 0L, System.currentTimeMillis(), false) |
||||
|
).build()).get(); |
||||
|
Assert.assertNotNull(alarms.getData()); |
||||
|
Assert.assertEquals(1, alarms.getData().size()); |
||||
|
Assert.assertEquals(created, alarms.getData().get(0)); |
||||
|
|
||||
|
alarmService.ackAlarm(created.getId(), System.currentTimeMillis()).get(); |
||||
|
created = alarmService.findAlarmByIdAsync(created.getId()).get(); |
||||
|
|
||||
|
alarms = alarmService.findAlarms(AlarmQuery.builder() |
||||
|
.affectedEntityId(childId) |
||||
|
.status(AlarmStatus.ACTIVE_ACK).pageLink( |
||||
|
new TimePageLink(1, 0L, System.currentTimeMillis(), false) |
||||
|
).build()).get(); |
||||
|
Assert.assertNotNull(alarms.getData()); |
||||
|
Assert.assertEquals(1, alarms.getData().size()); |
||||
|
Assert.assertEquals(created, alarms.getData().get(0)); |
||||
|
|
||||
|
// Check not existing relation
|
||||
|
alarms = alarmService.findAlarms(AlarmQuery.builder() |
||||
|
.affectedEntityId(childId) |
||||
|
.status(AlarmStatus.ACTIVE_UNACK).pageLink( |
||||
|
new TimePageLink(1, 0L, System.currentTimeMillis(), false) |
||||
|
).build()).get(); |
||||
|
Assert.assertNotNull(alarms.getData()); |
||||
|
Assert.assertEquals(0, alarms.getData().size()); |
||||
|
|
||||
|
alarmService.clearAlarm(created.getId(), System.currentTimeMillis()).get(); |
||||
|
created = alarmService.findAlarmByIdAsync(created.getId()).get(); |
||||
|
|
||||
|
alarms = alarmService.findAlarms(AlarmQuery.builder() |
||||
|
.affectedEntityId(childId) |
||||
|
.status(AlarmStatus.CLEARED_ACK).pageLink( |
||||
|
new TimePageLink(1, 0L, System.currentTimeMillis(), false) |
||||
|
).build()).get(); |
||||
|
Assert.assertNotNull(alarms.getData()); |
||||
|
Assert.assertEquals(1, alarms.getData().size()); |
||||
|
Assert.assertEquals(created, alarms.getData().get(0)); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,634 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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; |
||||
|
|
||||
|
import com.datastax.driver.core.utils.UUIDs; |
||||
|
import org.apache.commons.lang3.RandomStringUtils; |
||||
|
import org.junit.After; |
||||
|
import org.junit.Assert; |
||||
|
import org.junit.Before; |
||||
|
import org.junit.Test; |
||||
|
import org.thingsboard.server.common.data.Customer; |
||||
|
import org.thingsboard.server.common.data.Tenant; |
||||
|
import org.thingsboard.server.common.data.asset.Asset; |
||||
|
import org.thingsboard.server.common.data.asset.TenantAssetType; |
||||
|
import org.thingsboard.server.common.data.id.CustomerId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.page.TextPageData; |
||||
|
import org.thingsboard.server.common.data.page.TextPageLink; |
||||
|
import org.thingsboard.server.dao.exception.DataValidationException; |
||||
|
|
||||
|
import java.util.ArrayList; |
||||
|
import java.util.Collections; |
||||
|
import java.util.List; |
||||
|
|
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.NULL_UUID; |
||||
|
|
||||
|
public class BaseAssetServiceTest extends AbstractServiceTest { |
||||
|
|
||||
|
private IdComparator<Asset> idComparator = new IdComparator<>(); |
||||
|
|
||||
|
private TenantId tenantId; |
||||
|
|
||||
|
@Before |
||||
|
public void before() { |
||||
|
Tenant tenant = new Tenant(); |
||||
|
tenant.setTitle("My tenant"); |
||||
|
Tenant savedTenant = tenantService.saveTenant(tenant); |
||||
|
Assert.assertNotNull(savedTenant); |
||||
|
tenantId = savedTenant.getId(); |
||||
|
} |
||||
|
|
||||
|
@After |
||||
|
public void after() { |
||||
|
tenantService.deleteTenant(tenantId); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testSaveAsset() { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setTenantId(tenantId); |
||||
|
asset.setName("My asset"); |
||||
|
asset.setType("default"); |
||||
|
Asset savedAsset = assetService.saveAsset(asset); |
||||
|
|
||||
|
Assert.assertNotNull(savedAsset); |
||||
|
Assert.assertNotNull(savedAsset.getId()); |
||||
|
Assert.assertTrue(savedAsset.getCreatedTime() > 0); |
||||
|
Assert.assertEquals(asset.getTenantId(), savedAsset.getTenantId()); |
||||
|
Assert.assertNotNull(savedAsset.getCustomerId()); |
||||
|
Assert.assertEquals(NULL_UUID, savedAsset.getCustomerId().getId()); |
||||
|
Assert.assertEquals(asset.getName(), savedAsset.getName()); |
||||
|
|
||||
|
savedAsset.setName("My new asset"); |
||||
|
|
||||
|
assetService.saveAsset(savedAsset); |
||||
|
Asset foundAsset = assetService.findAssetById(savedAsset.getId()); |
||||
|
Assert.assertEquals(foundAsset.getName(), savedAsset.getName()); |
||||
|
|
||||
|
assetService.deleteAsset(savedAsset.getId()); |
||||
|
} |
||||
|
|
||||
|
@Test(expected = DataValidationException.class) |
||||
|
public void testSaveAssetWithEmptyName() { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setTenantId(tenantId); |
||||
|
asset.setType("default"); |
||||
|
assetService.saveAsset(asset); |
||||
|
} |
||||
|
|
||||
|
@Test(expected = DataValidationException.class) |
||||
|
public void testSaveAssetWithEmptyTenant() { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setName("My asset"); |
||||
|
asset.setType("default"); |
||||
|
assetService.saveAsset(asset); |
||||
|
} |
||||
|
|
||||
|
@Test(expected = DataValidationException.class) |
||||
|
public void testSaveAssetWithInvalidTenant() { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setName("My asset"); |
||||
|
asset.setType("default"); |
||||
|
asset.setTenantId(new TenantId(UUIDs.timeBased())); |
||||
|
assetService.saveAsset(asset); |
||||
|
} |
||||
|
|
||||
|
@Test(expected = DataValidationException.class) |
||||
|
public void testAssignAssetToNonExistentCustomer() { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setName("My asset"); |
||||
|
asset.setType("default"); |
||||
|
asset.setTenantId(tenantId); |
||||
|
asset = assetService.saveAsset(asset); |
||||
|
try { |
||||
|
assetService.assignAssetToCustomer(asset.getId(), new CustomerId(UUIDs.timeBased())); |
||||
|
} finally { |
||||
|
assetService.deleteAsset(asset.getId()); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Test(expected = DataValidationException.class) |
||||
|
public void testAssignAssetToCustomerFromDifferentTenant() { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setName("My asset"); |
||||
|
asset.setType("default"); |
||||
|
asset.setTenantId(tenantId); |
||||
|
asset = assetService.saveAsset(asset); |
||||
|
Tenant tenant = new Tenant(); |
||||
|
tenant.setTitle("Test different tenant"); |
||||
|
tenant = tenantService.saveTenant(tenant); |
||||
|
Customer customer = new Customer(); |
||||
|
customer.setTenantId(tenant.getId()); |
||||
|
customer.setTitle("Test different customer"); |
||||
|
customer = customerService.saveCustomer(customer); |
||||
|
try { |
||||
|
assetService.assignAssetToCustomer(asset.getId(), customer.getId()); |
||||
|
} finally { |
||||
|
assetService.deleteAsset(asset.getId()); |
||||
|
tenantService.deleteTenant(tenant.getId()); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindAssetById() { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setTenantId(tenantId); |
||||
|
asset.setName("My asset"); |
||||
|
asset.setType("default"); |
||||
|
Asset savedAsset = assetService.saveAsset(asset); |
||||
|
Asset foundAsset = assetService.findAssetById(savedAsset.getId()); |
||||
|
Assert.assertNotNull(foundAsset); |
||||
|
Assert.assertEquals(savedAsset, foundAsset); |
||||
|
assetService.deleteAsset(savedAsset.getId()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindAssetTypesByTenantId() throws Exception { |
||||
|
List<Asset> assets = new ArrayList<>(); |
||||
|
try { |
||||
|
for (int i=0;i<3;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setTenantId(tenantId); |
||||
|
asset.setName("My asset B"+i); |
||||
|
asset.setType("typeB"); |
||||
|
assets.add(assetService.saveAsset(asset)); |
||||
|
} |
||||
|
for (int i=0;i<7;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setTenantId(tenantId); |
||||
|
asset.setName("My asset C"+i); |
||||
|
asset.setType("typeC"); |
||||
|
assets.add(assetService.saveAsset(asset)); |
||||
|
} |
||||
|
for (int i=0;i<9;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setTenantId(tenantId); |
||||
|
asset.setName("My asset A"+i); |
||||
|
asset.setType("typeA"); |
||||
|
assets.add(assetService.saveAsset(asset)); |
||||
|
} |
||||
|
List<TenantAssetType> assetTypes = assetService.findAssetTypesByTenantId(tenantId).get(); |
||||
|
Assert.assertNotNull(assetTypes); |
||||
|
Assert.assertEquals(3, assetTypes.size()); |
||||
|
Assert.assertEquals("typeA", assetTypes.get(0).getType()); |
||||
|
Assert.assertEquals("typeB", assetTypes.get(1).getType()); |
||||
|
Assert.assertEquals("typeC", assetTypes.get(2).getType()); |
||||
|
} finally { |
||||
|
assets.forEach((asset) -> { assetService.deleteAsset(asset.getId()); }); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testDeleteAsset() { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setTenantId(tenantId); |
||||
|
asset.setName("My asset"); |
||||
|
asset.setType("default"); |
||||
|
Asset savedAsset = assetService.saveAsset(asset); |
||||
|
Asset foundAsset = assetService.findAssetById(savedAsset.getId()); |
||||
|
Assert.assertNotNull(foundAsset); |
||||
|
assetService.deleteAsset(savedAsset.getId()); |
||||
|
foundAsset = assetService.findAssetById(savedAsset.getId()); |
||||
|
Assert.assertNull(foundAsset); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindAssetsByTenantId() { |
||||
|
Tenant tenant = new Tenant(); |
||||
|
tenant.setTitle("Test tenant"); |
||||
|
tenant = tenantService.saveTenant(tenant); |
||||
|
|
||||
|
TenantId tenantId = tenant.getId(); |
||||
|
|
||||
|
List<Asset> assets = new ArrayList<>(); |
||||
|
for (int i=0;i<178;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setTenantId(tenantId); |
||||
|
asset.setName("Asset"+i); |
||||
|
asset.setType("default"); |
||||
|
assets.add(assetService.saveAsset(asset)); |
||||
|
} |
||||
|
|
||||
|
List<Asset> loadedAssets = new ArrayList<>(); |
||||
|
TextPageLink pageLink = new TextPageLink(23); |
||||
|
TextPageData<Asset> pageData = null; |
||||
|
do { |
||||
|
pageData = assetService.findAssetsByTenantId(tenantId, pageLink); |
||||
|
loadedAssets.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assets, idComparator); |
||||
|
Collections.sort(loadedAssets, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assets, loadedAssets); |
||||
|
|
||||
|
assetService.deleteAssetsByTenantId(tenantId); |
||||
|
|
||||
|
pageLink = new TextPageLink(33); |
||||
|
pageData = assetService.findAssetsByTenantId(tenantId, pageLink); |
||||
|
Assert.assertFalse(pageData.hasNext()); |
||||
|
Assert.assertTrue(pageData.getData().isEmpty()); |
||||
|
|
||||
|
tenantService.deleteTenant(tenantId); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindAssetsByTenantIdAndName() { |
||||
|
String title1 = "Asset title 1"; |
||||
|
List<Asset> assetsTitle1 = new ArrayList<>(); |
||||
|
for (int i=0;i<143;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setTenantId(tenantId); |
||||
|
String suffix = RandomStringUtils.randomAlphanumeric(15); |
||||
|
String name = title1+suffix; |
||||
|
name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); |
||||
|
asset.setName(name); |
||||
|
asset.setType("default"); |
||||
|
assetsTitle1.add(assetService.saveAsset(asset)); |
||||
|
} |
||||
|
String title2 = "Asset title 2"; |
||||
|
List<Asset> assetsTitle2 = new ArrayList<>(); |
||||
|
for (int i=0;i<175;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setTenantId(tenantId); |
||||
|
String suffix = RandomStringUtils.randomAlphanumeric(15); |
||||
|
String name = title2+suffix; |
||||
|
name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); |
||||
|
asset.setName(name); |
||||
|
asset.setType("default"); |
||||
|
assetsTitle2.add(assetService.saveAsset(asset)); |
||||
|
} |
||||
|
|
||||
|
List<Asset> loadedAssetsTitle1 = new ArrayList<>(); |
||||
|
TextPageLink pageLink = new TextPageLink(15, title1); |
||||
|
TextPageData<Asset> pageData = null; |
||||
|
do { |
||||
|
pageData = assetService.findAssetsByTenantId(tenantId, pageLink); |
||||
|
loadedAssetsTitle1.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assetsTitle1, idComparator); |
||||
|
Collections.sort(loadedAssetsTitle1, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assetsTitle1, loadedAssetsTitle1); |
||||
|
|
||||
|
List<Asset> loadedAssetsTitle2 = new ArrayList<>(); |
||||
|
pageLink = new TextPageLink(4, title2); |
||||
|
do { |
||||
|
pageData = assetService.findAssetsByTenantId(tenantId, pageLink); |
||||
|
loadedAssetsTitle2.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assetsTitle2, idComparator); |
||||
|
Collections.sort(loadedAssetsTitle2, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assetsTitle2, loadedAssetsTitle2); |
||||
|
|
||||
|
for (Asset asset : loadedAssetsTitle1) { |
||||
|
assetService.deleteAsset(asset.getId()); |
||||
|
} |
||||
|
|
||||
|
pageLink = new TextPageLink(4, title1); |
||||
|
pageData = assetService.findAssetsByTenantId(tenantId, pageLink); |
||||
|
Assert.assertFalse(pageData.hasNext()); |
||||
|
Assert.assertEquals(0, pageData.getData().size()); |
||||
|
|
||||
|
for (Asset asset : loadedAssetsTitle2) { |
||||
|
assetService.deleteAsset(asset.getId()); |
||||
|
} |
||||
|
|
||||
|
pageLink = new TextPageLink(4, title2); |
||||
|
pageData = assetService.findAssetsByTenantId(tenantId, pageLink); |
||||
|
Assert.assertFalse(pageData.hasNext()); |
||||
|
Assert.assertEquals(0, pageData.getData().size()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindAssetsByTenantIdAndType() { |
||||
|
String title1 = "Asset title 1"; |
||||
|
String type1 = "typeA"; |
||||
|
List<Asset> assetsType1 = new ArrayList<>(); |
||||
|
for (int i=0;i<143;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setTenantId(tenantId); |
||||
|
String suffix = RandomStringUtils.randomAlphanumeric(15); |
||||
|
String name = title1+suffix; |
||||
|
name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); |
||||
|
asset.setName(name); |
||||
|
asset.setType(type1); |
||||
|
assetsType1.add(assetService.saveAsset(asset)); |
||||
|
} |
||||
|
String title2 = "Asset title 2"; |
||||
|
String type2 = "typeB"; |
||||
|
List<Asset> assetsType2 = new ArrayList<>(); |
||||
|
for (int i=0;i<175;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setTenantId(tenantId); |
||||
|
String suffix = RandomStringUtils.randomAlphanumeric(15); |
||||
|
String name = title2+suffix; |
||||
|
name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); |
||||
|
asset.setName(name); |
||||
|
asset.setType(type2); |
||||
|
assetsType2.add(assetService.saveAsset(asset)); |
||||
|
} |
||||
|
|
||||
|
List<Asset> loadedAssetsType1 = new ArrayList<>(); |
||||
|
TextPageLink pageLink = new TextPageLink(15); |
||||
|
TextPageData<Asset> pageData = null; |
||||
|
do { |
||||
|
pageData = assetService.findAssetsByTenantIdAndType(tenantId, type1, pageLink); |
||||
|
loadedAssetsType1.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assetsType1, idComparator); |
||||
|
Collections.sort(loadedAssetsType1, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assetsType1, loadedAssetsType1); |
||||
|
|
||||
|
List<Asset> loadedAssetsType2 = new ArrayList<>(); |
||||
|
pageLink = new TextPageLink(4); |
||||
|
do { |
||||
|
pageData = assetService.findAssetsByTenantIdAndType(tenantId, type2, pageLink); |
||||
|
loadedAssetsType2.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assetsType2, idComparator); |
||||
|
Collections.sort(loadedAssetsType2, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assetsType2, loadedAssetsType2); |
||||
|
|
||||
|
for (Asset asset : loadedAssetsType1) { |
||||
|
assetService.deleteAsset(asset.getId()); |
||||
|
} |
||||
|
|
||||
|
pageLink = new TextPageLink(4); |
||||
|
pageData = assetService.findAssetsByTenantIdAndType(tenantId, type1, pageLink); |
||||
|
Assert.assertFalse(pageData.hasNext()); |
||||
|
Assert.assertEquals(0, pageData.getData().size()); |
||||
|
|
||||
|
for (Asset asset : loadedAssetsType2) { |
||||
|
assetService.deleteAsset(asset.getId()); |
||||
|
} |
||||
|
|
||||
|
pageLink = new TextPageLink(4); |
||||
|
pageData = assetService.findAssetsByTenantIdAndType(tenantId, type2, pageLink); |
||||
|
Assert.assertFalse(pageData.hasNext()); |
||||
|
Assert.assertEquals(0, pageData.getData().size()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindAssetsByTenantIdAndCustomerId() { |
||||
|
Tenant tenant = new Tenant(); |
||||
|
tenant.setTitle("Test tenant"); |
||||
|
tenant = tenantService.saveTenant(tenant); |
||||
|
|
||||
|
TenantId tenantId = tenant.getId(); |
||||
|
|
||||
|
Customer customer = new Customer(); |
||||
|
customer.setTitle("Test customer"); |
||||
|
customer.setTenantId(tenantId); |
||||
|
customer = customerService.saveCustomer(customer); |
||||
|
CustomerId customerId = customer.getId(); |
||||
|
|
||||
|
List<Asset> assets = new ArrayList<>(); |
||||
|
for (int i=0;i<278;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setTenantId(tenantId); |
||||
|
asset.setName("Asset"+i); |
||||
|
asset.setType("default"); |
||||
|
asset = assetService.saveAsset(asset); |
||||
|
assets.add(assetService.assignAssetToCustomer(asset.getId(), customerId)); |
||||
|
} |
||||
|
|
||||
|
List<Asset> loadedAssets = new ArrayList<>(); |
||||
|
TextPageLink pageLink = new TextPageLink(23); |
||||
|
TextPageData<Asset> pageData = null; |
||||
|
do { |
||||
|
pageData = assetService.findAssetsByTenantIdAndCustomerId(tenantId, customerId, pageLink); |
||||
|
loadedAssets.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assets, idComparator); |
||||
|
Collections.sort(loadedAssets, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assets, loadedAssets); |
||||
|
|
||||
|
assetService.unassignCustomerAssets(tenantId, customerId); |
||||
|
|
||||
|
pageLink = new TextPageLink(33); |
||||
|
pageData = assetService.findAssetsByTenantIdAndCustomerId(tenantId, customerId, pageLink); |
||||
|
Assert.assertFalse(pageData.hasNext()); |
||||
|
Assert.assertTrue(pageData.getData().isEmpty()); |
||||
|
|
||||
|
tenantService.deleteTenant(tenantId); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindAssetsByTenantIdCustomerIdAndName() { |
||||
|
|
||||
|
Customer customer = new Customer(); |
||||
|
customer.setTitle("Test customer"); |
||||
|
customer.setTenantId(tenantId); |
||||
|
customer = customerService.saveCustomer(customer); |
||||
|
CustomerId customerId = customer.getId(); |
||||
|
|
||||
|
String title1 = "Asset title 1"; |
||||
|
List<Asset> assetsTitle1 = new ArrayList<>(); |
||||
|
for (int i=0;i<175;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setTenantId(tenantId); |
||||
|
String suffix = RandomStringUtils.randomAlphanumeric(15); |
||||
|
String name = title1+suffix; |
||||
|
name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); |
||||
|
asset.setName(name); |
||||
|
asset.setType("default"); |
||||
|
asset = assetService.saveAsset(asset); |
||||
|
assetsTitle1.add(assetService.assignAssetToCustomer(asset.getId(), customerId)); |
||||
|
} |
||||
|
String title2 = "Asset title 2"; |
||||
|
List<Asset> assetsTitle2 = new ArrayList<>(); |
||||
|
for (int i=0;i<143;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setTenantId(tenantId); |
||||
|
String suffix = RandomStringUtils.randomAlphanumeric(15); |
||||
|
String name = title2+suffix; |
||||
|
name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); |
||||
|
asset.setName(name); |
||||
|
asset.setType("default"); |
||||
|
asset = assetService.saveAsset(asset); |
||||
|
assetsTitle2.add(assetService.assignAssetToCustomer(asset.getId(), customerId)); |
||||
|
} |
||||
|
|
||||
|
List<Asset> loadedAssetsTitle1 = new ArrayList<>(); |
||||
|
TextPageLink pageLink = new TextPageLink(15, title1); |
||||
|
TextPageData<Asset> pageData = null; |
||||
|
do { |
||||
|
pageData = assetService.findAssetsByTenantIdAndCustomerId(tenantId, customerId, pageLink); |
||||
|
loadedAssetsTitle1.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assetsTitle1, idComparator); |
||||
|
Collections.sort(loadedAssetsTitle1, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assetsTitle1, loadedAssetsTitle1); |
||||
|
|
||||
|
List<Asset> loadedAssetsTitle2 = new ArrayList<>(); |
||||
|
pageLink = new TextPageLink(4, title2); |
||||
|
do { |
||||
|
pageData = assetService.findAssetsByTenantIdAndCustomerId(tenantId, customerId, pageLink); |
||||
|
loadedAssetsTitle2.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assetsTitle2, idComparator); |
||||
|
Collections.sort(loadedAssetsTitle2, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assetsTitle2, loadedAssetsTitle2); |
||||
|
|
||||
|
for (Asset asset : loadedAssetsTitle1) { |
||||
|
assetService.deleteAsset(asset.getId()); |
||||
|
} |
||||
|
|
||||
|
pageLink = new TextPageLink(4, title1); |
||||
|
pageData = assetService.findAssetsByTenantIdAndCustomerId(tenantId, customerId, pageLink); |
||||
|
Assert.assertFalse(pageData.hasNext()); |
||||
|
Assert.assertEquals(0, pageData.getData().size()); |
||||
|
|
||||
|
for (Asset asset : loadedAssetsTitle2) { |
||||
|
assetService.deleteAsset(asset.getId()); |
||||
|
} |
||||
|
|
||||
|
pageLink = new TextPageLink(4, title2); |
||||
|
pageData = assetService.findAssetsByTenantIdAndCustomerId(tenantId, customerId, pageLink); |
||||
|
Assert.assertFalse(pageData.hasNext()); |
||||
|
Assert.assertEquals(0, pageData.getData().size()); |
||||
|
customerService.deleteCustomer(customerId); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindAssetsByTenantIdCustomerIdAndType() { |
||||
|
|
||||
|
Customer customer = new Customer(); |
||||
|
customer.setTitle("Test customer"); |
||||
|
customer.setTenantId(tenantId); |
||||
|
customer = customerService.saveCustomer(customer); |
||||
|
CustomerId customerId = customer.getId(); |
||||
|
|
||||
|
String title1 = "Asset title 1"; |
||||
|
String type1 = "typeC"; |
||||
|
List<Asset> assetsType1 = new ArrayList<>(); |
||||
|
for (int i=0;i<175;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setTenantId(tenantId); |
||||
|
String suffix = RandomStringUtils.randomAlphanumeric(15); |
||||
|
String name = title1+suffix; |
||||
|
name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); |
||||
|
asset.setName(name); |
||||
|
asset.setType(type1); |
||||
|
asset = assetService.saveAsset(asset); |
||||
|
assetsType1.add(assetService.assignAssetToCustomer(asset.getId(), customerId)); |
||||
|
} |
||||
|
String title2 = "Asset title 2"; |
||||
|
String type2 = "typeD"; |
||||
|
List<Asset> assetsType2 = new ArrayList<>(); |
||||
|
for (int i=0;i<143;i++) { |
||||
|
Asset asset = new Asset(); |
||||
|
asset.setTenantId(tenantId); |
||||
|
String suffix = RandomStringUtils.randomAlphanumeric(15); |
||||
|
String name = title2+suffix; |
||||
|
name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); |
||||
|
asset.setName(name); |
||||
|
asset.setType(type2); |
||||
|
asset = assetService.saveAsset(asset); |
||||
|
assetsType2.add(assetService.assignAssetToCustomer(asset.getId(), customerId)); |
||||
|
} |
||||
|
|
||||
|
List<Asset> loadedAssetsType1 = new ArrayList<>(); |
||||
|
TextPageLink pageLink = new TextPageLink(15); |
||||
|
TextPageData<Asset> pageData = null; |
||||
|
do { |
||||
|
pageData = assetService.findAssetsByTenantIdAndCustomerIdAndType(tenantId, customerId, type1, pageLink); |
||||
|
loadedAssetsType1.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assetsType1, idComparator); |
||||
|
Collections.sort(loadedAssetsType1, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assetsType1, loadedAssetsType1); |
||||
|
|
||||
|
List<Asset> loadedAssetsType2 = new ArrayList<>(); |
||||
|
pageLink = new TextPageLink(4); |
||||
|
do { |
||||
|
pageData = assetService.findAssetsByTenantIdAndCustomerIdAndType(tenantId, customerId, type2, pageLink); |
||||
|
loadedAssetsType2.addAll(pageData.getData()); |
||||
|
if (pageData.hasNext()) { |
||||
|
pageLink = pageData.getNextPageLink(); |
||||
|
} |
||||
|
} while (pageData.hasNext()); |
||||
|
|
||||
|
Collections.sort(assetsType2, idComparator); |
||||
|
Collections.sort(loadedAssetsType2, idComparator); |
||||
|
|
||||
|
Assert.assertEquals(assetsType2, loadedAssetsType2); |
||||
|
|
||||
|
for (Asset asset : loadedAssetsType1) { |
||||
|
assetService.deleteAsset(asset.getId()); |
||||
|
} |
||||
|
|
||||
|
pageLink = new TextPageLink(4); |
||||
|
pageData = assetService.findAssetsByTenantIdAndCustomerIdAndType(tenantId, customerId, type1, pageLink); |
||||
|
Assert.assertFalse(pageData.hasNext()); |
||||
|
Assert.assertEquals(0, pageData.getData().size()); |
||||
|
|
||||
|
for (Asset asset : loadedAssetsType2) { |
||||
|
assetService.deleteAsset(asset.getId()); |
||||
|
} |
||||
|
|
||||
|
pageLink = new TextPageLink(4); |
||||
|
pageData = assetService.findAssetsByTenantIdAndCustomerIdAndType(tenantId, customerId, type2, pageLink); |
||||
|
Assert.assertFalse(pageData.hasNext()); |
||||
|
Assert.assertEquals(0, pageData.getData().size()); |
||||
|
customerService.deleteCustomer(customerId); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,284 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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; |
||||
|
|
||||
|
import com.datastax.driver.core.utils.UUIDs; |
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
import org.junit.After; |
||||
|
import org.junit.Assert; |
||||
|
import org.junit.Before; |
||||
|
import org.junit.Test; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
import org.thingsboard.server.common.data.asset.Asset; |
||||
|
import org.thingsboard.server.common.data.id.AssetId; |
||||
|
import org.thingsboard.server.common.data.id.DeviceId; |
||||
|
import org.thingsboard.server.common.data.relation.EntityRelation; |
||||
|
import org.thingsboard.server.common.data.relation.RelationTypeGroup; |
||||
|
import org.thingsboard.server.dao.exception.DataValidationException; |
||||
|
import org.thingsboard.server.dao.relation.EntityRelationsQuery; |
||||
|
import org.thingsboard.server.dao.relation.EntitySearchDirection; |
||||
|
import org.thingsboard.server.dao.relation.EntityTypeFilter; |
||||
|
import org.thingsboard.server.dao.relation.RelationsSearchParameters; |
||||
|
|
||||
|
import java.util.Collections; |
||||
|
import java.util.List; |
||||
|
import java.util.concurrent.ExecutionException; |
||||
|
|
||||
|
public class RelationServiceTest extends AbstractServiceTest { |
||||
|
|
||||
|
@Before |
||||
|
public void before() { |
||||
|
} |
||||
|
|
||||
|
@After |
||||
|
public void after() { |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testSaveRelation() throws ExecutionException, InterruptedException { |
||||
|
AssetId parentId = new AssetId(UUIDs.timeBased()); |
||||
|
AssetId childId = new AssetId(UUIDs.timeBased()); |
||||
|
|
||||
|
EntityRelation relation = new EntityRelation(parentId, childId, EntityRelation.CONTAINS_TYPE); |
||||
|
|
||||
|
Assert.assertTrue(saveRelation(relation)); |
||||
|
|
||||
|
Assert.assertTrue(relationService.checkRelation(parentId, childId, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get()); |
||||
|
|
||||
|
Assert.assertFalse(relationService.checkRelation(parentId, childId, "NOT_EXISTING_TYPE", RelationTypeGroup.COMMON).get()); |
||||
|
|
||||
|
Assert.assertFalse(relationService.checkRelation(childId, parentId, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get()); |
||||
|
|
||||
|
Assert.assertFalse(relationService.checkRelation(childId, parentId, "NOT_EXISTING_TYPE", RelationTypeGroup.COMMON).get()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testDeleteRelation() throws ExecutionException, InterruptedException { |
||||
|
AssetId parentId = new AssetId(UUIDs.timeBased()); |
||||
|
AssetId childId = new AssetId(UUIDs.timeBased()); |
||||
|
AssetId subChildId = new AssetId(UUIDs.timeBased()); |
||||
|
|
||||
|
EntityRelation relationA = new EntityRelation(parentId, childId, EntityRelation.CONTAINS_TYPE); |
||||
|
EntityRelation relationB = new EntityRelation(childId, subChildId, EntityRelation.CONTAINS_TYPE); |
||||
|
|
||||
|
saveRelation(relationA); |
||||
|
saveRelation(relationB); |
||||
|
|
||||
|
Assert.assertTrue(relationService.deleteRelation(relationA).get()); |
||||
|
|
||||
|
Assert.assertFalse(relationService.checkRelation(parentId, childId, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get()); |
||||
|
|
||||
|
Assert.assertTrue(relationService.checkRelation(childId, subChildId, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get()); |
||||
|
|
||||
|
Assert.assertTrue(relationService.deleteRelation(childId, subChildId, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testDeleteEntityRelations() throws ExecutionException, InterruptedException { |
||||
|
AssetId parentId = new AssetId(UUIDs.timeBased()); |
||||
|
AssetId childId = new AssetId(UUIDs.timeBased()); |
||||
|
AssetId subChildId = new AssetId(UUIDs.timeBased()); |
||||
|
|
||||
|
EntityRelation relationA = new EntityRelation(parentId, childId, EntityRelation.CONTAINS_TYPE); |
||||
|
EntityRelation relationB = new EntityRelation(childId, subChildId, EntityRelation.CONTAINS_TYPE); |
||||
|
|
||||
|
saveRelation(relationA); |
||||
|
saveRelation(relationB); |
||||
|
|
||||
|
Assert.assertTrue(relationService.deleteEntityRelations(childId).get()); |
||||
|
|
||||
|
Assert.assertFalse(relationService.checkRelation(parentId, childId, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get()); |
||||
|
|
||||
|
Assert.assertFalse(relationService.checkRelation(childId, subChildId, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindFrom() throws ExecutionException, InterruptedException { |
||||
|
AssetId parentA = new AssetId(UUIDs.timeBased()); |
||||
|
AssetId parentB = new AssetId(UUIDs.timeBased()); |
||||
|
AssetId childA = new AssetId(UUIDs.timeBased()); |
||||
|
AssetId childB = new AssetId(UUIDs.timeBased()); |
||||
|
|
||||
|
EntityRelation relationA1 = new EntityRelation(parentA, childA, EntityRelation.CONTAINS_TYPE); |
||||
|
EntityRelation relationA2 = new EntityRelation(parentA, childB, EntityRelation.CONTAINS_TYPE); |
||||
|
|
||||
|
EntityRelation relationB1 = new EntityRelation(parentB, childA, EntityRelation.MANAGES_TYPE); |
||||
|
EntityRelation relationB2 = new EntityRelation(parentB, childB, EntityRelation.MANAGES_TYPE); |
||||
|
|
||||
|
saveRelation(relationA1); |
||||
|
saveRelation(relationA2); |
||||
|
|
||||
|
saveRelation(relationB1); |
||||
|
saveRelation(relationB2); |
||||
|
|
||||
|
List<EntityRelation> relations = relationService.findByFrom(parentA, RelationTypeGroup.COMMON).get(); |
||||
|
Assert.assertEquals(2, relations.size()); |
||||
|
for (EntityRelation relation : relations) { |
||||
|
Assert.assertEquals(EntityRelation.CONTAINS_TYPE, relation.getType()); |
||||
|
Assert.assertEquals(parentA, relation.getFrom()); |
||||
|
Assert.assertTrue(childA.equals(relation.getTo()) || childB.equals(relation.getTo())); |
||||
|
} |
||||
|
|
||||
|
relations = relationService.findByFromAndType(parentA, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get(); |
||||
|
Assert.assertEquals(2, relations.size()); |
||||
|
|
||||
|
relations = relationService.findByFromAndType(parentA, EntityRelation.MANAGES_TYPE, RelationTypeGroup.COMMON).get(); |
||||
|
Assert.assertEquals(0, relations.size()); |
||||
|
|
||||
|
relations = relationService.findByFrom(parentB, RelationTypeGroup.COMMON).get(); |
||||
|
Assert.assertEquals(2, relations.size()); |
||||
|
for (EntityRelation relation : relations) { |
||||
|
Assert.assertEquals(EntityRelation.MANAGES_TYPE, relation.getType()); |
||||
|
Assert.assertEquals(parentB, relation.getFrom()); |
||||
|
Assert.assertTrue(childA.equals(relation.getTo()) || childB.equals(relation.getTo())); |
||||
|
} |
||||
|
|
||||
|
relations = relationService.findByFromAndType(parentB, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get(); |
||||
|
Assert.assertEquals(0, relations.size()); |
||||
|
|
||||
|
relations = relationService.findByFromAndType(parentB, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get(); |
||||
|
Assert.assertEquals(0, relations.size()); |
||||
|
} |
||||
|
|
||||
|
private Boolean saveRelation(EntityRelation relationA1) throws ExecutionException, InterruptedException { |
||||
|
return relationService.saveRelation(relationA1).get(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testFindTo() throws ExecutionException, InterruptedException { |
||||
|
AssetId parentA = new AssetId(UUIDs.timeBased()); |
||||
|
AssetId parentB = new AssetId(UUIDs.timeBased()); |
||||
|
AssetId childA = new AssetId(UUIDs.timeBased()); |
||||
|
AssetId childB = new AssetId(UUIDs.timeBased()); |
||||
|
|
||||
|
EntityRelation relationA1 = new EntityRelation(parentA, childA, EntityRelation.CONTAINS_TYPE); |
||||
|
EntityRelation relationA2 = new EntityRelation(parentA, childB, EntityRelation.CONTAINS_TYPE); |
||||
|
|
||||
|
EntityRelation relationB1 = new EntityRelation(parentB, childA, EntityRelation.MANAGES_TYPE); |
||||
|
EntityRelation relationB2 = new EntityRelation(parentB, childB, EntityRelation.MANAGES_TYPE); |
||||
|
|
||||
|
saveRelation(relationA1); |
||||
|
saveRelation(relationA2); |
||||
|
|
||||
|
saveRelation(relationB1); |
||||
|
saveRelation(relationB2); |
||||
|
|
||||
|
// Data propagation to views is async
|
||||
|
Thread.sleep(3000); |
||||
|
|
||||
|
List<EntityRelation> relations = relationService.findByTo(childA, RelationTypeGroup.COMMON).get(); |
||||
|
Assert.assertEquals(2, relations.size()); |
||||
|
for (EntityRelation relation : relations) { |
||||
|
Assert.assertEquals(childA, relation.getTo()); |
||||
|
Assert.assertTrue(parentA.equals(relation.getFrom()) || parentB.equals(relation.getFrom())); |
||||
|
} |
||||
|
|
||||
|
relations = relationService.findByToAndType(childA, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get(); |
||||
|
Assert.assertEquals(1, relations.size()); |
||||
|
|
||||
|
relations = relationService.findByToAndType(childB, EntityRelation.MANAGES_TYPE, RelationTypeGroup.COMMON).get(); |
||||
|
Assert.assertEquals(1, relations.size()); |
||||
|
|
||||
|
relations = relationService.findByToAndType(parentA, EntityRelation.MANAGES_TYPE, RelationTypeGroup.COMMON).get(); |
||||
|
Assert.assertEquals(0, relations.size()); |
||||
|
|
||||
|
relations = relationService.findByToAndType(parentB, EntityRelation.MANAGES_TYPE, RelationTypeGroup.COMMON).get(); |
||||
|
Assert.assertEquals(0, relations.size()); |
||||
|
|
||||
|
relations = relationService.findByTo(childB, RelationTypeGroup.COMMON).get(); |
||||
|
Assert.assertEquals(2, relations.size()); |
||||
|
for (EntityRelation relation : relations) { |
||||
|
Assert.assertEquals(childB, relation.getTo()); |
||||
|
Assert.assertTrue(parentA.equals(relation.getFrom()) || parentB.equals(relation.getFrom())); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testCyclicRecursiveRelation() throws ExecutionException, InterruptedException { |
||||
|
// A -> B -> C -> A
|
||||
|
AssetId assetA = new AssetId(UUIDs.timeBased()); |
||||
|
AssetId assetB = new AssetId(UUIDs.timeBased()); |
||||
|
AssetId assetC = new AssetId(UUIDs.timeBased()); |
||||
|
|
||||
|
EntityRelation relationA = new EntityRelation(assetA, assetB, EntityRelation.CONTAINS_TYPE); |
||||
|
EntityRelation relationB = new EntityRelation(assetB, assetC, EntityRelation.CONTAINS_TYPE); |
||||
|
EntityRelation relationC = new EntityRelation(assetC, assetA, EntityRelation.CONTAINS_TYPE); |
||||
|
|
||||
|
saveRelation(relationA); |
||||
|
saveRelation(relationB); |
||||
|
saveRelation(relationC); |
||||
|
|
||||
|
EntityRelationsQuery query = new EntityRelationsQuery(); |
||||
|
query.setParameters(new RelationsSearchParameters(assetA, EntitySearchDirection.FROM, -1)); |
||||
|
query.setFilters(Collections.singletonList(new EntityTypeFilter(EntityRelation.CONTAINS_TYPE, Collections.singletonList(EntityType.ASSET)))); |
||||
|
List<EntityRelation> relations = relationService.findByQuery(query).get(); |
||||
|
Assert.assertEquals(3, relations.size()); |
||||
|
Assert.assertTrue(relations.contains(relationA)); |
||||
|
Assert.assertTrue(relations.contains(relationB)); |
||||
|
Assert.assertTrue(relations.contains(relationC)); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testRecursiveRelation() throws ExecutionException, InterruptedException { |
||||
|
// A -> B -> [C,D]
|
||||
|
AssetId assetA = new AssetId(UUIDs.timeBased()); |
||||
|
AssetId assetB = new AssetId(UUIDs.timeBased()); |
||||
|
AssetId assetC = new AssetId(UUIDs.timeBased()); |
||||
|
DeviceId deviceD = new DeviceId(UUIDs.timeBased()); |
||||
|
|
||||
|
EntityRelation relationAB = new EntityRelation(assetA, assetB, EntityRelation.CONTAINS_TYPE); |
||||
|
EntityRelation relationBC = new EntityRelation(assetB, assetC, EntityRelation.CONTAINS_TYPE); |
||||
|
EntityRelation relationBD = new EntityRelation(assetB, deviceD, EntityRelation.CONTAINS_TYPE); |
||||
|
|
||||
|
|
||||
|
saveRelation(relationAB); |
||||
|
saveRelation(relationBC); |
||||
|
saveRelation(relationBD); |
||||
|
|
||||
|
EntityRelationsQuery query = new EntityRelationsQuery(); |
||||
|
query.setParameters(new RelationsSearchParameters(assetA, EntitySearchDirection.FROM, -1)); |
||||
|
query.setFilters(Collections.singletonList(new EntityTypeFilter(EntityRelation.CONTAINS_TYPE, Collections.singletonList(EntityType.ASSET)))); |
||||
|
List<EntityRelation> relations = relationService.findByQuery(query).get(); |
||||
|
Assert.assertEquals(2, relations.size()); |
||||
|
Assert.assertTrue(relations.contains(relationAB)); |
||||
|
Assert.assertTrue(relations.contains(relationBC)); |
||||
|
} |
||||
|
|
||||
|
|
||||
|
@Test(expected = DataValidationException.class) |
||||
|
public void testSaveRelationWithEmptyFrom() throws ExecutionException, InterruptedException { |
||||
|
EntityRelation relation = new EntityRelation(); |
||||
|
relation.setTo(new AssetId(UUIDs.timeBased())); |
||||
|
relation.setType(EntityRelation.CONTAINS_TYPE); |
||||
|
Assert.assertTrue(saveRelation(relation)); |
||||
|
} |
||||
|
|
||||
|
@Test(expected = DataValidationException.class) |
||||
|
public void testSaveRelationWithEmptyTo() throws ExecutionException, InterruptedException { |
||||
|
EntityRelation relation = new EntityRelation(); |
||||
|
relation.setFrom(new AssetId(UUIDs.timeBased())); |
||||
|
relation.setType(EntityRelation.CONTAINS_TYPE); |
||||
|
Assert.assertTrue(saveRelation(relation)); |
||||
|
} |
||||
|
|
||||
|
@Test(expected = DataValidationException.class) |
||||
|
public void testSaveRelationWithEmptyType() throws ExecutionException, InterruptedException { |
||||
|
EntityRelation relation = new EntityRelation(); |
||||
|
relation.setFrom(new AssetId(UUIDs.timeBased())); |
||||
|
relation.setTo(new AssetId(UUIDs.timeBased())); |
||||
|
Assert.assertTrue(saveRelation(relation)); |
||||
|
} |
||||
|
} |
||||
@ -1 +1,7 @@ |
|||||
CASSANDRA_DATA_DIR=/home/docker/cassandra_volume |
CASSANDRA_DATA_DIR=/home/docker/cassandra_volume |
||||
|
|
||||
|
# cassandra schema container environment variables |
||||
|
CREATE_SCHEMA=true |
||||
|
ADD_SYSTEM_DATA=false |
||||
|
ADD_DEMO_DATA=false |
||||
|
CASSANDRA_URL=cassandra |
||||
@ -0,0 +1,10 @@ |
|||||
|
VERSION=1.2.4 |
||||
|
PROJECT=thingsboard |
||||
|
APP=cassandra |
||||
|
|
||||
|
build: |
||||
|
docker build --pull -t ${PROJECT}/${APP}:${VERSION} -t ${PROJECT}/${APP}:latest . |
||||
|
|
||||
|
push: build |
||||
|
docker push ${PROJECT}/${APP}:${VERSION} |
||||
|
docker push ${PROJECT}/${APP}:latest |
||||
@ -0,0 +1,132 @@ |
|||||
|
# |
||||
|
# Copyright © 2016-2017 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. |
||||
|
# |
||||
|
|
||||
|
apiVersion: v1 |
||||
|
kind: Service |
||||
|
metadata: |
||||
|
name: cassandra-headless |
||||
|
labels: |
||||
|
app: cassandra-headless |
||||
|
spec: |
||||
|
ports: |
||||
|
- port: 9042 |
||||
|
name: cql |
||||
|
clusterIP: None |
||||
|
selector: |
||||
|
app: cassandra |
||||
|
--- |
||||
|
apiVersion: "apps/v1beta1" |
||||
|
kind: StatefulSet |
||||
|
metadata: |
||||
|
name: cassandra |
||||
|
spec: |
||||
|
serviceName: cassandra-headless |
||||
|
replicas: 2 |
||||
|
template: |
||||
|
metadata: |
||||
|
labels: |
||||
|
app: cassandra |
||||
|
spec: |
||||
|
nodeSelector: |
||||
|
machinetype: other |
||||
|
affinity: |
||||
|
podAntiAffinity: |
||||
|
requiredDuringSchedulingIgnoredDuringExecution: |
||||
|
- labelSelector: |
||||
|
matchExpressions: |
||||
|
- key: "app" |
||||
|
operator: In |
||||
|
values: |
||||
|
- cassandra-headless |
||||
|
topologyKey: "kubernetes.io/hostname" |
||||
|
containers: |
||||
|
- name: cassandra |
||||
|
image: thingsboard/cassandra:1.2.4 |
||||
|
imagePullPolicy: Always |
||||
|
ports: |
||||
|
- containerPort: 7000 |
||||
|
name: intra-node |
||||
|
- containerPort: 7001 |
||||
|
name: tls-intra-node |
||||
|
- containerPort: 7199 |
||||
|
name: jmx |
||||
|
- containerPort: 9042 |
||||
|
name: cql |
||||
|
- containerPort: 9160 |
||||
|
name: thrift |
||||
|
securityContext: |
||||
|
capabilities: |
||||
|
add: |
||||
|
- IPC_LOCK |
||||
|
lifecycle: |
||||
|
preStop: |
||||
|
exec: |
||||
|
command: ["/bin/sh", "-c", "PID=$(pidof java) && kill $PID && while ps -p $PID > /dev/null; do sleep 1; done"] |
||||
|
env: |
||||
|
- name: MAX_HEAP_SIZE |
||||
|
value: 2048M |
||||
|
- name: HEAP_NEWSIZE |
||||
|
value: 100M |
||||
|
- name: CASSANDRA_SEEDS |
||||
|
value: "cassandra-0.cassandra-headless.default.svc.cluster.local" |
||||
|
- name: CASSANDRA_CLUSTER_NAME |
||||
|
value: "Thingsboard-Cluster" |
||||
|
- name: CASSANDRA_DC |
||||
|
value: "DC1-Thingsboard-Cluster" |
||||
|
- name: CASSANDRA_RACK |
||||
|
value: "Rack-Thingsboard-Cluster" |
||||
|
- name: CASSANDRA_AUTO_BOOTSTRAP |
||||
|
value: "false" |
||||
|
- name: POD_IP |
||||
|
valueFrom: |
||||
|
fieldRef: |
||||
|
fieldPath: status.podIP |
||||
|
- name: POD_NAMESPACE |
||||
|
valueFrom: |
||||
|
fieldRef: |
||||
|
fieldPath: metadata.namespace |
||||
|
readinessProbe: |
||||
|
exec: |
||||
|
command: |
||||
|
- /bin/bash |
||||
|
- -c |
||||
|
- /ready-probe.sh |
||||
|
initialDelaySeconds: 15 |
||||
|
timeoutSeconds: 5 |
||||
|
volumeMounts: |
||||
|
- name: cassandra-data |
||||
|
mountPath: /var/lib/cassandra/data |
||||
|
- name: cassandra-commitlog |
||||
|
mountPath: /var/lib/cassandra/commitlog |
||||
|
volumeClaimTemplates: |
||||
|
- metadata: |
||||
|
name: cassandra-data |
||||
|
annotations: |
||||
|
volume.beta.kubernetes.io/storage-class: fast |
||||
|
spec: |
||||
|
accessModes: [ "ReadWriteOnce" ] |
||||
|
resources: |
||||
|
requests: |
||||
|
storage: 3Gi |
||||
|
- metadata: |
||||
|
name: cassandra-commitlog |
||||
|
annotations: |
||||
|
volume.beta.kubernetes.io/storage-class: fast |
||||
|
spec: |
||||
|
accessModes: [ "ReadWriteOnce" ] |
||||
|
resources: |
||||
|
requests: |
||||
|
storage: 2Gi |
||||
@ -0,0 +1,14 @@ |
|||||
|
VERSION=1.2.4 |
||||
|
PROJECT=thingsboard |
||||
|
APP=tb-cassandra-schema |
||||
|
|
||||
|
build: |
||||
|
cp ../../dao/src/main/resources/schema.cql . |
||||
|
cp ../../dao/src/main/resources/demo-data.cql . |
||||
|
cp ../../dao/src/main/resources/system-data.cql . |
||||
|
docker build --pull -t ${PROJECT}/${APP}:${VERSION} -t ${PROJECT}/${APP}:latest . |
||||
|
rm schema.cql demo-data.cql system-data.cql |
||||
|
|
||||
|
push: build |
||||
|
docker push ${PROJECT}/${APP}:${VERSION} |
||||
|
docker push ${PROJECT}/${APP}:latest |
||||
@ -0,0 +1,39 @@ |
|||||
|
# |
||||
|
# Copyright © 2016-2017 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. |
||||
|
# |
||||
|
|
||||
|
apiVersion: v1 |
||||
|
kind: Pod |
||||
|
metadata: |
||||
|
name: tb-cassandra-schema |
||||
|
spec: |
||||
|
containers: |
||||
|
- name: tb-cassandra-schema |
||||
|
imagePullPolicy: Always |
||||
|
image: thingsboard/tb-cassandra-schema:1.2.4 |
||||
|
env: |
||||
|
- name: CREATE_SCHEMA |
||||
|
value: "true" |
||||
|
- name: ADD_SYSTEM_DATA |
||||
|
value: "true" |
||||
|
- name : ADD_DEMO_DATA |
||||
|
value: "true" |
||||
|
- name : CASSANDRA_URL |
||||
|
value: "cassandra-headless" |
||||
|
command: |
||||
|
- sh |
||||
|
- -c |
||||
|
- ./install-schema.sh |
||||
|
restartPolicy: Never |
||||
@ -1,6 +1,7 @@ |
|||||
#Thingsboard server configuration |
#Thingsboard server configuration |
||||
|
|
||||
CASSANDRA_URL=db:9042 |
TB_CASSANDRA_SCHEMA_URL=tb-cassandra-schema |
||||
|
CASSANDRA_URL=cassandra:9042 |
||||
ZOOKEEPER_URL=zk:2181 |
ZOOKEEPER_URL=zk:2181 |
||||
MQTT_BIND_ADDRESS=0.0.0.0 |
MQTT_BIND_ADDRESS=0.0.0.0 |
||||
MQTT_BIND_PORT=1883 |
MQTT_BIND_PORT=1883 |
||||
@ -0,0 +1,12 @@ |
|||||
|
VERSION=1.2.4 |
||||
|
PROJECT=thingsboard |
||||
|
APP=application |
||||
|
|
||||
|
build: |
||||
|
cp ../../application/target/thingsboard.deb . |
||||
|
docker build --pull -t ${PROJECT}/${APP}:${VERSION} -t ${PROJECT}/${APP}:latest . |
||||
|
rm thingsboard.deb |
||||
|
|
||||
|
push: build |
||||
|
docker push ${PROJECT}/${APP}:${VERSION} |
||||
|
docker push ${PROJECT}/${APP}:latest |
||||
@ -0,0 +1,121 @@ |
|||||
|
# |
||||
|
# Copyright © 2016-2017 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. |
||||
|
# |
||||
|
|
||||
|
--- |
||||
|
apiVersion: v1 |
||||
|
kind: Service |
||||
|
metadata: |
||||
|
name: tb-service |
||||
|
labels: |
||||
|
app: tb-service |
||||
|
spec: |
||||
|
ports: |
||||
|
- port: 8080 |
||||
|
name: ui |
||||
|
- port: 1883 |
||||
|
name: mqtt |
||||
|
- port: 5683 |
||||
|
name: coap |
||||
|
selector: |
||||
|
app: tb |
||||
|
type: LoadBalancer |
||||
|
--- |
||||
|
apiVersion: policy/v1beta1 |
||||
|
kind: PodDisruptionBudget |
||||
|
metadata: |
||||
|
name: tb-budget |
||||
|
spec: |
||||
|
selector: |
||||
|
matchLabels: |
||||
|
app: tb |
||||
|
minAvailable: 3 |
||||
|
--- |
||||
|
apiVersion: v1 |
||||
|
kind: ConfigMap |
||||
|
metadata: |
||||
|
name: tb-config |
||||
|
data: |
||||
|
zookeeper.enabled: "true" |
||||
|
zookeeper.url: "zk-headless" |
||||
|
cassandra.url: "cassandra-headless:9042" |
||||
|
--- |
||||
|
apiVersion: apps/v1beta1 |
||||
|
kind: StatefulSet |
||||
|
metadata: |
||||
|
name: tb |
||||
|
spec: |
||||
|
serviceName: "tb-service" |
||||
|
replicas: 3 |
||||
|
template: |
||||
|
metadata: |
||||
|
labels: |
||||
|
app: tb |
||||
|
spec: |
||||
|
nodeSelector: |
||||
|
machinetype: tb |
||||
|
affinity: |
||||
|
podAntiAffinity: |
||||
|
requiredDuringSchedulingIgnoredDuringExecution: |
||||
|
- labelSelector: |
||||
|
matchExpressions: |
||||
|
- key: "app" |
||||
|
operator: In |
||||
|
values: |
||||
|
- tb-service |
||||
|
topologyKey: "kubernetes.io/hostname" |
||||
|
containers: |
||||
|
- name: tb |
||||
|
imagePullPolicy: Always |
||||
|
image: thingsboard/application:1.2.4 |
||||
|
ports: |
||||
|
- containerPort: 8080 |
||||
|
name: ui |
||||
|
- containerPort: 1883 |
||||
|
name: mqtt |
||||
|
- containerPort: 5683 |
||||
|
name: coap |
||||
|
- containerPort: 9001 |
||||
|
name: rpc |
||||
|
env: |
||||
|
- name: ZOOKEEPER_ENABLED |
||||
|
valueFrom: |
||||
|
configMapKeyRef: |
||||
|
name: tb-config |
||||
|
key: zookeeper.enabled |
||||
|
- name: ZOOKEEPER_URL |
||||
|
valueFrom: |
||||
|
configMapKeyRef: |
||||
|
name: tb-config |
||||
|
key: zookeeper.url |
||||
|
- name : CASSANDRA_URL |
||||
|
valueFrom: |
||||
|
configMapKeyRef: |
||||
|
name: tb-config |
||||
|
key: cassandra.url |
||||
|
- name : RPC_HOST |
||||
|
valueFrom: |
||||
|
fieldRef: |
||||
|
fieldPath: status.podIP |
||||
|
command: |
||||
|
- sh |
||||
|
- -c |
||||
|
- ./run-application.sh |
||||
|
livenessProbe: |
||||
|
httpGet: |
||||
|
path: /login |
||||
|
port: ui-port |
||||
|
initialDelaySeconds: 120 |
||||
|
timeoutSeconds: 10 |
||||
@ -1,5 +0,0 @@ |
|||||
#Db schema configuration |
|
||||
|
|
||||
SKIP_SCHEMA_CREATION=false |
|
||||
SKIP_SYSTEM_DATA=false |
|
||||
SKIP_DEMO_DATA=false |
|
||||
@ -1,28 +0,0 @@ |
|||||
#!/bin/bash |
|
||||
# |
|
||||
# Copyright © 2016-2017 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. |
|
||||
# |
|
||||
|
|
||||
|
|
||||
cp ../../dao/src/main/resources/schema.cql schema.cql |
|
||||
cp ../../dao/src/main/resources/demo-data.cql demo-data.cql |
|
||||
cp ../../dao/src/main/resources/system-data.cql system-data.cql |
|
||||
|
|
||||
docker build -t thingsboard/thingsboard-db-schema:1.2.3 -t thingsboard/thingsboard-db-schema:latest . |
|
||||
|
|
||||
docker login |
|
||||
|
|
||||
docker push thingsboard/thingsboard-db-schema:1.2.3 |
|
||||
docker push thingsboard/thingsboard-db-schema:latest |
|
||||
@ -0,0 +1,71 @@ |
|||||
|
# |
||||
|
# Copyright © 2016-2017 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. |
||||
|
# |
||||
|
|
||||
|
FROM ubuntu:16.04 |
||||
|
ENV ZK_USER=zookeeper \ |
||||
|
ZK_DATA_DIR=/var/lib/zookeeper/data \ |
||||
|
ZK_DATA_LOG_DIR=/var/lib/zookeeper/log \ |
||||
|
ZK_LOG_DIR=/var/log/zookeeper \ |
||||
|
JAVA_HOME=/usr/lib/jvm/java-8-openjdk-amd64 |
||||
|
|
||||
|
ARG GPG_KEY=C823E3E5B12AF29C67F81976F5CECB3CB5E9BD2D |
||||
|
ARG ZK_DIST=zookeeper-3.4.9 |
||||
|
RUN set -x \ |
||||
|
&& apt-get update \ |
||||
|
&& apt-get install -y openjdk-8-jre-headless wget netcat-openbsd \ |
||||
|
&& wget -q "http://www.apache.org/dist/zookeeper/$ZK_DIST/$ZK_DIST.tar.gz" \ |
||||
|
&& wget -q "http://www.apache.org/dist/zookeeper/$ZK_DIST/$ZK_DIST.tar.gz.asc" \ |
||||
|
&& export GNUPGHOME="$(mktemp -d)" \ |
||||
|
&& gpg --keyserver ha.pool.sks-keyservers.net --recv-key "$GPG_KEY" \ |
||||
|
&& gpg --batch --verify "$ZK_DIST.tar.gz.asc" "$ZK_DIST.tar.gz" \ |
||||
|
&& tar -xzf "$ZK_DIST.tar.gz" -C /opt \ |
||||
|
&& rm -r "$GNUPGHOME" "$ZK_DIST.tar.gz" "$ZK_DIST.tar.gz.asc" \ |
||||
|
&& ln -s /opt/$ZK_DIST /opt/zookeeper \ |
||||
|
&& rm -rf /opt/zookeeper/CHANGES.txt \ |
||||
|
/opt/zookeeper/README.txt \ |
||||
|
/opt/zookeeper/NOTICE.txt \ |
||||
|
/opt/zookeeper/CHANGES.txt \ |
||||
|
/opt/zookeeper/README_packaging.txt \ |
||||
|
/opt/zookeeper/build.xml \ |
||||
|
/opt/zookeeper/config \ |
||||
|
/opt/zookeeper/contrib \ |
||||
|
/opt/zookeeper/dist-maven \ |
||||
|
/opt/zookeeper/docs \ |
||||
|
/opt/zookeeper/ivy.xml \ |
||||
|
/opt/zookeeper/ivysettings.xml \ |
||||
|
/opt/zookeeper/recipes \ |
||||
|
/opt/zookeeper/src \ |
||||
|
/opt/zookeeper/$ZK_DIST.jar.asc \ |
||||
|
/opt/zookeeper/$ZK_DIST.jar.md5 \ |
||||
|
/opt/zookeeper/$ZK_DIST.jar.sha1 \ |
||||
|
&& apt-get autoremove -y wget \ |
||||
|
&& rm -rf /var/lib/apt/lists/* |
||||
|
|
||||
|
#Copy configuration generator script to bin |
||||
|
COPY zk-gen-config.sh zk-ok.sh /opt/zookeeper/bin/ |
||||
|
|
||||
|
# Create a user for the zookeeper process and configure file system ownership |
||||
|
# for nessecary directories and symlink the distribution as a user executable |
||||
|
RUN set -x \ |
||||
|
&& useradd $ZK_USER \ |
||||
|
&& [ `id -u $ZK_USER` -eq 1000 ] \ |
||||
|
&& [ `id -g $ZK_USER` -eq 1000 ] \ |
||||
|
&& mkdir -p $ZK_DATA_DIR $ZK_DATA_LOG_DIR $ZK_LOG_DIR /usr/share/zookeeper /tmp/zookeeper /usr/etc/ \ |
||||
|
&& chown -R "$ZK_USER:$ZK_USER" /opt/$ZK_DIST $ZK_DATA_DIR $ZK_LOG_DIR $ZK_DATA_LOG_DIR /tmp/zookeeper \ |
||||
|
&& ln -s /opt/zookeeper/conf/ /usr/etc/zookeeper \ |
||||
|
&& ln -s /opt/zookeeper/bin/* /usr/bin \ |
||||
|
&& ln -s /opt/zookeeper/$ZK_DIST.jar /usr/share/zookeeper/ \ |
||||
|
&& ln -s /opt/zookeeper/lib/* /usr/share/zookeeper |
||||
@ -0,0 +1,10 @@ |
|||||
|
VERSION=1.2.4 |
||||
|
PROJECT=thingsboard |
||||
|
APP=zk |
||||
|
|
||||
|
build: |
||||
|
docker build --pull -t ${PROJECT}/${APP}:${VERSION} -t ${PROJECT}/${APP}:latest . |
||||
|
|
||||
|
push: build |
||||
|
docker push ${PROJECT}/${APP}:${VERSION} |
||||
|
docker push ${PROJECT}/${APP}:latest |
||||
@ -0,0 +1,153 @@ |
|||||
|
#!/usr/bin/env bash |
||||
|
# |
||||
|
# Copyright © 2016-2017 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. |
||||
|
# |
||||
|
|
||||
|
ZK_USER=${ZK_USER:-"zookeeper"} |
||||
|
ZK_LOG_LEVEL=${ZK_LOG_LEVEL:-"INFO"} |
||||
|
ZK_DATA_DIR=${ZK_DATA_DIR:-"/var/lib/zookeeper/data"} |
||||
|
ZK_DATA_LOG_DIR=${ZK_DATA_LOG_DIR:-"/var/lib/zookeeper/log"} |
||||
|
ZK_LOG_DIR=${ZK_LOG_DIR:-"var/log/zookeeper"} |
||||
|
ZK_CONF_DIR=${ZK_CONF_DIR:-"/opt/zookeeper/conf"} |
||||
|
ZK_CLIENT_PORT=${ZK_CLIENT_PORT:-2181} |
||||
|
ZK_SERVER_PORT=${ZK_SERVER_PORT:-2888} |
||||
|
ZK_ELECTION_PORT=${ZK_ELECTION_PORT:-3888} |
||||
|
ZK_TICK_TIME=${ZK_TICK_TIME:-2000} |
||||
|
ZK_INIT_LIMIT=${ZK_INIT_LIMIT:-10} |
||||
|
ZK_SYNC_LIMIT=${ZK_SYNC_LIMIT:-5} |
||||
|
ZK_HEAP_SIZE=${ZK_HEAP_SIZE:-2G} |
||||
|
ZK_MAX_CLIENT_CNXNS=${ZK_MAX_CLIENT_CNXNS:-60} |
||||
|
ZK_MIN_SESSION_TIMEOUT=${ZK_MIN_SESSION_TIMEOUT:- $((ZK_TICK_TIME*2))} |
||||
|
ZK_MAX_SESSION_TIMEOUT=${ZK_MAX_SESSION_TIMEOUT:- $((ZK_TICK_TIME*20))} |
||||
|
ZK_SNAP_RETAIN_COUNT=${ZK_SNAP_RETAIN_COUNT:-3} |
||||
|
ZK_PURGE_INTERVAL=${ZK_PURGE_INTERVAL:-0} |
||||
|
ID_FILE="$ZK_DATA_DIR/myid" |
||||
|
ZK_CONFIG_FILE="$ZK_CONF_DIR/zoo.cfg" |
||||
|
LOGGER_PROPS_FILE="$ZK_CONF_DIR/log4j.properties" |
||||
|
JAVA_ENV_FILE="$ZK_CONF_DIR/java.env" |
||||
|
HOST=`hostname -s` |
||||
|
DOMAIN=`hostname -d` |
||||
|
|
||||
|
function print_servers() { |
||||
|
for (( i=1; i<=$ZK_REPLICAS; i++ )) |
||||
|
do |
||||
|
echo "server.$i=$NAME-$((i-1)).$DOMAIN:$ZK_SERVER_PORT:$ZK_ELECTION_PORT" |
||||
|
done |
||||
|
} |
||||
|
|
||||
|
function validate_env() { |
||||
|
echo "Validating environment" |
||||
|
if [ -z $ZK_REPLICAS ]; then |
||||
|
echo "ZK_REPLICAS is a mandatory environment variable" |
||||
|
exit 1 |
||||
|
fi |
||||
|
|
||||
|
if [[ $HOST =~ (.*)-([0-9]+)$ ]]; then |
||||
|
NAME=${BASH_REMATCH[1]} |
||||
|
ORD=${BASH_REMATCH[2]} |
||||
|
else |
||||
|
echo "Failed to extract ordinal from hostname $HOST" |
||||
|
exit 1 |
||||
|
fi |
||||
|
MY_ID=$((ORD+1)) |
||||
|
echo "ZK_REPLICAS=$ZK_REPLICAS" |
||||
|
echo "MY_ID=$MY_ID" |
||||
|
echo "ZK_LOG_LEVEL=$ZK_LOG_LEVEL" |
||||
|
echo "ZK_DATA_DIR=$ZK_DATA_DIR" |
||||
|
echo "ZK_DATA_LOG_DIR=$ZK_DATA_LOG_DIR" |
||||
|
echo "ZK_LOG_DIR=$ZK_LOG_DIR" |
||||
|
echo "ZK_CLIENT_PORT=$ZK_CLIENT_PORT" |
||||
|
echo "ZK_SERVER_PORT=$ZK_SERVER_PORT" |
||||
|
echo "ZK_ELECTION_PORT=$ZK_ELECTION_PORT" |
||||
|
echo "ZK_TICK_TIME=$ZK_TICK_TIME" |
||||
|
echo "ZK_INIT_LIMIT=$ZK_INIT_LIMIT" |
||||
|
echo "ZK_SYNC_LIMIT=$ZK_SYNC_LIMIT" |
||||
|
echo "ZK_MAX_CLIENT_CNXNS=$ZK_MAX_CLIENT_CNXNS" |
||||
|
echo "ZK_MIN_SESSION_TIMEOUT=$ZK_MIN_SESSION_TIMEOUT" |
||||
|
echo "ZK_MAX_SESSION_TIMEOUT=$ZK_MAX_SESSION_TIMEOUT" |
||||
|
echo "ZK_HEAP_SIZE=$ZK_HEAP_SIZE" |
||||
|
echo "ZK_SNAP_RETAIN_COUNT=$ZK_SNAP_RETAIN_COUNT" |
||||
|
echo "ZK_PURGE_INTERVAL=$ZK_PURGE_INTERVAL" |
||||
|
echo "ENSEMBLE" |
||||
|
print_servers |
||||
|
echo "Environment validation successful" |
||||
|
} |
||||
|
|
||||
|
function create_config() { |
||||
|
rm -f $ZK_CONFIG_FILE |
||||
|
echo "Creating ZooKeeper configuration" |
||||
|
echo "#This file was autogenerated by zk DO NOT EDIT" >> $ZK_CONFIG_FILE |
||||
|
echo "clientPort=$ZK_CLIENT_PORT" >> $ZK_CONFIG_FILE |
||||
|
echo "dataDir=$ZK_DATA_DIR" >> $ZK_CONFIG_FILE |
||||
|
echo "dataLogDir=$ZK_DATA_LOG_DIR" >> $ZK_CONFIG_FILE |
||||
|
echo "tickTime=$ZK_TICK_TIME" >> $ZK_CONFIG_FILE |
||||
|
echo "initLimit=$ZK_INIT_LIMIT" >> $ZK_CONFIG_FILE |
||||
|
echo "syncLimit=$ZK_SYNC_LIMIT" >> $ZK_CONFIG_FILE |
||||
|
echo "maxClientCnxns=$ZK_MAX_CLIENT_CNXNS" >> $ZK_CONFIG_FILE |
||||
|
echo "minSessionTimeout=$ZK_MIN_SESSION_TIMEOUT" >> $ZK_CONFIG_FILE |
||||
|
echo "maxSessionTimeout=$ZK_MAX_SESSION_TIMEOUT" >> $ZK_CONFIG_FILE |
||||
|
echo "autopurge.snapRetainCount=$ZK_SNAP_RETAIN_COUNT" >> $ZK_CONFIG_FILE |
||||
|
echo "autopurge.purgeInteval=$ZK_PURGE_INTERVAL" >> $ZK_CONFIG_FILE |
||||
|
|
||||
|
if [ $ZK_REPLICAS -gt 1 ]; then |
||||
|
print_servers >> $ZK_CONFIG_FILE |
||||
|
fi |
||||
|
echo "Wrote ZooKeeper configuration file to $ZK_CONFIG_FILE" |
||||
|
} |
||||
|
|
||||
|
function create_data_dirs() { |
||||
|
echo "Creating ZooKeeper data directories and setting permissions" |
||||
|
if [ ! -d $ZK_DATA_DIR ]; then |
||||
|
mkdir -p $ZK_DATA_DIR |
||||
|
chown -R $ZK_USER:$ZK_USER $ZK_DATA_DIR |
||||
|
fi |
||||
|
|
||||
|
if [ ! -d $ZK_DATA_LOG_DIR ]; then |
||||
|
mkdir -p $ZK_DATA_LOG_DIR |
||||
|
chown -R $ZK_USER:$ZK_USER $ZK_DATA_LOG_DIR |
||||
|
fi |
||||
|
|
||||
|
if [ ! -d $ZK_LOG_DIR ]; then |
||||
|
mkdir -p $ZK_LOG_DIR |
||||
|
chown -R $ZK_USER:$ZK_USER $ZK_LOG_DIR |
||||
|
fi |
||||
|
if [ ! -f $ID_FILE ]; then |
||||
|
echo $MY_ID >> $ID_FILE |
||||
|
fi |
||||
|
echo "Created ZooKeeper data directories and set permissions in $ZK_DATA_DIR" |
||||
|
} |
||||
|
|
||||
|
function create_log_props () { |
||||
|
rm -f $LOGGER_PROPS_FILE |
||||
|
echo "Creating ZooKeeper log4j configuration" |
||||
|
echo "zookeeper.root.logger=CONSOLE" >> $LOGGER_PROPS_FILE |
||||
|
echo "zookeeper.console.threshold="$ZK_LOG_LEVEL >> $LOGGER_PROPS_FILE |
||||
|
echo "log4j.rootLogger=\${zookeeper.root.logger}" >> $LOGGER_PROPS_FILE |
||||
|
echo "log4j.appender.CONSOLE=org.apache.log4j.ConsoleAppender" >> $LOGGER_PROPS_FILE |
||||
|
echo "log4j.appender.CONSOLE.Threshold=\${zookeeper.console.threshold}" >> $LOGGER_PROPS_FILE |
||||
|
echo "log4j.appender.CONSOLE.layout=org.apache.log4j.PatternLayout" >> $LOGGER_PROPS_FILE |
||||
|
echo "log4j.appender.CONSOLE.layout.ConversionPattern=%d{ISO8601} [myid:%X{myid}] - %-5p [%t:%C{1}@%L] - %m%n" >> $LOGGER_PROPS_FILE |
||||
|
echo "Wrote log4j configuration to $LOGGER_PROPS_FILE" |
||||
|
} |
||||
|
|
||||
|
function create_java_env() { |
||||
|
rm -f $JAVA_ENV_FILE |
||||
|
echo "Creating JVM configuration file" |
||||
|
echo "ZOO_LOG_DIR=$ZK_LOG_DIR" >> $JAVA_ENV_FILE |
||||
|
echo "JVMFLAGS=\"-Xmx$ZK_HEAP_SIZE -Xms$ZK_HEAP_SIZE\"" >> $JAVA_ENV_FILE |
||||
|
echo "Wrote JVM configuration to $JAVA_ENV_FILE" |
||||
|
} |
||||
|
|
||||
|
validate_env && create_config && create_log_props && create_data_dirs && create_java_env |
||||
Some files were not shown because too many files changed in this diff
Loading…
Reference in new issue