Andrii Shvaika
3 years ago
103 changed files with 996 additions and 928 deletions
@ -0,0 +1,79 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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; |
||||
|
|
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.stereotype.Service; |
||||
|
import org.thingsboard.server.common.data.HasName; |
||||
|
import org.thingsboard.server.common.data.User; |
||||
|
import org.thingsboard.server.common.data.audit.ActionType; |
||||
|
import org.thingsboard.server.common.data.id.CustomerId; |
||||
|
import org.thingsboard.server.common.data.id.EntityId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.relation.EntityRelation; |
||||
|
import org.thingsboard.server.service.action.EntityActionService; |
||||
|
|
||||
|
@Slf4j |
||||
|
@Service |
||||
|
@RequiredArgsConstructor |
||||
|
public class DefaultTbLogEntityActionService implements TbLogEntityActionService { |
||||
|
|
||||
|
private final EntityActionService entityActionService; |
||||
|
|
||||
|
@Override |
||||
|
public <I extends EntityId> void logEntityAction(TenantId tenantId, I entityId, ActionType actionType, |
||||
|
User user, Exception e, Object... additionalInfo) { |
||||
|
logEntityAction(tenantId, entityId, null, null, actionType, user, e, additionalInfo); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public <E extends HasName, I extends EntityId> void logEntityAction(TenantId tenantId, I entityId, E entity, |
||||
|
ActionType actionType, User user, Object... additionalInfo) { |
||||
|
logEntityAction(tenantId, entityId, entity, null, actionType, user, null, additionalInfo); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public <E extends HasName, I extends EntityId> void logEntityAction(TenantId tenantId, I entityId, E entity, |
||||
|
ActionType actionType, User user, Exception e, |
||||
|
Object... additionalInfo) { |
||||
|
logEntityAction(tenantId, entityId, entity, null, actionType, user, e, additionalInfo); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public <E extends HasName, I extends EntityId> void logEntityAction(TenantId tenantId, I entityId, E entity, CustomerId customerId, |
||||
|
ActionType actionType, User user, Object... additionalInfo) { |
||||
|
logEntityAction(tenantId, entityId, entity, customerId, actionType, user, null, additionalInfo); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public <E extends HasName, I extends EntityId> void logEntityAction(TenantId tenantId, I entityId, E entity, |
||||
|
CustomerId customerId, ActionType actionType, |
||||
|
User user, Exception e, Object... additionalInfo) { |
||||
|
if (user != null) { |
||||
|
entityActionService.logEntityAction(user, entityId, entity, customerId, actionType, e, additionalInfo); |
||||
|
} else if (e == null) { |
||||
|
entityActionService.pushEntityActionToRuleEngine(entityId, entity, tenantId, customerId, actionType, null, additionalInfo); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void logEntityRelationAction(TenantId tenantId, CustomerId customerId, EntityRelation relation, User user, |
||||
|
ActionType actionType, Exception e, Object... additionalInfo) { |
||||
|
logEntityAction(tenantId, relation.getFrom(), null, customerId, actionType, user, e, additionalInfo); |
||||
|
logEntityAction(tenantId, relation.getTo(), null, customerId, actionType, user, e, additionalInfo); |
||||
|
} |
||||
|
} |
||||
@ -1,212 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2024 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; |
|
||||
|
|
||||
import lombok.RequiredArgsConstructor; |
|
||||
import lombok.extern.slf4j.Slf4j; |
|
||||
import org.springframework.stereotype.Service; |
|
||||
import org.thingsboard.common.util.JacksonUtil; |
|
||||
import org.thingsboard.server.common.msg.rule.engine.DeviceCredentialsUpdateNotificationMsg; |
|
||||
import org.thingsboard.server.cluster.TbClusterService; |
|
||||
import org.thingsboard.server.common.data.Device; |
|
||||
import org.thingsboard.server.common.data.HasName; |
|
||||
import org.thingsboard.server.common.data.Tenant; |
|
||||
import org.thingsboard.server.common.data.User; |
|
||||
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.id.CustomerId; |
|
||||
import org.thingsboard.server.common.data.id.DeviceId; |
|
||||
import org.thingsboard.server.common.data.id.EdgeId; |
|
||||
import org.thingsboard.server.common.data.id.EntityId; |
|
||||
import org.thingsboard.server.common.data.id.TenantId; |
|
||||
import org.thingsboard.server.common.data.msg.TbMsgType; |
|
||||
import org.thingsboard.server.common.data.plugin.ComponentLifecycleEvent; |
|
||||
import org.thingsboard.server.common.data.relation.EntityRelation; |
|
||||
import org.thingsboard.server.common.data.security.DeviceCredentials; |
|
||||
import org.thingsboard.server.common.msg.TbMsg; |
|
||||
import org.thingsboard.server.common.msg.TbMsgDataType; |
|
||||
import org.thingsboard.server.common.msg.TbMsgMetaData; |
|
||||
import org.thingsboard.server.service.action.EntityActionService; |
|
||||
import org.thingsboard.server.service.gateway_device.GatewayNotificationsService; |
|
||||
|
|
||||
@Slf4j |
|
||||
@Service |
|
||||
@RequiredArgsConstructor |
|
||||
public class DefaultTbNotificationEntityService implements TbNotificationEntityService { |
|
||||
|
|
||||
private final EntityActionService entityActionService; |
|
||||
private final TbClusterService tbClusterService; |
|
||||
private final GatewayNotificationsService gatewayNotificationsService; |
|
||||
|
|
||||
@Override |
|
||||
public <I extends EntityId> void logEntityAction(TenantId tenantId, I entityId, ActionType actionType, |
|
||||
User user, Exception e, Object... additionalInfo) { |
|
||||
logEntityAction(tenantId, entityId, null, null, actionType, user, e, additionalInfo); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public <E extends HasName, I extends EntityId> void logEntityAction(TenantId tenantId, I entityId, E entity, |
|
||||
ActionType actionType, User user, Object... additionalInfo) { |
|
||||
logEntityAction(tenantId, entityId, entity, null, actionType, user, null, additionalInfo); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public <E extends HasName, I extends EntityId> void logEntityAction(TenantId tenantId, I entityId, E entity, |
|
||||
ActionType actionType, User user, Exception e, |
|
||||
Object... additionalInfo) { |
|
||||
logEntityAction(tenantId, entityId, entity, null, actionType, user, e, additionalInfo); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public <E extends HasName, I extends EntityId> void logEntityAction(TenantId tenantId, I entityId, E entity, CustomerId customerId, |
|
||||
ActionType actionType, User user, Object... additionalInfo) { |
|
||||
logEntityAction(tenantId, entityId, entity, customerId, actionType, user, null, additionalInfo); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public <E extends HasName, I extends EntityId> void logEntityAction(TenantId tenantId, I entityId, E entity, |
|
||||
CustomerId customerId, ActionType actionType, |
|
||||
User user, Exception e, Object... additionalInfo) { |
|
||||
if (user != null) { |
|
||||
entityActionService.logEntityAction(user, entityId, entity, customerId, actionType, e, additionalInfo); |
|
||||
} else if (e == null) { |
|
||||
entityActionService.pushEntityActionToRuleEngine(entityId, entity, tenantId, customerId, actionType, null, additionalInfo); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void notifyCreateOrUpdateTenant(Tenant tenant, ComponentLifecycleEvent event) { |
|
||||
tbClusterService.onTenantChange(tenant, null); |
|
||||
tbClusterService.broadcastEntityStateChangeEvent(tenant.getId(), tenant.getId(), event); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void notifyDeleteTenant(Tenant tenant) { |
|
||||
tbClusterService.onTenantDelete(tenant, null); |
|
||||
tbClusterService.broadcastEntityStateChangeEvent(tenant.getId(), tenant.getId(), ComponentLifecycleEvent.DELETED); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void notifyCreateOrUpdateDevice(TenantId tenantId, DeviceId deviceId, CustomerId customerId, |
|
||||
Device device, Device oldDevice, ActionType actionType, |
|
||||
User user, Object... additionalInfo) { |
|
||||
tbClusterService.onDeviceUpdated(device, oldDevice); |
|
||||
logEntityAction(tenantId, deviceId, device, customerId, actionType, user, additionalInfo); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void notifyDeleteDevice(TenantId tenantId, DeviceId deviceId, CustomerId customerId, Device device, |
|
||||
User user, Object... additionalInfo) { |
|
||||
gatewayNotificationsService.onDeviceDeleted(device); |
|
||||
tbClusterService.onDeviceDeleted(tenantId, device, null); |
|
||||
logEntityAction(tenantId, deviceId, device, customerId, ActionType.DELETED, user, additionalInfo); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void notifyUpdateDeviceCredentials(TenantId tenantId, DeviceId deviceId, CustomerId customerId, Device device, |
|
||||
DeviceCredentials deviceCredentials, User user) { |
|
||||
tbClusterService.pushMsgToCore(new DeviceCredentialsUpdateNotificationMsg(tenantId, deviceCredentials.getDeviceId(), deviceCredentials), null); |
|
||||
logEntityAction(tenantId, deviceId, device, customerId, ActionType.CREDENTIALS_UPDATED, user, deviceCredentials); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void notifyAssignDeviceToTenant(TenantId tenantId, TenantId newTenantId, DeviceId deviceId, CustomerId customerId, |
|
||||
Device device, Tenant tenant, User user, Object... additionalInfo) { |
|
||||
tbClusterService.onDeviceAssignedToTenant(tenantId, device); |
|
||||
logEntityAction(tenantId, deviceId, device, customerId, ActionType.ASSIGNED_TO_TENANT, user, additionalInfo); |
|
||||
pushAssignedFromNotification(tenant, newTenantId, device); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void notifyCreateOrUpdateOrDeleteEdge(TenantId tenantId, EdgeId edgeId, CustomerId customerId, Edge edge, |
|
||||
ActionType actionType, User user, Object... additionalInfo) { |
|
||||
ComponentLifecycleEvent lifecycleEvent; |
|
||||
switch (actionType) { |
|
||||
case ADDED: |
|
||||
lifecycleEvent = ComponentLifecycleEvent.CREATED; |
|
||||
break; |
|
||||
case UPDATED: |
|
||||
lifecycleEvent = ComponentLifecycleEvent.UPDATED; |
|
||||
break; |
|
||||
case DELETED: |
|
||||
lifecycleEvent = ComponentLifecycleEvent.DELETED; |
|
||||
break; |
|
||||
default: |
|
||||
throw new IllegalArgumentException("Unknown actionType: " + actionType); |
|
||||
} |
|
||||
tbClusterService.broadcastEntityStateChangeEvent(tenantId, edgeId, lifecycleEvent); |
|
||||
logEntityAction(tenantId, edgeId, edge, customerId, actionType, user, additionalInfo); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void logEntityRelationAction(TenantId tenantId, CustomerId customerId, EntityRelation relation, User user, |
|
||||
ActionType actionType, Exception e, Object... additionalInfo) { |
|
||||
logEntityAction(tenantId, relation.getFrom(), null, customerId, actionType, user, e, additionalInfo); |
|
||||
logEntityAction(tenantId, relation.getTo(), null, customerId, actionType, user, e, additionalInfo); |
|
||||
} |
|
||||
|
|
||||
private void pushAssignedFromNotification(Tenant currentTenant, TenantId newTenantId, Device assignedDevice) { |
|
||||
String data = JacksonUtil.toString(JacksonUtil.valueToTree(assignedDevice)); |
|
||||
if (data != null) { |
|
||||
TbMsg tbMsg = TbMsg.newMsg(TbMsgType.ENTITY_ASSIGNED_FROM_TENANT, assignedDevice.getId(), |
|
||||
assignedDevice.getCustomerId(), getMetaDataForAssignedFrom(currentTenant), TbMsgDataType.JSON, data); |
|
||||
tbClusterService.pushMsgToRuleEngine(newTenantId, assignedDevice.getId(), tbMsg, null); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private TbMsgMetaData getMetaDataForAssignedFrom(Tenant tenant) { |
|
||||
TbMsgMetaData metaData = new TbMsgMetaData(); |
|
||||
metaData.putValue("assignedFromTenantId", tenant.getId().getId().toString()); |
|
||||
metaData.putValue("assignedFromTenantName", tenant.getName()); |
|
||||
return metaData; |
|
||||
} |
|
||||
|
|
||||
public static EdgeEventActionType edgeTypeByActionType(ActionType actionType) { |
|
||||
switch (actionType) { |
|
||||
case ADDED: |
|
||||
return EdgeEventActionType.ADDED; |
|
||||
case UPDATED: |
|
||||
return EdgeEventActionType.UPDATED; |
|
||||
case ALARM_ACK: |
|
||||
return EdgeEventActionType.ALARM_ACK; |
|
||||
case ALARM_CLEAR: |
|
||||
return EdgeEventActionType.ALARM_CLEAR; |
|
||||
case ALARM_ASSIGNED: |
|
||||
return EdgeEventActionType.ALARM_ASSIGNED; |
|
||||
case ALARM_UNASSIGNED: |
|
||||
return EdgeEventActionType.ALARM_UNASSIGNED; |
|
||||
case DELETED: |
|
||||
return EdgeEventActionType.DELETED; |
|
||||
case RELATION_ADD_OR_UPDATE: |
|
||||
return EdgeEventActionType.RELATION_ADD_OR_UPDATE; |
|
||||
case RELATION_DELETED: |
|
||||
return EdgeEventActionType.RELATION_DELETED; |
|
||||
case ASSIGNED_TO_CUSTOMER: |
|
||||
return EdgeEventActionType.ASSIGNED_TO_CUSTOMER; |
|
||||
case UNASSIGNED_FROM_CUSTOMER: |
|
||||
return EdgeEventActionType.UNASSIGNED_FROM_CUSTOMER; |
|
||||
case ASSIGNED_TO_EDGE: |
|
||||
return EdgeEventActionType.ASSIGNED_TO_EDGE; |
|
||||
case UNASSIGNED_FROM_EDGE: |
|
||||
return EdgeEventActionType.UNASSIGNED_FROM_EDGE; |
|
||||
case CREDENTIALS_UPDATED: |
|
||||
return EdgeEventActionType.CREDENTIALS_UPDATED; |
|
||||
default: |
|
||||
return null; |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
@ -0,0 +1,266 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 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; |
||||
|
|
||||
|
import com.fasterxml.jackson.core.type.TypeReference; |
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.stereotype.Component; |
||||
|
import org.springframework.transaction.event.TransactionalEventListener; |
||||
|
import org.thingsboard.common.util.JacksonUtil; |
||||
|
import org.thingsboard.server.cluster.TbClusterService; |
||||
|
import org.thingsboard.server.common.data.ApiUsageState; |
||||
|
import org.thingsboard.server.common.data.Device; |
||||
|
import org.thingsboard.server.common.data.DeviceProfile; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
import org.thingsboard.server.common.data.TbResource; |
||||
|
import org.thingsboard.server.common.data.TbResourceInfo; |
||||
|
import org.thingsboard.server.common.data.Tenant; |
||||
|
import org.thingsboard.server.common.data.TenantProfile; |
||||
|
import org.thingsboard.server.common.data.audit.ActionType; |
||||
|
import org.thingsboard.server.common.data.edge.Edge; |
||||
|
import org.thingsboard.server.common.data.edge.EdgeEvent; |
||||
|
import org.thingsboard.server.common.data.id.DeviceId; |
||||
|
import org.thingsboard.server.common.data.id.EdgeId; |
||||
|
import org.thingsboard.server.common.data.id.EntityId; |
||||
|
import org.thingsboard.server.common.data.id.RuleChainId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.msg.TbMsgType; |
||||
|
import org.thingsboard.server.common.data.notification.NotificationRequest; |
||||
|
import org.thingsboard.server.common.data.plugin.ComponentLifecycleEvent; |
||||
|
import org.thingsboard.server.common.data.rule.RuleChain; |
||||
|
import org.thingsboard.server.common.data.rule.RuleChainType; |
||||
|
import org.thingsboard.server.common.data.security.DeviceCredentials; |
||||
|
import org.thingsboard.server.common.msg.TbMsg; |
||||
|
import org.thingsboard.server.common.msg.TbMsgDataType; |
||||
|
import org.thingsboard.server.common.msg.TbMsgMetaData; |
||||
|
import org.thingsboard.server.common.msg.rule.engine.DeviceCredentialsUpdateNotificationMsg; |
||||
|
import org.thingsboard.server.dao.eventsourcing.ActionEntityEvent; |
||||
|
import org.thingsboard.server.dao.eventsourcing.DeleteEntityEvent; |
||||
|
import org.thingsboard.server.dao.eventsourcing.SaveEntityEvent; |
||||
|
|
||||
|
import javax.annotation.PostConstruct; |
||||
|
import java.util.Set; |
||||
|
|
||||
|
@Component |
||||
|
@RequiredArgsConstructor |
||||
|
@Slf4j |
||||
|
public class EntityStateSourcingListener { |
||||
|
|
||||
|
private final TbClusterService tbClusterService; |
||||
|
|
||||
|
@PostConstruct |
||||
|
public void init() { |
||||
|
log.info("EntityStateSourcingListener initiated"); |
||||
|
} |
||||
|
|
||||
|
@TransactionalEventListener(fallbackExecution = true) |
||||
|
public void handleEvent(SaveEntityEvent<?> event) { |
||||
|
log.trace("[{}] SaveEntityEvent called: {}", event.getTenantId(), event); |
||||
|
TenantId tenantId = event.getTenantId(); |
||||
|
EntityId entityId = event.getEntityId(); |
||||
|
EntityType entityType = entityId.getEntityType(); |
||||
|
boolean isCreated = event.getAdded() != null && event.getAdded(); |
||||
|
ComponentLifecycleEvent lifecycleEvent = isCreated ? ComponentLifecycleEvent.CREATED : ComponentLifecycleEvent.UPDATED; |
||||
|
|
||||
|
switch (entityType) { |
||||
|
case ASSET: |
||||
|
case ASSET_PROFILE: |
||||
|
case ENTITY_VIEW: |
||||
|
case NOTIFICATION_RULE: |
||||
|
tbClusterService.broadcastEntityStateChangeEvent(tenantId, entityId, lifecycleEvent); |
||||
|
break; |
||||
|
case RULE_CHAIN: |
||||
|
RuleChain ruleChain = (RuleChain) event.getEntity(); |
||||
|
if (RuleChainType.CORE.equals(ruleChain.getType())) { |
||||
|
tbClusterService.broadcastEntityStateChangeEvent(ruleChain.getTenantId(), ruleChain.getId(), lifecycleEvent); |
||||
|
} |
||||
|
break; |
||||
|
case TENANT: |
||||
|
Tenant tenant = (Tenant) event.getEntity(); |
||||
|
onTenantUpdate(tenant, lifecycleEvent); |
||||
|
break; |
||||
|
case TENANT_PROFILE: |
||||
|
TenantProfile tenantProfile = (TenantProfile) event.getEntity(); |
||||
|
onTenantProfileUpdate(tenantProfile, lifecycleEvent); |
||||
|
break; |
||||
|
case DEVICE: |
||||
|
onDeviceUpdate(event.getEntity(), event.getOldEntity()); |
||||
|
break; |
||||
|
case DEVICE_PROFILE: |
||||
|
DeviceProfile deviceProfile = (DeviceProfile) event.getEntity(); |
||||
|
onDeviceProfileUpdate(deviceProfile, event.getOldEntity(), isCreated); |
||||
|
break; |
||||
|
case EDGE: |
||||
|
handleEdgeEvent(tenantId, entityId, event.getEntity(), lifecycleEvent); |
||||
|
break; |
||||
|
case TB_RESOURCE: |
||||
|
TbResource tbResource = (TbResource) event.getEntity(); |
||||
|
tbClusterService.onResourceChange(tbResource, null); |
||||
|
break; |
||||
|
case API_USAGE_STATE: |
||||
|
ApiUsageState apiUsageState = (ApiUsageState) event.getEntity(); |
||||
|
tbClusterService.onApiStateChange(apiUsageState, null); |
||||
|
break; |
||||
|
default: |
||||
|
break; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@TransactionalEventListener(fallbackExecution = true) |
||||
|
public void handleEvent(DeleteEntityEvent<?> event) { |
||||
|
log.trace("[{}] DeleteEntityEvent called: {}", event.getTenantId(), event); |
||||
|
TenantId tenantId = event.getTenantId(); |
||||
|
EntityId entityId = event.getEntityId(); |
||||
|
EntityType entityType = entityId.getEntityType(); |
||||
|
|
||||
|
switch (entityType) { |
||||
|
case ASSET: |
||||
|
case ASSET_PROFILE: |
||||
|
case ENTITY_VIEW: |
||||
|
case CUSTOMER: |
||||
|
case EDGE: |
||||
|
case NOTIFICATION_RULE: |
||||
|
tbClusterService.broadcastEntityStateChangeEvent(tenantId, entityId, ComponentLifecycleEvent.DELETED); |
||||
|
break; |
||||
|
case NOTIFICATION_REQUEST: |
||||
|
NotificationRequest request = (NotificationRequest) event.getEntity(); |
||||
|
if (request.isScheduled()) { |
||||
|
tbClusterService.broadcastEntityStateChangeEvent(tenantId, entityId, ComponentLifecycleEvent.DELETED); |
||||
|
} |
||||
|
break; |
||||
|
case RULE_CHAIN: |
||||
|
RuleChain ruleChain = (RuleChain) event.getEntity(); |
||||
|
if (RuleChainType.CORE.equals(ruleChain.getType())) { |
||||
|
Set<RuleChainId> referencingRuleChainIds = JacksonUtil.fromString(event.getBody(), new TypeReference<>() {}); |
||||
|
if (referencingRuleChainIds != null) { |
||||
|
referencingRuleChainIds.forEach(referencingRuleChainId -> |
||||
|
tbClusterService.broadcastEntityStateChangeEvent(tenantId, referencingRuleChainId, ComponentLifecycleEvent.UPDATED)); |
||||
|
} |
||||
|
tbClusterService.broadcastEntityStateChangeEvent(tenantId, ruleChain.getId(), ComponentLifecycleEvent.DELETED); |
||||
|
} |
||||
|
break; |
||||
|
case TENANT: |
||||
|
Tenant tenant = (Tenant) event.getEntity(); |
||||
|
onTenantDeleted(tenant); |
||||
|
break; |
||||
|
case TENANT_PROFILE: |
||||
|
TenantProfile tenantProfile = (TenantProfile) event.getEntity(); |
||||
|
tbClusterService.onTenantProfileDelete(tenantProfile, null); |
||||
|
break; |
||||
|
case DEVICE: |
||||
|
Device device = (Device) event.getEntity(); |
||||
|
tbClusterService.onDeviceDeleted(tenantId, device, null); |
||||
|
break; |
||||
|
case DEVICE_PROFILE: |
||||
|
DeviceProfile deviceProfile = (DeviceProfile) event.getEntity(); |
||||
|
onDeviceProfileDelete(event.getTenantId(), event.getEntityId(), deviceProfile); |
||||
|
break; |
||||
|
case TB_RESOURCE: |
||||
|
TbResourceInfo tbResource = (TbResourceInfo) event.getEntity(); |
||||
|
tbClusterService.onResourceDeleted(tbResource, null); |
||||
|
break; |
||||
|
default: |
||||
|
break; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@TransactionalEventListener(fallbackExecution = true) |
||||
|
public void handleEvent(ActionEntityEvent<?> event) { |
||||
|
log.trace("[{}] ActionEntityEvent called: {}", event.getTenantId(), event); |
||||
|
if (ActionType.CREDENTIALS_UPDATED.equals(event.getActionType()) && |
||||
|
EntityType.DEVICE.equals(event.getEntityId().getEntityType()) |
||||
|
&& event.getEntity() instanceof DeviceCredentials) { |
||||
|
tbClusterService.pushMsgToCore(new DeviceCredentialsUpdateNotificationMsg(event.getTenantId(), |
||||
|
(DeviceId) event.getEntityId(), (DeviceCredentials) event.getEntity()), null); |
||||
|
} else if (ActionType.ASSIGNED_TO_TENANT.equals(event.getActionType()) && event.getEntity() instanceof Device) { |
||||
|
Device device = (Device) event.getEntity(); |
||||
|
Tenant tenant = JacksonUtil.fromString(event.getBody(), Tenant.class); |
||||
|
if (tenant != null) { |
||||
|
tbClusterService.onDeviceAssignedToTenant(tenant.getId(), device); |
||||
|
} |
||||
|
pushAssignedFromNotification(tenant, event.getTenantId(), device); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void onTenantUpdate(Tenant tenant, ComponentLifecycleEvent lifecycleEvent) { |
||||
|
tbClusterService.onTenantChange(tenant, null); |
||||
|
tbClusterService.broadcastEntityStateChangeEvent(tenant.getId(), tenant.getId(), lifecycleEvent); |
||||
|
} |
||||
|
|
||||
|
private void onTenantDeleted(Tenant tenant) { |
||||
|
tbClusterService.onTenantDelete(tenant, null); |
||||
|
tbClusterService.broadcastEntityStateChangeEvent(tenant.getId(), tenant.getId(), ComponentLifecycleEvent.DELETED); |
||||
|
} |
||||
|
|
||||
|
private void onTenantProfileUpdate(TenantProfile tenantProfile, ComponentLifecycleEvent lifecycleEvent) { |
||||
|
tbClusterService.onTenantProfileChange(tenantProfile, null); |
||||
|
tbClusterService.broadcastEntityStateChangeEvent(TenantId.SYS_TENANT_ID, tenantProfile.getId(), lifecycleEvent); |
||||
|
} |
||||
|
|
||||
|
private void onDeviceProfileUpdate(DeviceProfile deviceProfile, Object oldEntity, boolean isCreated) { |
||||
|
DeviceProfile oldDeviceProfile = null; |
||||
|
if (!isCreated) { |
||||
|
oldDeviceProfile = getOldDeviceProfile(oldEntity); |
||||
|
} |
||||
|
tbClusterService.onDeviceProfileChange(deviceProfile, oldDeviceProfile, null); |
||||
|
} |
||||
|
|
||||
|
private DeviceProfile getOldDeviceProfile(Object oldEntity) { |
||||
|
if (oldEntity instanceof DeviceProfile) { |
||||
|
return (DeviceProfile) oldEntity; |
||||
|
} |
||||
|
return null; |
||||
|
} |
||||
|
|
||||
|
private void onDeviceProfileDelete(TenantId tenantId, EntityId entityId, DeviceProfile deviceProfile) { |
||||
|
tbClusterService.onDeviceProfileDelete(deviceProfile, null); |
||||
|
tbClusterService.broadcastEntityStateChangeEvent(tenantId, entityId, ComponentLifecycleEvent.DELETED); |
||||
|
} |
||||
|
|
||||
|
private void onDeviceUpdate(Object entity, Object oldEntity) { |
||||
|
Device device = (Device) entity; |
||||
|
Device oldDevice = null; |
||||
|
if (oldEntity instanceof Device) { |
||||
|
oldDevice = (Device) oldEntity; |
||||
|
} |
||||
|
tbClusterService.onDeviceUpdated(device, oldDevice); |
||||
|
} |
||||
|
|
||||
|
private void handleEdgeEvent(TenantId tenantId, EntityId entityId, Object entity, ComponentLifecycleEvent lifecycleEvent) { |
||||
|
if (entity instanceof Edge) { |
||||
|
tbClusterService.broadcastEntityStateChangeEvent(tenantId, entityId, lifecycleEvent); |
||||
|
} else if (entity instanceof EdgeEvent) { |
||||
|
tbClusterService.onEdgeEventUpdate(tenantId, (EdgeId) entityId); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void pushAssignedFromNotification(Tenant currentTenant, TenantId newTenantId, Device assignedDevice) { |
||||
|
String data = JacksonUtil.toString(JacksonUtil.valueToTree(assignedDevice)); |
||||
|
if (data != null) { |
||||
|
TbMsg tbMsg = TbMsg.newMsg(TbMsgType.ENTITY_ASSIGNED_FROM_TENANT, assignedDevice.getId(), |
||||
|
assignedDevice.getCustomerId(), getMetaDataForAssignedFrom(currentTenant), TbMsgDataType.JSON, data); |
||||
|
tbClusterService.pushMsgToRuleEngine(newTenantId, assignedDevice.getId(), tbMsg, null); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private TbMsgMetaData getMetaDataForAssignedFrom(Tenant tenant) { |
||||
|
TbMsgMetaData metaData = new TbMsgMetaData(); |
||||
|
metaData.putValue("assignedFromTenantId", tenant.getId().getId().toString()); |
||||
|
metaData.putValue("assignedFromTenantName", tenant.getName()); |
||||
|
return metaData; |
||||
|
} |
||||
|
} |
||||
Some files were not shown because too many files changed in this diff
Loading…
Reference in new issue