netconf/tools/netconf-test-perf/src/main/java/org/opendaylight/test/endtoend/NcmountServiceImpl.java

   1 /*
   2  * Copyright (c) 2021 Pantheon Technologies, s.r.o. 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 package org.opendaylight.test.endtoend;
   9
  10 import static java.util.Objects.requireNonNull;
  11
  12 import com.google.common.util.concurrent.ListenableFuture;
  13 import com.google.common.util.concurrent.MoreExecutors;
  14 import java.util.Map;
  15 import java.util.Optional;
  16 import org.opendaylight.mdsal.binding.api.DataBroker;
  17 import org.opendaylight.mdsal.binding.api.MountPoint;
  18 import org.opendaylight.mdsal.binding.api.MountPointService;
  19 import org.opendaylight.mdsal.binding.api.WriteTransaction;
  20 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
  21 import org.opendaylight.yang.gen.v1.http.cisco.com.ns.yang.cisco.ios.xr.ip._static.cfg.rev130722.RouterStatic;
  22 import org.opendaylight.yang.gen.v1.http.cisco.com.ns.yang.cisco.ios.xr.ip._static.cfg.rev130722.address.family.AddressFamilyBuilder;
  23 import org.opendaylight.yang.gen.v1.http.cisco.com.ns.yang.cisco.ios.xr.ip._static.cfg.rev130722.address.family.address.family.Vrfipv4Builder;
  24 import org.opendaylight.yang.gen.v1.http.cisco.com.ns.yang.cisco.ios.xr.ip._static.cfg.rev130722.router._static.Vrfs;
  25 import org.opendaylight.yang.gen.v1.http.cisco.com.ns.yang.cisco.ios.xr.ip._static.cfg.rev130722.router._static.vrfs.Vrf;
  26 import org.opendaylight.yang.gen.v1.http.cisco.com.ns.yang.cisco.ios.xr.ip._static.cfg.rev130722.router._static.vrfs.VrfBuilder;
  27 import org.opendaylight.yang.gen.v1.http.cisco.com.ns.yang.cisco.ios.xr.ip._static.cfg.rev130722.router._static.vrfs.VrfKey;
  28 import org.opendaylight.yang.gen.v1.http.cisco.com.ns.yang.cisco.ios.xr.ip._static.cfg.rev130722.vrf.prefix.table.VrfPrefixesBuilder;
  29 import org.opendaylight.yang.gen.v1.http.cisco.com.ns.yang.cisco.ios.xr.ip._static.cfg.rev130722.vrf.prefix.table.vrf.prefixes.VrfPrefix;
  30 import org.opendaylight.yang.gen.v1.http.cisco.com.ns.yang.cisco.ios.xr.ip._static.cfg.rev130722.vrf.prefix.table.vrf.prefixes.VrfPrefixBuilder;
  31 import org.opendaylight.yang.gen.v1.http.cisco.com.ns.yang.cisco.ios.xr.ip._static.cfg.rev130722.vrf.prefix.table.vrf.prefixes.VrfPrefixKey;
  32 import org.opendaylight.yang.gen.v1.http.cisco.com.ns.yang.cisco.ios.xr.ip._static.cfg.rev130722.vrf.route.VrfRouteBuilder;
  33 import org.opendaylight.yang.gen.v1.http.cisco.com.ns.yang.cisco.ios.xr.ip._static.cfg.rev130722.vrf.route.vrf.route.VrfNextHopsBuilder;
  34 import org.opendaylight.yang.gen.v1.http.cisco.com.ns.yang.cisco.ios.xr.ip._static.cfg.rev130722.vrf.route.vrf.route.vrf.next.hops.NextHopAddressBuilder;
  35 import org.opendaylight.yang.gen.v1.http.cisco.com.ns.yang.cisco.ios.xr.ip._static.cfg.rev130722.vrf.unicast.VrfUnicastBuilder;
  36 import org.opendaylight.yang.gen.v1.http.cisco.com.ns.yang.cisco.xr.types.rev150119.CiscoIosXrString;
  37 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.network.topology.topology.topology.types.TopologyNetconf;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ncmount.rev150105.ListNodesInput;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ncmount.rev150105.ListNodesOutput;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ncmount.rev150105.NcmountService;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ncmount.rev150105.ShowNodeInput;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ncmount.rev150105.ShowNodeOutput;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ncmount.rev150105.WriteRoutesInput;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ncmount.rev150105.WriteRoutesOutput;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ncmount.rev150105.WriteRoutesOutputBuilder;
  47 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
  48 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
  49 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
  50 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
  51 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
  52 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  53 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
  54 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  55 import org.opendaylight.yangtools.yang.binding.util.BindingMap;
  56 import org.opendaylight.yangtools.yang.common.ErrorType;
  57 import org.opendaylight.yangtools.yang.common.RpcResult;
  58 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
  59
  60 public class NcmountServiceImpl implements NcmountService {
  61     private static final InstanceIdentifier<Topology> NETCONF_TOPO_IID =
  62             InstanceIdentifier.create(NetworkTopology.class)
  63                     .child(Topology.class, new TopologyKey(new TopologyId(TopologyNetconf.QNAME.getLocalName())));
  64
  65     private final MountPointService mountPointService;
  66
  67     public NcmountServiceImpl(final MountPointService mountPointService) {
  68         this.mountPointService = requireNonNull(mountPointService);
  69     }
  70
  71     @Override
  72     public ListenableFuture<RpcResult<WriteRoutesOutput>> writeRoutes(final WriteRoutesInput input) {
  73         final Optional<MountPoint> optMountPoint = mountPointService.getMountPoint(
  74             NETCONF_TOPO_IID.child(Node.class, new NodeKey(new NodeId(input.getMountName()))));
  75         if (optMountPoint.isEmpty()) {
  76             return RpcResultBuilder.<WriteRoutesOutput>failed()
  77                 .withError(ErrorType.TRANSPORT, "Mount point not present")
  78                 .buildFuture();
  79         }
  80
  81         final Optional<DataBroker> optBroker = optMountPoint.orElseThrow().getService(DataBroker.class);
  82         if (optBroker.isEmpty()) {
  83             return RpcResultBuilder.<WriteRoutesOutput>failed()
  84                 .withError(ErrorType.TRANSPORT, "Mount point does not provide DataBroker service")
  85                 .buildFuture();
  86         }
  87
  88         final Map<VrfPrefixKey, VrfPrefix> routes = input.nonnullRoute().entrySet().stream()
  89             .map(entry -> new VrfPrefixBuilder()
  90                 .withKey(new VrfPrefixKey(new IpAddress(entry.getValue().getIpv4Prefix()),
  91                     entry.getValue().getIpv4PrefixLength().toUint32()))
  92                 .setVrfRoute(new VrfRouteBuilder()
  93                     .setVrfNextHops(new VrfNextHopsBuilder()
  94                         .setNextHopAddress(BindingMap.of(new NextHopAddressBuilder()
  95                             .setNextHopAddress(new IpAddress(entry.getValue().getIpv4NextHop()))
  96                             .build()))
  97                         .build())
  98                     .build())
  99                 .build())
 100             .collect(BindingMap.toOrderedMap());
 101
 102         final Vrf vrf = new VrfBuilder()
 103             .withKey(new VrfKey(new CiscoIosXrString(input.getVrfId())))
 104             .setAddressFamily(new AddressFamilyBuilder()
 105                 .setVrfipv4(new Vrfipv4Builder()
 106                     .setVrfUnicast(new VrfUnicastBuilder()
 107                         .setVrfPrefixes(new VrfPrefixesBuilder()
 108                             // FIXME: inline routes once it is unambiguous
 109                             .setVrfPrefix(routes)
 110                             .build())
 111                         .build())
 112                     .build())
 113                 .build())
 114             .build();
 115
 116         final WriteTransaction writeTransaction = optBroker.orElseThrow().newWriteOnlyTransaction();
 117         writeTransaction.merge(LogicalDatastoreType.CONFIGURATION,
 118             InstanceIdentifier.create(RouterStatic.class).child(Vrfs.class).child(Vrf.class, vrf.key()), vrf);
 119
 120         return writeTransaction.commit().transform(
 121             info -> RpcResultBuilder.success(new WriteRoutesOutputBuilder().build()).build(),
 122             MoreExecutors.directExecutor());
 123     }
 124
 125     @Override
 126     public ListenableFuture<RpcResult<ShowNodeOutput>> showNode(final ShowNodeInput input) {
 127         return RpcResultBuilder.<ShowNodeOutput>failed()
 128             .withError(ErrorType.APPLICATION, "Not implemented")
 129             .buildFuture();
 130     }
 131
 132     @Override
 133     public ListenableFuture<RpcResult<ListNodesOutput>> listNodes(final ListNodesInput input) {
 134         return RpcResultBuilder.<ListNodesOutput>failed()
 135             .withError(ErrorType.APPLICATION, "Not implemented")
 136             .buildFuture();
 137     }
 138 }