Viacheslav Klimov
2 years ago
committed by
GitHub
GitHub
179 changed files with 620 additions and 3559 deletions
@ -1,58 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2024 ThingsBoard, Inc. |
|
||||
* |
|
||||
* 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.edqs.data; |
|
||||
|
|
||||
import org.thingsboard.server.common.data.EntityType; |
|
||||
import org.thingsboard.server.common.data.edqs.fields.EntityGroupFields; |
|
||||
|
|
||||
import java.util.Collection; |
|
||||
import java.util.UUID; |
|
||||
import java.util.concurrent.ConcurrentHashMap; |
|
||||
import java.util.concurrent.ConcurrentMap; |
|
||||
|
|
||||
public class EntityGroupData extends BaseEntityData<EntityGroupFields> { |
|
||||
|
|
||||
private final ConcurrentMap<UUID, EntityData<?>> entitiesById = new ConcurrentHashMap<>(); |
|
||||
|
|
||||
public EntityGroupData(UUID entityId) { |
|
||||
super(entityId); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public EntityType getEntityType() { |
|
||||
return EntityType.ENTITY_GROUP; |
|
||||
} |
|
||||
|
|
||||
public Collection<EntityData<?>> getEntities() { |
|
||||
return entitiesById.values(); |
|
||||
} |
|
||||
|
|
||||
public boolean addOrUpdate(EntityData<?> ed) { |
|
||||
return entitiesById.put(ed.getId(), ed) == null; |
|
||||
} |
|
||||
|
|
||||
public boolean remove(EntityData<?> ed) { |
|
||||
return entitiesById.remove(ed.getId()) != null; |
|
||||
} |
|
||||
|
|
||||
public EntityData<?> getEntity(UUID entityId) { |
|
||||
return entitiesById.get(entityId); |
|
||||
} |
|
||||
|
|
||||
public boolean remove(UUID toId) { |
|
||||
return entitiesById.remove(toId) != null; |
|
||||
} |
|
||||
} |
|
||||
@ -1,75 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2024 ThingsBoard, Inc. |
|
||||
* |
|
||||
* 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.edqs.query.processor; |
|
||||
|
|
||||
import org.thingsboard.server.common.data.permission.QueryContext; |
|
||||
import org.thingsboard.server.common.data.query.EntityFilter; |
|
||||
import org.thingsboard.server.edqs.data.EntityGroupData; |
|
||||
import org.thingsboard.server.edqs.query.EdqsQuery; |
|
||||
import org.thingsboard.server.edqs.query.SortableEntityData; |
|
||||
import org.thingsboard.server.edqs.repo.TenantRepo; |
|
||||
|
|
||||
import java.util.ArrayList; |
|
||||
import java.util.HashMap; |
|
||||
import java.util.List; |
|
||||
import java.util.Map; |
|
||||
import java.util.UUID; |
|
||||
|
|
||||
public abstract class AbstractEntityGroupQueryProcessor<T extends EntityFilter> extends AbstractSingleEntityTypeQueryProcessor<T> { |
|
||||
|
|
||||
public AbstractEntityGroupQueryProcessor(TenantRepo repo, QueryContext ctx, EdqsQuery query, T filter) { |
|
||||
super(repo, ctx, query, filter); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected List<SortableEntityData> processCustomerGenericReadWithGroups(UUID customerId, boolean readAttrPermissions, boolean readTsPermissions, List<GroupPermissions> groupPermissions) { |
|
||||
var genericReadResults = processCustomerGenericRead(customerId, readAttrPermissions, readTsPermissions); |
|
||||
Map<UUID, SortableEntityData> mergedResult = new HashMap<>(genericReadResults.size()); |
|
||||
for (SortableEntityData sd : genericReadResults) { |
|
||||
mergedResult.put(sd.getId(), sd); |
|
||||
} |
|
||||
for (GroupPermissions permissions : groupPermissions) { |
|
||||
SortableEntityData alreadyAdded = mergedResult.get(permissions.groupId); |
|
||||
if (alreadyAdded != null) { |
|
||||
alreadyAdded.setReadAttrs(alreadyAdded.isReadAttrs() || permissions.readAttrs); |
|
||||
alreadyAdded.setReadTs(alreadyAdded.isReadTs() || permissions.readTs); |
|
||||
} else { |
|
||||
EntityGroupData egData = repository.getEntityGroup(permissions.groupId); |
|
||||
if (matches(egData)) { |
|
||||
SortableEntityData sortData = toSortData(egData, permissions); |
|
||||
mergedResult.put(egData.getId(), sortData); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
return new ArrayList<>(mergedResult.values()); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected CombinedPermissions getCombinedPermissionsInternal(UUID id, boolean read, boolean readAttrs, boolean readTs, List<GroupPermissions> groupPermissions) { |
|
||||
for (GroupPermissions eg : groupPermissions) { |
|
||||
if (read && readAttrs && readTs) { |
|
||||
break; |
|
||||
} |
|
||||
if (eg.groupId.equals(id)) { |
|
||||
read = true; |
|
||||
readAttrs = readAttrs || eg.readAttrs; |
|
||||
readTs = readTs || eg.readTs; |
|
||||
} |
|
||||
} |
|
||||
return new CombinedPermissions(read, readAttrs, readTs); |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
@ -1,25 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2024 ThingsBoard, Inc. |
|
||||
* |
|
||||
* 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.edqs.query.processor; |
|
||||
|
|
||||
import lombok.Data; |
|
||||
|
|
||||
@Data |
|
||||
public class CombinedPermissions implements Permissions { |
|
||||
private final boolean read; |
|
||||
private final boolean readAttrs; |
|
||||
private final boolean readTs; |
|
||||
} |
|
||||
@ -1,121 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2024 ThingsBoard, Inc. |
|
||||
* |
|
||||
* 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.edqs.query.processor; |
|
||||
|
|
||||
import org.thingsboard.server.common.data.EntityType; |
|
||||
import org.thingsboard.server.common.data.edqs.fields.EntityGroupFields; |
|
||||
import org.thingsboard.server.common.data.permission.QueryContext; |
|
||||
import org.thingsboard.server.common.data.query.EntitiesByGroupNameFilter; |
|
||||
import org.thingsboard.server.edqs.data.CustomerData; |
|
||||
import org.thingsboard.server.edqs.data.EntityData; |
|
||||
import org.thingsboard.server.edqs.query.EdqsQuery; |
|
||||
import org.thingsboard.server.edqs.query.SortableEntityData; |
|
||||
import org.thingsboard.server.edqs.repo.TenantRepo; |
|
||||
import org.thingsboard.server.edqs.util.RepositoryUtils; |
|
||||
|
|
||||
import java.util.ArrayList; |
|
||||
import java.util.Collection; |
|
||||
import java.util.HashSet; |
|
||||
import java.util.List; |
|
||||
import java.util.UUID; |
|
||||
import java.util.function.Consumer; |
|
||||
import java.util.regex.Pattern; |
|
||||
|
|
||||
import static org.thingsboard.server.common.data.EntityType.ENTITY_GROUP; |
|
||||
import static org.thingsboard.server.edqs.util.RepositoryUtils.getSortValue; |
|
||||
|
|
||||
public class EntitiesByGroupNameQueryProcessor extends AbstractSingleEntityTypeQueryProcessor<EntitiesByGroupNameFilter> { |
|
||||
|
|
||||
private final String groupType; |
|
||||
private final UUID ownerId; |
|
||||
private final Pattern pattern; |
|
||||
|
|
||||
public EntitiesByGroupNameQueryProcessor(TenantRepo repo, QueryContext ctx, EdqsQuery query) { |
|
||||
super(repo, ctx, query, (EntitiesByGroupNameFilter) query.getEntityFilter()); |
|
||||
this.groupType = filter.getGroupType().name(); |
|
||||
this.ownerId = filter.getOwnerId() != null ? filter.getOwnerId().getId() : null; |
|
||||
this.pattern = RepositoryUtils.toSqlLikePattern(filter.getEntityGroupNameFilter()); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void processCustomerGenericRead(UUID customerId, Consumer<EntityData<?>> processor) { |
|
||||
var customers = repository.getEntityMap(EntityType.CUSTOMER); |
|
||||
for (UUID cId : repository.getAllCustomers(customerId)) { |
|
||||
var customerData = (CustomerData) customers.get(cId); |
|
||||
if (customerData != null) { |
|
||||
process(customerData.getEntities(ENTITY_GROUP), processor); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected List<SortableEntityData> processCustomerGenericReadWithGroups(UUID customerId, boolean readAttrPermissions, boolean readTsPermissions, List<GroupPermissions> groupPermissions) { |
|
||||
List<SortableEntityData> result = new ArrayList<>(getProbableResultSize()); |
|
||||
var customers = repository.getAllCustomers(customerId); |
|
||||
processAll(ed -> { |
|
||||
CombinedPermissions permissions = getCombinedPermissions(ed.getId(), checkCustomerHierarchy(customers, ed), readAttrPermissions, readTsPermissions, groupPermissions); |
|
||||
if (permissions.isRead()) { |
|
||||
SortableEntityData sortData = new SortableEntityData(ed); |
|
||||
sortData.setSortValue(getSortValue(ed, sortKey)); |
|
||||
sortData.setReadAttrs(permissions.isReadAttrs()); |
|
||||
sortData.setReadTs(permissions.isReadTs()); |
|
||||
result.add(sortData); |
|
||||
} |
|
||||
}); |
|
||||
return result; |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void processGroupsOnly(List<GroupPermissions> groupPermissions, Consumer<EntityData<?>> processor) { |
|
||||
Collection<EntityData<?>> entities = new HashSet<>(getProbableResultSize()); |
|
||||
for (GroupPermissions groupPermission : groupPermissions) { |
|
||||
entities.add(repository.getEntityGroup(groupPermission.groupId)); |
|
||||
} |
|
||||
process(entities, processor); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void processAll(Consumer<EntityData<?>> processor) { |
|
||||
process(repository.getEntitySet(ENTITY_GROUP), processor); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void process(Collection<EntityData<?>> entities, Consumer<EntityData<?>> processor) { |
|
||||
for (EntityData<?> ed : entities) { |
|
||||
if (matches(ed)) { |
|
||||
Collection<EntityData<?>> groupEntities = repository.getEntityGroup(ed.getId()).getEntities(); |
|
||||
for (EntityData<?> groupEntity : groupEntities) { |
|
||||
processor.accept(groupEntity); |
|
||||
} |
|
||||
return; |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected boolean matches(EntityData ed) { |
|
||||
EntityGroupFields fields = (EntityGroupFields)ed.getFields(); |
|
||||
return super.matches(ed) && groupType.equals(fields.getType()) |
|
||||
&& (pattern == null || pattern.matcher(fields.getName()).matches()) |
|
||||
&& (ownerId == null || ownerId.equals(fields.getOwnerId())); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected int getProbableResultSize() { |
|
||||
return 1024; |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
@ -1,96 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2024 ThingsBoard, Inc. |
|
||||
* |
|
||||
* 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.edqs.query.processor; |
|
||||
|
|
||||
import org.thingsboard.server.common.data.permission.QueryContext; |
|
||||
import org.thingsboard.server.common.data.query.EntityGroupFilter; |
|
||||
import org.thingsboard.server.edqs.data.EntityData; |
|
||||
import org.thingsboard.server.edqs.data.EntityGroupData; |
|
||||
import org.thingsboard.server.edqs.query.EdqsQuery; |
|
||||
import org.thingsboard.server.edqs.query.SortableEntityData; |
|
||||
import org.thingsboard.server.edqs.repo.TenantRepo; |
|
||||
|
|
||||
import java.util.ArrayList; |
|
||||
import java.util.List; |
|
||||
import java.util.UUID; |
|
||||
import java.util.function.Consumer; |
|
||||
|
|
||||
import static org.thingsboard.server.edqs.util.RepositoryUtils.getSortValue; |
|
||||
|
|
||||
public class EntitiesByGroupQueryProcessor extends AbstractSingleEntityTypeQueryProcessor<EntityGroupFilter> { |
|
||||
|
|
||||
private final String groupType; |
|
||||
private final UUID groupId; |
|
||||
|
|
||||
public EntitiesByGroupQueryProcessor(TenantRepo repo, QueryContext ctx, EdqsQuery query) { |
|
||||
super(repo, ctx, query, (EntityGroupFilter) query.getEntityFilter()); |
|
||||
groupId = UUID.fromString(filter.getEntityGroup()); |
|
||||
groupType = filter.getGroupType().name(); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void processCustomerGenericRead(UUID customerId, Consumer<EntityData<?>> processor) { |
|
||||
var customers = repository.getAllCustomers(customerId); |
|
||||
processAll(ed -> { |
|
||||
if (checkCustomerHierarchy(customers, ed)) { |
|
||||
processor.accept(ed); |
|
||||
} |
|
||||
}); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected List<SortableEntityData> processCustomerGenericReadWithGroups(UUID customerId, boolean readAttrPermissions, boolean readTsPermissions, List<GroupPermissions> groupPermissions) { |
|
||||
List<SortableEntityData> result = new ArrayList<>(getProbableResultSize()); |
|
||||
var customers = repository.getAllCustomers(customerId); |
|
||||
processAll(ed -> { |
|
||||
CombinedPermissions permissions = getCombinedPermissions(ed.getId(), checkCustomerHierarchy(customers, ed), readAttrPermissions, readTsPermissions, groupPermissions); |
|
||||
if (permissions.isRead()) { |
|
||||
SortableEntityData sortData = new SortableEntityData(ed); |
|
||||
sortData.setSortValue(getSortValue(ed, sortKey)); |
|
||||
sortData.setReadAttrs(permissions.isReadAttrs()); |
|
||||
sortData.setReadTs(permissions.isReadTs()); |
|
||||
result.add(sortData); |
|
||||
} |
|
||||
}); |
|
||||
return result; |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void processGroupsOnly(List<GroupPermissions> groupPermissions, Consumer<EntityData<?>> processor) { |
|
||||
processAll(processor); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void processAll(Consumer<EntityData<?>> processor) { |
|
||||
EntityGroupData entityGroup = repository.getEntityGroup(groupId); |
|
||||
if (matches(entityGroup)) { |
|
||||
for (EntityData<?> ed : entityGroup.getEntities()) { |
|
||||
processor.accept(ed); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected boolean matches(EntityData ed) { |
|
||||
return super.matches(ed) && groupType.equals(ed.getFields().getType()); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected int getProbableResultSize() { |
|
||||
return 1024; |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
@ -1,84 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2024 ThingsBoard, Inc. |
|
||||
* |
|
||||
* 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.edqs.query.processor; |
|
||||
|
|
||||
import org.thingsboard.server.common.data.permission.QueryContext; |
|
||||
import org.thingsboard.server.common.data.query.EntityGroupListFilter; |
|
||||
import org.thingsboard.server.edqs.data.EntityData; |
|
||||
import org.thingsboard.server.edqs.data.EntityGroupData; |
|
||||
import org.thingsboard.server.edqs.query.EdqsQuery; |
|
||||
import org.thingsboard.server.edqs.repo.TenantRepo; |
|
||||
|
|
||||
import java.util.List; |
|
||||
import java.util.Set; |
|
||||
import java.util.UUID; |
|
||||
import java.util.function.Consumer; |
|
||||
import java.util.stream.Collectors; |
|
||||
|
|
||||
public class EntityGroupListQueryProcessor extends AbstractEntityGroupQueryProcessor<EntityGroupListFilter> { |
|
||||
|
|
||||
private final String groupType; |
|
||||
private final Set<UUID> groupIds; |
|
||||
|
|
||||
public EntityGroupListQueryProcessor(TenantRepo repo, QueryContext ctx, EdqsQuery query) { |
|
||||
super(repo, ctx, query, (EntityGroupListFilter) query.getEntityFilter()); |
|
||||
this.groupType = filter.getGroupType().name(); |
|
||||
this.groupIds = filter.getEntityGroupList().stream().map(UUID::fromString).collect(Collectors.toSet()); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void processCustomerGenericRead(UUID customerId, Consumer<EntityData<?>> processor) { |
|
||||
var customers = repository.getAllCustomers(customerId); |
|
||||
processAll(ed -> { |
|
||||
if (checkCustomerHierarchy(customers, ed)) { |
|
||||
processor.accept(ed); |
|
||||
} |
|
||||
}); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void processGroupsOnly(List<GroupPermissions> groupPermissions, Consumer<EntityData<?>> processor) { |
|
||||
Set<UUID> allowedGroupIds = groupPermissions.stream().map(GroupPermissions::getGroupId) |
|
||||
.filter(this.groupIds::contains).collect(Collectors.toSet()); |
|
||||
|
|
||||
checkGroupIds(allowedGroupIds, processor); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void processAll(Consumer<EntityData<?>> processor) { |
|
||||
checkGroupIds(groupIds, processor); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected int getProbableResultSize() { |
|
||||
return groupIds.size(); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected boolean matches(EntityData ed) { |
|
||||
return super.matches(ed) && groupType.equals(ed.getFields().getType()); |
|
||||
} |
|
||||
|
|
||||
private void checkGroupIds(Set<UUID> allowedGroupIds, Consumer<EntityData<?>> processor) { |
|
||||
for (UUID groupId : allowedGroupIds) { |
|
||||
EntityGroupData entityGroup = repository.getEntityGroup(groupId); |
|
||||
if (matches(entityGroup)) { |
|
||||
processor.accept(entityGroup); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
@ -1,83 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2024 ThingsBoard, Inc. |
|
||||
* |
|
||||
* 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.edqs.query.processor; |
|
||||
|
|
||||
import org.thingsboard.server.common.data.EntityType; |
|
||||
import org.thingsboard.server.common.data.permission.QueryContext; |
|
||||
import org.thingsboard.server.common.data.query.EntityGroupNameFilter; |
|
||||
import org.thingsboard.server.edqs.data.CustomerData; |
|
||||
import org.thingsboard.server.edqs.data.EntityData; |
|
||||
import org.thingsboard.server.edqs.data.EntityGroupData; |
|
||||
import org.thingsboard.server.edqs.query.EdqsQuery; |
|
||||
import org.thingsboard.server.edqs.repo.TenantRepo; |
|
||||
import org.thingsboard.server.edqs.util.RepositoryUtils; |
|
||||
|
|
||||
import java.util.List; |
|
||||
import java.util.UUID; |
|
||||
import java.util.function.Consumer; |
|
||||
import java.util.regex.Pattern; |
|
||||
|
|
||||
import static org.thingsboard.server.common.data.EntityType.ENTITY_GROUP; |
|
||||
|
|
||||
public class EntityGroupNameQueryProcessor extends AbstractEntityGroupQueryProcessor<EntityGroupNameFilter> { |
|
||||
|
|
||||
private final String groupType; |
|
||||
private final Pattern groupNamePattern; |
|
||||
|
|
||||
public EntityGroupNameQueryProcessor(TenantRepo repo, QueryContext ctx, EdqsQuery query) { |
|
||||
super(repo, ctx, query, (EntityGroupNameFilter) query.getEntityFilter()); |
|
||||
this.groupType = filter.getGroupType().name(); |
|
||||
this.groupNamePattern = RepositoryUtils.toSqlLikePattern(filter.getEntityGroupNameFilter()); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void processCustomerGenericRead(UUID customerId, Consumer<EntityData<?>> processor) { |
|
||||
var customers = repository.getEntityMap(EntityType.CUSTOMER); |
|
||||
for (UUID cId : repository.getAllCustomers(customerId)) { |
|
||||
var customerData = (CustomerData) customers.get(cId); |
|
||||
if (customerData != null) { |
|
||||
process(customerData.getEntities(ENTITY_GROUP), processor); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void processGroupsOnly(List<GroupPermissions> groupPermissions, Consumer<EntityData<?>> processor) { |
|
||||
for (GroupPermissions groupPermission : groupPermissions) { |
|
||||
EntityGroupData entityGroup = repository.getEntityGroup(groupPermission.groupId); |
|
||||
if (matches(entityGroup)) { |
|
||||
processor.accept(entityGroup); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void processAll(Consumer<EntityData<?>> processor) { |
|
||||
process(repository.getEntitySet(ENTITY_GROUP), processor); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected boolean matches(EntityData ed) { |
|
||||
return super.matches(ed) && (groupNamePattern == null || groupNamePattern.matcher(ed.getFields().getName()).matches()) |
|
||||
&& groupType.equals(ed.getFields().getType()); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected int getProbableResultSize() { |
|
||||
return 1024; |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
@ -1,29 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2024 ThingsBoard, Inc. |
|
||||
* |
|
||||
* 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.edqs.query.processor; |
|
||||
|
|
||||
import lombok.Data; |
|
||||
|
|
||||
import java.util.UUID; |
|
||||
|
|
||||
@Data |
|
||||
public class GroupPermissions implements Permissions { |
|
||||
|
|
||||
protected final UUID groupId; |
|
||||
protected final boolean readAttrs; |
|
||||
protected final boolean readTs; |
|
||||
|
|
||||
} |
|
||||
@ -1,24 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2024 ThingsBoard, Inc. |
|
||||
* |
|
||||
* 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.edqs.query.processor; |
|
||||
|
|
||||
public interface Permissions { |
|
||||
|
|
||||
boolean isReadAttrs(); |
|
||||
|
|
||||
boolean isReadTs(); |
|
||||
|
|
||||
} |
|
||||
@ -1,33 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2024 ThingsBoard, Inc. |
|
||||
* |
|
||||
* 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.edqs.query.processor; |
|
||||
|
|
||||
import lombok.Builder; |
|
||||
import lombok.Getter; |
|
||||
|
|
||||
import java.util.List; |
|
||||
|
|
||||
@Getter |
|
||||
@Builder |
|
||||
public class RelationQueryPermissions implements Permissions { |
|
||||
|
|
||||
private final boolean readEntity; |
|
||||
private final boolean readAttrs; |
|
||||
private final boolean readTs; |
|
||||
private final boolean hasGroups; |
|
||||
private final List<GroupPermissions> groupPermissions; |
|
||||
|
|
||||
} |
|
||||
@ -1,37 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2024 ThingsBoard, Inc. |
|
||||
* |
|
||||
* 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.edqs.query.processor; |
|
||||
|
|
||||
import org.thingsboard.server.common.data.EntityType; |
|
||||
import org.thingsboard.server.common.data.permission.QueryContext; |
|
||||
import org.thingsboard.server.common.data.query.SchedulerEventFilter; |
|
||||
import org.thingsboard.server.edqs.data.EntityData; |
|
||||
import org.thingsboard.server.edqs.query.EdqsQuery; |
|
||||
import org.thingsboard.server.edqs.repo.TenantRepo; |
|
||||
|
|
||||
public class SchedulerEventQueryProcessor extends AbstractSimpleQueryProcessor<SchedulerEventFilter> { |
|
||||
|
|
||||
public SchedulerEventQueryProcessor(TenantRepo repo, QueryContext ctx, EdqsQuery query) { |
|
||||
super(repo, ctx, query, (SchedulerEventFilter) query.getEntityFilter(), EntityType.SCHEDULER_EVENT); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected boolean matches(EntityData<?> ed) { |
|
||||
return super.matches(ed) && (filter.getEventType() == null || filter.getEventType().equals(ed.getFields().getType())) |
|
||||
&& (filter.getOriginator() == null || filter.getOriginator().equals(ed.getFields().getOriginatorId())); |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
@ -1,91 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2024 ThingsBoard, Inc. |
|
||||
* |
|
||||
* 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.edqs.query.processor; |
|
||||
|
|
||||
import org.thingsboard.server.common.data.EntityType; |
|
||||
import org.thingsboard.server.common.data.id.EntityId; |
|
||||
import org.thingsboard.server.common.data.permission.QueryContext; |
|
||||
import org.thingsboard.server.common.data.query.StateEntityOwnerFilter; |
|
||||
import org.thingsboard.server.edqs.data.EntityData; |
|
||||
import org.thingsboard.server.edqs.query.EdqsQuery; |
|
||||
import org.thingsboard.server.edqs.query.SortableEntityData; |
|
||||
import org.thingsboard.server.edqs.repo.TenantRepo; |
|
||||
|
|
||||
import java.util.Collections; |
|
||||
import java.util.List; |
|
||||
import java.util.UUID; |
|
||||
import java.util.function.Consumer; |
|
||||
|
|
||||
public class StateEntityOwnerQueryProcessor extends AbstractSingleEntityTypeQueryProcessor<StateEntityOwnerFilter> { |
|
||||
|
|
||||
private final EntityId entityId; |
|
||||
|
|
||||
public StateEntityOwnerQueryProcessor(TenantRepo repo, QueryContext ctx, EdqsQuery query) { |
|
||||
super(repo, ctx, query, (StateEntityOwnerFilter) query.getEntityFilter()); |
|
||||
this.entityId = filter.getSingleEntity(); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void processCustomerGenericRead(UUID customerId, Consumer<EntityData<?>> processor) { |
|
||||
EntityData ed = repository.getEntityMap(entityId.getEntityType()).get(entityId.getId()); |
|
||||
if (ed != null && ed.getCustomerId() != null && matches(ed)) { |
|
||||
if (customerId.equals(ed.getCustomerId()) || repository.getAllCustomers(customerId).contains(ed.getCustomerId())) { |
|
||||
processor.accept(ed); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
|
|
||||
@Override |
|
||||
protected List<SortableEntityData> processCustomerGenericReadWithGroups(UUID customerId, boolean readAttrPermissions, boolean readTsPermissions, List<GroupPermissions> groupPermissions) { |
|
||||
EntityData ed = repository.getEntityMap(entityId.getEntityType()).get(entityId.getId()); |
|
||||
if (!matches(ed)) { |
|
||||
return Collections.emptyList(); |
|
||||
} else { |
|
||||
boolean genericRead = customerId.equals(ed.getCustomerId()) || repository.getAllCustomers(customerId).contains(ed.getCustomerId()); |
|
||||
CombinedPermissions permissions = getCombinedPermissions(ed.getId(), genericRead, readAttrPermissions, readTsPermissions, groupPermissions); |
|
||||
if (permissions.isRead()) { |
|
||||
SortableEntityData sortData = toSortData(ed, permissions); |
|
||||
return Collections.singletonList(sortData); |
|
||||
} else { |
|
||||
return Collections.emptyList(); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void processGroupsOnly(List<GroupPermissions> groupPermissions, Consumer<EntityData<?>> processor) { |
|
||||
processAll(processor); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected void processAll(Consumer<EntityData<?>> processor) { |
|
||||
EntityData ed = repository.getEntityMap(entityId.getEntityType()).get(entityId.getId()); |
|
||||
if (ed != null) { |
|
||||
if (ed.getCustomerId() != null) { |
|
||||
processor.accept(repository.getEntityMap(EntityType.CUSTOMER).get(ed.getCustomerId())); |
|
||||
} else { |
|
||||
processor.accept(repository.getEntityMap(EntityType.TENANT).get(repository.getTenantId().getId())); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
protected int getProbableResultSize() { |
|
||||
return 1; |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
Some files were not shown because too many files changed in this diff
Loading…
Reference in new issue