Merge "Datastore: Reuse successful future"
[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
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,String shardName) {
30         super(shardActor, schemaContext, shardName);
31         this.transaction = transaction;
32     }
33
34     @Override
35     public void handleReceive(Object message) throws Exception {
36         if(ReadData.SERIALIZABLE_CLASS.equals(message.getClass())) {
37             readData(transaction, ReadData.fromSerializable(message));
38         } else if(DataExists.SERIALIZABLE_CLASS.equals(message.getClass())) {
39             dataExists(transaction, DataExists.fromSerializable(message));
40         } else {
41             super.handleReceive(message);
42         }
43     }
44
45     @Override
46     protected DOMStoreTransaction getDOMStoreTransaction() {
47         return transaction;
48     }
49 }