ProducerImpl.java

package de.rwth.idsg.bikeman.ixsi.endpoint;

import de.rwth.idsg.bikeman.ixsi.CommunicationContext;
import de.rwth.idsg.bikeman.ixsi.endpoint.Parser;
import de.rwth.idsg.bikeman.ixsi.endpoint.Producer;
import de.rwth.idsg.bikeman.ixsi.store.WebSocketSessionStore;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import xjc.schema.ixsi.IxsiMessageType;

import java.util.Set;

/**
 * Created by max on 08/09/14.
 */
@Slf4j
@Component
public class ProducerImpl implements Producer {

    @Autowired private Parser parser;
    @Autowired private WebSocketSessionStore webSocketSessionStore;

    @Override
    public void send(CommunicationContext context) {
        try {
            String str = parser.marshal(context.getOutgoingIxsi());
            TextMessage out = new TextMessage(str);

            WebSocketSession session = context.getSession();
            log.info("[id: {}] Sending message: {}", session.getId(), str);
            session.sendMessage(out);

        } catch (Exception e) {
            log.error("Exception happened", e);
        }
    }

    @Override
    public void send(IxsiMessageType ixsi, Set<String> systemIdSet) {
        String str = parser.marshal(ixsi);
        TextMessage out = new TextMessage(str);

        for (String systemId : systemIdSet) {
            push(systemId, out);
        }
    }

    /**
     * Push message exceptions should be handled silently, that is they should not take the ongoing process/thread or
     * system down. With this, we just log the exception and allow the application to continue with the next system in
     * line when the current push fails.
     */
    private void push(String systemId, TextMessage out) {
        try {
            WebSocketSession session = webSocketSessionStore.getNext(systemId);
            log.info("[id: {}] Sending message: {}", session.getId(), out.getPayload());
            session.sendMessage(out);

        } catch (Exception e) {
            log.error("Exception happened", e);
        }
    }
}