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.Collections;
  17 import java.util.List;
  18 import org.opendaylight.controller.cluster.datastore.DataStoreVersions;
  19 import org.opendaylight.controller.cluster.datastore.messages.VersionedExternalizableMessage;
  20 import org.opendaylight.controller.cluster.datastore.node.utils.stream.NormalizedNodeInputOutput;
  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     private List<Modification> immutableModifications = null;
  33
  34     public MutableCompositeModification() {
  35         this(DataStoreVersions.CURRENT_VERSION);
  36     }
  37
  38     public MutableCompositeModification(short version) {
  39         super(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     /**
  62      * Add a new Modification to the list of Modifications represented by this composite.
  63      *
  64      * @param modification the modification to add.
  65      */
  66     public void addModification(Modification modification) {
  67         Preconditions.checkNotNull(modification);
  68         modifications.add(modification);
  69     }
  70
  71     public void addModifications(Iterable<Modification> newMods) {
  72         for (Modification mod : newMods) {
  73             addModification(mod);
  74         }
  75     }
  76
  77     @Override
  78     public List<Modification> getModifications() {
  79         if (immutableModifications == null) {
  80             immutableModifications = Collections.unmodifiableList(modifications);
  81         }
  82
  83         return immutableModifications;
  84     }
  85
  86     @Override
  87     public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException {
  88         super.readExternal(in);
  89
  90         int size = in.readInt();
  91
  92         if (size > 1) {
  93             SerializationUtils.REUSABLE_READER_TL.set(NormalizedNodeInputOutput.newDataInputWithoutValidation(in));
  94         }
  95
  96         try {
  97             for (int i = 0; i < size; i++) {
  98                 byte type = in.readByte();
  99                 switch (type) {
 100                     case Modification.WRITE:
 101                         modifications.add(WriteModification.fromStream(in, getVersion()));
 102                         break;
 103
 104                     case Modification.MERGE:
 105                         modifications.add(MergeModification.fromStream(in, getVersion()));
 106                         break;
 107
 108                     case Modification.DELETE:
 109                         modifications.add(DeleteModification.fromStream(in, getVersion()));
 110                         break;
 111                     default:
 112                         break;
 113                 }
 114             }
 115         } finally {
 116             SerializationUtils.REUSABLE_READER_TL.remove();
 117         }
 118     }
 119
 120     @Override
 121     public void writeExternal(ObjectOutput out) throws IOException {
 122         super.writeExternal(out);
 123
 124         out.writeInt(modifications.size());
 125
 126         if (modifications.size() > 1) {
 127             SerializationUtils.REUSABLE_WRITER_TL.set(NormalizedNodeInputOutput.newDataOutput(out));
 128         }
 129
 130         try {
 131             for (Modification mod: modifications) {
 132                 out.writeByte(mod.getType());
 133                 mod.writeExternal(out);
 134             }
 135         } finally {
 136             SerializationUtils.REUSABLE_WRITER_TL.remove();
 137         }
 138     }
 139
 140     public static MutableCompositeModification fromSerializable(Object serializable) {
 141         Preconditions.checkArgument(serializable instanceof MutableCompositeModification);
 142         return (MutableCompositeModification)serializable;
 143     }
 144 }