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