nickAS21
4 years ago
8 changed files with 310 additions and 384 deletions
@ -1,284 +0,0 @@ |
|||||
/** |
|
||||
* 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.edge.EdgeEventActionType; |
|
||||
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.notifyDeleteRuleChain(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 saveDefaultByName(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 previousRootRuleChainId = previousRootRuleChain.getId(); |
|
||||
RuleChainId ruleChainId = ruleChain.getId(); |
|
||||
try { |
|
||||
if (ruleChainService.setRootRuleChain(tenantId, ruleChainId)) { |
|
||||
if (previousRootRuleChain != null) { |
|
||||
previousRootRuleChain = ruleChainService.findRuleChainById(tenantId, previousRootRuleChainId); |
|
||||
|
|
||||
tbClusterService.broadcastEntityStateChangeEvent(tenantId, previousRootRuleChainId, |
|
||||
ComponentLifecycleEvent.UPDATED); |
|
||||
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null, previousRootRuleChainId, |
|
||||
previousRootRuleChain, user, ActionType.UPDATED, false, null); |
|
||||
} |
|
||||
ruleChain = ruleChainService.findRuleChainById(tenantId, ruleChainId); |
|
||||
|
|
||||
tbClusterService.broadcastEntityStateChangeEvent(tenantId, ruleChainId, |
|
||||
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(); |
|
||||
RuleChainId ruleChainMetaDataId = ruleChainMetaData.getRuleChainId(); |
|
||||
try { |
|
||||
RuleChainUpdateResult result = ruleChainService.saveRuleChainMetaData(tenantId, ruleChainMetaData); |
|
||||
checkNotNull(result.isSuccess() ? true : null); |
|
||||
|
|
||||
List<RuleChain> updatedRuleChains; |
|
||||
if (updateRelated && result.isSuccess()) { |
|
||||
updatedRuleChains = tbRuleChainService.updateRelatedRuleChains(tenantId, ruleChainMetaDataId, result); |
|
||||
} else { |
|
||||
updatedRuleChains = Collections.emptyList(); |
|
||||
} |
|
||||
|
|
||||
RuleChainMetaData savedRuleChainMetaData = checkNotNull(ruleChainService.loadRuleChainMetaData(tenantId, ruleChainMetaDataId)); |
|
||||
|
|
||||
if (RuleChainType.CORE.equals(ruleChain.getType())) { |
|
||||
tbClusterService.broadcastEntityStateChangeEvent(tenantId, ruleChainId, ComponentLifecycleEvent.UPDATED); |
|
||||
updatedRuleChains.forEach(updatedRuleChain -> { |
|
||||
tbClusterService.broadcastEntityStateChangeEvent(tenantId, updatedRuleChain.getId(), ComponentLifecycleEvent.UPDATED); |
|
||||
}); |
|
||||
} |
|
||||
|
|
||||
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null, ruleChainId, |
|
||||
ruleChain, user, ActionType.UPDATED, false, null, ruleChainMetaData); |
|
||||
|
|
||||
if (RuleChainType.EDGE.equals(ruleChain.getType())) { |
|
||||
notificationEntityService.notifySendMsgToEdgeService(tenantId, ruleChain, EdgeEventActionType.UPDATED); |
|
||||
} |
|
||||
|
|
||||
updatedRuleChains.forEach(updatedRuleChain -> { |
|
||||
if (RuleChainType.EDGE.equals(ruleChain.getType())) { |
|
||||
notificationEntityService.notifySendMsgToEdgeService(tenantId, updatedRuleChain, EdgeEventActionType.UPDATED); |
|
||||
} else { |
|
||||
try { |
|
||||
RuleChainMetaData updatedRuleChainMetaData = checkNotNull(ruleChainService.loadRuleChainMetaData(tenantId, updatedRuleChain.getId())); |
|
||||
notificationEntityService.notifyCreateOrUpdateOrDelete(tenantId, null, updatedRuleChain.getId(), |
|
||||
updatedRuleChain, user, ActionType.UPDATED, false, null, updatedRuleChainMetaData); |
|
||||
} catch (ThingsboardException e) { |
|
||||
e.printStackTrace(); |
|
||||
} |
|
||||
} |
|
||||
}); |
|
||||
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, RuleChain ruleChain, Edge edge, SecurityUser user) throws ThingsboardException { |
|
||||
RuleChainId ruleChainId = ruleChain.getId(); |
|
||||
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 unassignRuleChainToEdge(TenantId tenantId, RuleChain ruleChain, Edge edge, SecurityUser user) throws ThingsboardException { |
|
||||
RuleChainId ruleChainId = ruleChain.getId(); |
|
||||
try { |
|
||||
RuleChain savedRuleChain = checkNotNull(ruleChainService.unassignRuleChainFromEdge(tenantId, ruleChainId, edge.getId(), 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 setEdgeTemplateRootRuleChain(TenantId tenantId, RuleChain ruleChain, SecurityUser user) throws ThingsboardException { |
|
||||
RuleChainId ruleChainId = ruleChain.getId(); |
|
||||
try { |
|
||||
ruleChainService.setEdgeTemplateRootRuleChain(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 setAutoAssignToEdgeRuleChain(TenantId tenantId, RuleChain ruleChain, SecurityUser user) throws ThingsboardException { |
|
||||
RuleChainId ruleChainId = ruleChain.getId(); |
|
||||
try { |
|
||||
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); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
@ -1,46 +0,0 @@ |
|||||
/** |
|
||||
* 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.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 saveDefaultByName(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, RuleChain ruleChain, Edge edge, |
|
||||
SecurityUser user) throws ThingsboardException; |
|
||||
RuleChain unassignRuleChainToEdge(TenantId tenantId, RuleChain ruleChain, Edge edge, |
|
||||
SecurityUser user) throws ThingsboardException; |
|
||||
|
|
||||
RuleChain setEdgeTemplateRootRuleChain(TenantId tenantId, RuleChain ruleChain, SecurityUser user) throws ThingsboardException; |
|
||||
|
|
||||
RuleChain setAutoAssignToEdgeRuleChain(TenantId tenantId, RuleChain ruleChain, SecurityUser user) throws ThingsboardException; |
|
||||
|
|
||||
RuleChain unsetAutoAssignToEdgeRuleChain(TenantId tenantId, RuleChain ruleChain, SecurityUser user) throws ThingsboardException; |
|
||||
} |
|
||||
Loading…
Reference in new issue