X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=benchmark%2Fdsbenchmark%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fdsbenchmark%2FDatastoreAbstractWriter.java;h=ef4d04d8606d3a228263ef51dd3efa5e732a461f;hb=625723b875800d5c421061f09cd5f387ccb59e62;hp=93a4876340bf85e5e7fd6b9659aada5865144861;hpb=bc740310bca93dcefcf546f7c4a627d8153b3739;p=controller.git

diff --git a/benchmark/dsbenchmark/src/main/java/org/opendaylight/dsbenchmark/DatastoreAbstractWriter.java b/benchmark/dsbenchmark/src/main/java/org/opendaylight/dsbenchmark/DatastoreAbstractWriter.java
index 93a4876340..ef4d04d860 100644
--- a/benchmark/dsbenchmark/src/main/java/org/opendaylight/dsbenchmark/DatastoreAbstractWriter.java
+++ b/benchmark/dsbenchmark/src/main/java/org/opendaylight/dsbenchmark/DatastoreAbstractWriter.java
@@ -8,27 +8,41 @@
 
 package org.opendaylight.dsbenchmark;
 
+import java.util.Random;
+
+import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.dsbenchmark.rev150105.StartTestInput;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.dsbenchmark.rev150105.StartTestInput.DataStore;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 public abstract class DatastoreAbstractWriter {
+    private static final Logger LOG = LoggerFactory.getLogger(DatastoreAbstractWriter.class);
+
     protected final int outerListElem;
     protected final int innerListElem;
     protected final long writesPerTx;
     protected final StartTestInput.Operation oper;
+    protected final StartTestInput.DataStore dataStore;
+    protected final Random rn = new Random();
+
 
     protected int txOk = 0;
     protected int txError = 0;
 
 
     public DatastoreAbstractWriter(StartTestInput.Operation oper,
-                                   int outerListElem, int innerListElem, long writesPerTx) {
+                                   int outerListElem, int innerListElem, long writesPerTx, DataStore dataStore) {
         this.outerListElem = outerListElem;
         this.innerListElem = innerListElem;
         this.writesPerTx = writesPerTx;
         this.oper = oper;
+        this.dataStore = dataStore;
+        LOG.info("DatastoreAbstractWriter created: {}", this);
     }
 
     public abstract void createList();
+
     public abstract void executeList();
 
     public int getTxError() {
@@ -39,4 +53,19 @@ public abstract class DatastoreAbstractWriter {
         return txOk;
     }
 
+    protected LogicalDatastoreType getDataStoreType() {
+        final LogicalDatastoreType dsType;
+        if (dataStore == DataStore.CONFIG) {
+            dsType = LogicalDatastoreType.CONFIGURATION;
+        } else if (dataStore == DataStore.OPERATIONAL) {
+            dsType = LogicalDatastoreType.OPERATIONAL;
+        } else {
+            if (rn.nextBoolean() == true) {
+                dsType = LogicalDatastoreType.OPERATIONAL;
+            } else {
+                dsType = LogicalDatastoreType.CONFIGURATION;
+            }
+        }
+        return dsType;
+    }
 }