170 changed files with 5763 additions and 3232 deletions
@ -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,59 @@ |
|||
/** |
|||
* 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 toName; |
|||
|
|||
public EntityRelationInfo() { |
|||
super(); |
|||
} |
|||
|
|||
public EntityRelationInfo(EntityRelation entityRelation) { |
|||
super(entityRelation); |
|||
} |
|||
|
|||
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,66 @@ |
|||
/** |
|||
* 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.util.concurrent.ListenableFuture; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.springframework.stereotype.Service; |
|||
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.page.TimePageData; |
|||
|
|||
import java.util.Optional; |
|||
|
|||
@Service |
|||
@Slf4j |
|||
public class BaseAlarmService implements AlarmService { |
|||
|
|||
@Override |
|||
public Alarm findAlarmById(AlarmId alarmId) { |
|||
return null; |
|||
} |
|||
|
|||
@Override |
|||
public ListenableFuture<Alarm> findAlarmByIdAsync(AlarmId alarmId) { |
|||
return null; |
|||
} |
|||
|
|||
@Override |
|||
public Optional<Alarm> saveIfNotExists(Alarm alarm) { |
|||
return null; |
|||
} |
|||
|
|||
@Override |
|||
public ListenableFuture<Boolean> updateAlarm(Alarm alarm) { |
|||
return null; |
|||
} |
|||
|
|||
@Override |
|||
public ListenableFuture<Boolean> ackAlarm(Alarm alarm) { |
|||
return null; |
|||
} |
|||
|
|||
@Override |
|||
public ListenableFuture<Boolean> clearAlarm(AlarmId alarmId) { |
|||
return null; |
|||
} |
|||
|
|||
@Override |
|||
public ListenableFuture<TimePageData<Alarm>> findAlarms(AlarmQuery query) { |
|||
return null; |
|||
} |
|||
} |
|||
@ -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,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); |
|||
} |
|||
|
|||
} |
|||
@ -1 +1,7 @@ |
|||
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,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: cassandra:3.9 |
|||
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,13 @@ |
|||
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} . |
|||
rm schema.cql demo-data.cql system-data.cql |
|||
|
|||
push: build |
|||
docker push ${PROJECT}/${APP}:${VERSION} |
|||
@ -1,6 +1,7 @@ |
|||
#Thingsboard server configuration |
|||
|
|||
CASSANDRA_URL=db:9042 |
|||
TB_CASSANDRA_SCHEMA_URL=tb-cassandra-schema |
|||
CASSANDRA_URL=cassandra:9042 |
|||
ZOOKEEPER_URL=zk:2181 |
|||
MQTT_BIND_ADDRESS=0.0.0.0 |
|||
MQTT_BIND_PORT=1883 |
|||
@ -0,0 +1,11 @@ |
|||
VERSION=1.2.4 |
|||
PROJECT=thingsboard |
|||
APP=application |
|||
|
|||
build: |
|||
cp ../../application/target/thingsboard.deb . |
|||
docker build --pull -t ${PROJECT}/${APP}:${VERSION} . |
|||
rm thingsboard.deb |
|||
|
|||
push: build |
|||
docker push ${PROJECT}/${APP}:${VERSION} |
|||
@ -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 |
|||
@ -1,26 +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 ../../application/target/thingsboard.deb thingsboard.deb |
|||
|
|||
docker build -t thingsboard/application:1.2.3 -t thingsboard/application:latest . |
|||
|
|||
docker login |
|||
|
|||
docker push thingsboard/application:1.2.3 |
|||
docker push thingsboard/application: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,9 @@ |
|||
VERSION=1.2.4 |
|||
PROJECT=thingsboard |
|||
APP=zk |
|||
|
|||
build: |
|||
docker build --pull -t ${PROJECT}/${APP}:${VERSION} . |
|||
|
|||
push: build |
|||
docker push ${PROJECT}/${APP}:${VERSION} |
|||
@ -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 |
|||
@ -0,0 +1,190 @@ |
|||
# |
|||
# 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: zk-headless |
|||
labels: |
|||
app: zk-headless |
|||
spec: |
|||
ports: |
|||
- port: 2888 |
|||
name: server |
|||
- port: 3888 |
|||
name: leader-election |
|||
clusterIP: None |
|||
selector: |
|||
app: zk |
|||
--- |
|||
apiVersion: v1 |
|||
kind: ConfigMap |
|||
metadata: |
|||
name: zk-config |
|||
data: |
|||
ensemble: "zk-0;zk-1;zk-2" |
|||
replicas: "3" |
|||
jvm.heap: "500m" |
|||
tick: "2000" |
|||
init: "10" |
|||
sync: "5" |
|||
client.cnxns: "60" |
|||
snap.retain: "3" |
|||
purge.interval: "1" |
|||
client.port: "2181" |
|||
server.port: "2888" |
|||
election.port: "3888" |
|||
--- |
|||
apiVersion: policy/v1beta1 |
|||
kind: PodDisruptionBudget |
|||
metadata: |
|||
name: zk-budget |
|||
spec: |
|||
selector: |
|||
matchLabels: |
|||
app: zk |
|||
minAvailable: 3 |
|||
--- |
|||
apiVersion: apps/v1beta1 |
|||
kind: StatefulSet |
|||
metadata: |
|||
name: zk |
|||
spec: |
|||
serviceName: zk-headless |
|||
replicas: 3 |
|||
template: |
|||
metadata: |
|||
labels: |
|||
app: zk |
|||
annotations: |
|||
pod.alpha.kubernetes.io/initialized: "true" |
|||
spec: |
|||
nodeSelector: |
|||
machinetype: other |
|||
affinity: |
|||
podAntiAffinity: |
|||
requiredDuringSchedulingIgnoredDuringExecution: |
|||
- labelSelector: |
|||
matchExpressions: |
|||
- key: "app" |
|||
operator: In |
|||
values: |
|||
- zk-headless |
|||
topologyKey: "kubernetes.io/hostname" |
|||
containers: |
|||
- name: zk |
|||
imagePullPolicy: Always |
|||
image: thingsboard/zk:1.2.4 |
|||
ports: |
|||
- containerPort: 2181 |
|||
name: client |
|||
- containerPort: 2888 |
|||
name: server |
|||
- containerPort: 3888 |
|||
name: leader-election |
|||
env: |
|||
- name : ZK_ENSEMBLE |
|||
valueFrom: |
|||
configMapKeyRef: |
|||
name: zk-config |
|||
key: ensemble |
|||
- name : ZK_REPLICAS |
|||
valueFrom: |
|||
configMapKeyRef: |
|||
name: zk-config |
|||
key: replicas |
|||
- name : ZK_HEAP_SIZE |
|||
valueFrom: |
|||
configMapKeyRef: |
|||
name: zk-config |
|||
key: jvm.heap |
|||
- name : ZK_TICK_TIME |
|||
valueFrom: |
|||
configMapKeyRef: |
|||
name: zk-config |
|||
key: tick |
|||
- name : ZK_INIT_LIMIT |
|||
valueFrom: |
|||
configMapKeyRef: |
|||
name: zk-config |
|||
key: init |
|||
- name : ZK_SYNC_LIMIT |
|||
valueFrom: |
|||
configMapKeyRef: |
|||
name: zk-config |
|||
key: tick |
|||
- name : ZK_MAX_CLIENT_CNXNS |
|||
valueFrom: |
|||
configMapKeyRef: |
|||
name: zk-config |
|||
key: client.cnxns |
|||
- name: ZK_SNAP_RETAIN_COUNT |
|||
valueFrom: |
|||
configMapKeyRef: |
|||
name: zk-config |
|||
key: snap.retain |
|||
- name: ZK_PURGE_INTERVAL |
|||
valueFrom: |
|||
configMapKeyRef: |
|||
name: zk-config |
|||
key: purge.interval |
|||
- name: ZK_CLIENT_PORT |
|||
valueFrom: |
|||
configMapKeyRef: |
|||
name: zk-config |
|||
key: client.port |
|||
- name: ZK_SERVER_PORT |
|||
valueFrom: |
|||
configMapKeyRef: |
|||
name: zk-config |
|||
key: server.port |
|||
- name: ZK_ELECTION_PORT |
|||
valueFrom: |
|||
configMapKeyRef: |
|||
name: zk-config |
|||
key: election.port |
|||
command: |
|||
- sh |
|||
- -c |
|||
- zk-gen-config.sh && zkServer.sh start-foreground |
|||
readinessProbe: |
|||
exec: |
|||
command: |
|||
- "zk-ok.sh" |
|||
initialDelaySeconds: 15 |
|||
timeoutSeconds: 5 |
|||
livenessProbe: |
|||
exec: |
|||
command: |
|||
- "zk-ok.sh" |
|||
initialDelaySeconds: 15 |
|||
timeoutSeconds: 5 |
|||
volumeMounts: |
|||
- name: zkdatadir |
|||
mountPath: /var/lib/zookeeper |
|||
securityContext: |
|||
runAsUser: 1000 |
|||
fsGroup: 1000 |
|||
volumeClaimTemplates: |
|||
- metadata: |
|||
name: zkdatadir |
|||
annotations: |
|||
volume.beta.kubernetes.io/storage-class: slow |
|||
spec: |
|||
accessModes: [ "ReadWriteOnce" ] |
|||
resources: |
|||
requests: |
|||
storage: 1Gi |
|||
@ -1,248 +0,0 @@ |
|||
/* |
|||
* 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. |
|||
*/ |
|||
import './datasource-device.scss'; |
|||
|
|||
import 'md-color-picker'; |
|||
import tinycolor from 'tinycolor2'; |
|||
import $ from 'jquery'; |
|||
import thingsboardTypes from '../common/types.constant'; |
|||
import thingsboardDatakeyConfigDialog from './datakey-config-dialog.controller'; |
|||
import thingsboardTruncate from './truncate.filter'; |
|||
|
|||
/* eslint-disable import/no-unresolved, import/default */ |
|||
|
|||
import datasourceDeviceTemplate from './datasource-device.tpl.html'; |
|||
import datakeyConfigDialogTemplate from './datakey-config-dialog.tpl.html'; |
|||
|
|||
/* eslint-enable import/no-unresolved, import/default */ |
|||
|
|||
/* eslint-disable angular/angularelement */ |
|||
|
|||
export default angular.module('thingsboard.directives.datasourceDevice', [thingsboardTruncate, thingsboardTypes, thingsboardDatakeyConfigDialog]) |
|||
.directive('tbDatasourceDevice', DatasourceDevice) |
|||
.name; |
|||
|
|||
/*@ngInject*/ |
|||
function DatasourceDevice($compile, $templateCache, $q, $mdDialog, $window, $document, $mdColorPicker, $mdConstant, types) { |
|||
|
|||
var linker = function (scope, element, attrs, ngModelCtrl) { |
|||
var template = $templateCache.get(datasourceDeviceTemplate); |
|||
element.html(template); |
|||
|
|||
scope.ngModelCtrl = ngModelCtrl; |
|||
scope.types = types; |
|||
|
|||
scope.selectedTimeseriesDataKey = null; |
|||
scope.timeseriesDataKeySearchText = null; |
|||
|
|||
scope.selectedAttributeDataKey = null; |
|||
scope.attributeDataKeySearchText = null; |
|||
|
|||
scope.updateValidity = function () { |
|||
if (ngModelCtrl.$viewValue) { |
|||
var value = ngModelCtrl.$viewValue; |
|||
var dataValid = angular.isDefined(value) && value != null; |
|||
ngModelCtrl.$setValidity('deviceData', dataValid); |
|||
if (dataValid) { |
|||
ngModelCtrl.$setValidity('deviceAlias', |
|||
angular.isDefined(value.deviceAliasId) && |
|||
value.deviceAliasId != null); |
|||
ngModelCtrl.$setValidity('deviceKeys', |
|||
angular.isDefined(value.dataKeys) && |
|||
value.dataKeys != null && |
|||
value.dataKeys.length > 0); |
|||
} |
|||
} |
|||
}; |
|||
|
|||
scope.$watch('deviceAlias', function () { |
|||
if (ngModelCtrl.$viewValue) { |
|||
if (scope.deviceAlias) { |
|||
ngModelCtrl.$viewValue.deviceAliasId = scope.deviceAlias.id; |
|||
} else { |
|||
ngModelCtrl.$viewValue.deviceAliasId = null; |
|||
} |
|||
scope.updateValidity(); |
|||
scope.selectedDeviceAliasChange(); |
|||
} |
|||
}); |
|||
|
|||
scope.$watch('timeseriesDataKeys', function () { |
|||
if (ngModelCtrl.$viewValue) { |
|||
var dataKeys = []; |
|||
dataKeys = dataKeys.concat(scope.timeseriesDataKeys); |
|||
dataKeys = dataKeys.concat(scope.attributeDataKeys); |
|||
ngModelCtrl.$viewValue.dataKeys = dataKeys; |
|||
scope.updateValidity(); |
|||
} |
|||
}, true); |
|||
|
|||
scope.$watch('attributeDataKeys', function () { |
|||
if (ngModelCtrl.$viewValue) { |
|||
var dataKeys = []; |
|||
dataKeys = dataKeys.concat(scope.timeseriesDataKeys); |
|||
dataKeys = dataKeys.concat(scope.attributeDataKeys); |
|||
ngModelCtrl.$viewValue.dataKeys = dataKeys; |
|||
scope.updateValidity(); |
|||
} |
|||
}, true); |
|||
|
|||
ngModelCtrl.$render = function () { |
|||
if (ngModelCtrl.$viewValue) { |
|||
var deviceAliasId = ngModelCtrl.$viewValue.deviceAliasId; |
|||
if (scope.deviceAliases[deviceAliasId]) { |
|||
scope.deviceAlias = {id: deviceAliasId, alias: scope.deviceAliases[deviceAliasId].alias, |
|||
deviceId: scope.deviceAliases[deviceAliasId].deviceId}; |
|||
} else { |
|||
scope.deviceAlias = null; |
|||
} |
|||
var timeseriesDataKeys = []; |
|||
var attributeDataKeys = []; |
|||
for (var d in ngModelCtrl.$viewValue.dataKeys) { |
|||
var dataKey = ngModelCtrl.$viewValue.dataKeys[d]; |
|||
if (dataKey.type === types.dataKeyType.timeseries) { |
|||
timeseriesDataKeys.push(dataKey); |
|||
} else if (dataKey.type === types.dataKeyType.attribute) { |
|||
attributeDataKeys.push(dataKey); |
|||
} |
|||
} |
|||
scope.timeseriesDataKeys = timeseriesDataKeys; |
|||
scope.attributeDataKeys = attributeDataKeys; |
|||
} |
|||
}; |
|||
|
|||
scope.textIsNotEmpty = function(text) { |
|||
return (text && text != null && text.length > 0) ? true : false; |
|||
} |
|||
|
|||
scope.selectedDeviceAliasChange = function () { |
|||
if (!scope.timeseriesDataKeySearchText || scope.timeseriesDataKeySearchText === '') { |
|||
scope.timeseriesDataKeySearchText = scope.timeseriesDataKeySearchText === '' ? null : ''; |
|||
} |
|||
if (!scope.attributeDataKeySearchText || scope.attributeDataKeySearchText === '') { |
|||
scope.attributeDataKeySearchText = scope.attributeDataKeySearchText === '' ? null : ''; |
|||
} |
|||
}; |
|||
|
|||
scope.transformTimeseriesDataKeyChip = function (chip) { |
|||
return scope.generateDataKey({chip: chip, type: types.dataKeyType.timeseries}); |
|||
}; |
|||
|
|||
scope.transformAttributeDataKeyChip = function (chip) { |
|||
return scope.generateDataKey({chip: chip, type: types.dataKeyType.attribute}); |
|||
}; |
|||
|
|||
scope.showColorPicker = function (event, dataKey) { |
|||
$mdColorPicker.show({ |
|||
value: dataKey.color, |
|||
defaultValue: '#fff', |
|||
random: tinycolor.random(), |
|||
clickOutsideToClose: false, |
|||
hasBackdrop: false, |
|||
skipHide: true, |
|||
preserveScope: false, |
|||
|
|||
mdColorAlphaChannel: true, |
|||
mdColorSpectrum: true, |
|||
mdColorSliders: true, |
|||
mdColorGenericPalette: false, |
|||
mdColorMaterialPalette: true, |
|||
mdColorHistory: false, |
|||
mdColorDefaultTab: 2, |
|||
|
|||
$event: event |
|||
|
|||
}).then(function (color) { |
|||
dataKey.color = color; |
|||
ngModelCtrl.$setDirty(); |
|||
}); |
|||
} |
|||
|
|||
scope.editDataKey = function (event, dataKey, index) { |
|||
|
|||
$mdDialog.show({ |
|||
controller: 'DatakeyConfigDialogController', |
|||
controllerAs: 'vm', |
|||
templateUrl: datakeyConfigDialogTemplate, |
|||
locals: { |
|||
dataKey: angular.copy(dataKey), |
|||
dataKeySettingsSchema: scope.datakeySettingsSchema, |
|||
deviceAlias: scope.deviceAlias, |
|||
deviceAliases: scope.deviceAliases |
|||
}, |
|||
parent: angular.element($document[0].body), |
|||
fullscreen: true, |
|||
targetEvent: event, |
|||
skipHide: true, |
|||
onComplete: function () { |
|||
var w = angular.element($window); |
|||
w.triggerHandler('resize'); |
|||
} |
|||
}).then(function (dataKey) { |
|||
if (dataKey.type === types.dataKeyType.timeseries) { |
|||
scope.timeseriesDataKeys[index] = dataKey; |
|||
} else if (dataKey.type === types.dataKeyType.attribute) { |
|||
scope.attributeDataKeys[index] = dataKey; |
|||
} |
|||
ngModelCtrl.$setDirty(); |
|||
}, function () { |
|||
}); |
|||
}; |
|||
|
|||
scope.dataKeysSearch = function (searchText, type) { |
|||
if (scope.deviceAlias) { |
|||
var deferred = $q.defer(); |
|||
scope.fetchDeviceKeys({deviceAliasId: scope.deviceAlias.id, query: searchText, type: type}) |
|||
.then(function (dataKeys) { |
|||
deferred.resolve(dataKeys); |
|||
}, function (e) { |
|||
deferred.reject(e); |
|||
}); |
|||
return deferred.promise; |
|||
} else { |
|||
return $q.when([]); |
|||
} |
|||
}; |
|||
|
|||
scope.createKey = function (event, chipsId) { |
|||
var chipsChild = $(chipsId, element)[0].firstElementChild; |
|||
var el = angular.element(chipsChild); |
|||
var chipBuffer = el.scope().$mdChipsCtrl.getChipBuffer(); |
|||
event.preventDefault(); |
|||
event.stopPropagation(); |
|||
el.scope().$mdChipsCtrl.appendChip(chipBuffer.trim()); |
|||
el.scope().$mdChipsCtrl.resetChipBuffer(); |
|||
} |
|||
|
|||
$compile(element.contents())(scope); |
|||
} |
|||
|
|||
return { |
|||
restrict: "E", |
|||
require: "^ngModel", |
|||
scope: { |
|||
widgetType: '=', |
|||
deviceAliases: '=', |
|||
datakeySettingsSchema: '=', |
|||
generateDataKey: '&', |
|||
fetchDeviceKeys: '&', |
|||
onCreateDeviceAlias: '&' |
|||
}, |
|||
link: linker |
|||
}; |
|||
} |
|||
|
|||
/* eslint-enable angular/angularelement */ |
|||
@ -1,137 +0,0 @@ |
|||
<!-- |
|||
|
|||
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. |
|||
|
|||
--> |
|||
<section flex layout='column' layout-align="center" layout-gt-sm='row' layout-align-gt-sm="start center"> |
|||
<tb-device-alias-select |
|||
tb-required="true" |
|||
device-aliases="deviceAliases" |
|||
ng-model="deviceAlias" |
|||
on-create-device-alias="onCreateDeviceAlias({event: event, alias: alias})"> |
|||
</tb-device-alias-select> |
|||
<section flex layout='column'> |
|||
<section flex layout='column' layout-align="center" style="padding-left: 4px;"> |
|||
<md-chips flex |
|||
id="timeseries_datakey_chips" |
|||
ng-required="true" |
|||
ng-model="timeseriesDataKeys" md-autocomplete-snap |
|||
md-transform-chip="transformTimeseriesDataKeyChip($chip)" |
|||
md-require-match="false"> |
|||
<md-autocomplete |
|||
md-no-cache="true" |
|||
id="timeseries_datakey" |
|||
md-selected-item="selectedTimeseriesDataKey" |
|||
md-search-text="timeseriesDataKeySearchText" |
|||
md-items="item in dataKeysSearch(timeseriesDataKeySearchText, types.dataKeyType.timeseries)" |
|||
md-item-text="item.name" |
|||
md-min-length="0" |
|||
placeholder="{{'datakey.timeseries' | translate }}" |
|||
md-menu-class="tb-timeseries-datakey-autocomplete"> |
|||
<span md-highlight-text="timeseriesDataKeySearchText" md-highlight-flags="^i">{{item}}</span> |
|||
<md-not-found> |
|||
<div class="tb-not-found"> |
|||
<div class="tb-no-entries" ng-if="!textIsNotEmpty(timeseriesDataKeySearchText)"> |
|||
<span translate>device.no-keys-found</span> |
|||
</div> |
|||
<div ng-if="textIsNotEmpty(timeseriesDataKeySearchText)"> |
|||
<span translate translate-values='{ key: "{{timeseriesDataKeySearchText | truncate:true:6:'...'}}" }'>device.no-key-matching</span> |
|||
<span> |
|||
<a translate ng-click="createKey($event, '#timeseries_datakey_chips')">device.create-new-key</a> |
|||
</span> |
|||
</div> |
|||
</div> |
|||
</md-not-found> |
|||
</md-autocomplete> |
|||
<md-chip-template> |
|||
<div layout="row" layout-align="start center" class="tb-attribute-chip"> |
|||
<div class="tb-color-preview" ng-click="showColorPicker($event, $chip, $index)" style="margin-right: 5px;"> |
|||
<div class="tb-color-result" ng-style="{background: $chip.color}"></div> |
|||
</div> |
|||
<div layout="row" flex> |
|||
<div class="tb-chip-label"> |
|||
{{$chip.label}} |
|||
</div> |
|||
<div class="tb-chip-separator">: </div> |
|||
<div class="tb-chip-label"> |
|||
<strong ng-if="!$chip.postFuncBody">{{$chip.name}}</strong> |
|||
<strong ng-if="$chip.postFuncBody">f({{$chip.name}})</strong> |
|||
</div> |
|||
</div> |
|||
<md-button ng-click="editDataKey($event, $chip, $index)" class="md-icon-button tb-md-32"> |
|||
<md-icon aria-label="edit" class="material-icons tb-md-20">edit</md-icon> |
|||
</md-button> |
|||
</div> |
|||
</md-chip-template> |
|||
</md-chips> |
|||
<md-chips flex ng-if="widgetType === types.widgetType.latest.value" |
|||
id="attribute_datakey_chips" |
|||
ng-required="true" |
|||
ng-model="attributeDataKeys" md-autocomplete-snap |
|||
md-transform-chip="transformAttributeDataKeyChip($chip)" |
|||
md-require-match="false"> |
|||
<md-autocomplete |
|||
md-no-cache="true" |
|||
id="attribute_datakey" |
|||
md-selected-item="selectedAttributeDataKey" |
|||
md-search-text="attributeDataKeySearchText" |
|||
md-items="item in dataKeysSearch(attributeDataKeySearchText, types.dataKeyType.attribute)" |
|||
md-item-text="item.name" |
|||
md-min-length="0" |
|||
placeholder="{{'datakey.attributes' | translate }}" |
|||
md-menu-class="tb-attribute-datakey-autocomplete"> |
|||
<span md-highlight-text="attributeDataKeySearchText" md-highlight-flags="^i">{{item}}</span> |
|||
<md-not-found> |
|||
<div class="tb-not-found"> |
|||
<div class="tb-no-entries" ng-if="!textIsNotEmpty(attributeDataKeySearchText)"> |
|||
<span translate>device.no-keys-found</span> |
|||
</div> |
|||
<div ng-if="textIsNotEmpty(attributeDataKeySearchText)"> |
|||
<span translate translate-values='{ key: "{{attributeDataKeySearchText | truncate:true:6:'...'}}" }'>device.no-key-matching</span> |
|||
<span> |
|||
<a translate ng-click="createKey($event, '#attribute_datakey_chips')">device.create-new-key</a> |
|||
</span> |
|||
</div> |
|||
</div> |
|||
</md-not-found> |
|||
</md-autocomplete> |
|||
<md-chip-template> |
|||
<div layout="row" layout-align="start center" class="tb-attribute-chip"> |
|||
<div class="tb-color-preview" ng-click="showColorPicker($event, $chip, $index)" style="margin-right: 5px;"> |
|||
<div class="tb-color-result" ng-style="{background: $chip.color}"></div> |
|||
</div> |
|||
<div layout="row" flex> |
|||
<div class="tb-chip-label"> |
|||
{{$chip.label}} |
|||
</div> |
|||
<div class="tb-chip-separator">: </div> |
|||
<div class="tb-chip-label"> |
|||
<strong ng-if="!$chip.postFuncBody">{{$chip.name}}</strong> |
|||
<strong ng-if="$chip.postFuncBody">f({{$chip.name}})</strong> |
|||
</div> |
|||
</div> |
|||
<md-button ng-click="editDataKey($event, $chip, $index)" class="md-icon-button tb-md-32"> |
|||
<md-icon aria-label="edit" class="material-icons tb-md-20">edit</md-icon> |
|||
</md-button> |
|||
</div> |
|||
</md-chip-template> |
|||
</md-chips> |
|||
</section> |
|||
<div class="tb-error-messages" ng-messages="ngModelCtrl.$error" role="alert"> |
|||
<div translate ng-message="deviceKeys" ng-if="widgetType === types.widgetType.timeseries.value" class="tb-error-message">datakey.timeseries-required</div> |
|||
<div translate ng-message="deviceKeys" ng-if="widgetType === types.widgetType.latest.value" class="tb-error-message">datakey.timeseries-or-attributes-required</div> |
|||
</div> |
|||
</section> |
|||
</section> |
|||
@ -1,144 +0,0 @@ |
|||
/* |
|||
* 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. |
|||
*/ |
|||
import $ from 'jquery'; |
|||
|
|||
import './device-alias-select.scss'; |
|||
|
|||
/* eslint-disable import/no-unresolved, import/default */ |
|||
|
|||
import deviceAliasSelectTemplate from './device-alias-select.tpl.html'; |
|||
|
|||
/* eslint-enable import/no-unresolved, import/default */ |
|||
|
|||
|
|||
/* eslint-disable angular/angularelement */ |
|||
|
|||
export default angular.module('thingsboard.directives.deviceAliasSelect', []) |
|||
.directive('tbDeviceAliasSelect', DeviceAliasSelect) |
|||
.name; |
|||
|
|||
/*@ngInject*/ |
|||
function DeviceAliasSelect($compile, $templateCache, $mdConstant) { |
|||
|
|||
var linker = function (scope, element, attrs, ngModelCtrl) { |
|||
var template = $templateCache.get(deviceAliasSelectTemplate); |
|||
element.html(template); |
|||
|
|||
scope.tbRequired = angular.isDefined(scope.tbRequired) ? scope.tbRequired : false; |
|||
|
|||
scope.ngModelCtrl = ngModelCtrl; |
|||
scope.deviceAliasList = []; |
|||
scope.deviceAlias = null; |
|||
|
|||
scope.updateValidity = function () { |
|||
var value = ngModelCtrl.$viewValue; |
|||
var valid = angular.isDefined(value) && value != null || !scope.tbRequired; |
|||
ngModelCtrl.$setValidity('deviceAlias', valid); |
|||
}; |
|||
|
|||
scope.$watch('deviceAliases', function () { |
|||
scope.deviceAliasList = []; |
|||
for (var aliasId in scope.deviceAliases) { |
|||
var deviceAlias = {id: aliasId, alias: scope.deviceAliases[aliasId].alias, deviceId: scope.deviceAliases[aliasId].deviceId}; |
|||
scope.deviceAliasList.push(deviceAlias); |
|||
} |
|||
}, true); |
|||
|
|||
scope.$watch('deviceAlias', function () { |
|||
scope.updateView(); |
|||
}); |
|||
|
|||
scope.deviceAliasSearch = function (deviceAliasSearchText) { |
|||
return deviceAliasSearchText ? scope.deviceAliasList.filter( |
|||
scope.createFilterForDeviceAlias(deviceAliasSearchText)) : scope.deviceAliasList; |
|||
}; |
|||
|
|||
scope.createFilterForDeviceAlias = function (query) { |
|||
var lowercaseQuery = angular.lowercase(query); |
|||
return function filterFn(deviceAlias) { |
|||
return (angular.lowercase(deviceAlias.alias).indexOf(lowercaseQuery) === 0); |
|||
}; |
|||
}; |
|||
|
|||
scope.updateView = function () { |
|||
ngModelCtrl.$setViewValue(scope.deviceAlias); |
|||
scope.updateValidity(); |
|||
} |
|||
|
|||
ngModelCtrl.$render = function () { |
|||
if (ngModelCtrl.$viewValue) { |
|||
scope.deviceAlias = ngModelCtrl.$viewValue; |
|||
} |
|||
} |
|||
|
|||
scope.textIsNotEmpty = function(text) { |
|||
return (text && text != null && text.length > 0) ? true : false; |
|||
} |
|||
|
|||
scope.deviceAliasEnter = function($event) { |
|||
if ($event.keyCode === $mdConstant.KEY_CODE.ENTER) { |
|||
$event.preventDefault(); |
|||
if (!scope.deviceAlias) { |
|||
var found = scope.deviceAliasSearch(scope.deviceAliasSearchText); |
|||
found = found.length > 0; |
|||
if (!found) { |
|||
scope.createDeviceAlias($event, scope.deviceAliasSearchText); |
|||
} |
|||
} |
|||
} |
|||
} |
|||
|
|||
scope.createDeviceAlias = function (event, alias) { |
|||
var autoChild = $('#device-autocomplete', element)[0].firstElementChild; |
|||
var el = angular.element(autoChild); |
|||
el.scope().$mdAutocompleteCtrl.hidden = true; |
|||
el.scope().$mdAutocompleteCtrl.hasNotFound = false; |
|||
event.preventDefault(); |
|||
var promise = scope.onCreateDeviceAlias({event: event, alias: alias}); |
|||
if (promise) { |
|||
promise.then( |
|||
function success(newAlias) { |
|||
el.scope().$mdAutocompleteCtrl.hasNotFound = true; |
|||
if (newAlias) { |
|||
scope.deviceAliasList.push(newAlias); |
|||
scope.deviceAlias = newAlias; |
|||
} |
|||
}, |
|||
function fail() { |
|||
el.scope().$mdAutocompleteCtrl.hasNotFound = true; |
|||
} |
|||
); |
|||
} else { |
|||
el.scope().$mdAutocompleteCtrl.hasNotFound = true; |
|||
} |
|||
}; |
|||
|
|||
$compile(element.contents())(scope); |
|||
} |
|||
|
|||
return { |
|||
restrict: "E", |
|||
require: "^ngModel", |
|||
link: linker, |
|||
scope: { |
|||
tbRequired: '=?', |
|||
deviceAliases: '=', |
|||
onCreateDeviceAlias: '&' |
|||
} |
|||
}; |
|||
} |
|||
|
|||
/* eslint-enable angular/angularelement */ |
|||
Some files were not shown because too many files changed in this diff
Loading…
Reference in new issue