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