OpenDaylightOFFlowJSONSerializer.java

package org.opennaas.extensions.openflowswitch.driver.opendaylight.protocol.client.serializers.json;

import java.io.IOException;
import java.util.List;
import org.codehaus.jackson.JsonGenerator;
import org.codehaus.jackson.JsonProcessingException;
import org.codehaus.jackson.map.JsonSerializer;
import org.codehaus.jackson.map.SerializerProvider;
import org.opennaas.extensions.openflowswitch.model.FloodlightOFAction;
import org.opennaas.extensions.openflowswitch.model.FloodlightOFMatch;
import org.opennaas.extensions.openflowswitch.model.OpenDaylightOFFlow;

public class OpenDaylightOFFlowJSONSerializer extends
        JsonSerializer<OpenDaylightOFFlow> {

    @Override
    public void serialize(OpenDaylightOFFlow flow, JsonGenerator jGen,
            SerializerProvider serializer) throws IOException,
            JsonProcessingException {

        jGen.writeStartObject();
        jGen.writeStringField("installInHw", "true");
        jGen.writeObjectFieldStart("node");
        if (flow.getSwitchId() != null) {
            jGen.writeStringField("id", flow.getSwitchId());
            jGen.writeStringField("type", "OF");
        }
        jGen.writeEndObject();
        if (flow.getName() != null)
            jGen.writeStringField("name", flow.getName());
        if (flow.getPriority() != null)
            jGen.writeStringField("priority", flow.getPriority());
        jGen.writeStringField("active", String.valueOf(flow.isActive()));

        if (flow.getMatch() != null)
            serializeMatch(flow.getMatch(), jGen, serializer);
        jGen.writeFieldName("actions");
        jGen.writeStartArray();//[
        if (flow.getActions() != null) {
            serializeActions(flow.getActions(), jGen, serializer);
        }
        jGen.writeEndArray();
        jGen.writeEndObject();
    }

    /**
     * Fields in a Match are serialized as fields in the flow (no Match object
     * separators)
     *
     * @param match
     * @param jGen
     * @param serializer
     * @throws IOException
     * @throws JsonProcessingException
     */
    private void serializeMatch(FloodlightOFMatch match, JsonGenerator jGen,
            SerializerProvider serializer) throws IOException,
            JsonProcessingException {

        if (match == null) {
            return;
        }

        /*		if (match.getWildcards() != null && !match.getWildcards().isEmpty())
         jGen.writeStringField("wildcards", match.getWildcards());
         */
        if (match.getIngressPort() != null && !match.getIngressPort().isEmpty())
            jGen.writeStringField("ingressPort", match.getIngressPort());

        if (match.getSrcMac() != null && !match.getSrcMac().isEmpty())
            jGen.writeStringField("dlSrc", match.getSrcMac());

        if (match.getDstMac() != null && !match.getDstMac().isEmpty())
            jGen.writeStringField("dlDst", match.getDstMac());

        if (match.getVlanId() != null && !match.getVlanId().isEmpty())
            jGen.writeStringField("vlan-id", match.getVlanId());

        if (match.getVlanPriority() != null && !match.getVlanPriority().isEmpty())
            jGen.writeStringField("vlan-priority", match.getVlanPriority());

        if (match.getEtherType() != null && !match.getEtherType().isEmpty())
            jGen.writeStringField("etherType", match.getEtherType());

        if (match.getTosBits() != null && !match.getTosBits().isEmpty())
            jGen.writeStringField("tos-bits", match.getTosBits());

        if (match.getProtocol() != null && !match.getProtocol().isEmpty())
            jGen.writeStringField("protocol", match.getProtocol());
            
        if (match.getSrcIp() != null && !match.getSrcIp().isEmpty())
            jGen.writeStringField("nwSrc", match.getSrcIp());

        if (match.getDstIp() != null && !match.getDstIp().isEmpty())
            jGen.writeStringField("nwDst", match.getDstIp());

        if (match.getSrcPort() != null && !match.getSrcPort().isEmpty())
            jGen.writeStringField("src-port", match.getSrcPort());

        if (match.getDstPort() != null && !match.getDstPort().isEmpty())
            jGen.writeStringField("dst-port", match.getDstPort());
    }

    /**
     * The list of actions is serialized as a comma separated list of type=value
     * elements: "actions":"output=2,set-vlan-id=1"
     *
     * @param actions
     * @param jGen
     * @param serializer
     * @throws IOException
     * @throws JsonProcessingException
     */
    private void serializeActions(List<FloodlightOFAction> actions, JsonGenerator jGen,
            SerializerProvider serializer) throws IOException,
            JsonProcessingException {

        if (actions == null || actions.isEmpty())
            return;

        StringBuilder sb = new StringBuilder();
        for (FloodlightOFAction action : actions) {
            sb.append(",");
            sb.append(action.getType());
            if (action.getValue() != null) {
                sb.append("=");
                sb.append(action.getValue());
            }
        }
        sb.deleteCharAt(0); // delete first comma
        jGen.writeString(sb.toString());
    }

    @Override
    public Class<OpenDaylightOFFlow> handledType() {
        return OpenDaylightOFFlow.class;
    }
}