openstack/net-virt-sfc/it/src/test/java/org/opendaylight/ovsdb/openstack/netvirt/sfc/utils/ServiceFunctionUtils.java

   1 /*
   2  * Copyright © 2015 Red Hat, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.ovsdb.openstack.netvirt.sfc.utils;
  10
  11 import java.util.ArrayList;
  12 import java.util.List;
  13 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.common.rev151017.SfDataPlaneLocatorName;
  14 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.common.rev151017.SfName;
  15 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.common.rev151017.SffName;
  16 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.sf.rev140701.ServiceFunctionsBuilder;
  17 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.sf.rev140701.service.function.entry.SfDataPlaneLocator;
  18 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.sf.rev140701.service.function.entry.SfDataPlaneLocatorBuilder;
  19 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.sf.rev140701.service.functions.ServiceFunction;
  20 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.sf.rev140701.service.functions.ServiceFunctionBuilder;
  21 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.sft.rev140701.Firewall;
  22 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.sft.rev140701.ServiceFunctionTypeIdentity;
  23 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.sl.rev140701.VxlanGpe;
  24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
  25
  26 public class ServiceFunctionUtils extends AbstractUtils {
  27     public SfDataPlaneLocatorBuilder sfDataPlaneLocatorBuilder(SfDataPlaneLocatorBuilder sfDataPlaneLocatorBuilder,
  28                                                                String ip, int port, String dplName, String sffName) {
  29         return sfDataPlaneLocatorBuilder
  30                 .setLocatorType(ipBuilder(ip, port).build())
  31                 .setName(new SfDataPlaneLocatorName(dplName))
  32                 .setTransport(VxlanGpe.class)
  33                 .setServiceFunctionForwarder(new SffName(sffName));
  34     }
  35
  36     public ServiceFunctionBuilder serviceFunctionBuilder(ServiceFunctionBuilder serviceFunctionBuilder,
  37                                                          String ip, String sfName,
  38                                                          List<SfDataPlaneLocator> sfDataPlaneLocatorList,
  39                                                          Class<? extends ServiceFunctionTypeIdentity> type) {
  40         return serviceFunctionBuilder
  41                 .setSfDataPlaneLocator(sfDataPlaneLocatorList)
  42                 .setName(new SfName(sfName))
  43                 .setIpMgmtAddress(new IpAddress(ip.toCharArray()))
  44                 .setType(type)
  45                 .setNshAware(true);
  46     }
  47
  48     public ServiceFunctionsBuilder serviceFunctionsBuilder(ServiceFunctionsBuilder serviceFunctionsBuilder,
  49                                                            List<ServiceFunction> serviceFunctionList) {
  50         return serviceFunctionsBuilder.setServiceFunction(serviceFunctionList);
  51     }
  52
  53     public ServiceFunctionBuilder serviceFunctionBuilder(String sfIp, int port, String sf1DplName,
  54                                                          String sffname, String sfName) {
  55         SfDataPlaneLocatorBuilder sfDataPlaneLocator =
  56                 sfDataPlaneLocatorBuilder(new SfDataPlaneLocatorBuilder(), sfIp, port, sf1DplName, sffname);
  57         List<SfDataPlaneLocator> sfDataPlaneLocatorList =
  58                 list(new ArrayList<SfDataPlaneLocator>(), sfDataPlaneLocator);
  59         return serviceFunctionBuilder(
  60                 new ServiceFunctionBuilder(), sfIp, sfName, sfDataPlaneLocatorList, Firewall.class);
  61     }
  62
  63
  64 }