net.sf.jrtps.builtin.SubscriptionDataMarshaller Maven / Gradle / Ivy

Go to download
package net.sf.jrtps.builtin;

import java.util.Set;

import net.sf.jrtps.InconsistentPolicy;
import net.sf.jrtps.Marshaller;
import net.sf.jrtps.message.DataEncapsulation;
import net.sf.jrtps.message.ParameterListEncapsulation;
import net.sf.jrtps.message.parameter.BuiltinTopicKey;
import net.sf.jrtps.message.parameter.DataReaderPolicy;
import net.sf.jrtps.message.parameter.Parameter;
import net.sf.jrtps.message.parameter.ParameterList;
import net.sf.jrtps.message.parameter.QosPolicy;
import net.sf.jrtps.message.parameter.Sentinel;
import net.sf.jrtps.message.parameter.TopicName;
import net.sf.jrtps.message.parameter.TypeName;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Marshaller for builtin data for topic DCPSSubscription. With jRTPS,
 * instances of this topic is of type SubscriptionData.
 * 
 * @author mcr70
 * 
 */
public class SubscriptionDataMarshaller implements Marshaller {
    private static final Logger log = LoggerFactory.getLogger(SubscriptionDataMarshaller.class);

    /**
     * SubscriptionData has always a key.
     * 
     * @return true
     */
    @Override
    public boolean hasKey() {
        return true; // Always true. Key is PID_KEY_HASH.
    }

    /**
     * Extracts the key from PublicationData. Guid of the reader represented
     * by SubscriptionData is the key.
     * 
     * @return Guid as byte array
     */
    @Override
    public byte[] extractKey(SubscriptionData data) {
        return data.getBuiltinTopicKey().getBytes();
    }

    @Override
    public SubscriptionData unmarshall(DataEncapsulation data) {
        ParameterListEncapsulation plEnc = (ParameterListEncapsulation) data;
        SubscriptionData rd = null;
        try {
            rd = new SubscriptionData(plEnc.getParameterList());
        } catch (InconsistentPolicy e) {
            log.error("Could not resolve inconsistent policies for ReaderData", e);
        }

        return rd;
    }

    @Override
    public DataEncapsulation marshall(SubscriptionData rd) {
        ParameterList payloadParams = new ParameterList();

        payloadParams.add(new BuiltinTopicKey(rd.getBuiltinTopicKey()));
        payloadParams.add(new TopicName(rd.getTopicName()));
        payloadParams.add(new TypeName(rd.getTypeName()));

        addQoS(rd, payloadParams);
        payloadParams.add(new Sentinel());

        return new ParameterListEncapsulation(payloadParams);
    }

    private void addQoS(SubscriptionData rd, ParameterList payloadParams) {
        Set policies = rd.getQualityOfService().getReaderPolicies();
        for (QosPolicy qp : policies) {
            payloadParams.add((Parameter) qp);
        }
    }
}