Split out TransactionContext classes
[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / main / java / org / opendaylight / controller / cluster / datastore / ShardReadTransaction.java
1 /*
2  *
3  *  Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
4  *
5  *  This program and the accompanying materials are made available under the
6  *  terms of the Eclipse Public License v1.0 which accompanies this distribution,
7  *  and is available at http://www.eclipse.org/legal/epl-v10.html
8  *
9  */
10
11 package org.opendaylight.controller.cluster.datastore;
12
13 import akka.actor.ActorRef;
14 import org.opendaylight.controller.cluster.datastore.jmx.mbeans.shard.ShardStats;
15 import org.opendaylight.controller.cluster.datastore.messages.DataExists;
16 import org.opendaylight.controller.cluster.datastore.messages.ReadData;
17 import org.opendaylight.controller.sal.core.spi.data.DOMStoreReadTransaction;
18 import org.opendaylight.controller.sal.core.spi.data.DOMStoreTransaction;
19 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
20
21 /**
22  * @author: syedbahm
23  * Date: 8/6/14
24  */
25 public class ShardReadTransaction extends ShardTransaction {
26     private final DOMStoreReadTransaction transaction;
27
28     public ShardReadTransaction(DOMStoreReadTransaction transaction, ActorRef shardActor,
29             SchemaContext schemaContext, ShardStats shardStats, String transactionID,
30             short clientTxVersion) {
31         super(shardActor, schemaContext, shardStats, transactionID, clientTxVersion);
32         this.transaction = transaction;
33     }
34
35     @Override
36     public void handleReceive(Object message) throws Exception {
37         if(message instanceof ReadData) {
38             readData(transaction, (ReadData) message, !SERIALIZED_REPLY);
39
40         } else if (message instanceof DataExists) {
41             dataExists(transaction, (DataExists) message, !SERIALIZED_REPLY);
42
43         } else if(ReadData.SERIALIZABLE_CLASS.equals(message.getClass())) {
44             readData(transaction, ReadData.fromSerializable(message), SERIALIZED_REPLY);
45
46         } else if(DataExists.SERIALIZABLE_CLASS.equals(message.getClass())) {
47             dataExists(transaction, DataExists.fromSerializable(message), SERIALIZED_REPLY);
48
49         } else {
50             super.handleReceive(message);
51         }
52     }
53
54     @Override
55     protected DOMStoreTransaction getDOMStoreTransaction() {
56         return transaction;
57     }
58 }