ch.rasc.wampspring.message.EventMessage.java Source code

Introduction

Here is the source code for ch.rasc.wampspring.message.EventMessage.java

Source

/**
 * Copyright 2014-2015 Ralph Schaer <ralphschaer@gmail.com>
 *
 * 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 ch.rasc.wampspring.message;

import java.io.IOException;
import java.io.StringWriter;
import java.util.Set;

import ch.rasc.wampspring.config.WampSession;

import com.fasterxml.jackson.core.JsonFactory;
import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.core.JsonParser;
import com.fasterxml.jackson.core.JsonToken;

/**
 * Subscribers receive PubSub events published by other components of the system via the
 * EVENT message. The EVENT message contains the topicURI, the topic under which the event
 * was published, and the event, the PubSub payload.
 *
 * <p>
 * Server-to-Client message
 *
 * @see <a href="http://wamp.ws/spec/wamp1/#event_message">WAMP specification</a>
 */
public class EventMessage extends PubSubMessage {

    private final Object event;

    private Set<String> excludeWebSocketSessionIds;

    private Set<String> eligibleWebSocketSessionIds;

    public EventMessage(EventMessage originEventMessage, String receiverWebSocketSessionId) {
        super(WampMessageType.EVENT, originEventMessage.getTopicURI());
        this.event = originEventMessage.getEvent();

        setWebSocketSessionId(receiverWebSocketSessionId);
        setPrincipal(originEventMessage.getPrincipal());
        setWampSession(originEventMessage.getWampSession());
    }

    public EventMessage(PublishMessage publishMessage, String receiverWebSocketSessionId) {
        super(WampMessageType.EVENT, publishMessage.getTopicURI());
        this.event = publishMessage.getEvent();

        setWebSocketSessionId(receiverWebSocketSessionId);
        setPrincipal(publishMessage.getPrincipal());
        setWampSession(publishMessage.getWampSession());
    }

    public EventMessage(String topicURI, Object event) {
        super(WampMessageType.EVENT, topicURI);
        this.event = event;
    }

    public EventMessage(JsonParser jp) throws IOException {
        this(jp, null);
    }

    public EventMessage(JsonParser jp, WampSession wampSession) throws IOException {
        super(WampMessageType.EVENT);

        if (jp.nextToken() != JsonToken.VALUE_STRING) {
            throw new IOException();
        }
        setTopicURI(replacePrefix(jp.getValueAsString(), wampSession));

        jp.nextToken();
        this.event = jp.readValueAs(Object.class);
    }

    public Object getEvent() {
        return this.event;
    }

    public Set<String> getExcludeWebSocketSessionIds() {
        return this.excludeWebSocketSessionIds;
    }

    public void setExcludeWebSocketSessionIds(Set<String> excludeSessionIds) {
        this.excludeWebSocketSessionIds = excludeSessionIds;
    }

    public Set<String> getEligibleWebSocketSessionIds() {
        return this.eligibleWebSocketSessionIds;
    }

    public void setEligibleWebSocketSessionIds(Set<String> eligibleSessionIds) {
        this.eligibleWebSocketSessionIds = eligibleSessionIds;
    }

    @Override
    public String toJson(JsonFactory jsonFactory) throws IOException {
        try (StringWriter sw = new StringWriter(); JsonGenerator jg = jsonFactory.createGenerator(sw)) {
            jg.writeStartArray();
            jg.writeNumber(getTypeId());
            jg.writeString(getTopicURI());
            jg.writeObject(this.event);
            jg.writeEndArray();
            jg.close();

            return sw.toString();
        }
    }

    @Override
    public String toString() {
        return "EventMessage [topicURI=" + getTopicURI() + ", event=" + this.event + "]";
    }

}