dom/mdsal-dom-inmemory-datastore-benchmark/src/main/java/org/opendaylight/mdsal/dom/store/inmemory/benchmark/AbstractInMemoryWriteTransactionBenchmark.java

   1 /*
   2  * Copyright (c) 2014, 2017 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 package org.opendaylight.mdsal.dom.store.inmemory.benchmark;
   9
  10 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  11 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  12 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
  13 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
  14 import org.opendaylight.yangtools.yang.data.api.schema.MapNode;
  15 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  16 import org.opendaylight.yangtools.yang.data.api.schema.SystemMapNode;
  17 import org.opendaylight.yangtools.yang.data.api.schema.builder.CollectionNodeBuilder;
  18 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
  19 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
  20 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
  21
  22 public abstract class AbstractInMemoryWriteTransactionBenchmark {
  23     protected static final int OUTER_LIST_100K = 100000;
  24     protected static final int OUTER_LIST_50K = 50000;
  25     protected static final int OUTER_LIST_10K = 10000;
  26
  27     protected static final YangInstanceIdentifier[] OUTER_LIST_100K_PATHS = initOuterListPaths(OUTER_LIST_100K);
  28     protected static final YangInstanceIdentifier[] OUTER_LIST_50K_PATHS = initOuterListPaths(OUTER_LIST_50K);
  29     protected static final YangInstanceIdentifier[] OUTER_LIST_10K_PATHS = initOuterListPaths(OUTER_LIST_10K);
  30
  31     private static YangInstanceIdentifier[] initOuterListPaths(final int outerListPathsCount) {
  32         final YangInstanceIdentifier[] paths = new YangInstanceIdentifier[outerListPathsCount];
  33
  34         for (int outerListKey = 0; outerListKey < outerListPathsCount; ++outerListKey) {
  35             paths[outerListKey] = YangInstanceIdentifier.builder(BenchmarkModel.OUTER_LIST_PATH)
  36                     .nodeWithKey(BenchmarkModel.OUTER_LIST_QNAME, BenchmarkModel.ID_QNAME, outerListKey).build();
  37         }
  38         return paths;
  39     }
  40
  41     protected static final int WARMUP_ITERATIONS = 6;
  42     protected static final int MEASUREMENT_ITERATIONS = 6;
  43
  44     protected static final MapNode ONE_ITEM_INNER_LIST = initInnerListItems(1);
  45     protected static final MapNode TWO_ITEM_INNER_LIST = initInnerListItems(2);
  46     protected static final MapNode TEN_ITEM_INNER_LIST = initInnerListItems(10);
  47
  48     private static MapNode initInnerListItems(final int count) {
  49         final CollectionNodeBuilder<MapEntryNode, SystemMapNode> mapEntryBuilder = ImmutableNodes
  50                 .mapNodeBuilder(BenchmarkModel.INNER_LIST_QNAME);
  51
  52         for (int i = 1; i <= count; ++i) {
  53             mapEntryBuilder
  54                     .withChild(ImmutableNodes.mapEntry(BenchmarkModel.INNER_LIST_QNAME, BenchmarkModel.NAME_QNAME, i));
  55         }
  56         return mapEntryBuilder.build();
  57     }
  58
  59     protected static final NormalizedNode[] OUTER_LIST_ONE_ITEM_INNER_LIST = initOuterListItems(OUTER_LIST_100K,
  60             ONE_ITEM_INNER_LIST);
  61     protected static final NormalizedNode[] OUTER_LIST_TWO_ITEM_INNER_LIST = initOuterListItems(OUTER_LIST_50K,
  62             TWO_ITEM_INNER_LIST);
  63     protected static final NormalizedNode[] OUTER_LIST_TEN_ITEM_INNER_LIST = initOuterListItems(OUTER_LIST_10K,
  64             TEN_ITEM_INNER_LIST);
  65
  66     private static NormalizedNode[] initOuterListItems(final int outerListItemsCount, final MapNode innerList) {
  67         final NormalizedNode[] outerListItems = new NormalizedNode[outerListItemsCount];
  68
  69         for (int i = 0; i < outerListItemsCount; ++i) {
  70             int outerListKey = i;
  71             outerListItems[i] = ImmutableNodes
  72                     .mapEntryBuilder(BenchmarkModel.OUTER_LIST_QNAME, BenchmarkModel.ID_QNAME, outerListKey)
  73                     .withChild(innerList).build();
  74         }
  75         return outerListItems;
  76     }
  77
  78     protected EffectiveModelContext schemaContext;
  79
  80     public abstract void setUp() throws Exception;
  81
  82     public abstract void tearDown();
  83
  84     protected static DataContainerChild provideOuterListNode() {
  85         return Builders.containerBuilder()
  86                 .withNodeIdentifier(new NodeIdentifier(BenchmarkModel.TEST_QNAME))
  87                 .withChild(ImmutableNodes.mapNodeBuilder(BenchmarkModel.OUTER_LIST_QNAME).build()).build();
  88     }
  89 }