Improvement to have only one place where we schedule the session timeout message

application/src/main/java/org/thingsboard/server/actors/app/AppActor.java

@@ -18,9 +18,12 @@ package org.thingsboard.server.actors.app;
 import lombok.extern.slf4j.Slf4j;
 import org.thingsboard.server.actors.ActorSystemContext;
 import org.thingsboard.server.actors.TbActor;
+import org.thingsboard.server.actors.TbActorCtx;
+import org.thingsboard.server.actors.TbActorException;
 import org.thingsboard.server.actors.TbActorId;
 import org.thingsboard.server.actors.TbActorRef;
 import org.thingsboard.server.actors.TbEntityActorId;
+import org.thingsboard.server.actors.device.SessionTimeoutCheckMsg;
 import org.thingsboard.server.actors.service.ContextAwareActor;
 import org.thingsboard.server.actors.service.ContextBasedCreator;
 import org.thingsboard.server.actors.service.DefaultActorService;
@@ -64,6 +67,15 @@ public class AppActor extends ContextAwareActor {
         this.deletedTenants = new HashSet<>();
     }
 
+    @Override
+    public void init(TbActorCtx ctx) throws TbActorException {
+        super.init(ctx);
+        if (systemContext.getServiceInfoProvider().isService(ServiceType.TB_CORE)) {
+            systemContext.schedulePeriodicMsgWithDelay(ctx, SessionTimeoutCheckMsg.instance(),
+                    systemContext.getSessionReportTimeout(), systemContext.getSessionReportTimeout());
+        }
+    }
+
     @Override
     protected boolean doProcess(TbActorMsg msg) {
         if (!ruleChainsInitialized) {
@@ -101,6 +113,9 @@ public class AppActor extends ContextAwareActor {
             case EDGE_EVENT_UPDATE_TO_EDGE_SESSION_MSG:
                 onToTenantActorMsg((EdgeEventUpdateMsg) msg);
                 break;
+            case SESSION_TIMEOUT_MSG:
+                ctx.broadcastToChildrenByType(msg, EntityType.TENANT);
+                break;
             default:
                 return false;
         }

application/src/main/java/org/thingsboard/server/actors/tenant/TenantActor.java

@@ -85,8 +85,6 @@ public class TenantActor extends RuleChainManagerActor {
                 cantFindTenant = true;
                 log.info("[{}] Started tenant actor for missing tenant.", tenantId);
             } else {
-                systemContext.schedulePeriodicMsgWithDelay(ctx, SessionTimeoutCheckMsg.instance(), systemContext.getSessionReportTimeout(), systemContext.getSessionReportTimeout());
-
                 apiUsageState = new ApiUsageState(systemContext.getApiUsageStateService().getApiUsageState(tenant.getId()));
 
                 // This Service may be started for specific tenant only.
@@ -174,7 +172,7 @@ public class TenantActor extends RuleChainManagerActor {
                 onToDeviceActorMsg((DeviceAwareMsg) msg, true);
                 break;
             case SESSION_TIMEOUT_MSG:
-                broadcastToAllDeviceActors(msg);
+                ctx.broadcastToChildrenByType(msg, EntityType.DEVICE);
                 break;
             case RULE_CHAIN_INPUT_MSG:
             case RULE_CHAIN_OUTPUT_MSG:
@@ -190,11 +188,6 @@ public class TenantActor extends RuleChainManagerActor {
         return true;
     }
 
-    private void broadcastToAllDeviceActors(TbActorMsg msg) {
-        ctx.broadcastToChildren(msg, actorId -> actorId instanceof TbEntityActorId
-                && EntityType.DEVICE.equals(((TbEntityActorId) actorId).getEntityId().getEntityType()));
-    }
-
     private boolean isMyPartition(EntityId entityId) {
         return systemContext.resolve(ServiceType.TB_CORE, tenantId, entityId).isMyPartition();
     }

common/actor/src/main/java/org/thingsboard/server/actors/TbActorCtx.java

@@ -15,6 +15,7 @@
  */
 package org.thingsboard.server.actors;
 
+import org.thingsboard.server.common.data.EntityType;
 import org.thingsboard.server.common.msg.TbActorMsg;
 
 import java.util.List;
@@ -35,6 +36,8 @@ public interface TbActorCtx extends TbActorRef {
 
     void broadcastToChildren(TbActorMsg msg);
 
+    void broadcastToChildrenByType(TbActorMsg msg, EntityType entityType);
+
     void broadcastToChildren(TbActorMsg msg, Predicate<TbActorId> childFilter);
 
     List<TbActorId> filterChildren(Predicate<TbActorId> childFilter);

common/actor/src/main/java/org/thingsboard/server/actors/TbActorId.java

@@ -15,6 +15,16 @@
  */
 package org.thingsboard.server.actors;
 
+import org.thingsboard.server.common.data.EntityType;
+
 public interface TbActorId {
 
+    /**
+     * Returns entity type of the actor.
+     * May return null if the actor does not belong to any entity.
+     * This method is added for performance optimization.
+     *
+     */
+    EntityType getEntityType();
+
 }

common/actor/src/main/java/org/thingsboard/server/actors/TbActorMailbox.java

@@ -17,6 +17,7 @@ package org.thingsboard.server.actors;
 
 import lombok.Data;
 import lombok.extern.slf4j.Slf4j;
+import org.thingsboard.server.common.data.EntityType;
 import org.thingsboard.server.common.msg.MsgType;
 import org.thingsboard.server.common.msg.TbActorMsg;
 import org.thingsboard.server.common.msg.TbActorStopReason;
@@ -177,6 +178,11 @@ public final class TbActorMailbox implements TbActorCtx {
         system.broadcastToChildren(selfId, msg);
     }
 
+    @Override
+    public void broadcastToChildrenByType(TbActorMsg msg, EntityType entityType) {
+        broadcastToChildren(msg, actorId -> entityType.equals(actorId.getEntityType()));
+    }
+
     @Override
     public void broadcastToChildren(TbActorMsg msg, Predicate<TbActorId> childFilter) {
         system.broadcastToChildren(selfId, childFilter, msg);

common/actor/src/main/java/org/thingsboard/server/actors/TbEntityActorId.java

@@ -16,6 +16,7 @@
 package org.thingsboard.server.actors;
 
 import lombok.Getter;
+import org.thingsboard.server.common.data.EntityType;
 import org.thingsboard.server.common.data.id.EntityId;
 
 import java.util.Objects;
@@ -46,4 +47,9 @@ public class TbEntityActorId implements TbActorId {
     public int hashCode() {
         return Objects.hash(entityId);
     }
+
+    @Override
+    public EntityType getEntityType() {
+        return entityId.getEntityType();
+    }
 }

common/actor/src/main/java/org/thingsboard/server/actors/TbStringActorId.java

@@ -15,6 +15,8 @@
  */
 package org.thingsboard.server.actors;
 
+import org.thingsboard.server.common.data.EntityType;
+
 import java.util.Objects;
 
 public class TbStringActorId implements TbActorId {
@@ -42,4 +44,9 @@ public class TbStringActorId implements TbActorId {
     public int hashCode() {
         return Objects.hash(id);
     }
+
+    @Override
+    public EntityType getEntityType() {
+        return null;
+    }
 }