Added support for data store type (OPERATIONAL or CONFIG or BOTH)
[controller.git] / benchmark / dsbenchmark / src / main / java / org / opendaylight / dsbenchmark / txchain / TxchainBaRead.java
1 /*
2  * Copyright (c) 2015 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
9 package org.opendaylight.dsbenchmark.txchain;
10
11 import com.google.common.base.Optional;
12 import com.google.common.util.concurrent.CheckedFuture;
13
14 import org.opendaylight.controller.md.sal.binding.api.BindingTransactionChain;
15 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
16 import org.opendaylight.controller.md.sal.binding.api.ReadTransaction;
17 import org.opendaylight.controller.md.sal.common.api.data.*;
18 import org.opendaylight.dsbenchmark.DatastoreAbstractWriter;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.dsbenchmark.rev150105.StartTestInput;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.dsbenchmark.rev150105.StartTestInput.DataStore;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.dsbenchmark.rev150105.TestExec;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.dsbenchmark.rev150105.test.exec.OuterList;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.dsbenchmark.rev150105.test.exec.OuterListKey;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.dsbenchmark.rev150105.test.exec.outer.list.InnerList;
25 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
26 import org.slf4j.Logger;
27 import org.slf4j.LoggerFactory;
28
29 public class TxchainBaRead extends DatastoreAbstractWriter implements TransactionChainListener{
30     private static final Logger LOG = (Logger) LoggerFactory.getLogger(TxchainBaRead.class);
31     private DataBroker bindingDataBroker;
32
33     public TxchainBaRead(DataBroker bindingDataBroker, int outerListElem, int innerListElem, long writesPerTx, DataStore dataStore) {
34         super(StartTestInput.Operation.DELETE, outerListElem, innerListElem, writesPerTx, dataStore);
35         this.bindingDataBroker = bindingDataBroker;
36         LOG.info("Created TxchainBaRead");
37     }
38
39     @Override
40     public void createList() {
41         LOG.info("TxchainBaRead: reading data in the data store");
42
43         // Dump the whole list into the data store in a single transaction
44         // with <outerListElem> PUTs on the transaction
45         TxchainBaWrite dd = new TxchainBaWrite(bindingDataBroker,
46                                                StartTestInput.Operation.PUT,
47                                                outerListElem,
48                                                innerListElem,
49                                                outerListElem,
50                                                dataStore);
51         dd.createList();
52         dd.executeList();
53     }
54
55     @Override
56     public void executeList() {
57
58         BindingTransactionChain chain = bindingDataBroker.createTransactionChain(this);
59         ReadTransaction tx = bindingDataBroker.newReadOnlyTransaction();
60
61         for (long l = 0; l < outerListElem; l++) {
62
63             OuterList outerList;
64             InstanceIdentifier<OuterList> iid = InstanceIdentifier.create(TestExec.class)
65                     .child(OuterList.class, new OuterListKey((int) l));
66             Optional<OuterList> optionalDataObject;
67             CheckedFuture<Optional<OuterList>, ReadFailedException> submitFuture = tx.read(LogicalDatastoreType.CONFIGURATION, iid);
68
69             try {
70                 optionalDataObject = submitFuture.checkedGet();
71                 /*if (optionalDataObject != null && optionalDataObject.isPresent()) {
72                     ret = optionalDataObject.get();
73                     txOk++;
74                 }*/
75                 if (optionalDataObject != null && optionalDataObject.isPresent()) {
76                     outerList = optionalDataObject.get();
77
78                     String[] objectsArray = new String[outerList.getInnerList().size()];
79
80                     //LOG.info("innerList element: " + objectsArray );
81                     for (InnerList innerList : outerList.getInnerList()) {
82                         if (objectsArray[innerList.getName()] != null) {
83                             LOG.error("innerList: DUPLICATE name: {}, value: {}", innerList.getName(), innerList.getValue());
84                         }
85                         objectsArray[innerList.getName()] = innerList.getValue();
86                         // LOG.info("innerList: name: {}, value: {}", innerList.getName(), innerList.getValue());
87                     }
88                     boolean foundAll = true;
89                     for (int i = 0; i < outerList.getInnerList().size(); i++) {
90                         String itemStr = objectsArray[i];
91                         if (!itemStr.contentEquals("Item-" + String.valueOf(l) + "-" + String.valueOf(i))) {
92                             foundAll = false;
93                             LOG.error("innerList: name: {}, value: {}", i, itemStr);
94                             break;
95                         }
96                     }
97                     txOk++;
98                 }
99                 else {
100                     txError++;
101                 }
102             } catch (ReadFailedException e) {
103                 LOG.warn("failed to ....", e);
104                 txError++;
105             }
106         }
107     }
108
109     @Override
110     public void onTransactionChainFailed(TransactionChain<?, ?> chain,
111                                          AsyncTransaction<?, ?> transaction, Throwable cause) {
112         LOG.error("Broken chain {} in TxchainBaDelete, transaction {}, cause {}",
113                 chain, transaction.getIdentifier(), cause);
114     }
115
116     @Override
117     public void onTransactionChainSuccessful(TransactionChain<?, ?> chain) {
118         LOG.info("TxchainBaDelete closed successfully, chain {}", chain);
119     }
120
121 }