Search in sources :

Example 1 with AssetProcessingException

use of org.openremote.manager.asset.AssetProcessingException in project openremote by openremote.

the class GatewayService method processAssetUpdate.

@Override
public boolean processAssetUpdate(EntityManager em, Asset<?> asset, Attribute<?> attribute, AttributeEvent.Source source) throws AssetProcessingException {
    // If the update was generated by a gateway then we don't want to process it here
    if (source == AttributeEvent.Source.GATEWAY) {
        return false;
    }
    GatewayConnector connector = gatewayConnectorMap.get(asset.getId().toLowerCase(Locale.ROOT));
    if (connector != null) {
        LOG.fine("Attribute event for a locally registered gateway asset (Asset ID=" + asset.getId() + "): " + attribute);
        GatewayAsset gatewayAsset = (GatewayAsset) asset;
        // This is a change to a locally registered gateway
        if (GatewayAsset.DISABLED.getName().equals(attribute.getName())) {
            boolean disabled = attribute.getValueAs(Boolean.class).orElse(false);
            boolean isAlreadyDisabled = gatewayAsset.getDisabled().orElse(false);
            // Ensure we update state
            gatewayAsset.setDisabled(disabled);
            if (disabled != isAlreadyDisabled) {
                createUpdateGatewayServiceUser(gatewayAsset);
                if (disabled) {
                    connector.sendMessageToGateway(new GatewayDisconnectEvent(GatewayDisconnectEvent.Reason.DISABLED));
                }
                connector.setDisabled(disabled);
            }
        }
        if (GatewayAsset.CLIENT_SECRET.getName().equals(attribute.getName())) {
            String newSecret = attribute.getValueAs(String.class).orElse(null);
            if (!TextUtil.isNullOrEmpty(newSecret)) {
                LOG.fine("Gateway client secret attribute updated so updating gateway service user secret to match: (Gateway ID=" + asset.getId() + ")");
                User gatewayServiceUser = identityProvider.getUserByUsername(asset.getRealm(), User.SERVICE_ACCOUNT_PREFIX + ((GatewayAsset) asset).getClientId().orElse(""));
                if (gatewayServiceUser != null) {
                    identityProvider.resetSecret(asset.getRealm(), gatewayServiceUser.getId(), newSecret);
                } else {
                    LOG.info("Couldn't retrieve gateway service user to update secret: (Gateway ID=" + asset.getId() + ")");
                }
            } else {
                // Push old secret back
                assetProcessingService.sendAttributeEvent(new AttributeEvent(asset.getId(), GatewayAsset.CLIENT_SECRET, gatewayAsset.getClientSecret().orElseThrow(() -> new IllegalStateException("Gateway client secret is null which was not expected"))));
            }
        }
    } else {
        String gatewayId = assetIdGatewayIdMap.get(asset.getId());
        if (gatewayId != null) {
            LOG.fine("Attribute event for a gateway descendant asset (Asset<?> ID=" + asset.getId() + ", Gateway ID=" + gatewayId + "): " + attribute);
            connector = gatewayConnectorMap.get(gatewayId.toLowerCase(Locale.ROOT));
            if (connector == null) {
                LOG.warning("Gateway not found for descendant asset, this should not happen!!! (Asset<?> ID=" + asset.getId() + ", Gateway ID=" + gatewayId + ")");
            } else {
                if (!connector.isConnected()) {
                    LOG.info("Gateway is not connected so attribute event for descendant asset will be dropped (Asset<?> ID=" + asset.getId() + ", Gateway ID=" + gatewayId + "): " + attribute);
                    throw new AssetProcessingException(AttributeWriteFailure.GATEWAY_DISCONNECTED, "Gateway is not connected: Gateway ID=" + connector.gatewayId);
                }
                LOG.fine("Attribute event for a gateway descendant asset being forwarded to the gateway (Asset<?> ID=" + asset.getId() + ", Gateway ID=" + gatewayId + "): " + attribute);
                connector.sendMessageToGateway(new AttributeEvent(mapAssetId(gatewayId, asset.getId(), true), attribute.getName(), attribute.getValue().orElse(null), attribute.getTimestamp().orElse(0L)).setParentId(mapAssetId(gatewayId, asset.getParentId(), true)).setRealm(asset.getRealm()));
            }
            // Consume this event as it is for a gateway descendant and we've sent it to that gateway for processing
            return true;
        }
    }
    // Don't consume event for non gateway descendant assets
    return false;
}
Also used : User(org.openremote.model.security.User) GatewayDisconnectEvent(org.openremote.model.gateway.GatewayDisconnectEvent) GatewayAsset(org.openremote.model.asset.impl.GatewayAsset) AtomicBoolean(java.util.concurrent.atomic.AtomicBoolean) AssetProcessingException(org.openremote.manager.asset.AssetProcessingException)

Aggregations

AtomicBoolean (java.util.concurrent.atomic.AtomicBoolean)1 AssetProcessingException (org.openremote.manager.asset.AssetProcessingException)1 GatewayAsset (org.openremote.model.asset.impl.GatewayAsset)1 GatewayDisconnectEvent (org.openremote.model.gateway.GatewayDisconnectEvent)1 User (org.openremote.model.security.User)1