SetTpSrcActionSerializer.java example

Explorer
openflowplugin-master
/*
 * Copyright (c) 2016 Pantheon Technologies s.r.o. and others.  All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 */

package org.opendaylight.openflowplugin.impl.protocol.serialization.actions;

import java.util.Optional;
import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.common.IPProtocols;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.Action;
import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.action.SetFieldCase;
import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.action.SetFieldCaseBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.action.SetTpSrcActionCase;
import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.action.set.field._case.SetFieldBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.action.set.tp.src.action._case.SetTpSrcAction;
import org.opendaylight.yang.gen.v1.urn.opendaylight.model.match.types.rev131026.match.Icmpv4MatchBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.model.match.types.rev131026.match.Icmpv6MatchBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.model.match.types.rev131026.match.layer._4.match.TcpMatchBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.model.match.types.rev131026.match.layer._4.match.UdpMatchBuilder;

public class SetTpSrcActionSerializer extends AbstractSetFieldActionSerializer {

    @Override
    protected SetFieldCase buildAction(Action input) {
        final SetTpSrcAction setTpSrcAction = SetTpSrcActionCase.class.cast(input).getSetTpSrcAction();
        final PortNumber port = setTpSrcAction.getPort();
        final SetFieldBuilder builder = new SetFieldBuilder();

        Optional.ofNullable(IPProtocols.fromProtocolNum(setTpSrcAction.getIpProtocol())).ifPresent(proto -> {
            switch (proto) {
                case ICMP: {
                    builder.setIcmpv4Match(new Icmpv4MatchBuilder()
                            .setIcmpv4Type((short) (0xFF & port.getValue()))
                            .build());
                    break;
                }
                case ICMPV6: {
                    builder.setIcmpv6Match(new Icmpv6MatchBuilder()
                            .setIcmpv6Type((short) (0xFF & port.getValue()))
                            .build());
                    break;
                }
                case TCP: {
                    builder.setLayer4Match(new TcpMatchBuilder()
                            .setTcpSourcePort(port)
                            .build());
                    break;
                }
                case UDP: {
                    builder.setLayer4Match(new UdpMatchBuilder()
                            .setUdpSourcePort(port)
                            .build());
                    break;
                }
            }
        });

        return new SetFieldCaseBuilder().setSetField(builder.build()).build();
    }

}