gerrit.fd Code Review - hc2vpp.git/blob

   1 /*
   2  * Copyright (c) 2018 Bell Canada, Pantheon Technologies 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
  18 package io.fd.hc2vpp.fib.management.read;
  19
  20 import io.fd.hc2vpp.common.translate.util.ByteDataTranslator;
  21 import io.fd.honeycomb.translate.read.ReadContext;
  22 import io.fd.honeycomb.translate.read.ReadFailedException;
  23 import io.fd.honeycomb.translate.spi.read.Initialized;
  24 import io.fd.honeycomb.translate.spi.read.InitializingListReaderCustomizer;
  25 import io.fd.honeycomb.translate.util.read.cache.DumpCacheManager;
  26 import io.fd.vpp.jvpp.core.dto.Ip6FibDetailsReplyDump;
  27 import io.fd.vpp.jvpp.core.dto.IpFibDetailsReplyDump;
  28 import java.util.List;
  29 import java.util.stream.Collectors;
  30 import java.util.stream.Stream;
  31 import javax.annotation.Nonnull;
  32 import org.opendaylight.yang.gen.v1.http.fd.io.hc2vpp.yang.vpp.fib.table.management.rev180521.AddressFamilyIdentity;
  33 import org.opendaylight.yang.gen.v1.http.fd.io.hc2vpp.yang.vpp.fib.table.management.rev180521.Ipv4;
  34 import org.opendaylight.yang.gen.v1.http.fd.io.hc2vpp.yang.vpp.fib.table.management.rev180521.Ipv6;
  35 import org.opendaylight.yang.gen.v1.http.fd.io.hc2vpp.yang.vpp.fib.table.management.rev180521.VniReference;
  36 import org.opendaylight.yang.gen.v1.http.fd.io.hc2vpp.yang.vpp.fib.table.management.rev180521.vpp.fib.table.management.FibTablesBuilder;
  37 import org.opendaylight.yang.gen.v1.http.fd.io.hc2vpp.yang.vpp.fib.table.management.rev180521.vpp.fib.table.management.fib.tables.Table;
  38 import org.opendaylight.yang.gen.v1.http.fd.io.hc2vpp.yang.vpp.fib.table.management.rev180521.vpp.fib.table.management.fib.tables.TableBuilder;
  39 import org.opendaylight.yang.gen.v1.http.fd.io.hc2vpp.yang.vpp.fib.table.management.rev180521.vpp.fib.table.management.fib.tables.TableKey;
  40 import org.opendaylight.yangtools.concepts.Builder;
  41 import org.opendaylight.yangtools.yang.binding.DataObject;
  42 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  43
  44 class FibTableCustomizer implements InitializingListReaderCustomizer<Table, TableKey, TableBuilder>, ByteDataTranslator {
  45     private final DumpCacheManager<IpFibDetailsReplyDump, Void> ipv4DumpManager;
  46     private final DumpCacheManager<Ip6FibDetailsReplyDump, Void> ipv6DumpManager;
  47
  48     FibTableCustomizer(final DumpCacheManager<IpFibDetailsReplyDump, Void> ipv4DumpManager,
  49                        final DumpCacheManager<Ip6FibDetailsReplyDump, Void> ipv6DumpManager) {
  50
  51         this.ipv4DumpManager = ipv4DumpManager;
  52         this.ipv6DumpManager = ipv6DumpManager;
  53     }
  54
  55     @Nonnull
  56     @Override
  57     public List<TableKey> getAllIds(@Nonnull final InstanceIdentifier<Table> instanceIdentifier,
  58                                     @Nonnull final ReadContext readContext) throws ReadFailedException {
  59         return Stream.concat(ipv4DumpManager.getDump(instanceIdentifier, readContext.getModificationCache())
  60                         .or(new IpFibDetailsReplyDump())
  61                         .ipFibDetails.stream()
  62                         .filter(ipFibDetails -> ipFibDetails.tableId >= 0)
  63                         .map(ipFibDetails -> new TableKey(Ipv4.class, new VniReference((long) ipFibDetails.tableId)))
  64                         .distinct(),
  65                 ipv6DumpManager.getDump(instanceIdentifier, readContext.getModificationCache())
  66                         .or(new Ip6FibDetailsReplyDump())
  67                         .ip6FibDetails.stream()
  68                         .filter(ip6FibDetails -> ip6FibDetails.tableId >= 0)
  69                         .map(ipFibDetails -> new TableKey(Ipv6.class, new VniReference((long) ipFibDetails.tableId))))
  70                 .distinct()
  71                 .collect(Collectors.toList());
  72     }
  73
  74     @Override
  75     public void merge(@Nonnull final Builder<? extends DataObject> builder, @Nonnull final List<Table> list) {
  76         ((FibTablesBuilder) builder).setTable(list);
  77     }
  78
  79     @Nonnull
  80     @Override
  81     public TableBuilder getBuilder(@Nonnull final InstanceIdentifier<Table> instanceIdentifier) {
  82         return new TableBuilder();
  83     }
  84
  85     @Override
  86     public void readCurrentAttributes(@Nonnull final InstanceIdentifier<Table> instanceIdentifier,
  87                                       @Nonnull final TableBuilder builder, @Nonnull final ReadContext readContext)
  88             throws ReadFailedException {
  89         TableKey tableKey = instanceIdentifier.firstKeyOf(Table.class);
  90
  91         if (tableKey.getAddressFamily().equals(Ipv4.class)) {
  92             ipv4DumpManager.getDump(instanceIdentifier, readContext.getModificationCache())
  93                     .or(new IpFibDetailsReplyDump())
  94                     .ipFibDetails.stream()
  95                     .filter(ipFibDetails -> ipFibDetails.tableId == tableKey.getTableId().getValue().intValue())
  96                     .findFirst().ifPresent(
  97                     ipFibDetails -> parseFibDetails(ipFibDetails.tableId, ipFibDetails.tableName, Ipv4.class, builder));
  98
  99         } else {
 100             ipv6DumpManager.getDump(instanceIdentifier, readContext.getModificationCache())
 101                     .or(new Ip6FibDetailsReplyDump())
 102                     .ip6FibDetails.stream()
 103                     .filter(ipFibDetails -> ipFibDetails.tableId == tableKey.getTableId().getValue().intValue())
 104                     .findFirst().ifPresent(
 105                     ipFibDetails -> parseFibDetails(ipFibDetails.tableId, ipFibDetails.tableName, Ipv6.class, builder));
 106         }
 107     }
 108
 109     private void parseFibDetails(final Integer tableId, final byte[] tableName,
 110                                  final Class<? extends AddressFamilyIdentity> addressFamily,
 111                                  final TableBuilder builder) {
 112         builder.setAddressFamily(addressFamily)
 113                 .setTableId(new VniReference(Integer.toUnsignedLong(tableId)));
 114
 115         if (tableName != null) {
 116             // table name is optional
 117             builder.setName(toString(tableName));
 118         }
 119     }
 120
 121     @Nonnull
 122     @Override
 123     public Initialized<Table> init(@Nonnull final InstanceIdentifier<Table> id,
 124                                    @Nonnull final Table readValue,
 125                                    @Nonnull final ReadContext ctx) {
 126         return Initialized.create(id, readValue);
 127     }
 128 }