opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/modification/MutableCompositeModification.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.modification;
  10
  11 import java.io.IOException;
  12 import java.io.ObjectInput;
  13 import java.io.ObjectOutput;
  14 import java.util.ArrayList;
  15 import java.util.List;
  16 import org.opendaylight.controller.cluster.datastore.DataStoreVersions;
  17 import org.opendaylight.controller.cluster.datastore.node.utils.stream.NormalizedNodeInputStreamReader;
  18 import org.opendaylight.controller.cluster.datastore.node.utils.stream.NormalizedNodeOutputStreamWriter;
  19 import org.opendaylight.controller.cluster.datastore.utils.SerializationUtils;
  20 import org.opendaylight.controller.protobuff.messages.persistent.PersistentMessages;
  21 import org.opendaylight.controller.sal.core.spi.data.DOMStoreWriteTransaction;
  22 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeModification;
  23
  24 /**
  25  * MutableCompositeModification is just a mutable version of a
  26  * CompositeModification {@link org.opendaylight.controller.cluster.datastore.modification.MutableCompositeModification#addModification(Modification)}
  27  */
  28 public class MutableCompositeModification implements CompositeModification {
  29     private static final long serialVersionUID = 1L;
  30
  31     private final List<Modification> modifications = new ArrayList<>();
  32     private short version;
  33
  34     public MutableCompositeModification() {
  35         this(DataStoreVersions.CURRENT_VERSION);
  36     }
  37
  38     public MutableCompositeModification(short version) {
  39         this.version = version;
  40     }
  41
  42     @Override
  43     public void apply(DOMStoreWriteTransaction transaction) {
  44         for (Modification modification : modifications) {
  45             modification.apply(transaction);
  46         }
  47     }
  48
  49     @Override
  50     public void apply(DataTreeModification transaction) {
  51         for (Modification modification : modifications) {
  52             modification.apply(transaction);
  53         }
  54     }
  55
  56     @Override
  57     public byte getType() {
  58         return COMPOSITE;
  59     }
  60
  61     public short getVersion() {
  62         return version;
  63     }
  64
  65     public void setVersion(short version) {
  66         this.version = version;
  67     }
  68
  69     /**
  70      * Add a new Modification to the list of Modifications represented by this
  71      * composite
  72      *
  73      * @param modification
  74      */
  75     public void addModification(Modification modification) {
  76         modifications.add(modification);
  77     }
  78
  79     @Override
  80     public List<Modification> getModifications() {
  81         return modifications;
  82     }
  83
  84     @Override
  85     public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException {
  86         version = in.readShort();
  87
  88         int size = in.readInt();
  89
  90         if(size > 1) {
  91             SerializationUtils.REUSABLE_READER_TL.set(new NormalizedNodeInputStreamReader(in));
  92         }
  93
  94         try {
  95             for(int i = 0; i < size; i++) {
  96                 byte type = in.readByte();
  97                 switch(type) {
  98                 case Modification.WRITE:
  99                     modifications.add(WriteModification.fromStream(in, version));
 100                     break;
 101
 102                 case Modification.MERGE:
 103                     modifications.add(MergeModification.fromStream(in, version));
 104                     break;
 105
 106                 case Modification.DELETE:
 107                     modifications.add(DeleteModification.fromStream(in, version));
 108                     break;
 109                 }
 110             }
 111         } finally {
 112             SerializationUtils.REUSABLE_READER_TL.remove();
 113         }
 114     }
 115
 116     @Override
 117     public void writeExternal(ObjectOutput out) throws IOException {
 118         out.writeShort(version);
 119
 120         out.writeInt(modifications.size());
 121
 122         if(modifications.size() > 1) {
 123             SerializationUtils.REUSABLE_WRITER_TL.set(new NormalizedNodeOutputStreamWriter(out));
 124         }
 125
 126         try {
 127             for(Modification mod: modifications) {
 128                 out.writeByte(mod.getType());
 129                 mod.writeExternal(out);
 130             }
 131         } finally {
 132             SerializationUtils.REUSABLE_WRITER_TL.remove();
 133         }
 134     }
 135
 136     @Override
 137     @Deprecated
 138     public Object toSerializable() {
 139         PersistentMessages.CompositeModification.Builder builder =
 140                 PersistentMessages.CompositeModification.newBuilder();
 141
 142         builder.setTimeStamp(System.nanoTime());
 143
 144         for (Modification m : modifications) {
 145             builder.addModification((PersistentMessages.Modification) m.toSerializable());
 146         }
 147
 148         return builder.build();
 149     }
 150
 151     public static MutableCompositeModification fromSerializable(Object serializable) {
 152         if(serializable instanceof MutableCompositeModification) {
 153             return (MutableCompositeModification)serializable;
 154         } else {
 155             return fromLegacySerializable(serializable);
 156         }
 157     }
 158
 159     private static MutableCompositeModification fromLegacySerializable(Object serializable) {
 160         PersistentMessages.CompositeModification o = (PersistentMessages.CompositeModification) serializable;
 161         MutableCompositeModification compositeModification = new MutableCompositeModification();
 162
 163         for(PersistentMessages.Modification m : o.getModificationList()){
 164             if(m.getType().equals(DeleteModification.class.toString())){
 165                 compositeModification.addModification(DeleteModification.fromSerializable(m));
 166             } else if(m.getType().equals(WriteModification.class.toString())){
 167                 compositeModification.addModification(WriteModification.fromSerializable(m));
 168             } else if(m.getType().equals(MergeModification.class.toString())){
 169                 compositeModification.addModification(MergeModification.fromSerializable(m));
 170             }
 171         }
 172
 173         return compositeModification;
 174     }
 175 }