gerrit.fd Code Review - hc2vpp.git/blob

   1 /*
   2  * Copyright (c) 2016 Cisco and/or its affiliates.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at:
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package io.fd.honeycomb.v3po.translate.impl.write;
  18
  19 import com.google.common.base.Function;
  20 import com.google.common.collect.ImmutableMap;
  21 import com.google.common.collect.Maps;
  22 import com.google.common.collect.Sets;
  23 import io.fd.honeycomb.v3po.translate.spi.write.ListWriterCustomizer;
  24 import io.fd.honeycomb.v3po.translate.write.ChildWriter;
  25 import io.fd.honeycomb.v3po.translate.write.WriteContext;
  26 import io.fd.honeycomb.v3po.translate.util.RWUtils;
  27 import java.util.List;
  28 import java.util.Map;
  29 import javax.annotation.Nonnull;
  30 import org.opendaylight.yangtools.yang.binding.Augmentation;
  31 import org.opendaylight.yangtools.yang.binding.ChildOf;
  32 import org.opendaylight.yangtools.yang.binding.DataObject;
  33 import org.opendaylight.yangtools.yang.binding.Identifiable;
  34 import org.opendaylight.yangtools.yang.binding.Identifier;
  35 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  36
  37 public class CompositeListWriter<D extends DataObject & Identifiable<K>, K extends Identifier<D>> extends
  38     AbstractCompositeWriter<D>
  39     implements ChildWriter<D> {
  40
  41     public static final Function<DataObject, Object> INDEX_FUNCTION = new Function<DataObject, Object>() {
  42         @Override
  43         public Object apply(final DataObject input) {
  44             return input instanceof Identifiable<?>
  45                 ? ((Identifiable<?>) input).getKey()
  46                 : input;
  47         }
  48     };
  49
  50
  51     private final ListWriterCustomizer<D, K> customizer;
  52
  53     public CompositeListWriter(@Nonnull final Class<D> type,
  54                                @Nonnull final List<ChildWriter<? extends ChildOf<D>>> childWriters,
  55                                @Nonnull final List<ChildWriter<? extends Augmentation<D>>> augWriters,
  56                                @Nonnull final ListWriterCustomizer<D, K> customizer) {
  57         super(type, childWriters, augWriters);
  58         this.customizer = customizer;
  59     }
  60
  61     public CompositeListWriter(@Nonnull final Class<D> type,
  62                                @Nonnull final List<ChildWriter<? extends ChildOf<D>>> childWriters,
  63                                @Nonnull final ListWriterCustomizer<D, K> customizer) {
  64         this(type, childWriters, RWUtils.<D>emptyAugWriterList(), customizer);
  65     }
  66
  67     public CompositeListWriter(@Nonnull final Class<D> type,
  68                                @Nonnull final ListWriterCustomizer<D, K> customizer) {
  69         this(type, RWUtils.<D>emptyChildWriterList(), RWUtils.<D>emptyAugWriterList(), customizer);
  70
  71     }
  72
  73     @Override
  74     protected void writeCurrentAttributes(@Nonnull final InstanceIdentifier<D> id, @Nonnull final D data,
  75                                           @Nonnull final WriteContext ctx) {
  76         customizer.writeCurrentAttributes(id, data, ctx.getContext());
  77     }
  78
  79     @Override
  80     protected void deleteCurrentAttributes(@Nonnull final InstanceIdentifier<D> id, @Nonnull final D dataBefore,
  81                                            @Nonnull final WriteContext ctx) {
  82         customizer.deleteCurrentAttributes(id, dataBefore, ctx.getContext());
  83     }
  84
  85     @Override
  86     protected void updateCurrentAttributes(@Nonnull final InstanceIdentifier<D> id, @Nonnull final D dataBefore,
  87                                            @Nonnull final D dataAfter, @Nonnull final WriteContext ctx) {
  88         customizer.updateCurrentAttributes(id, dataBefore, dataAfter, ctx.getContext());
  89     }
  90
  91     @Override
  92     public void writeChild(@Nonnull final InstanceIdentifier<? extends DataObject> parentId,
  93                            @Nonnull final DataObject parentData,
  94                            @Nonnull final WriteContext ctx) {
  95         final InstanceIdentifier<D> currentId = RWUtils.appendTypeToId(parentId, getManagedDataObjectType());
  96         final List<D> currentData = customizer.extract(currentId, parentData);
  97         for (D entry : currentData) {
  98             writeCurrent(currentId, entry, ctx);
  99         }
 100     }
 101
 102     @Override
 103     public void deleteChild(@Nonnull final InstanceIdentifier<? extends DataObject> parentId,
 104                             @Nonnull final DataObject parentDataBefore,
 105                             @Nonnull final WriteContext ctx) {
 106         final InstanceIdentifier<D> currentId = RWUtils.appendTypeToId(parentId, getManagedDataObjectType());
 107         final List<D> dataBefore = customizer.extract(currentId, parentDataBefore);
 108         for (D entry : dataBefore) {
 109             deleteCurrent(currentId, entry, ctx);
 110         }
 111     }
 112
 113     @Override
 114     public void updateChild(@Nonnull final InstanceIdentifier<? extends DataObject> parentId,
 115                             @Nonnull final DataObject parentDataBefore, @Nonnull final DataObject parentDataAfter,
 116                             @Nonnull final WriteContext ctx) {
 117         final InstanceIdentifier<D> currentId = RWUtils.appendTypeToId(parentId, getManagedDataObjectType());
 118         final ImmutableMap<Object, D>
 119             dataBefore = Maps.uniqueIndex(customizer.extract(currentId, parentDataBefore), INDEX_FUNCTION);
 120         final ImmutableMap<Object, D>
 121             dataAfter = Maps.uniqueIndex(customizer.extract(currentId, parentDataAfter), INDEX_FUNCTION);
 122
 123         for (Map.Entry<Object, D> after : dataAfter.entrySet()) {
 124             final D before = dataBefore.get(after.getKey());
 125             if(before == null) {
 126                 writeCurrent(currentId, after.getValue(), ctx);
 127             } else {
 128                 updateCurrent(currentId, before, after.getValue(), ctx);
 129             }
 130         }
 131
 132         // Delete the rest in dataBefore
 133         for (Object deletedNodeKey : Sets.difference(dataBefore.keySet(), dataAfter.keySet())) {
 134             final D deleted = dataBefore.get(deletedNodeKey);
 135             deleteCurrent(currentId, deleted, ctx);
 136         }
 137
 138     }
 139
 140     @Override
 141     protected void writeCurrent(final InstanceIdentifier<D> id, final D data, final WriteContext ctx) {
 142         // Make sure the key is present
 143         if(isWildcarded(id)) {
 144             super.writeCurrent(getSpecificId(id, data), data, ctx);
 145         } else {
 146             super.writeCurrent(id, data, ctx);
 147         }
 148     }
 149
 150     @Override
 151     protected void updateCurrent(final InstanceIdentifier<D> id, final D dataBefore, final D dataAfter,
 152                                  final WriteContext ctx) {
 153         // Make sure the key is present
 154         if(isWildcarded(id)) {
 155             super.updateCurrent(getSpecificId(id, dataBefore), dataBefore, dataAfter, ctx);
 156         } else {
 157             super.updateCurrent(id, dataBefore, dataAfter, ctx);
 158         }
 159     }
 160
 161     @Override
 162     protected void deleteCurrent(final InstanceIdentifier<D> id, final D dataBefore, final WriteContext ctx) {
 163         // Make sure the key is present
 164         if(isWildcarded(id)) {
 165             super.deleteCurrent(getSpecificId(id, dataBefore), dataBefore, ctx);
 166         } else {
 167             super.deleteCurrent(id, dataBefore, ctx);
 168         }
 169     }
 170
 171     private boolean isWildcarded(final InstanceIdentifier<D> id) {
 172         return id.firstIdentifierOf(getManagedDataObjectType().getTargetType()).isWildcarded();
 173     }
 174
 175     private InstanceIdentifier<D> getSpecificId(final InstanceIdentifier<D> currentId, final D current) {
 176         return RWUtils.replaceLastInId(currentId,
 177             new InstanceIdentifier.IdentifiableItem<>(currentId.getTargetType(), current.getKey()));
 178     }
 179 }