-import org.opendaylight.controller.cluster.datastore.utils.InstanceIdentifierUtils;
-import org.opendaylight.controller.protobuff.messages.transaction.ShardTransactionMessages;
-import org.opendaylight.yangtools.yang.data.api.InstanceIdentifier;
-
-public class ReadData {
- public static final Class SERIALIZABLE_CLASS = ShardTransactionMessages.ReadData.class;
- private final InstanceIdentifier path;
-
- public ReadData(InstanceIdentifier path) {
- this.path = path;
- }
-
- public InstanceIdentifier getPath() {
- return path;
- }
-
- public Object toSerializable(){
- return ShardTransactionMessages.ReadData.newBuilder()
- .setInstanceIdentifierPathArguments(path.toString())
- .build();
- }
-
- public static ReadData fromSerializable(Object serializable){
- ShardTransactionMessages.ReadData o = (ShardTransactionMessages.ReadData) serializable;
- return new ReadData(InstanceIdentifierUtils.from(o.getInstanceIdentifierPathArguments()));
- }
+import com.google.common.base.Optional;
+import com.google.common.base.Preconditions;
+import com.google.common.util.concurrent.CheckedFuture;
+import com.google.common.util.concurrent.SettableFuture;
+import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
+import org.opendaylight.controller.sal.core.spi.data.DOMStoreReadTransaction;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
+import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
+
+public class ReadData extends AbstractRead<Optional<NormalizedNode<?, ?>>> {
+ private static final long serialVersionUID = 1L;
+
+ public ReadData() {
+ }
+
+ public ReadData(final YangInstanceIdentifier path, short version) {
+ super(path, version);
+ }
+
+ @Override
+ public CheckedFuture<Optional<NormalizedNode<?, ?>>, ReadFailedException> apply(
+ DOMStoreReadTransaction readDelegate) {
+ return readDelegate.read(getPath());
+ }
+
+ @Override
+ public void processResponse(Object readResponse, SettableFuture<Optional<NormalizedNode<?, ?>>> returnFuture) {
+ if (ReadDataReply.isSerializedType(readResponse)) {
+ ReadDataReply reply = ReadDataReply.fromSerializable(readResponse);
+ returnFuture.set(Optional.<NormalizedNode<?, ?>>fromNullable(reply.getNormalizedNode()));
+ } else {
+ returnFuture.setException(new ReadFailedException("Invalid response reading data for path " + getPath()));
+ }
+ }
+
+ @Override
+ protected AbstractRead<Optional<NormalizedNode<?, ?>>> newInstance(short withVersion) {
+ return new ReadData(getPath(), withVersion);
+ }
+
+ public static ReadData fromSerializable(final Object serializable) {
+ Preconditions.checkArgument(serializable instanceof ReadData);
+ return (ReadData)serializable;
+ }
+
+ public static boolean isSerializedType(Object message) {
+ return message instanceof ReadData;
+ }