0c358e0b4b3bd120871832a0cdf680cce5b1721f
[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / main / java / org / opendaylight / controller / cluster / datastore / ShardReadWriteTransaction.java
1 /*
2  * Copyright (c) 2014 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.controller.cluster.datastore;
10
11 import akka.actor.ActorRef;
12 import org.opendaylight.controller.cluster.datastore.jmx.mbeans.shard.ShardStats;
13 import org.opendaylight.controller.cluster.datastore.messages.DataExists;
14 import org.opendaylight.controller.cluster.datastore.messages.ReadData;
15
16 /**
17  * @author: syedbahm
18  * Date: 8/6/14
19  */
20 public class ShardReadWriteTransaction extends ShardWriteTransaction {
21     public ShardReadWriteTransaction(ReadWriteShardDataTreeTransaction transaction, ActorRef shardActor,
22             ShardStats shardStats, String transactionID, short clientTxVersion) {
23         super(transaction, shardActor, shardStats, transactionID, clientTxVersion);
24     }
25
26     @Override
27     public void handleReceive(Object message) throws Exception {
28         if (message instanceof ReadData) {
29             readData((ReadData) message, !SERIALIZED_REPLY);
30
31         } else if (message instanceof DataExists) {
32             dataExists((DataExists) message, !SERIALIZED_REPLY);
33
34         } else if(ReadData.SERIALIZABLE_CLASS.equals(message.getClass())) {
35             readData(ReadData.fromSerializable(message), SERIALIZED_REPLY);
36
37         } else if(DataExists.SERIALIZABLE_CLASS.equals(message.getClass())) {
38             dataExists(DataExists.fromSerializable(message), SERIALIZED_REPLY);
39         } else {
40             super.handleReceive(message);
41         }
42     }
43 }