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