opendaylight/md-sal/benchmark-data-store/src/main/java/org/opendaylight/controller/md/sal/dom/store/benchmark/AbstractInMemoryWriteTransactionBenchmark.java

   1 /*
   2  * Copyright (c) 2014 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)
  35                 .build();
  36         }
  37         return paths;
  38     }
  39
  40     protected static final int WARMUP_ITERATIONS = 20;
  41     protected static final int MEASUREMENT_ITERATIONS = 20;
  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, MapNode> 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, ONE_ITEM_INNER_LIST);
  59     protected static final NormalizedNode<?, ?>[] OUTER_LIST_TWO_ITEM_INNER_LIST = initOuterListItems(OUTER_LIST_50K, TWO_ITEM_INNER_LIST);
  60     protected static final NormalizedNode<?, ?>[] OUTER_LIST_TEN_ITEM_INNER_LIST = initOuterListItems(OUTER_LIST_10K, TEN_ITEM_INNER_LIST);
  61
  62     private static NormalizedNode<?,?>[] initOuterListItems(final int outerListItemsCount, final MapNode innerList) {
  63         final NormalizedNode<?,?>[] outerListItems = new NormalizedNode[outerListItemsCount];
  64
  65         for (int i = 0; i < outerListItemsCount; ++i) {
  66             int outerListKey = i;
  67             outerListItems[i] = ImmutableNodes.mapEntryBuilder(BenchmarkModel.OUTER_LIST_QNAME, BenchmarkModel.ID_QNAME, outerListKey)
  68                 .withChild(innerList).build();
  69         }
  70         return outerListItems;
  71     }
  72
  73     protected SchemaContext schemaContext;
  74     abstract public void setUp() throws Exception;
  75     abstract public void tearDown();
  76
  77     protected static DataContainerChild<?, ?> provideOuterListNode() {
  78         return ImmutableContainerNodeBuilder
  79             .create()
  80             .withNodeIdentifier(new YangInstanceIdentifier.NodeIdentifier(BenchmarkModel.TEST_QNAME))
  81             .withChild(
  82                 ImmutableNodes.mapNodeBuilder(BenchmarkModel.OUTER_LIST_QNAME)
  83                     .build()).build();
  84     }
  85 }