Search in sources :

Example 1 with EventSubscription

use of org.openremote.model.event.shared.EventSubscription in project openremote by openremote.

the class ClientEventService method init.

@Override
public void init(Container container) throws Exception {
    timerService = container.getService(TimerService.class);
    messageBrokerService = container.getService(MessageBrokerService.class);
    identityService = container.getService(ManagerIdentityService.class);
    eventSubscriptions = new EventSubscriptions(container.getService(TimerService.class), container.getService(ManagerExecutorService.class));
    MessageBrokerSetupService messageBrokerSetupService = container.getService(MessageBrokerSetupService.class);
    messageBrokerSetupService.getContext().getTypeConverterRegistry().addTypeConverters(new EventTypeConverters());
    messageBrokerSetupService.getContext().addRoutes(new RouteBuilder() {

        @Override
        public void configure() throws Exception {
            from("websocket://" + WEBSOCKET_EVENTS).routeId("FromClientWebsocketEvents").choice().when(header(WebsocketConstants.SESSION_OPEN)).process(exchange -> {
            // Do nothing except stop the exchanges
            }).stop().when(or(header(WebsocketConstants.SESSION_CLOSE), header(WebsocketConstants.SESSION_CLOSE_ERROR))).process(exchange -> {
                String sessionKey = getSessionKey(exchange);
                eventSubscriptions.cancelAll(sessionKey);
            }).stop().end().choice().when(bodyAs(String.class).startsWith(EventSubscription.MESSAGE_PREFIX)).convertBodyTo(EventSubscription.class).process(exchange -> {
                String sessionKey = getSessionKey(exchange);
                EventSubscription subscription = exchange.getIn().getBody(EventSubscription.class);
                AuthContext authContext = exchange.getIn().getHeader(Constants.AUTH_CONTEXT, AuthContext.class);
                if (eventSubscriptionAuthorizers.stream().anyMatch(authorizer -> authorizer.apply(authContext, subscription))) {
                    boolean restrictedUser = identityService.getIdentityProvider().isRestrictedUser(authContext.getUserId());
                    eventSubscriptions.update(sessionKey, restrictedUser, subscription);
                } else {
                    LOG.warning("Unauthorized subscription from '" + authContext.getUsername() + "' in realm '" + authContext.getAuthenticatedRealm() + "': " + subscription);
                    sendToSession(sessionKey, new UnauthorizedEventSubscription(subscription.getEventType()));
                }
            }).when(bodyAs(String.class).startsWith(CancelEventSubscription.MESSAGE_PREFIX)).convertBodyTo(CancelEventSubscription.class).process(exchange -> {
                String sessionKey = getSessionKey(exchange);
                eventSubscriptions.cancel(sessionKey, exchange.getIn().getBody(CancelEventSubscription.class));
            }).when(bodyAs(String.class).startsWith(SharedEvent.MESSAGE_PREFIX)).convertBodyTo(SharedEvent.class).process(exchange -> {
                SharedEvent event = exchange.getIn().getBody(SharedEvent.class);
                // If there is no timestamp in event, set to system time
                if (event.getTimestamp() <= 0) {
                    event.setTimestamp(timerService.getCurrentTimeMillis());
                }
            }).to(ClientEventService.CLIENT_EVENT_TOPIC).otherwise().process(exchange -> LOG.fine("Unsupported message body: " + exchange.getIn().getBody())).end();
            from(ClientEventService.CLIENT_EVENT_QUEUE).routeId("ToClientWebsocketEvents").choice().when(body().isInstanceOf(SharedEvent.class)).split(method(eventSubscriptions, "splitForSubscribers")).to("websocket://" + WEBSOCKET_EVENTS).end();
        }
    });
}
Also used : MessageBrokerService(org.openremote.container.message.MessageBrokerService) ManagerIdentityService(org.openremote.manager.security.ManagerIdentityService) AuthContext(org.openremote.container.security.AuthContext) Collection(java.util.Collection) CancelEventSubscription(org.openremote.model.event.shared.CancelEventSubscription) Exchange(org.apache.camel.Exchange) Constants(org.openremote.model.Constants) CopyOnWriteArraySet(java.util.concurrent.CopyOnWriteArraySet) Logger(java.util.logging.Logger) MessageBrokerSetupService(org.openremote.container.message.MessageBrokerSetupService) Container(org.openremote.container.Container) WebsocketConstants(org.openremote.container.web.socket.WebsocketConstants) SyslogEvent(org.openremote.model.syslog.SyslogEvent) RouteBuilder(org.apache.camel.builder.RouteBuilder) ContainerService(org.openremote.container.ContainerService) TimerService(org.openremote.container.timer.TimerService) UnauthorizedEventSubscription(org.openremote.model.event.shared.UnauthorizedEventSubscription) PredicateBuilder.or(org.apache.camel.builder.PredicateBuilder.or) ManagerExecutorService(org.openremote.manager.concurrent.ManagerExecutorService) EventSubscription(org.openremote.model.event.shared.EventSubscription) SharedEvent(org.openremote.model.event.shared.SharedEvent) Collections(java.util.Collections) CancelEventSubscription(org.openremote.model.event.shared.CancelEventSubscription) MessageBrokerSetupService(org.openremote.container.message.MessageBrokerSetupService) RouteBuilder(org.apache.camel.builder.RouteBuilder) AuthContext(org.openremote.container.security.AuthContext) UnauthorizedEventSubscription(org.openremote.model.event.shared.UnauthorizedEventSubscription) TimerService(org.openremote.container.timer.TimerService) ManagerIdentityService(org.openremote.manager.security.ManagerIdentityService) CancelEventSubscription(org.openremote.model.event.shared.CancelEventSubscription) UnauthorizedEventSubscription(org.openremote.model.event.shared.UnauthorizedEventSubscription) EventSubscription(org.openremote.model.event.shared.EventSubscription) SharedEvent(org.openremote.model.event.shared.SharedEvent) MessageBrokerService(org.openremote.container.message.MessageBrokerService)

Aggregations

Collection (java.util.Collection)1 Collections (java.util.Collections)1 CopyOnWriteArraySet (java.util.concurrent.CopyOnWriteArraySet)1 Logger (java.util.logging.Logger)1 Exchange (org.apache.camel.Exchange)1 PredicateBuilder.or (org.apache.camel.builder.PredicateBuilder.or)1 RouteBuilder (org.apache.camel.builder.RouteBuilder)1 Container (org.openremote.container.Container)1 ContainerService (org.openremote.container.ContainerService)1 MessageBrokerService (org.openremote.container.message.MessageBrokerService)1 MessageBrokerSetupService (org.openremote.container.message.MessageBrokerSetupService)1 AuthContext (org.openremote.container.security.AuthContext)1 TimerService (org.openremote.container.timer.TimerService)1 WebsocketConstants (org.openremote.container.web.socket.WebsocketConstants)1 ManagerExecutorService (org.openremote.manager.concurrent.ManagerExecutorService)1 ManagerIdentityService (org.openremote.manager.security.ManagerIdentityService)1 Constants (org.openremote.model.Constants)1 CancelEventSubscription (org.openremote.model.event.shared.CancelEventSubscription)1 EventSubscription (org.openremote.model.event.shared.EventSubscription)1 SharedEvent (org.openremote.model.event.shared.SharedEvent)1