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 com.google.common.base.Preconditions;
  12 import java.io.IOException;
  13 import java.io.ObjectInput;
  14 import java.io.ObjectOutput;
  15 import java.util.ArrayList;
  16 import java.util.List;
  17 import org.opendaylight.controller.cluster.datastore.DataStoreVersions;
  18 import org.opendaylight.controller.cluster.datastore.messages.VersionedExternalizableMessage;
  19 import org.opendaylight.controller.cluster.datastore.node.utils.stream.NormalizedNodeInputOutput;
  20 import org.opendaylight.controller.cluster.datastore.node.utils.stream.NormalizedNodeInputStreamReader;
  21 import org.opendaylight.controller.cluster.datastore.node.utils.stream.SerializationUtils;
  22 import org.opendaylight.controller.sal.core.spi.data.DOMStoreWriteTransaction;
  23 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeModification;
  24
  25 /**
  26  * MutableCompositeModification is just a mutable version of a CompositeModification.
  27  */
  28 public class MutableCompositeModification extends VersionedExternalizableMessage implements CompositeModification {
  29     private static final long serialVersionUID = 1L;
  30
  31     private final List<Modification> modifications = new ArrayList<>();
  32
  33     public MutableCompositeModification() {
  34         this(DataStoreVersions.CURRENT_VERSION);
  35     }
  36
  37     public MutableCompositeModification(short version) {
  38         super(version);
  39     }
  40
  41     @Override
  42     public void apply(DOMStoreWriteTransaction transaction) {
  43         for (Modification modification : modifications) {
  44             modification.apply(transaction);
  45         }
  46     }
  47
  48     @Override
  49     public void apply(DataTreeModification transaction) {
  50         for (Modification modification : modifications) {
  51             modification.apply(transaction);
  52         }
  53     }
  54
  55     @Override
  56     public byte getType() {
  57         return COMPOSITE;
  58     }
  59
  60     /**
  61      * Add a new Modification to the list of Modifications represented by this composite.
  62      *
  63      * @param modification the modification to add.
  64      */
  65     public void addModification(Modification modification) {
  66         modifications.add(modification);
  67     }
  68
  69     @Override
  70     public List<Modification> getModifications() {
  71         return modifications;
  72     }
  73
  74     @Override
  75     public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException {
  76         super.readExternal(in);
  77
  78         int size = in.readInt();
  79
  80         if (size > 1) {
  81             SerializationUtils.REUSABLE_READER_TL.set(new NormalizedNodeInputStreamReader(in));
  82         }
  83
  84         try {
  85             for (int i = 0; i < size; i++) {
  86                 byte type = in.readByte();
  87                 switch (type) {
  88                     case Modification.WRITE:
  89                         modifications.add(WriteModification.fromStream(in, getVersion()));
  90                         break;
  91
  92                     case Modification.MERGE:
  93                         modifications.add(MergeModification.fromStream(in, getVersion()));
  94                         break;
  95
  96                     case Modification.DELETE:
  97                         modifications.add(DeleteModification.fromStream(in, getVersion()));
  98                         break;
  99                     default:
 100                         break;
 101                 }
 102             }
 103         } finally {
 104             SerializationUtils.REUSABLE_READER_TL.remove();
 105         }
 106     }
 107
 108     @Override
 109     public void writeExternal(ObjectOutput out) throws IOException {
 110         super.writeExternal(out);
 111
 112         out.writeInt(modifications.size());
 113
 114         if (modifications.size() > 1) {
 115             SerializationUtils.REUSABLE_WRITER_TL.set(NormalizedNodeInputOutput.newDataOutput(out));
 116         }
 117
 118         try {
 119             for (Modification mod: modifications) {
 120                 out.writeByte(mod.getType());
 121                 mod.writeExternal(out);
 122             }
 123         } finally {
 124             SerializationUtils.REUSABLE_WRITER_TL.remove();
 125         }
 126     }
 127
 128     public static MutableCompositeModification fromSerializable(Object serializable) {
 129         Preconditions.checkArgument(serializable instanceof MutableCompositeModification);
 130         return (MutableCompositeModification)serializable;
 131     }
 132 }