Examples with Subscription org.thingsboard.server.extensions.core.plugin.telemetry.sub.Subscription used on opensource projects

Example 6 with Subscription

use of org.thingsboard.server.extensions.core.plugin.telemetry.sub.Subscription in project thingsboard by thingsboard.

the class SubscriptionManager method cleanupWsSessionSubscriptions.

private void cleanupWsSessionSubscriptions(PluginContext ctx, String sessionId, boolean localSession) {
    log.debug("[{}] Removing all subscriptions for particular session.", sessionId);
    Map<Integer, Subscription> sessionSubscriptions = subscriptionsByWsSessionId.get(sessionId);
    if (sessionSubscriptions != null) {
        int sessionSubscriptionSize = sessionSubscriptions.size();
        for (Subscription subscription : sessionSubscriptions.values()) {
            EntityId entityId = subscription.getEntityId();
            Set<Subscription> deviceSubscriptions = subscriptionsByEntityId.get(entityId);
            deviceSubscriptions.remove(subscription);
            if (deviceSubscriptions.isEmpty()) {
                subscriptionsByEntityId.remove(entityId);
            }
        }
        subscriptionsByWsSessionId.remove(sessionId);
        log.debug("[{}] Removed {} subscriptions for particular session.", sessionId, sessionSubscriptionSize);
        if (localSession) {
            notifyWsSubscriptionClosed(ctx, sessionId, sessionSubscriptions);
        }
    } else {
        log.debug("[{}] No subscriptions found!", sessionId);
    }
}

Example 7 with Subscription

use of org.thingsboard.server.extensions.core.plugin.telemetry.sub.Subscription in project thingsboard by thingsboard.

the class SubscriptionManager method processSubscriptionRemoval.

private void processSubscriptionRemoval(PluginContext ctx, String sessionId, Map<Integer, Subscription> sessionSubscriptions, Subscription subscription) {
    EntityId entityId = subscription.getEntityId();
    if (subscription.isLocal() && subscription.getServer() != null) {
        rpcHandler.onSubscriptionClose(ctx, subscription.getServer(), sessionId, subscription.getSubscriptionId());
    }
    if (sessionSubscriptions.isEmpty()) {
        log.debug("[{}] Removed last subscription for particular session.", sessionId);
        subscriptionsByWsSessionId.remove(sessionId);
    } else {
        log.debug("[{}] Removed session subscription.", sessionId);
    }
    Set<Subscription> deviceSubscriptions = subscriptionsByEntityId.get(entityId);
    if (deviceSubscriptions != null) {
        boolean result = deviceSubscriptions.remove(subscription);
        if (result) {
            if (deviceSubscriptions.size() == 0) {
                log.debug("[{}] Removed last subscription for particular device.", sessionId);
                subscriptionsByEntityId.remove(entityId);
            } else {
                log.debug("[{}] Removed device subscription.", sessionId);
            }
        } else {
            log.debug("[{}] Subscription not found!", sessionId);
        }
    } else {
        log.debug("[{}] No device subscriptions found!", sessionId);
    }
}

Example 8 with Subscription

use of org.thingsboard.server.extensions.core.plugin.telemetry.sub.Subscription in project thingsboard by thingsboard.

the class SubscriptionManager method addLocalWsSubscription.

public void addLocalWsSubscription(PluginContext ctx, String sessionId, EntityId entityId, SubscriptionState sub) {
    Optional<ServerAddress> server = ctx.resolve(entityId);
    Subscription subscription;
    if (server.isPresent()) {
        ServerAddress address = server.get();
        log.trace("[{}] Forwarding subscription [{}] for device [{}] to [{}]", sessionId, sub.getSubscriptionId(), entityId, address);
        subscription = new Subscription(sub, true, address);
        rpcHandler.onNewSubscription(ctx, address, sessionId, subscription);
    } else {
        log.trace("[{}] Registering local subscription [{}] for device [{}]", sessionId, sub.getSubscriptionId(), entityId);
        subscription = new Subscription(sub, true);
    }
    registerSubscription(sessionId, entityId, subscription);
}

Example 9 with Subscription

use of org.thingsboard.server.extensions.core.plugin.telemetry.sub.Subscription in project thingsboard by thingsboard.

the class SubscriptionManager method getSubscription.

private Optional<Subscription> getSubscription(String sessionId, int subscriptionId) {
    Subscription state = null;
    Map<Integer, Subscription> subMap = subscriptionsByWsSessionId.get(sessionId);
    if (subMap != null) {
        state = subMap.get(subscriptionId);
    }
    return Optional.ofNullable(state);
}

Example 10 with Subscription

use of org.thingsboard.server.extensions.core.plugin.telemetry.sub.Subscription in project thingsboard by thingsboard.

the class SubscriptionManager method checkSubsciptionsPrevAddress.

private void checkSubsciptionsPrevAddress(Set<Subscription> subscriptions) {
    Iterator<Subscription> subscriptionIterator = subscriptions.iterator();
    while (subscriptionIterator.hasNext()) {
        Subscription s = subscriptionIterator.next();
        if (s.isLocal()) {
            if (s.getServer() != null) {
                log.trace("[{}] Local subscription is no longer handled on remote server address [{}]", s.getWsSessionId(), s.getServer());
                s.setServer(null);
            }
        } else {
            log.trace("[{}] Remote subscription is on up to date server address.", s.getWsSessionId());
        }
    }
}