Added support for data store type (OPERATIONAL or CONFIG or BOTH)
[controller.git] / benchmark / dsbenchmark / src / main / java / org / opendaylight / dsbenchmark / txchain / TxchainDomRead.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.common.api.data.*;
15 import org.opendaylight.controller.md.sal.dom.api.DOMDataBroker;
16 import org.opendaylight.controller.md.sal.dom.api.DOMDataReadOnlyTransaction;
17 import org.opendaylight.controller.md.sal.dom.api.DOMTransactionChain;
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.yangtools.yang.common.QName;
24 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
25 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
26 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
27 import org.slf4j.Logger;
28 import org.slf4j.LoggerFactory;
29
30 public class TxchainDomRead extends DatastoreAbstractWriter implements TransactionChainListener{
31     private static final Logger LOG = LoggerFactory.getLogger(TxchainDomRead.class);
32     private final DOMDataBroker domDataBroker;
33
34     public TxchainDomRead(DOMDataBroker domDataBroker, int outerListElem, int innerListElem, 
35             long writesPerTx, DataStore dataStore) {
36         super(StartTestInput.Operation.DELETE, outerListElem, innerListElem, writesPerTx, dataStore);
37         this.domDataBroker = domDataBroker;
38         LOG.info("Created TxchainDomDelete");
39     }
40
41     @Override
42     public void createList() {
43         LOG.info("TxchainDomDelete: creating data in the data store");
44
45         // Dump the whole list into the data store in a single transaction
46         // with <outerListElem> PUTs on the transaction
47         TxchainDomWrite dd = new TxchainDomWrite(domDataBroker,
48                                                  StartTestInput.Operation.PUT,
49                                                      outerListElem,
50                                                      innerListElem,
51                                                      outerListElem,
52                                                      dataStore);
53         dd.createList();
54         dd.executeList();
55     }
56
57     @Override
58     public void executeList() {
59
60         org.opendaylight.yangtools.yang.common.QName OL_ID = QName.create(OuterList.QNAME, "id");
61         DOMTransactionChain chain = domDataBroker.createTransactionChain(this);
62         DOMDataReadOnlyTransaction tx = domDataBroker.newReadOnlyTransaction();
63
64         YangInstanceIdentifier pid = YangInstanceIdentifier.builder().node(TestExec.QNAME).node(OuterList.QNAME).build();
65         for (int l = 0; l < outerListElem; l++) {
66             NormalizedNode<?,?> ret;
67             YangInstanceIdentifier yid = pid.node(new NodeIdentifierWithPredicates(OuterList.QNAME, OL_ID, l));
68             Optional<NormalizedNode<?,?>> optionalDataObject;
69             CheckedFuture<Optional<NormalizedNode<?,?>>, ReadFailedException> submitFuture = tx.read(LogicalDatastoreType.CONFIGURATION, yid);
70             try {
71                 optionalDataObject = submitFuture.checkedGet();
72                 if (optionalDataObject != null && optionalDataObject.isPresent()) {
73                     ret = optionalDataObject.get();
74                     txOk++;
75                 }
76             } catch (ReadFailedException e) {
77                 LOG.warn("failed to ....", e);
78                 txError++;
79             }
80         }
81
82
83     }
84
85     @Override
86     public void onTransactionChainFailed(TransactionChain<?, ?> chain,
87                                          AsyncTransaction<?, ?> transaction, Throwable cause) {
88         LOG.error("Broken chain {} in TxchainDomDelete, transaction {}, cause {}",
89                 chain, transaction.getIdentifier(), cause);
90     }
91
92     @Override
93     public void onTransactionChainSuccessful(TransactionChain<?, ?> chain) {
94         LOG.info("TxchainDomDelete closed successfully, chain {}", chain);
95     }
96 }