benchmark/dsbenchmark/src/main/java/org/opendaylight/dsbenchmark/simpletx/SimpletxDomWrite.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems 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.dsbenchmark.simpletx;
   9
  10 import static java.util.Objects.requireNonNull;
  11
  12 import java.util.List;
  13 import java.util.concurrent.ExecutionException;
  14 import org.opendaylight.dsbenchmark.DatastoreAbstractWriter;
  15 import org.opendaylight.dsbenchmark.DomListBuilder;
  16 import org.opendaylight.mdsal.dom.api.DOMDataBroker;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.dsbenchmark.rev150105.StartTestInput;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.dsbenchmark.rev150105.StartTestInput.DataStore;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.dsbenchmark.rev150105.TestExec;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.dsbenchmark.rev150105.test.exec.OuterList;
  21 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  22 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
  23 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
  24 import org.slf4j.Logger;
  25 import org.slf4j.LoggerFactory;
  26
  27 public class SimpletxDomWrite extends DatastoreAbstractWriter {
  28     private static final Logger LOG = LoggerFactory.getLogger(SimpletxDomWrite.class);
  29
  30     private final DOMDataBroker dataBroker;
  31     private List<MapEntryNode> list = null;
  32
  33     public SimpletxDomWrite(final DOMDataBroker dataBroker, final StartTestInput.Operation oper,
  34             final int outerListElem, final int innerListElem, final long putsPerTx, final DataStore dataStore) {
  35         super(oper, outerListElem, innerListElem, putsPerTx, dataStore);
  36         this.dataBroker = requireNonNull(dataBroker);
  37         LOG.debug("Created SimpletxDomWrite");
  38     }
  39
  40     @Override
  41     public void createList() {
  42         list = DomListBuilder.buildOuterList(outerListElem, innerListElem);
  43     }
  44
  45     @Override
  46     public void executeList() {
  47         final var dsType = getDataStoreType();
  48         final var pid = YangInstanceIdentifier.of(TestExec.QNAME, OuterList.QNAME);
  49
  50         var tx = dataBroker.newWriteOnlyTransaction();
  51         long writeCnt = 0;
  52
  53         for (var element : list) {
  54             final var yid = pid.node(NodeIdentifierWithPredicates.of(OuterList.QNAME, element.name().asMap()));
  55
  56             if (oper == StartTestInput.Operation.PUT) {
  57                 tx.put(dsType, yid, element);
  58             } else {
  59                 tx.merge(dsType, yid, element);
  60             }
  61
  62             writeCnt++;
  63
  64             if (writeCnt == writesPerTx) {
  65                 try {
  66                     tx.commit().get();
  67                     txOk++;
  68                 } catch (final InterruptedException | ExecutionException e) {
  69                     LOG.error("Transaction failed", e);
  70                     txError++;
  71                 }
  72                 tx = dataBroker.newWriteOnlyTransaction();
  73                 writeCnt = 0;
  74             }
  75         }
  76
  77         if (writeCnt != 0) {
  78             try {
  79                 tx.commit().get();
  80             } catch (final InterruptedException | ExecutionException e) {
  81                 LOG.error("Transaction failed", e);
  82             }
  83         }
  84     }
  85 }