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 io.fd.honeycomb.v3po.translate.impl.TraversalType;
  20 import io.fd.honeycomb.v3po.translate.spi.write.RootWriterCustomizer;
  21 import io.fd.honeycomb.v3po.translate.util.RWUtils;
  22 import io.fd.honeycomb.v3po.translate.write.ChildWriter;
  23 import io.fd.honeycomb.v3po.translate.write.WriteContext;
  24 import io.fd.honeycomb.v3po.translate.write.WriteFailedException;
  25 import java.util.List;
  26 import javax.annotation.Nonnull;
  27 import org.opendaylight.yangtools.yang.binding.Augmentation;
  28 import org.opendaylight.yangtools.yang.binding.ChildOf;
  29 import org.opendaylight.yangtools.yang.binding.DataObject;
  30 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  31
  32 public class CompositeRootWriter<D extends DataObject> extends AbstractCompositeWriter<D> {
  33
  34     private final RootWriterCustomizer<D> customizer;
  35
  36     public CompositeRootWriter(@Nonnull final Class<D> type,
  37                                @Nonnull final List<ChildWriter<? extends ChildOf<D>>> childWriters,
  38                                @Nonnull final List<ChildWriter<? extends Augmentation<D>>> augWriters,
  39                                @Nonnull final RootWriterCustomizer<D> customizer) {
  40         this(type, childWriters, augWriters, customizer, TraversalType.PREORDER);
  41     }
  42
  43     public CompositeRootWriter(@Nonnull final Class<D> type,
  44                                @Nonnull final List<ChildWriter<? extends ChildOf<D>>> childWriters,
  45                                @Nonnull final List<ChildWriter<? extends Augmentation<D>>> augWriters,
  46                                @Nonnull final RootWriterCustomizer<D> customizer,
  47                                @Nonnull final TraversalType traversalType) {
  48         super(type, childWriters, augWriters, traversalType);
  49         this.customizer = customizer;
  50     }
  51
  52     public CompositeRootWriter(@Nonnull final Class<D> type,
  53                                @Nonnull final List<ChildWriter<? extends ChildOf<D>>> childWriters,
  54                                @Nonnull final RootWriterCustomizer<D> customizer) {
  55         this(type, childWriters, RWUtils.<D>emptyAugWriterList(), customizer);
  56     }
  57
  58     public CompositeRootWriter(@Nonnull final Class<D> type,
  59                                @Nonnull final RootWriterCustomizer<D> customizer) {
  60         this(type, RWUtils.<D>emptyChildWriterList(), RWUtils.<D>emptyAugWriterList(), customizer);
  61     }
  62
  63     @Override
  64     protected void writeCurrentAttributes(@Nonnull final InstanceIdentifier<D> id, @Nonnull final D data,
  65                                           @Nonnull final WriteContext ctx) throws WriteFailedException {
  66         // TODO wrap all customizer invocations in try catch, and wrap runtime exceptions in ReadFailed
  67         // TODO same for readers
  68         customizer.writeCurrentAttributes(id, data, ctx.getContext());
  69     }
  70
  71     @Override
  72     protected void deleteCurrentAttributes(@Nonnull final InstanceIdentifier<D> id, @Nonnull final D dataBefore,
  73                                            @Nonnull final WriteContext ctx) throws WriteFailedException {
  74         customizer.deleteCurrentAttributes(id, dataBefore, ctx.getContext());
  75     }
  76
  77     @Override
  78     protected void updateCurrentAttributes(@Nonnull final InstanceIdentifier<D> id,
  79                                            @Nonnull final D dataBefore,
  80                                            @Nonnull final D dataAfter,
  81                                            @Nonnull final WriteContext ctx) throws WriteFailedException {
  82         customizer.updateCurrentAttributes(id, dataBefore, dataAfter, ctx.getContext());
  83     }
  84 }