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