bgp/benchmark-app/src/main/java/org/opendaylight/protocol/bgp/benchmark/app/AppPeerBenchmark.java

   1 /*
   2  * Copyright (c) 2016 Cisco Systems, 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.protocol.bgp.benchmark.app;
  10
  11 import static java.util.Objects.requireNonNull;
  12 import static org.opendaylight.protocol.bgp.parser.spi.PathIdUtil.NON_PATH_ID;
  13
  14 import com.google.common.annotations.VisibleForTesting;
  15 import com.google.common.base.Stopwatch;
  16 import com.google.common.net.InetAddresses;
  17 import com.google.common.util.concurrent.FutureCallback;
  18 import com.google.common.util.concurrent.ListenableFuture;
  19 import com.google.common.util.concurrent.MoreExecutors;
  20 import java.util.Collections;
  21 import java.util.List;
  22 import java.util.concurrent.ExecutionException;
  23 import java.util.concurrent.TimeUnit;
  24 import org.opendaylight.mdsal.binding.api.DataBroker;
  25 import org.opendaylight.mdsal.binding.api.RpcProviderService;
  26 import org.opendaylight.mdsal.binding.api.Transaction;
  27 import org.opendaylight.mdsal.binding.api.TransactionChain;
  28 import org.opendaylight.mdsal.binding.api.TransactionChainListener;
  29 import org.opendaylight.mdsal.binding.api.WriteTransaction;
  30 import org.opendaylight.mdsal.common.api.CommitInfo;
  31 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
  32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.application.rib.tables.routes.Ipv4RoutesCaseBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.bgp.rib.rib.loc.rib.tables.routes.Ipv4RoutesCase;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.routes.Ipv4Routes;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.routes.Ipv4RoutesBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.routes.ipv4.routes.Ipv4Route;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.routes.ipv4.routes.Ipv4RouteBuilder;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.routes.ipv4.routes.Ipv4RouteKey;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.Attributes;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.AttributesBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.AsPath;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.AsPathBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.LocalPref;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.LocalPrefBuilder;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.MultiExitDisc;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.MultiExitDiscBuilder;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.Origin;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.OriginBuilder;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.ApplicationRib;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.ApplicationRibBuilder;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.ApplicationRibId;
  54 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.ApplicationRibKey;
  55 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.Tables;
  56 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.TablesBuilder;
  57 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.TablesKey;
  58 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.BgpOrigin;
  59 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.Ipv4AddressFamily;
  60 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.UnicastSubsequentAddressFamily;
  61 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.next.hop.c.next.hop.Ipv4NextHopCaseBuilder;
  62 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.next.hop.c.next.hop.ipv4.next.hop._case.Ipv4NextHopBuilder;
  63 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp.app.peer.benchmark.rev160309.AddPrefixInput;
  64 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp.app.peer.benchmark.rev160309.AddPrefixOutput;
  65 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp.app.peer.benchmark.rev160309.AddPrefixOutputBuilder;
  66 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp.app.peer.benchmark.rev160309.DeletePrefixInput;
  67 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp.app.peer.benchmark.rev160309.DeletePrefixOutput;
  68 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp.app.peer.benchmark.rev160309.DeletePrefixOutputBuilder;
  69 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp.app.peer.benchmark.rev160309.OdlBgpAppPeerBenchmarkService;
  70 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp.app.peer.benchmark.rev160309.output.Result;
  71 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp.app.peer.benchmark.rev160309.output.ResultBuilder;
  72 import org.opendaylight.yangtools.concepts.ObjectRegistration;
  73 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  74 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
  75 import org.opendaylight.yangtools.yang.common.RpcResult;
  76 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
  77 import org.slf4j.Logger;
  78 import org.slf4j.LoggerFactory;
  79
  80 public class AppPeerBenchmark implements OdlBgpAppPeerBenchmarkService, TransactionChainListener, AutoCloseable {
  81
  82     private static final Logger LOG = LoggerFactory.getLogger(AppPeerBenchmark.class);
  83
  84     private static final AsPath AS_PATH = new AsPathBuilder().build();
  85     private static final Origin ORIGIN = new OriginBuilder().setValue(BgpOrigin.Igp).build();
  86     private static final MultiExitDisc MED = new MultiExitDiscBuilder().setMed(0L).build();
  87     private static final LocalPref LOC_PREF = new LocalPrefBuilder().setPref(100L).build();
  88     private static final List<Tables> EMPTY_TABLES = Collections.singletonList(new TablesBuilder()
  89         .setAfi(Ipv4AddressFamily.class).setSafi(UnicastSubsequentAddressFamily.class).setRoutes(
  90             new Ipv4RoutesCaseBuilder().setIpv4Routes(new Ipv4RoutesBuilder().setIpv4Route(Collections.emptyList())
  91                 .build()).build()).build());
  92
  93     private static final String SLASH = "/";
  94     private static final String PREFIX = SLASH + "32";
  95
  96     private final TransactionChain txChain;
  97     private final ObjectRegistration<OdlBgpAppPeerBenchmarkService> rpcRegistration;
  98     private final InstanceIdentifier<ApplicationRib> appIID;
  99     private final InstanceIdentifier<Ipv4Routes> routesIId;
 100     private final String appRibId;
 101
 102     public AppPeerBenchmark(final DataBroker bindingDataBroker, final RpcProviderService rpcProviderRegistry,
 103             final String appRibId) {
 104         this.appRibId = requireNonNull(appRibId);
 105         this.txChain = bindingDataBroker.createMergingTransactionChain(this);
 106
 107         this.appIID = InstanceIdentifier.builder(ApplicationRib.class,
 108             new ApplicationRibKey(new ApplicationRibId(appRibId))).build();
 109         this.routesIId = this.appIID
 110             .child(Tables.class, new TablesKey(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class))
 111             .child(Ipv4RoutesCase.class, Ipv4Routes.class);
 112         this.rpcRegistration = rpcProviderRegistry.registerRpcImplementation(OdlBgpAppPeerBenchmarkService.class, this);
 113         LOG.info("BGP Application Peer Benchmark Application started.");
 114     }
 115
 116     public void start() {
 117         LOG.debug("Instantiating App Peer Benchmark : {}", this.appRibId);
 118         final ApplicationRib appRib = new ApplicationRibBuilder().setId(new ApplicationRibId(
 119             new ApplicationRibId(this.appRibId))).setTables(EMPTY_TABLES).build();
 120
 121         final WriteTransaction wTx = this.txChain.newWriteOnlyTransaction();
 122         wTx.put(LogicalDatastoreType.CONFIGURATION, this.appIID, appRib);
 123         wTx.commit().addCallback(new FutureCallback<CommitInfo>() {
 124             @Override
 125             public void onSuccess(final CommitInfo result) {
 126                 LOG.info("Empty Structure created for Application Peer Benchmark {}", AppPeerBenchmark.this.appRibId);
 127             }
 128
 129             @Override
 130             public void onFailure(final Throwable throwable) {
 131                 LOG.error("Failed to create Empty Structure for Application Peer Benchmark {}",
 132                     AppPeerBenchmark.this.appRibId, throwable);
 133             }
 134         }, MoreExecutors.directExecutor());
 135     }
 136
 137     @Override
 138     public void onTransactionChainFailed(final TransactionChain chain, final Transaction transaction,
 139             final Throwable cause) {
 140         LOG.error("Broken chain {} in DatastoreBaAbstractWrite, transaction {}", chain, transaction.getIdentifier(),
 141             cause);
 142         close();
 143     }
 144
 145     @Override
 146     public void onTransactionChainSuccessful(final TransactionChain chain) {
 147         LOG.debug("DatastoreBaAbstractWrite closed successfully, chain {}", chain);
 148     }
 149
 150     @Override
 151     public ListenableFuture<RpcResult<AddPrefixOutput>> addPrefix(final AddPrefixInput input) {
 152         final long duration = addRoute(input.getPrefix(), input.getNexthop(), input.getCount(), input.getBatchsize());
 153         final long rate = countRate(duration, input.getCount());
 154
 155         final AddPrefixOutputBuilder outputbuilder = new AddPrefixOutputBuilder();
 156         outputbuilder.setResult(createResult(input.getCount(), duration, rate));
 157         final AddPrefixOutput output = outputbuilder.build();
 158         return RpcResultBuilder.success(output).buildFuture();
 159     }
 160
 161     @Override
 162     public ListenableFuture<RpcResult<DeletePrefixOutput>> deletePrefix(final DeletePrefixInput input) {
 163         final long duration = deleteRoute(input.getPrefix(), input.getCount(), input.getBatchsize());
 164         final long rate = countRate(duration, input.getCount());
 165
 166         final DeletePrefixOutputBuilder outputbuilder = new DeletePrefixOutputBuilder();
 167         outputbuilder.setResult(createResult(input.getCount(), duration, rate));
 168         final DeletePrefixOutput output = outputbuilder.build();
 169         return RpcResultBuilder.success(output).buildFuture();
 170     }
 171
 172     @Override
 173     public void close() {
 174         this.rpcRegistration.close();
 175         final WriteTransaction dTx = this.txChain.newWriteOnlyTransaction();
 176         dTx.delete(LogicalDatastoreType.CONFIGURATION, this.appIID);
 177         try {
 178             dTx.commit().get();
 179         } catch (final InterruptedException | ExecutionException e) {
 180             LOG.warn("Failed to clean-up BGP Application RIB.", e);
 181         }
 182         this.txChain.close();
 183         LOG.info("BGP Application Peer Benchmark Application closed.");
 184     }
 185
 186     @VisibleForTesting
 187     InstanceIdentifier<Ipv4Routes> getIpv4RoutesIID() {
 188         return this.routesIId;
 189     }
 190
 191     private long addRoute(final Ipv4Prefix ipv4Prefix, final Ipv4Address nextHop, final long count, final long batch) {
 192         final AttributesBuilder attributesBuilder = new AttributesBuilder();
 193         attributesBuilder.setCNextHop(new Ipv4NextHopCaseBuilder().setIpv4NextHop(
 194                 new Ipv4NextHopBuilder().setGlobal(new Ipv4Address(nextHop)).build()).build());
 195         attributesBuilder.setMultiExitDisc(MED);
 196         attributesBuilder.setLocalPref(LOC_PREF);
 197         attributesBuilder.setOrigin(ORIGIN);
 198         attributesBuilder.setAsPath(AS_PATH);
 199         final Attributes attributes = attributesBuilder.build();
 200         return processRoutes(ipv4Prefix, count, batch, attributes);
 201     }
 202
 203     private long deleteRoute(final Ipv4Prefix ipv4Prefix, final long count, final long batch) {
 204         return processRoutes(ipv4Prefix, count, batch, null);
 205     }
 206
 207     private long processRoutes(final Ipv4Prefix ipv4Prefix, final long count, final long batch,
 208         final Attributes attributes) {
 209         WriteTransaction wt = this.txChain.newWriteOnlyTransaction();
 210         String address = getAdddressFromPrefix(ipv4Prefix);
 211         final Stopwatch stopwatch = Stopwatch.createStarted();
 212         for (int i = 1; i <= count; i++) {
 213             final Ipv4RouteKey routeKey = new Ipv4RouteKey(NON_PATH_ID, createKey(address));
 214             final KeyedInstanceIdentifier<Ipv4Route, Ipv4RouteKey> routeIId =
 215                 this.routesIId.child(Ipv4Route.class, routeKey);
 216             if (attributes != null) {
 217                 final Ipv4RouteBuilder ipv4RouteBuilder = new Ipv4RouteBuilder();
 218                 ipv4RouteBuilder.setRouteKey(routeKey.getRouteKey());
 219                 ipv4RouteBuilder.setPrefix(new Ipv4Prefix(routeKey.getRouteKey()));
 220                 ipv4RouteBuilder.withKey(routeKey);
 221                 ipv4RouteBuilder.setAttributes(attributes);
 222                 final Ipv4Route ipv4Route = ipv4RouteBuilder.build();
 223                 wt.put(LogicalDatastoreType.CONFIGURATION, routeIId,
 224                         ipv4Route);
 225             } else {
 226                 wt.delete(LogicalDatastoreType.CONFIGURATION, routeIId);
 227             }
 228             if (i % batch == 0) {
 229                 wt.commit().addCallback(new FutureCallback<CommitInfo>() {
 230                     @Override
 231                     public void onSuccess(final CommitInfo result) {
 232                         LOG.trace("Successful commit");
 233                     }
 234
 235                     @Override
 236                     public void onFailure(final Throwable trw) {
 237                         LOG.error("Failed commit", trw);
 238                     }
 239                 }, MoreExecutors.directExecutor());
 240                 wt = this.txChain.newWriteOnlyTransaction();
 241             }
 242             address = increasePrefix(address);
 243         }
 244         wt.commit().addCallback(new FutureCallback<CommitInfo>() {
 245             @Override
 246             public void onSuccess(final CommitInfo result) {
 247                 LOG.trace("Route batch stored.");
 248             }
 249
 250             @Override
 251             public void onFailure(final Throwable throwable) {
 252                 LOG.error("Failed to store route batch.", throwable);
 253             }
 254         }, MoreExecutors.directExecutor());
 255         return stopwatch.stop().elapsed(TimeUnit.MILLISECONDS);
 256     }
 257
 258     private static long countRate(final long durationMillis, final long count) {
 259         final long durationSec = TimeUnit.MILLISECONDS.toSeconds(durationMillis);
 260         if (durationSec != 0) {
 261             return count / durationSec;
 262         }
 263         return count;
 264     }
 265
 266     private static String increasePrefix(final String prefix) {
 267         return InetAddresses.increment(InetAddresses.forString(prefix)).getHostAddress();
 268     }
 269
 270     private static Result createResult(final long count, final long duration, final long rate) {
 271         return new ResultBuilder().setCount(count).setDuration(duration).setRate(rate).build();
 272     }
 273
 274     private static String getAdddressFromPrefix(final Ipv4Prefix prefix) {
 275         return prefix.getValue().split(SLASH)[0];
 276     }
 277
 278     private static String createKey(final String address) {
 279         return address + PREFIX;
 280     }
 281 }