opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/messages/ReadyLocalTransactionSerializer.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 package org.opendaylight.controller.cluster.datastore.messages;
   9
  10 import akka.serialization.JSerializer;
  11 import com.google.common.base.Optional;
  12 import com.google.common.base.Preconditions;
  13 import java.util.ArrayDeque;
  14 import java.util.Deque;
  15 import javax.annotation.Nonnull;
  16 import org.apache.commons.lang3.SerializationUtils;
  17 import org.opendaylight.controller.cluster.datastore.DataStoreVersions;
  18 import org.opendaylight.controller.cluster.datastore.modification.DeleteModification;
  19 import org.opendaylight.controller.cluster.datastore.modification.MergeModification;
  20 import org.opendaylight.controller.cluster.datastore.modification.WriteModification;
  21 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  22 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
  23 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  24 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeModificationCursor;
  25
  26 /**
  27  * Specialized message transformer, which transforms a {@link ReadyLocalTransaction}
  28  * into a {@link BatchedModifications} message. This serializer needs to be plugged
  29  * into akka serialization to allow forwarding of ReadyLocalTransaction to remote
  30  * shards.
  31  */
  32 public final class ReadyLocalTransactionSerializer extends JSerializer {
  33     @Override
  34     public int identifier() {
  35         return 97439437;
  36     }
  37
  38     @Override
  39     public boolean includeManifest() {
  40         return false;
  41     }
  42
  43     @Override
  44     public byte[] toBinary(final Object obj) {
  45         Preconditions.checkArgument(obj instanceof ReadyLocalTransaction, "Unsupported object type %s", obj.getClass());
  46         final ReadyLocalTransaction msg = (ReadyLocalTransaction) obj;
  47         final BatchedModifications batched = new BatchedModifications(msg.getTransactionID(),
  48                 DataStoreVersions.CURRENT_VERSION, "");
  49         batched.setDoCommitOnReady(msg.isDoCommitOnReady());
  50         batched.setTotalMessagesSent(1);
  51         batched.setReady(true);
  52
  53         msg.getModification().applyToCursor(new BatchedCursor(batched));
  54
  55         return SerializationUtils.serialize(batched);
  56     }
  57
  58     @Override
  59     public Object fromBinaryJava(final byte[] bytes, final Class<?> clazz) {
  60         return SerializationUtils.deserialize(bytes);
  61     }
  62
  63     private static final class BatchedCursor implements DataTreeModificationCursor {
  64         private final Deque<YangInstanceIdentifier> stack = new ArrayDeque<>();
  65         private final BatchedModifications message;
  66
  67         BatchedCursor(final BatchedModifications message) {
  68             this.message = Preconditions.checkNotNull(message);
  69             stack.push(YangInstanceIdentifier.EMPTY);
  70         }
  71
  72         @Override
  73         public void delete(final PathArgument child) {
  74             message.addModification(new DeleteModification(stack.peek().node(child)));
  75         }
  76
  77         @Override
  78         public void merge(final PathArgument child, final NormalizedNode<?, ?> data) {
  79             message.addModification(new MergeModification(stack.peek().node(child), data));
  80         }
  81
  82         @Override
  83         public void write(final PathArgument child, final NormalizedNode<?, ?> data) {
  84             message.addModification(new WriteModification(stack.peek().node(child), data));
  85         }
  86
  87         @Override
  88         public void enter(@Nonnull final PathArgument child) {
  89             stack.push(stack.peek().node(child));
  90         }
  91
  92         @Override
  93         public void enter(@Nonnull final PathArgument... path) {
  94             for (PathArgument arg : path) {
  95                 enter(arg);
  96             }
  97         }
  98
  99         @Override
 100         public void enter(@Nonnull final Iterable<PathArgument> path) {
 101             for (PathArgument arg : path) {
 102                 enter(arg);
 103             }
 104         }
 105
 106         @Override
 107         public void exit() {
 108             stack.pop();
 109         }
 110
 111         @Override
 112         public void exit(final int depth) {
 113             Preconditions.checkArgument(depth < stack.size(), "Stack holds only %s elements, cannot exit %s levels", stack.size(), depth);
 114             for (int i = 0; i < depth; ++i) {
 115                 stack.pop();
 116             }
 117         }
 118
 119         @Override
 120         public Optional<NormalizedNode<?, ?>> readNode(@Nonnull final PathArgument child) {
 121             throw new UnsupportedOperationException("Not implemented");
 122         }
 123
 124         @Override
 125         public void close() {
 126             // No-op
 127         }
 128     }
 129 }