nickAS21
4 years ago
13 changed files with 433 additions and 288 deletions
@ -0,0 +1,266 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2022 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.service.entitiy.ruleChain; |
||||
|
|
||||
|
import lombok.AllArgsConstructor; |
||||
|
import org.springframework.stereotype.Service; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
import org.thingsboard.server.common.data.audit.ActionType; |
||||
|
import org.thingsboard.server.common.data.edge.Edge; |
||||
|
import org.thingsboard.server.common.data.exception.ThingsboardException; |
||||
|
import org.thingsboard.server.common.data.id.EdgeId; |
||||
|
import org.thingsboard.server.common.data.id.RuleChainId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.plugin.ComponentLifecycleEvent; |
||||
|
import org.thingsboard.server.common.data.rule.DefaultRuleChainCreateRequest; |
||||
|
import org.thingsboard.server.common.data.rule.RuleChain; |
||||
|
import org.thingsboard.server.common.data.rule.RuleChainMetaData; |
||||
|
import org.thingsboard.server.common.data.rule.RuleChainType; |
||||
|
import org.thingsboard.server.common.data.rule.RuleChainUpdateResult; |
||||
|
import org.thingsboard.server.common.data.rule.RuleNode; |
||||
|
import org.thingsboard.server.queue.util.TbCoreComponent; |
||||
|
import org.thingsboard.server.service.entitiy.AbstractTbEntityService; |
||||
|
import org.thingsboard.server.service.security.model.SecurityUser; |
||||
|
|
||||
|
import java.util.Collections; |
||||
|
import java.util.List; |
||||
|
import java.util.Set; |
||||
|
import java.util.stream.Collectors; |
||||
|
|
||||
|
@Service |
||||
|
@TbCoreComponent |
||||
|
@AllArgsConstructor |
||||
|
public class DefaultTbRuleChainNotifyService extends AbstractTbEntityService implements TbRuleChainNotifyService { |
||||
|
|
||||
|
@Override |
||||
|
public RuleChain save(RuleChain ruleChain, SecurityUser user) throws ThingsboardException { |
||||
|
TenantId tenantId = ruleChain.getTenantId(); |
||||
|
ActionType actionType = ruleChain.getId() == null ? ActionType.ADDED : ActionType.UPDATED; |
||||
|
try { |
||||
|
RuleChain savedRuleChain = checkNotNull(ruleChainService.saveRuleChain(ruleChain)); |
||||
|
|
||||
|
if (RuleChainType.CORE.equals(savedRuleChain.getType())) { |
||||
|
tbClusterService.broadcastEntityStateChangeEvent(tenantId, savedRuleChain.getId(), |
||||
|
actionType.equals(ActionType.ADDED) ? ComponentLifecycleEvent.CREATED : ComponentLifecycleEvent.UPDATED); |
||||
|
} |
||||
|
boolean isSendMsg = RuleChainType.EDGE.equals(savedRuleChain.getType()) && actionType.equals(ActionType.UPDATED); |
||||
|
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null, savedRuleChain.getId(), |
||||
|
savedRuleChain, user, actionType, isSendMsg, null); |
||||
|
return savedRuleChain; |
||||
|
} catch (Exception e) { |
||||
|
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null, emptyId(EntityType.RULE_CHAIN), |
||||
|
ruleChain, user, actionType, false, e); |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void delete(RuleChain ruleChain, SecurityUser user) throws ThingsboardException { |
||||
|
TenantId tenantId = ruleChain.getTenantId(); |
||||
|
RuleChainId ruleChainId = ruleChain.getId(); |
||||
|
try { |
||||
|
|
||||
|
|
||||
|
List<RuleNode> referencingRuleNodes = ruleChainService.getReferencingRuleChainNodes(tenantId, ruleChainId); |
||||
|
|
||||
|
Set<RuleChainId> referencingRuleChainIds = referencingRuleNodes.stream().map(RuleNode::getRuleChainId).collect(Collectors.toSet()); |
||||
|
|
||||
|
List<EdgeId> relatedEdgeIds = null; |
||||
|
if (RuleChainType.EDGE.equals(ruleChain.getType())) { |
||||
|
relatedEdgeIds = findRelatedEdgeIds(tenantId, ruleChainId); |
||||
|
} |
||||
|
|
||||
|
ruleChainService.deleteRuleChainById(tenantId, ruleChainId); |
||||
|
|
||||
|
referencingRuleChainIds.remove(ruleChain.getId()); |
||||
|
|
||||
|
if (RuleChainType.CORE.equals(ruleChain.getType())) { |
||||
|
referencingRuleChainIds.forEach(referencingRuleChainId -> |
||||
|
tbClusterService.broadcastEntityStateChangeEvent(tenantId, referencingRuleChainId, ComponentLifecycleEvent.UPDATED)); |
||||
|
|
||||
|
tbClusterService.broadcastEntityStateChangeEvent(tenantId, ruleChain.getId(), ComponentLifecycleEvent.DELETED); |
||||
|
} |
||||
|
|
||||
|
notificationEntityService.notifyDeleteRuleChane(tenantId, ruleChain, relatedEdgeIds, user); |
||||
|
} catch (Exception e) { |
||||
|
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null, emptyId(EntityType.RULE_CHAIN), |
||||
|
null, user, ActionType.DELETED, false, e, ruleChainId.toString()); |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public RuleChain saveRuleChain(TenantId tenantId, DefaultRuleChainCreateRequest request, SecurityUser user) throws ThingsboardException { |
||||
|
try { |
||||
|
RuleChain savedRuleChain = installScripts.createDefaultRuleChain(tenantId, request.getName()); |
||||
|
tbClusterService.broadcastEntityStateChangeEvent(tenantId, savedRuleChain.getId(), ComponentLifecycleEvent.CREATED); |
||||
|
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null, savedRuleChain.getId(), |
||||
|
savedRuleChain, user, ActionType.ADDED, false, null); |
||||
|
return savedRuleChain; |
||||
|
} catch (Exception e) { |
||||
|
RuleChain ruleChain = new RuleChain(); |
||||
|
ruleChain.setName(request.getName()); |
||||
|
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null, emptyId(EntityType.RULE_CHAIN), |
||||
|
ruleChain, user, ActionType.ADDED, false, e); |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public RuleChain setRootRuleChain(TenantId tenantId, RuleChain ruleChain, SecurityUser user) throws ThingsboardException { |
||||
|
RuleChain previousRootRuleChain = ruleChainService.getRootTenantRuleChain(tenantId); |
||||
|
RuleChainId ruleChainId = previousRootRuleChain.getId(); |
||||
|
try { |
||||
|
if (ruleChainService.setRootRuleChain(tenantId, ruleChainId)) { |
||||
|
if (previousRootRuleChain != null) { |
||||
|
previousRootRuleChain = ruleChainService.findRuleChainById(tenantId, previousRootRuleChain.getId()); |
||||
|
|
||||
|
tbClusterService.broadcastEntityStateChangeEvent(tenantId, previousRootRuleChain.getId(), |
||||
|
ComponentLifecycleEvent.UPDATED); |
||||
|
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null,ruleChainId, |
||||
|
previousRootRuleChain, user, ActionType.UPDATED, false, null); |
||||
|
} |
||||
|
ruleChain = ruleChainService.findRuleChainById(tenantId, ruleChainId); |
||||
|
|
||||
|
tbClusterService.broadcastEntityStateChangeEvent(tenantId, ruleChain.getId(), |
||||
|
ComponentLifecycleEvent.UPDATED); |
||||
|
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null, ruleChainId, |
||||
|
ruleChain, user, ActionType.UPDATED, false, null); |
||||
|
} |
||||
|
return ruleChain; |
||||
|
} catch (Exception e) { |
||||
|
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null, emptyId(EntityType.RULE_CHAIN), |
||||
|
ruleChain, user, ActionType.UPDATED, false, e, ruleChainId.toString()); |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public RuleChainMetaData saveRuleChainMetaData(TenantId tenantId, RuleChain ruleChain, RuleChainMetaData ruleChainMetaData, |
||||
|
boolean updateRelated, SecurityUser user) throws ThingsboardException { |
||||
|
RuleChainId ruleChainId = ruleChain.getId(); |
||||
|
try { |
||||
|
|
||||
|
RuleChainUpdateResult result = ruleChainService.saveRuleChainMetaData(tenantId, ruleChainMetaData); |
||||
|
checkNotNull(result.isSuccess() ? true : null); |
||||
|
|
||||
|
List<RuleChain> updatedRuleChains; |
||||
|
if (updateRelated && result.isSuccess()) { |
||||
|
updatedRuleChains = tbRuleChainService.updateRelatedRuleChains(tenantId, ruleChainMetaData.getRuleChainId(), result); |
||||
|
} else { |
||||
|
updatedRuleChains = Collections.emptyList(); |
||||
|
} |
||||
|
|
||||
|
RuleChainMetaData savedRuleChainMetaData = checkNotNull(ruleChainService.loadRuleChainMetaData(tenantId, ruleChainMetaData.getRuleChainId())); |
||||
|
|
||||
|
if (RuleChainType.CORE.equals(ruleChain.getType())) { |
||||
|
tbClusterService.broadcastEntityStateChangeEvent(ruleChain.getTenantId(), ruleChain.getId(), ComponentLifecycleEvent.UPDATED); |
||||
|
updatedRuleChains.forEach(updatedRuleChain -> { |
||||
|
tbClusterService.broadcastEntityStateChangeEvent(updatedRuleChain.getTenantId(), updatedRuleChain.getId(), ComponentLifecycleEvent.UPDATED); |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null, ruleChainId, |
||||
|
ruleChain, user, ActionType.UPDATED, false, null, ruleChainMetaData); |
||||
|
|
||||
|
if (RuleChainType.EDGE.equals(ruleChain.getType())) { |
||||
|
notificationEntityService.notifySaveRuleChainMetaData(tenantId, ruleChain); |
||||
|
} |
||||
|
|
||||
|
for (RuleChain updatedRuleChain : updatedRuleChains) { |
||||
|
RuleChainMetaData updatedRuleChainMetaData = checkNotNull(ruleChainService.loadRuleChainMetaData(tenantId, |
||||
|
updatedRuleChain.getId())); |
||||
|
notificationEntityService.notifySaveRuleChainMetaData(tenantId, ruleChain, updatedRuleChain, |
||||
|
updatedRuleChainMetaData, user); |
||||
|
} |
||||
|
|
||||
|
return savedRuleChainMetaData; |
||||
|
} catch (Exception e) { |
||||
|
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null, emptyId(EntityType.RULE_CHAIN), |
||||
|
null, user, ActionType.ADDED, false, e, ruleChainMetaData); |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public RuleChain assignRuleChainToEdge(TenantId tenantId, RuleChainId ruleChainId, Edge edge, SecurityUser user) throws ThingsboardException { |
||||
|
try { |
||||
|
RuleChain savedRuleChain = checkNotNull(ruleChainService.assignRuleChainToEdge(tenantId, ruleChainId, edge.getId())); |
||||
|
notificationEntityService.notifyAssignOrUnassignEntityToEdge(tenantId, ruleChainId, |
||||
|
null, edge.getId(), |
||||
|
savedRuleChain, ActionType.ASSIGNED_TO_EDGE, |
||||
|
user, ruleChainId.toString(), edge.getId().toString(), edge.getName()); |
||||
|
return savedRuleChain; |
||||
|
} catch (Exception e) { |
||||
|
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null, emptyId(EntityType.RULE_CHAIN), |
||||
|
null, user, ActionType.ASSIGNED_TO_EDGE, false, e, ruleChainId.toString(), edge.getId().toString()); |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public RuleChain unassignRuleChainFromEdge(TenantId tenantId, RuleChain ruleChain, Edge edge, SecurityUser user) throws ThingsboardException { |
||||
|
RuleChainId ruleChainId = ruleChain.getId(); |
||||
|
EdgeId edgeId = edge.getId(); |
||||
|
try { |
||||
|
RuleChain savedRuleChain = checkNotNull(ruleChainService.unassignRuleChainFromEdge(tenantId, ruleChainId, edgeId, false)); |
||||
|
notificationEntityService.notifyAssignOrUnassignEntityToEdge(tenantId, ruleChainId, |
||||
|
null, edge.getId(), |
||||
|
savedRuleChain, ActionType.UNASSIGNED_FROM_EDGE, |
||||
|
user, ruleChainId.toString(), edge.getId().toString(), edge.getName()); |
||||
|
return savedRuleChain; |
||||
|
} catch (Exception e) { |
||||
|
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null, emptyId(EntityType.RULE_CHAIN), |
||||
|
null, user, ActionType.UNASSIGNED_FROM_EDGE, false, e, ruleChainId.toString(), edge.getId().toString()); |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public RuleChain setEdgeRuleChain(TenantId tenantId, RuleChain ruleChain, SecurityUser user, boolean isSetEdgeTemplateRootRuleChain) throws ThingsboardException { |
||||
|
RuleChainId ruleChainId = ruleChain.getId(); |
||||
|
try { |
||||
|
if (isSetEdgeTemplateRootRuleChain) { |
||||
|
ruleChainService.setEdgeTemplateRootRuleChain(tenantId, ruleChainId); |
||||
|
} else { |
||||
|
ruleChainService.setAutoAssignToEdgeRuleChain(tenantId, ruleChainId); |
||||
|
} |
||||
|
|
||||
|
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null, ruleChainId, |
||||
|
ruleChain, user, ActionType.UPDATED, false, null); |
||||
|
return ruleChain; |
||||
|
} catch (Exception e) { |
||||
|
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null, emptyId(EntityType.RULE_CHAIN), |
||||
|
null, user, ActionType.UPDATED, false, e, ruleChainId.toString()); |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public RuleChain unsetAutoAssignToEdgeRuleChain(TenantId tenantId, RuleChain ruleChain, SecurityUser user) throws ThingsboardException { |
||||
|
RuleChainId ruleChainId = ruleChain.getId(); |
||||
|
try { |
||||
|
ruleChainService.unsetAutoAssignToEdgeRuleChain(tenantId, ruleChainId); |
||||
|
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null, ruleChainId, |
||||
|
ruleChain, user, ActionType.UPDATED, false, null); |
||||
|
return ruleChain; |
||||
|
} catch (Exception e) { |
||||
|
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null, emptyId(EntityType.RULE_CHAIN), |
||||
|
null, user, ActionType.UPDATED, false, e, ruleChainId.toString()); |
||||
|
throw handleException(e); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,44 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2022 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.service.entitiy.ruleChain; |
||||
|
|
||||
|
import org.thingsboard.server.common.data.edge.Edge; |
||||
|
import org.thingsboard.server.common.data.exception.ThingsboardException; |
||||
|
import org.thingsboard.server.common.data.id.RuleChainId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.rule.DefaultRuleChainCreateRequest; |
||||
|
import org.thingsboard.server.common.data.rule.RuleChain; |
||||
|
import org.thingsboard.server.common.data.rule.RuleChainMetaData; |
||||
|
import org.thingsboard.server.service.entitiy.SimpleTbEntityService; |
||||
|
import org.thingsboard.server.service.security.model.SecurityUser; |
||||
|
|
||||
|
public interface TbRuleChainNotifyService extends SimpleTbEntityService<RuleChain> { |
||||
|
|
||||
|
RuleChain saveRuleChain(TenantId tenantId, DefaultRuleChainCreateRequest request, SecurityUser user) throws ThingsboardException; |
||||
|
|
||||
|
RuleChain setRootRuleChain(TenantId tenantId, RuleChain ruleChain, SecurityUser user) throws ThingsboardException; |
||||
|
|
||||
|
RuleChainMetaData saveRuleChainMetaData(TenantId tenantId, RuleChain ruleChain, RuleChainMetaData ruleChainMetaData, |
||||
|
boolean updateRelated, SecurityUser user) throws ThingsboardException; |
||||
|
|
||||
|
RuleChain assignRuleChainToEdge(TenantId tenantId, RuleChainId ruleChainId, Edge edge, SecurityUser user) throws ThingsboardException; |
||||
|
|
||||
|
RuleChain unassignRuleChainFromEdge(TenantId tenantId, RuleChain ruleChain, Edge edge, SecurityUser user) throws ThingsboardException; |
||||
|
|
||||
|
RuleChain setEdgeRuleChain(TenantId tenantId, RuleChain ruleChain, SecurityUser user, boolean isSetEdgeTemplateRootRuleChain) throws ThingsboardException; |
||||
|
|
||||
|
RuleChain unsetAutoAssignToEdgeRuleChain(TenantId tenantId, RuleChain ruleChain, SecurityUser user) throws ThingsboardException; |
||||
|
} |
||||
Loading…
Reference in new issue